Update agents/language_agent.py

agents/language_agent.py

@@ -1,10 +1,8 @@
-from langchain.chains import RetrievalQA
-from langchain.chat_models import ChatOpenAI
-from agents.retriever_agent import create_vectorstore
+from transformers import pipeline
 
-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)
+# Load once at module level
+summarizer = pipeline("summarization", model="google/flan-t5-small")
+
+def generate_brief(text):
+    response = summarizer(text, max_length=80, min_length=20, do_sample=False)
+    return response[0]['summary_text']