/
igormayer
/
langgraph_sql_agent_web
Обзор
Документация
Войти
/
igormayer
/
langgraph_sql_agent_web
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
api/sqlagentservice.py
37 строк
1 KB
Igor Mayer
docker support was added
07 фев 2026, 18:48
07 фев 2026, 18:48
7c59ea2
Код
Авторство
О чём код?
import json from fastapi import FastAPI, Body from pydantic import BaseModel from typing import Dict, Any from fastapi.responses import StreamingResponse from backend.sqlagent.graph import get_agent_stream app = FastAPI() @app.post("/chat") async def chat_endpoint(payload: Dict[str, Any] = Body(...)): settings = payload.get("settings", {}) prompt = payload.get("prompt", "") async def event_generator(): try: async for mode, chunk in get_agent_stream(prompt, settings): if mode == "custom": # Данные от StreamWriter из узла payload = {"type": "status", "data": chunk} else: # Данные о завершении узлов (updates) payload = {"type": "update", "data": chunk} yield f"data: {json.dumps(payload)}\n\n" except Exception as e: print(f"Error in stream: {e}") yield f"data: Error: {str(e)}\n\n" return StreamingResponse(event_generator(), media_type="text/event-stream") if __name__ == "__main__": import uvicorn uvicorn.run(app, host="0.0.0.0", port=8000)