Initial commit
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
from fastapi import FastAPI
|
||||
from pydantic import BaseModel
|
||||
from .backend import BDAgent
|
||||
from .backend import orquestrador
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
@@ -26,7 +26,7 @@ class QueryResponse(BaseModel):
|
||||
|
||||
@app.post("/agent", response_model=QueryResponse)
|
||||
def run_agent(request: QueryRequest):
|
||||
result = BDAgent.main(request.query, request.history, request.model, request.base)
|
||||
result = orquestrador.main(request.query, request.history, request.model, request.base)
|
||||
return QueryResponse(
|
||||
response=result["response"],
|
||||
input_tokens=result["input_tokens"],
|
||||
|
||||
Reference in New Issue
Block a user