| import httpx |
| from fastapi import FastAPI, Request |
| from fastapi.responses import StreamingResponse |
| from fastapi.middleware.cors import CORSMiddleware |
| from pydantic import BaseModel |
| from routes_auth import router as auth_router |
| from service_chat import stream_chat_pipeline |
| from connector_browser import sync_user_cookies |
|
|
| app = FastAPI(title="Omni 4.8 Connectors API") |
|
|
| app.add_middleware( |
| CORSMiddleware, |
| allow_origins=["*"], |
| allow_credentials=True, |
| allow_methods=["*"], |
| allow_headers=["*"], |
| ) |
|
|
| if hasattr(auth_router, "prefix") and not auth_router.prefix: |
| app.include_router(auth_router, prefix="/auth") |
| else: |
| app.include_router(auth_router) |
|
|
| class CookieSyncRequest(BaseModel): |
| user_id: str |
| cookies: str |
| url: str |
|
|
| @app.get("/") |
| async def root(): |
| return {"status": "ok", "service": "Omni 4.8 Backend"} |
|
|
| @app.post("/browser/sync-cookies") |
| async def sync_cookies_endpoint(data: CookieSyncRequest): |
| success = await sync_user_cookies(data.user_id, data.cookies, data.url) |
| return {"status": "success" if success else "failed", "synced": success} |
|
|
| @app.post("/generate") |
| async def generate_endpoint(request: Request): |
| body = await request.json() |
| client = httpx.AsyncClient(timeout=180.0) |
|
|
| async def stream_wrapper(): |
| try: |
| async for chunk in stream_chat_pipeline(request, body, client): |
| yield chunk |
| finally: |
| await client.aclose() |
|
|
| return StreamingResponse(stream_wrapper(), media_type="text/event-stream") |
|
|
