wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #4248 from omChauhanDev/add-openai-custom-tools-support

Browse Source 
Add custom_tools support for OpenAI adapters
This commit is contained in:
kompfner
2026-04-10 10:27:28 -04:00
committed by GitHub
parent 094d36904c ef4dcca4f1
commit d07eebff20
6 changed files with 157 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

130
tests/test_function_calling_adapters.py
View File

@@ -7,6 +7,8 @@
import unittest
from openai.types.chat import ChatCompletionToolParam
from openai.types.responses.function_tool_param import FunctionToolParam
from openai.types.responses.tool_search_tool_param import ToolSearchToolParam
from pipecat.adapters.schemas.function_schema import FunctionSchema
from pipecat.adapters.schemas.tools_schema import AdapterType, ToolsSchema
@@ -16,6 +18,7 @@ from pipecat.adapters.services.gemini_adapter import GeminiLLMAdapter
from pipecat.adapters.services.inworld_realtime_adapter import InworldRealtimeLLMAdapter
from pipecat.adapters.services.open_ai_adapter import OpenAILLMAdapter
from pipecat.adapters.services.open_ai_realtime_adapter import OpenAIRealtimeLLMAdapter
from pipecat.adapters.services.open_ai_responses_adapter import OpenAIResponsesLLMAdapter
class TestFunctionAdapters(unittest.TestCase):
@@ -203,6 +206,133 @@ class TestFunctionAdapters(unittest.TestCase):
tools_def.custom_tools = {AdapterType.GEMINI: [search_tool]}
assert GeminiLLMAdapter().to_provider_tools_format(tools_def) == expected
def test_openai_adapter_with_custom_tools(self):
"""Test OpenAI adapter appends custom tools."""
tool_search = {"type": "tool_search"}
expected = [
ChatCompletionToolParam(
type="function",
function={
"name": "get_weather",
"description": "Get the weather in a given location",
"parameters": {
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "The city, e.g. San Francisco",
},
"format": {
"type": "string",
"enum": ["celsius", "fahrenheit"],
"description": "The temperature unit to use.",
},
},
"required": ["location", "format"],
},
},
),
tool_search,
]
tools_def = self.tools_def
tools_def.custom_tools = {AdapterType.OPENAI: [tool_search]}
assert OpenAILLMAdapter().to_provider_tools_format(tools_def) == expected
def test_openai_responses_adapter_with_custom_tools(self):
"""Test OpenAI Responses adapter appends custom tools."""
tool_search = {"type": "tool_search"}
expected = [
FunctionToolParam(
type="function",
name="get_weather",
description="Get the weather in a given location",
parameters={
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "The city, e.g. San Francisco",
},
"format": {
"type": "string",
"enum": ["celsius", "fahrenheit"],
"description": "The temperature unit to use.",
},
},
"required": ["location", "format"],
},
strict=None,
),
ToolSearchToolParam(type="tool_search"),
]
tools_def = self.tools_def
tools_def.custom_tools = {AdapterType.OPENAI: [tool_search]}
assert OpenAIResponsesLLMAdapter().to_provider_tools_format(tools_def) == expected
def test_openai_responses_adapter(self):
"""Test OpenAI Responses adapter format transformation."""
expected = [
{
"type": "function",
"name": "get_weather",
"description": "Get the weather in a given location",
"parameters": {
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "The city, e.g. San Francisco",
},
"format": {
"type": "string",
"enum": ["celsius", "fahrenheit"],
"description": "The temperature unit to use.",
},
},
"required": ["location", "format"],
},
"strict": None,
}
]
assert OpenAIResponsesLLMAdapter().to_provider_tools_format(self.tools_def) == expected
def test_openai_realtime_adapter_with_custom_tools(self):
"""Test OpenAI Realtime adapter appends custom tools."""
tool_search = {"type": "tool_search"}
expected = [
{
"type": "function",
"name": "get_weather",
"description": "Get the weather in a given location",
"parameters": {
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "The city, e.g. San Francisco",
},
"format": {
"type": "string",
"enum": ["celsius", "fahrenheit"],
"description": "The temperature unit to use.",
},
},
"required": ["location", "format"],
},
},
tool_search,
]
tools_def = self.tools_def
tools_def.custom_tools = {AdapterType.OPENAI: [tool_search]}
assert OpenAIRealtimeLLMAdapter().to_provider_tools_format(tools_def) == expected
def test_openai_adapter_ignores_other_adapter_custom_tools(self):
"""Test that OpenAI adapter ignores custom tools for other adapters."""
expected = OpenAILLMAdapter().to_provider_tools_format(self.tools_def)
tools_def = self.tools_def
tools_def.custom_tools = {AdapterType.GEMINI: [{"google_search": {}}]}
assert OpenAILLMAdapter().to_provider_tools_format(tools_def) == expected
def test_bedrock_adapter(self):
"""Test AWS Bedrock adapter format transformation."""
expected = [