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.
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
"""Chat REST route — POST /chat fallback when WS is unavailable."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from fastapi import APIRouter, Request
|
|
from fastapi.responses import JSONResponse
|
|
|
|
from shared.schemas import ChatRequest
|
|
|
|
from app.deep_agent import run_home
|
|
from shared.ws_context import clear_current_user, set_current_user
|
|
|
|
router = APIRouter(prefix="/chat", tags=["chat"])
|
|
|
|
|
|
@router.post("")
|
|
async def chat(body: ChatRequest, request: Request) -> JSONResponse:
|
|
"""REST fallback for home chat.
|
|
|
|
In the microservices setup, Traefik ForwardAuth has already validated
|
|
the JWT and injected X-User-Id / X-User-Email / X-User-Tier headers.
|
|
"""
|
|
user_id = request.headers.get("X-User-Id", "")
|
|
if not user_id:
|
|
return JSONResponse(status_code=401, content={"detail": "Missing X-User-Id header"})
|
|
|
|
set_current_user(user_id)
|
|
try:
|
|
response = await run_home(
|
|
user_id=user_id,
|
|
message=body.message,
|
|
context=body.context.model_dump(),
|
|
)
|
|
finally:
|
|
clear_current_user()
|
|
|
|
return JSONResponse(content={"response": response})
|