Merge pull request #122 from rgbkrk/parallel-tool-calling

provide tools for multiple function calling

CHANGELOG.md

@@ -5,6 +5,10 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.3.0]
+
+- Support tool call format from `FunctionRegistry`. Enables parallel function calling (note: not in `Chat` yet). https://github.com/rgbkrk/chatlab/pull/122
+
 ## [1.2.1]
 
 - Drop Noteable builtin

chatlab/__init__.py

@@ -30,6 +30,7 @@
     narrate,
     system,
     user,
+    tool_result
 )
 from .registry import FunctionRegistry
 from .views.markdown import Markdown
@@ -85,6 +86,7 @@ def __init__(self, *args, **kwargs):
     "assistant",
     "assistant_function_call",
     "function_result",
+    "tool_result",
     "models",
     "Session",
     "Chat",

chatlab/messaging.py

@@ -12,7 +12,7 @@
 
 from typing import Optional
 
-from openai.types.chat import ChatCompletionMessageParam
+from openai.types.chat import ChatCompletionMessageParam, ChatCompletionToolMessageParam
 
 
 def assistant(content: str) -> ChatCompletionMessageParam:
@@ -100,6 +100,22 @@ def function_result(name: str, content: str) -> ChatCompletionMessageParam:
     }
 
 
+def tool_result(tool_call_id: str, content: str) -> ChatCompletionToolMessageParam:
+    """Create a tool result message.
+
+    Args:
+        tool_call_id: The ID of the tool call.
+        content: The content of the message.
+
+    Returns:
+        A dictionary representing a tool result message.
+    """
+    return {
+        "role": "tool",
+        "content": content,
+        "tool_call_id": tool_call_id,
+    }
+
 # Aliases
 narrate = system
 human = user

chatlab/registry.py

@@ -428,6 +428,10 @@ def api_manifest(self, function_call_option: FunctionCall = "auto") -> APIManife
             "function_call": function_call_option,
         }
 
+    @property
+    def tools(self):
+        return [{"type": "function", "function": adapt_function_definition(f)} for f in self.__schemas.values()]
+
     async def call(self, name: str, arguments: Optional[str] = None) -> Any:
         """Call a function by name with the given parameters."""
         if name is None: