File size: 665 Bytes
efd486d 42838b0 b656794 efd486d 42838b0 efd486d 42838b0 |
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 |
from fastapi import APIRouter
from app.models.prediction_models import PredictionRequest, PredictionResponse
from typing import List
from app.agents.real_state_agent import real_state_agent
import os
import openai
openai.api_key = os.getenv("OPENAI_API_KEY")
router = APIRouter()
## MODIFICAR ESTO:
@router.post("/", response_model=List[PredictionResponse])
def predict(request: PredictionRequest):
# response = real_state_agent.run(request.question)
response = real_state_agent.run(request)
# ver el type esto sigue siendo list, pero ahora dije que queiro una response string:
# response_data = [{"response": response}]
return response
|