Update agents/language_agent.py

agents/language_agent.py

@@ -1,21 +1,10 @@
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+from transformers import pipeline
 from langchain.llms import HuggingFacePipeline
 from langchain.chains import RetrievalQA
 from agents.retriever_agent import create_vectorstore
 
 def generate_brief(question):
-    model_id = "google/flan-t5-small"
-
-    tokenizer = AutoTokenizer.from_pretrained(model_id)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
-
-    pipe = pipeline(
-        "text2text-generation",
-        model=model,
-        tokenizer=tokenizer,
-        max_length=512,
-        temperature=0.7
-    )
+    pipe = pipeline("text2text-generation", model="google/flan-t5-small")
 
     llm = HuggingFacePipeline(pipeline=pipe)
 
@@ -23,4 +12,4 @@ def generate_brief(question):
     retriever = vectordb.as_retriever()
 
     qa_chain = RetrievalQA.from_chain_type(llm=llm, retriever=retriever)
-    return qa_chain.run(question)
+    return qa_chain.run(question)