First Commit

2025-03-06 17:32:42 +00:00
parent f97d731c82
commit 0c1f372e17
2 changed files with 58 additions and 5 deletions
@@ -70,7 +70,7 @@ def setup_rag_system(file_path, llm, embedding_function):
    question_answer_chain = create_stuff_documents_chain(llm, prompt)
    return create_retrieval_chain(retriever, question_answer_chain)

-def get_answer(rag_chain, user_input):
+def get_calculations(rag_chain, user_input):
    answer = rag_chain.invoke({"input": user_input})
    return answer['answer']