Feat: Adds base project
This commit is contained in:
32
back/app/api.py
Normal file
32
back/app/api.py
Normal file
@@ -0,0 +1,32 @@
|
||||
from fastapi import FastAPI
|
||||
from pydantic import BaseModel
|
||||
from .backend import orquestrador
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
|
||||
@app.get("/")
|
||||
def health():
|
||||
return {"status": "ok"}
|
||||
|
||||
|
||||
class QueryRequest(BaseModel):
|
||||
query: str
|
||||
history: str = ""
|
||||
|
||||
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 = orquestrador.main(request.query, request.history)
|
||||
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