d856dfd28c
Move duplicated files from chat + batch-agent into shared/: - shared/ws_context.py — Redis-based tool call round-trip - shared/llm.py — LiteLLM factory (get_llm, embed) - shared/agents/ — 4 domain agents (task, note, project, timeline) Update all service imports to use shared.* instead of app.*. Delete 12 duplicated files across both services.
80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
"""Batch Agent Service — FastAPI application.
|
|
|
|
Owns: agent_runner (local directory + cloud connectors), journey builder,
|
|
filesystem_agent, integrations (Gmail, MS Graph).
|
|
|
|
Communicates with WS Gateway via Redis:
|
|
- Subscribes to batch:request:{user_id} (journey_start, journey_message)
|
|
- Publishes to ws:out:{user_id} (journey replies + tool calls)
|
|
- BRPOP on tool:result:{call_id} (tool-call round-trip, 30s timeout)
|
|
- SET+EX on journey:{user_id} (journey session state, TTL 1800s)
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import logging
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
# Ensure the repo root is on sys.path so ``shared`` is importable when
|
|
# running locally (in Docker the COPY already places it at /app/shared/).
|
|
_repo_root = str(Path(__file__).resolve().parents[3])
|
|
if _repo_root not in sys.path:
|
|
sys.path.insert(0, _repo_root)
|
|
|
|
from contextlib import asynccontextmanager
|
|
from typing import AsyncGenerator
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
from app.redis_consumer import start_consumer
|
|
from app.routes import router
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
|
|
# Initialise Langfuse tracing (no-op if keys are missing)
|
|
from app.tracing import init_langfuse
|
|
init_langfuse()
|
|
|
|
logger.info("batch-agent: starting Redis consumer")
|
|
task = asyncio.create_task(start_consumer())
|
|
yield
|
|
task.cancel()
|
|
try:
|
|
await task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
|
|
from app.tracing import shutdown as shutdown_langfuse
|
|
shutdown_langfuse()
|
|
|
|
from shared.db import engine
|
|
await engine.dispose()
|
|
|
|
from shared.redis import redis_client
|
|
await redis_client.aclose()
|
|
|
|
logger.info("batch-agent: Redis consumer stopped")
|
|
|
|
|
|
app = FastAPI(title="Adiuva Batch Agent Service", lifespan=lifespan)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_methods=["GET", "POST"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
app.include_router(router)
|
|
|
|
|
|
@app.get("/health")
|
|
async def health() -> dict[str, str]:
|
|
return {"status": "ok", "service": "batch-agent"}
|