Adds final version with new prompt

This commit is contained in:
2026-02-11 14:45:00 -03:00
parent c31d089efb
commit 82ac556ecc
9 changed files with 283 additions and 92 deletions

35
code/app/api.py Normal file
View File

@@ -0,0 +1,35 @@
from fastapi import FastAPI
from pydantic import BaseModel
from .backend import BDAgent
app = FastAPI()
@app.get("/")
def health():
return {"status": "ok"}
class QueryRequest(BaseModel):
query: str
history: str = ""
model: str = "anthropic.claude-haiku-4-5-20251001-v1:0"
base: str = "bacio_transacional_loja_app"
class QueryResponse(BaseModel):
response: str
input_tokens: int
output_tokens: int
total_tokens: int
@app.post("/agent", response_model=QueryResponse)
def run_agent(request: QueryRequest):
result = BDAgent.main(request.query, request.history, request.model, request.base)
return QueryResponse(
response=result["response"],
input_tokens=result["input_tokens"],
output_tokens=result["output_tokens"],
total_tokens=result["total_tokens"],
)