tool update

app.py

@@ -59,7 +59,7 @@ class BasicAgent:
         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. Question: {question}
+        in the list is a number or a string. Use the tools available to you to answer the question. Question: {question}
         For answering the question use this context: {context}, if no context is provided
         use your knowledge to answer the question."""
         # Your answer must be in the following format:
@@ -71,6 +71,7 @@ class BasicAgent:
         # Call the LLM
         messages = [HumanMessage(content=prompt)]
         response = (llm_with_tools.invoke(messages)).content
+        print("response: ", response)
         # cleaned_text = re.sub(r"<think>.*?</think>", "", response.content, flags=re.DOTALL)
 
         # json_start = response.find('{')
@@ -85,7 +86,7 @@ class BasicAgent:
         builder.add_node("tools", ToolNode(tools = tools))
 
         builder.add_edge(START, "agent")
-        builder.add_conditional_edges("agent", tools_condition)
+        builder.add_conditional_edges("agent", tools_condition, ["agent", END])
         builder.add_edge("agent", END)
         return builder.compile()