Commit 52c48df8

Robert Craigie <robert@craigie.dev>
2025-08-09 01:14:16
fix(types): re-export more tool call types
1 parent 7aa3c78
src/openai/types/chat/chat_completion_message_tool_call.py
@@ -5,9 +5,9 @@ from typing_extensions import Annotated, TypeAlias
 
 from ..._utils import PropertyInfo
 from .chat_completion_message_custom_tool_call import ChatCompletionMessageCustomToolCall
-from .chat_completion_message_function_tool_call import ChatCompletionMessageFunctionToolCall
+from .chat_completion_message_function_tool_call import Function as Function, ChatCompletionMessageFunctionToolCall
 
-__all__ = ["ChatCompletionMessageToolCall"]
+__all__ = ["ChatCompletionMessageToolCall", "Function"]
 
 ChatCompletionMessageToolCall: TypeAlias = Annotated[
     Union[ChatCompletionMessageFunctionToolCall, ChatCompletionMessageCustomToolCall],
src/openai/types/chat/chat_completion_tool_param.py
@@ -4,8 +4,11 @@ from __future__ import annotations
 
 from typing_extensions import TypeAlias
 
-from .chat_completion_function_tool_param import ChatCompletionFunctionToolParam
+from .chat_completion_function_tool_param import (
+    FunctionDefinition as FunctionDefinition,
+    ChatCompletionFunctionToolParam,
+)
 
-__all__ = ["ChatCompletionToolParam"]
+__all__ = ["ChatCompletionToolParam", "FunctionDefinition"]
 
 ChatCompletionToolParam: TypeAlias = ChatCompletionFunctionToolParam