Update agent.py

agent.py

@@ -2,7 +2,7 @@ from langchain_core.messages import HumanMessage, AIMessage
 from langchain_community.tools import DuckDuckGoSearchRun
 from langchain.prompts import PromptTemplate
 from langchain_openai import ChatOpenAI
-from langchain.chains import LLMChain
+from langchain_core.runnables import RunnableSequence
 import os
 
 class AdvancedAgent:
@@ -30,18 +30,23 @@ class AdvancedAgent:
             Answer: 
             """
         )
-        self.chain = LLMChain(llm=self.llm, prompt=self.prompt_template)
+        self.chain = self.prompt_template | self.llm
 
     def __call__(self, question: str) -> str:
         print(f"Agent processing question (first 50 chars): {question[:50]}...")
         try:
-            # Perform a web search to gather context
-            search_query = f"{question}"
-            search_results = self.search_tool.run(search_query)
-            context = search_results[:1000]  # Limit context length
+            # Avoid rate limiting with a delay
+            time.sleep(1)  # Adjust delay based on testing
+    
+            # Perform web search
+            search_results = self.search_tool.run(question)
+            if not search_results:
+                context = "No search results found."
+            else:
+                context = search_results[:1000]  # Limit context length
 
             # Generate answer using LLM and context
-            response = self.chain.run(question=question, context=context)
+            response = self.chain.invoke({"question": question, "context": context})
             print(f"Agent generated answer: {response[:50]}...")
             return response.strip()
         except Exception as e: