INNER CODE UNIT · Python

chat

grumpyp/aixplora · backend/main.py:222

def chat(request: Request, question: Question, document: Document, usebrain: bool = Body(...)):
    apikey = request.headers.get("apikey", False)
    email = request.headers.get("email", False)
    genie = Genie()
    if apikey and email:
        genie = Genie(remote_db=True, apikey=apikey, email=email)
    answer = genie.query(query_texts=question.question, specific_doc=document.document, use_brain=usebrain)
    print(answer)
    return {"question": question.question, "answer": answer["answer"], "meta_data": answer["meta_data"]}


@app.post("/summarize/")
def test(document: Document):
    from llm.summarize import Summarize
    from database.models.summary import Summary
    db = Database().get_session()
    indexed_summaries = db.execute(text("SELECT * FROM summaries WHERE file_name = :file_name"),
                                   {"file_name": document.document}).first()

View source record →