from fastapi import FastAPI, Form
from app.langchain_lhia import chat,vector_chat,milvus_vector
app = FastAPI()


@app.post('/pregunta')
def envio_pregunta(pregunta:str= Form(...), sesion:str= Form(...)):
    return chat.pregunta(pregunta,sesion)


@app.post('/pregunta_pgvector')
def envio_pregunta_pgvector(pregunta:str= Form(...)):
    return vector_chat.pregunta_pgvector(pregunta)

@app.post('/pregunta_milvus')
def envio_pregunta_milvus(pregunta:str= Form(...)):
    return milvus_vector.pregunta_milvusr(pregunta)

@app.post('/pregunta_milvus_danec_ventas')
def envio_pregunta_milvus_danec_ventas(pregunta:str= Form(...)):
    return milvus_vector.pregunta_milvus_danec_ventas(pregunta)


@app.post('/pregunta_milvus_vcisne')
def envio_pregunta_milvus_vcisne(pregunta:str= Form(...)):
    return milvus_vector.pregunta_milvus_vcisne(pregunta)

@app.post('/pregunta_milvus_academico')
def envio_pregunta_milvus_academico(pregunta:str= Form(...)):
    return milvus_vector.pregunta_milvus_academico(pregunta)

@app.post('/pregunta_coop_ambato')
def envio_pregunta_coop_ambato(pregunta:str= Form(...)):
    return milvus_vector.pregunta_coop_ambato(pregunta)

@app.post('/pregunta_coop_andalucia')
def envio_pregunta_coop_andalucia(pregunta:str= Form(...)):
    return milvus_vector.pregunta_coop_andalucia(pregunta)