Adds final version with new prompt
This commit is contained in:
35
code/app/api.py
Normal file
35
code/app/api.py
Normal 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"],
|
||||
)
|
||||
Reference in New Issue
Block a user