Remove intermediate message, return function call result as well

Author: viktorvirtuals

examples/game/chat_agent.py

@@ -68,18 +68,16 @@ def check_crypto_price(currency: str):
 
     user_message = input("Enter a message: ")
 
-    response, function_report_response = chat.next(user_message)
+    response = chat.next(user_message)
+
+    if response.function_call:
+        print(f"Function call: {response.function_call.fn_name}")
 
     if response.message:
         print(f"Response: {response.message}")
 
-    if function_report_response:
-        print(f"Response: {function_report_response}")
-
     if response.is_finished:
         chat_continue = False
         break
 
-
-
 print("Chat ended")

src/game_sdk/game/README.md

@@ -141,41 +141,22 @@ chat = chat_agent.create_chat(
     partner_id="user123",
     partner_name="User Name",
     action_space=[list_of_functions],  # Optional
-    message_handler=lambda message: print(f"Agent: {message.message}")  # Required callback for handling agent messages
     get_state_fn=lambda: {...}  # Optional, allows to push state of the environment to the agent
 )
 
 # Run conversation
-while chat.next("User message"):
-    # Continue conversation
-    pass
+chat_continue = True
+while chat_continue:
+    user_message = input("Enter a message: ")
+    response = chat.next(user_message)
+    ...
+    if response.is_finished:
+        chat_continue = False
 
 # End chat
 chat.end("Optional ending message")
 ```
 
-### Message Handler
-
-The `message_handler` is a required callback function that processes messages from the agent. It receives an `AgentMessage` object containing:
-- `message`: The text response from the agent
-- `chat_id`: The ID of the current chat session
-
-Example message handlers:
-
-```python
-# Simple print handler
-def print_handler(message):
-    print(f"Agent: {message.message}")
-
-# Custom logging handler
-def log_handler(message):
-    logger.info(f"Chat {message.chat_id}: {message.message}")
-    
-# UI update handler
-def ui_handler(message):
-    update_chat_window(message.message)
-```
-
 ### Chat Termination
 
 The chat can be terminated in two ways:

src/game_sdk/game/chat_agent.py

@@ -1,5 +1,7 @@
 from typing import Any, Callable, Dict, List, Optional, Tuple
 from game_sdk.game.custom_types import (
+    ChatResponse,
+    FunctionCallResponse,
     FunctionResult,
     GameChatResponse,
     Function,
@@ -23,7 +25,7 @@ def __init__(
         )
         self.get_state_fn = get_state_fn
 
-    def next(self, message: str) -> Tuple[GameChatResponse, Optional[str]]:
+    def next(self, message: str) -> ChatResponse:
 
         convo_response = self._update_conversation(message)
 
@@ -46,12 +48,21 @@ def next(self, message: str) -> Tuple[GameChatResponse, Optional[str]]:
                     "args": convo_response.function_call.args,
                 }
             )
-            function_report_response = self._report_function_result(result)
-            
+            response_message = self._report_function_result(result)
+            function_call_response = FunctionCallResponse(
+                fn_name=fn_name,
+                fn_args=convo_response.function_call.args,
+                result=result,
+            )
         else:
-            function_report_response = None
-        
-        return convo_response, function_report_response
+            response_message = convo_response.message or ""
+            function_call_response = None
+
+        return ChatResponse(
+            message=response_message,
+            is_finished=convo_response.is_finished,
+            function_call=function_call_response,
+        )
 
     def end(self, message: Optional[str] = None):
         self.client.end_chat(
@@ -96,10 +107,10 @@ def __init__(
         self,
         api_key: str,
         prompt: str,
-    ):  
+    ):
         self._api_key = api_key
         self.prompt = prompt
-        
+
         if api_key.startswith("apt-"):
             self.client = GAMEClientV2(api_key)
         else:
@@ -112,7 +123,7 @@ def create_chat(
         action_space: Optional[List[Function]] = None,
         get_state_fn: Optional[Callable[[], Dict[str, Any]]] = None,
     ) -> Chat:
-        
+
         chat_id = self.client.create_chat(
             {
                 "prompt": self.prompt,
@@ -121,9 +132,4 @@ def create_chat(
             },
         )
 
-        return Chat(
-            chat_id,
-            self.client,
-            action_space,
-            get_state_fn
-        )
+        return Chat(chat_id, self.client, action_space, get_state_fn)

src/game_sdk/game/custom_types.py

@@ -253,3 +253,13 @@ class GameChatResponse(BaseModel):
 class AgentMessage(BaseModel):
     message: str
     chat_id: str
+
+class FunctionCallResponse(BaseModel):
+    fn_name: str
+    fn_args: Dict[str, Any]
+    result: FunctionResult
+
+class ChatResponse(BaseModel):
+    message: str
+    is_finished: bool
+    function_call: Optional[FunctionCallResponse] = None