fixing freeplay logging

api/scripts/freeplay_playground.py

@@ -36,9 +36,7 @@ variables = {
 # ]
 history = [
     {'role': 'user', 'content': 'tell me about some players in everglade fc'},
-    # {'role': 'assistant', 'content': ''},
-    {'role': 'tool_call', 'content': '{"number": 23, "name": "Brian Davis", "age": 27, "nationality": "USA", "shirt_number"...c image full of energy and anticipation."}'},
-    # {'role': 'tool', 'content': '{"number": 10, "name": "Matthew Martin", "age": 24, "nationality": "USA", "shirt_numb...he dramatic flair he brings to the game."}'},
+    {'role': 'tool', 'content': '{"number": 23, "name": "Brian Davis", "age": 27, "nationality": "USA", "shirt_number"...c image full of energy and anticipation."}', 'tool_call_id': 'abc123'},
     {'role': 'assistant', 'content': 'Everglade FC is bursting with talent! Here are some standout players to watch:\n\n1. **...rglade FC brings to the field! Go, team! 🌟'},
 ]
 

api/utils/freeplay_helpers.py

@@ -5,10 +5,14 @@ from functools import lru_cache
 from typing import Union, Optional, List
 from freeplay import Freeplay, RecordPayload, ResponseInfo, CallInfo
 from freeplay.resources.prompts import FormattedPrompt
-from langchain_core.messages import BaseMessage
+from langchain_core.messages import BaseMessage, ToolMessage
 
 FREEPLAY_PROJECT_ID = os.getenv("FREEPLAY_PROJECT_ID")
-
+_role_map = {
+    'human': 'user',
+    'ai': 'assistant',
+    'tool': 'tool',
+}
 
 # @lru_cache(maxsize=1)
 def _get_fp_client():
@@ -102,9 +106,19 @@ class FreeplayClient:
 
         # convert messages to Freeplay format
         if state['messages'] and isinstance(state['messages'][0], dict):
+            # if it's a dict leave it alone and just send it on
             all_messages = state['messages']
         else:
-            all_messages = [{'role': m.type, 'content': m.content} for m in state['messages'] if m.content]
+            # otherwise assume it's langchain messages and we need to parse them for freeplay
+            # all_messages = [{'role': _role_map[m.type], 'content': m.content} for m in state['messages'] if m.content]
+            all_messages = [
+                {
+                    'role': _role_map[m.type],
+                    'content': m.content,
+                    **({'tool_call_id': m.tool_call_id} if isinstance(m, ToolMessage) else {})
+                }
+                for m in state['messages'] if m.content
+            ]
 
         # fix session if we landed here and it's missing
         if not self.session:

api/workflows/base.py

@@ -124,8 +124,7 @@ async def should_continue(state: AgentState,
             messages=messages,
         )
         # inform freeplay of final response
-        # TODO: wait for feedback from freeplay about tool messages
-        # freeplay_client.record_session(state)
+        freeplay_client.record_session(state)
         # trigger on_workflow_end callback
         if hasattr(handler, 'on_workflow_end'):
             await handler.on_workflow_end(state)