Update agent.py

agent.py

@@ -111,7 +111,7 @@ def build_graph():
             messages = [SystemMessage(content=system_prompt)] + messages
         
         result = llm_with_tools.invoke(messages)
-        print(f"---Assistant Response: {result}") # Added print for debugging
+        # print(f"---Assistant Response: {result}") # Added print for debugging
         return {"messages": [result]}
 
     builder = StateGraph(MessagesState)
@@ -178,6 +178,33 @@ if __name__ == "__main__":
         try:
             # LangGraph returns the final state after execution completes or hits recursion limit
             final_state = graph.invoke({"messages": [SystemMessage(content=system_prompt), HumanMessage(content=question)]})
+            
+            # 在这里添加您的处理答案代码
+            def process_answer(answer): 
+                """处理最终答案，去除可能的解释性文本""" 
+                # 如果答案包含"FINAL ANSWER:"，提取实际答案部分 
+                if "FINAL ANSWER:" in answer.upper(): 
+                    import re 
+                    match = re.search(r'(?i)FINAL ANSWER:\s*(.*)', answer) 
+                    if match: 
+                        return match.group(1).strip() 
+                
+                # 如果答案较长且包含多个句子，尝试提取最后一句作为答案 
+                if len(answer.split()) > 15 and "." in answer: 
+                    sentences = answer.split(".") 
+                    # 过滤掉空字符串 
+                    sentences = [s.strip() for s in sentences if s.strip()] 
+                    if sentences: 
+                        return sentences[-1].strip() 
+                
+                return answer.strip() 
+            
+            # 在提交答案前应用处理 
+            final_answer = final_state["messages"][-1].content 
+            processed_answer = process_answer(final_answer) 
+            # 打印处理后的答案
+            print(f"\n--- Processed Answer: {processed_answer}")
+        
             print("\n--- Final State Messages ---")
             for m in final_state["messages"]:
                  m.pretty_print()