Merge pull request #132 from rgbkrk/fix-tool-type

fix the type on FunctionRegistry.tools
rgbkrk · Feb 27, 2024 · f700287 · f700287
2 parents 2b0e0cc + c4c3138
commit f700287
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/chatlab/registry.py b/chatlab/registry.py
@@ -62,6 +62,8 @@ class WhatTime(BaseModel):
 from openai.types.chat.completion_create_params import Function, FunctionCall
 from pydantic import BaseModel, create_model
 
+from openai.types.chat import ChatCompletionToolParam
+
 from .decorators import ChatlabMetadata
 
 
@@ -429,7 +431,7 @@ def api_manifest(self, function_call_option: FunctionCall = "auto") -> APIManife
         }
 
     @property
-    def tools(self):
+    def tools(self) -> Iterable[ChatCompletionToolParam]:
         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: