wip

Arize-ai · Feb 6, 2025 · ed3b1aa · ed3b1aa
1 parent ae8c048
commit ed3b1aa
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/phoenix/server/api/helpers/prompts/models.py b/src/phoenix/server/api/helpers/prompts/models.py
@@ -353,8 +353,8 @@ class AnthropicToolDefinition(PromptModel):
 
 def normalize_tools(
     schemas: list[dict[str, Any]],
-    tool_choice: Optional[Union[str, Mapping[str, Any]]],
     model_provider: str,
+    tool_choice: Optional[Union[str, Mapping[str, Any]]] = None,
 ) -> PromptToolsV1:
     tools: list[PromptFunctionToolV1]
     if model_provider.lower() == "openai":

diff --git a/src/phoenix/server/api/mutations/prompt_mutations.py b/src/phoenix/server/api/mutations/prompt_mutations.py
@@ -94,7 +94,11 @@ async def create_chat_prompt(
         )
         try:
             tools = (
-                normalize_tools(tool_definitions, tool_choice, input_prompt_version.model_provider)
+                normalize_tools(
+                    tool_definitions,
+                    input_prompt_version.model_provider,
+                    tool_choice,
+                )
                 if tool_definitions
                 else None
             )
@@ -160,7 +164,11 @@ async def create_chat_prompt_version(
         )
         try:
             tools = (
-                normalize_tools(tool_definitions, tool_choice, input_prompt_version.model_provider)
+                normalize_tools(
+                    tool_definitions,
+                    input_prompt_version.model_provider,
+                    tool_choice,
+                )
                 if tool_definitions
                 else None
             )