File size: 2,272 Bytes
4402d23 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 | import os
import threading
from fastapi import FastAPI, Response
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import RedirectResponse
from pydantic import BaseModel
from typing import Optional
from .agent import ShoppingAgent
from .feedback import salvar_feedback
from .logger import salvar_log_busca
def _cors_origins():
base = [
"http://127.0.0.1:5173",
"http://localhost:5173",
"http://127.0.0.1:3000",
"http://localhost:3000",
]
extra = os.getenv("CORS_ORIGINS", "")
if extra.strip():
base.extend(o.strip() for o in extra.split(",") if o.strip())
return base
app = FastAPI(title="TCC2 Agent API")
app.add_middleware(
CORSMiddleware,
allow_origins=_cors_origins(),
allow_origin_regex=r"https://.*\.hf\.space$",
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
agent = None
agent_lock = threading.Lock()
def get_agent():
global agent
if agent is None:
with agent_lock:
if agent is None:
agent = ShoppingAgent()
return agent
class ChatRequest(BaseModel):
query: Optional[str] = None
message: Optional[str] = None
top_k: int = 5
class FeedbackRequest(BaseModel):
query: str
product_id: str
product_name: str
rating: Optional[int] = None
is_helpful: Optional[bool] = None
@app.get("/health")
def health():
return {"status": "ok", "agent_ready": agent is not None}
@app.get("/", include_in_schema=False)
def root():
return RedirectResponse(url="/docs")
@app.get("/favicon.ico", include_in_schema=False)
def favicon():
return Response(status_code=204)
@app.post("/chat")
def chat(request: ChatRequest):
texto = request.query or request.message
if not texto:
return {"error": "query ou message deve ser informado"}
resultado = get_agent().responder(texto, top_k=request.top_k)
salvar_log_busca(resultado)
return resultado
@app.post("/feedback")
def feedback(request: FeedbackRequest):
return salvar_feedback(
query=request.query,
product_id=request.product_id,
product_name=request.product_name,
rating=request.rating,
is_helpful=request.is_helpful
) |