Update app.py

app.py

@@ -29,20 +29,10 @@ class BasicAgent:
         
         chat = ChatOpenAI(model="gpt-4", temperature=0)
 
-        # llm = HuggingFaceEndpoint(
-        #     repo_id="mistralai/Mistral-7B-Instruct-v0.2",
-        #     huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN"),
-        # )
-        
-        # chat = ChatHuggingFace(llm=llm, verbose=True)
-
         # Set up tools
         search_tool = DuckDuckGoSearchRun()
         tools = [search_tool]
         
-        # # Bind LLM with tools
-        # chat_with_tools = chat.bind_tools(tools)
-
         # Assistant function (process one step)
         def assistant(state: AgentState):
             return {
@@ -66,10 +56,10 @@ class BasicAgent:
         
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        
-        messages = [HumanMessage(content=question)]
+
+        system_prompt = "you are an expert and talkless person. please response shortly."
+        messages = [SystemMessage(content=system_prompt), HumanMessage(content=question)]
         response = self.agent.invoke({"messages": messages})
-        print('response:', response)
         
         # Access final answer correctly
         if isinstance(response, dict) and "messages" in response: