Update agents/language_agent.py

agents/language_agent.py

@@ -0,0 +1,10 @@
+from langchain.chains import RetrievalQA
+from langchain.chat_models import ChatOpenAI
+from agents.retriever_agent import create_vectorstore
+
+def generate_brief(question):
+    vectordb = create_vectorstore()
+    retriever = vectordb.as_retriever()
+    llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.2)
+    qa_chain = RetrievalQA.from_chain_type(llm=llm, retriever=retriever)
+    return qa_chain.run(question)