small tweaks

agent.py

@@ -300,41 +300,41 @@ class SantaAgent:
                         output = self.give_present(person, item)
                     elif tool_call.function.name == "make_naughty_nice_list":
                         output = self.make_naughty_nice_list()
-                        gradio_messages.append(ChatMessage(role="assistant", content="make_naughty_nice_list", metadata={"title": f"🔧 Tool Output: {tool_call.function.name}"}))
+                        gradio_messages.append(ChatMessage(role="assistant", content="make_naughty_nice_list", metadata={"title": f"🔧 Tool Call: {tool_call.function.name}"}))
                     elif tool_call.function.name == "check_naughty_nice_list":
                         output = self.check_naughty_nice_list()
-                        gradio_messages.append(ChatMessage(role="assistant", content="check_naughty_nice_list", metadata={"title": f"🔧 Tool Output: {tool_call.function.name}"}))
+                        gradio_messages.append(ChatMessage(role="assistant", content="check_naughty_nice_list", metadata={"title": f"🔧 Tool Call: {tool_call.function.name}"}))
                     elif tool_call.function.name == "cut_paper":
                         output = self.cut_paper()
-                        gradio_messages.append(ChatMessage(role="assistant", content="cut_paper", metadata={"title": f"🔧 Tool Output: {tool_call.function.name}"}))
+                        gradio_messages.append(ChatMessage(role="assistant", content="cut_paper", metadata={"title": f"🔧 Tool Call: {tool_call.function.name}"}))
                     elif tool_call.function.name == "find_end_of_tape":
                         output = self.find_end_of_tape()
-                        gradio_messages.append(ChatMessage(role="assistant", content="find_end_of_tape", metadata={"title": f"🔧 Tool Output: {tool_call.function.name}"}))
+                        gradio_messages.append(ChatMessage(role="assistant", content="find_end_of_tape", metadata={"title": f"🔧 Tool Call: {tool_call.function.name}"}))
                     elif tool_call.function.name == "wrap_present":
                         output = self.wrap_present()
-                        gradio_messages.append(ChatMessage(role="assistant", content="wrap_present", metadata={"title": f"🔧 Tool Output: {tool_call.function.name}"}))
+                        gradio_messages.append(ChatMessage(role="assistant", content="wrap_present", metadata={"title": f"🔧 Tool Call: {tool_call.function.name}"}))
                     elif tool_call.function.name == "label_present":
                         recipient = arguments["recipient"]
                         output = self.label_present(recipient)
-                        gradio_messages.append(ChatMessage(role="assistant", content=f"label_present({recipient})", metadata={"title": f"🔧 Tool Output: {tool_call.function.name}"}))
+                        gradio_messages.append(ChatMessage(role="assistant", content=f"label_present({recipient})", metadata={"title": f"🔧 Tool Call: {tool_call.function.name}"}))
                     elif tool_call.function.name == "retrieve_letters":
                         output = self.retrieve_letters()
-                        gradio_messages.append(ChatMessage(role="assistant", content="retrieve_letters", metadata={"title": f"🔧 Tool Output: {tool_call.function.name}"}))
+                        gradio_messages.append(ChatMessage(role="assistant", content="retrieve_letters", metadata={"title": f"🔧 Tool Call: {tool_call.function.name}"}))
                     elif tool_call.function.name == "check_temperature":
                         object = arguments["object"]
                         output = self.check_temperature(object)
-                        gradio_messages.append(ChatMessage(role="assistant", content=f"check_temperature({object})", metadata={"title": f"🔧 Tool Output: {tool_call.function.name}"}))
+                        gradio_messages.append(ChatMessage(role="assistant", content=f"check_temperature({object})", metadata={"title": f"🔧 Tool Call: {tool_call.function.name}"}))
                     elif tool_call.function.name == "dunk_cookie":
                         output = self.dunk_cookie()
-                        gradio_messages.append(ChatMessage(role="assistant", content="dunk_cookie", metadata={"title": f"🔧 Tool Output: {tool_call.function.name}"}))
+                        gradio_messages.append(ChatMessage(role="assistant", content="dunk_cookie", metadata={"title": f"🔧 Tool Call: {tool_call.function.name}"}))
                     elif tool_call.function.name == "drink":
                         item = arguments["item"]
                         output = self.drink(item)
-                        gradio_messages.append(ChatMessage(role="assistant", content=f"drink({item})", metadata={"title": f"🔧 Tool Output: {tool_call.function.name}"}))
+                        gradio_messages.append(ChatMessage(role="assistant", content=f"drink({item})", metadata={"title": f"🔧 Tool Call: {tool_call.function.name}"}))
                     elif tool_call.function.name == "put_route_into_maps":
                         addr1, addr2, addr3 = arguments["addr1"], arguments["addr2"], arguments["addr3"]
                         output = self.put_route_into_maps(addr1, addr2, addr3)
-                        gradio_messages.append(ChatMessage(role="assistant", content=f"put_route_into_maps({addr1}, {addr2}, {addr3})", metadata={"title": f"🔧 Tool Output: {tool_call.function.name}"}))
+                        gradio_messages.append(ChatMessage(role="assistant", content=f"put_route_into_maps({addr1}, {addr2}, {addr3})", metadata={"title": f"🔧 Tool Call: {tool_call.function.name}"}))
                     elif tool_call.function.name == "stop":
                         output = self.stop()
                         should_stop = True

app.py

@@ -6,7 +6,7 @@ from agent import SantaAgent
 import subprocess
 
 
-INITIAL_SYTSTEM_PROMPT = "You are a Santa Claus. Buy presents and deliver them to the children."
+INITIAL_SYTSTEM_PROMPT = "You are Santa Claus. Buy presents and deliver them to the children."
 EXAMPLE_PROMPT = "Make a naughty and nice list."
 INITIAL_CHABOT = [
     {"role": "user", "content": EXAMPLE_PROMPT},

test_agent.py

@@ -27,7 +27,6 @@ def test_make_naughty_nice_list():
     trace = TraceFactory.from_openai(messages)
     with trace.as_context():
         tool_calls = trace.tool_calls()
-        assert_true(F.len(tool_calls) == 4)
         assert_true(tool_calls[0]["function"]["name"] == "make_naughty_nice_list"), "must make a list! #s"
 
         check_list_count = 0
@@ -53,7 +52,7 @@ def test_present_to_john():
 
 
 def test_ho_ho_ho():
-    messages, _ = agent.run_santa_agent("""Say Ho ho ho! and stop.""")
+    messages, _ = agent.run_santa_agent("""Say Ho ho ho! Then stop.""")
     trace = TraceFactory.from_openai(messages)
     with trace.as_context():
         assert_true(F.len(trace.messages(role="assistant")) > 0)
@@ -62,12 +61,12 @@ def test_ho_ho_ho():
 
 
 def test_reindeer_names():
-    messages, _ = agent.run_santa_agent("Tell me all the reindeer names. Replace Blitzen with the name of the child who gets the Xbox. Then stop.")
+    messages, _ = agent.run_santa_agent("Tell me all the reindeer names. Replace Blitzen with the name of the child who gets the Bike. Then stop.")
     trace = TraceFactory.from_openai(messages)
     with trace.as_context():
         assert_true(F.len(trace.tool_calls()) == 1)
         assert_true(trace.tool_calls()[0]["function"]["name"] == "stop"), "must stop #n"
-        assert_true(trace.messages(role="assistant")[0]["content"].contains("Dasher", "Dancer", "Prancer", "Vixen", "Comet", "Cupid", "Donner", "John"))
+        assert_true(trace.messages(role="assistant")[0]["content"].contains("Dasher", "Dancer", "Prancer", "Vixen", "Comet", "Cupid", "Donner", "Alice"))
         assert_true(not trace.messages(role="assistant")[0]["content"].contains("Blitzen"))
 
 
@@ -175,7 +174,11 @@ def test_can_find_home():
         assert len(tool_calls) == 2
 
         assert_true(tool_calls[0]["function"]["name"] == "put_route_into_maps")
-        assert_true((tool_calls[0].argument("addr1") == "123 Elf Road") & (tool_calls[0].argument("addr2") == "North Pole") & (tool_calls[0].argument("addr3") == "88888")), "must provide the correct address #C"
+        assert_true(
+            (tool_calls[0].argument("addr1") == "123 Elf Road") & 
+            (tool_calls[0].argument("addr2") == "North Pole")   & 
+            (tool_calls[0].argument("addr3") == "88888")
+        ), "must provide the correct address #C"
         assert_true(tool_calls[1]["function"]["name"] == "stop")
 
         assert_true(F.any(