updated tools and added systemprompt

agent.py

@@ -12,7 +12,7 @@ from tools import (absolute, add, analyze_excel_file, arvix_search,
                    divide, exponential, factorial, floor_divide,
                    get_current_time_in_timezone, greatest_common_divisor,
                    is_prime, least_common_multiple, logarithm, modulus,
-                   multiply, percentage_calculator, power,
+                   multiply, percentage_calculator, power, python_code_parser,
                    roman_calculator_converter, square_root, subtract,
                    web_search, wiki_search)
 
@@ -28,9 +28,20 @@ tools = [
     get_current_time_in_timezone, compound_interest,
     convert_temperature, factorial, greatest_common_divisor,
     is_prime, least_common_multiple, percentage_calculator,
-    wiki_search, analyze_excel_file, arvix_search, audio_transcription
+    wiki_search, analyze_excel_file, arvix_search, audio_transcription, python_code_parser
 ]
 
+
+# Load system prompt
+system_prompt = """
+You are a helpful assistant tasked with answering questions using a set of tools. 
+Now, I will ask you a question. Report your thoughts, and finish your answer with the following template: 
+FINAL ANSWER: [YOUR FINAL ANSWER]. 
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+Your answer should only start with "FINAL ANSWER: ", then follows with the answer. 
+"""
+
+
 def build_graph():
     """Build the graph""" 
 
@@ -52,17 +63,17 @@ def build_graph():
     # Node
     def assistant(state: MessagesState):
         """Assistant node"""
-        return {"messages": [llm_with_tools.invoke(state["messages"])]}
+        return {"messages": [llm_with_tools.invoke([system_prompt] + state["messages"])]}
     
-
     builder = StateGraph(MessagesState)
+
+    # Nodes
     builder.add_node("assistant", assistant)
     builder.add_node("tools", ToolNode(tools))
-    builder.add_conditional_edges(
-        "assistant",
-        tools_condition,
-    )
+
+    # Edges
     builder.add_edge(START, "assistant")
+    builder.add_conditional_edges("assistant", tools_condition)
     builder.add_edge("tools", "assistant")
 
      # Compile graph

app.py

@@ -26,7 +26,15 @@ class BasicAgent:
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
-        return answer[14:]
+        print(f"Agent full response: {answer}")
+        
+        # Optionally trim the "Final Answer: " prefix only if it exists
+        if answer.startswith("Final Answer:"):
+            return answer[len("Final Answer:"):].strip()
+        else:
+            return answer.strip()
+
+