arize-phoenix 7.12.3__py3-none-any.whl → 8.0.1__py3-none-any.whl

phoenix/server/api/helpers/prompts/conversions/anthropic.py

@@ -0,0 +1,87 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Optional, Union
+
+from typing_extensions import assert_never
+
+if TYPE_CHECKING:
+    from anthropic.types import (
+        ToolChoiceAnyParam,
+        ToolChoiceAutoParam,
+        ToolChoiceParam,
+        ToolChoiceToolParam,
+    )
+
+    from phoenix.server.api.helpers.prompts.models import (
+        PromptToolChoiceOneOrMore,
+        PromptToolChoiceSpecificFunctionTool,
+        PromptToolChoiceZeroOrMore,
+    )
+
+
+class AnthropicToolChoiceConversion:
+    @staticmethod
+    def to_anthropic(
+        obj: Union[
+            PromptToolChoiceZeroOrMore,
+            PromptToolChoiceOneOrMore,
+            PromptToolChoiceSpecificFunctionTool,
+        ],
+        disable_parallel_tool_use: Optional[bool] = None,
+    ) -> ToolChoiceParam:
+        if obj.type == "zero_or_more":
+            choice_auto: ToolChoiceAutoParam = {"type": "auto"}
+            if disable_parallel_tool_use is not None:
+                choice_auto["disable_parallel_tool_use"] = disable_parallel_tool_use
+            return choice_auto
+        if obj.type == "one_or_more":
+            choice_any: ToolChoiceAnyParam = {"type": "any"}
+            if disable_parallel_tool_use is not None:
+                choice_any["disable_parallel_tool_use"] = disable_parallel_tool_use
+            return choice_any
+        if obj.type == "specific_function":
+            choice_tool: ToolChoiceToolParam = {"type": "tool", "name": obj.function_name}
+            if disable_parallel_tool_use is not None:
+                choice_tool["disable_parallel_tool_use"] = disable_parallel_tool_use
+            return choice_tool
+        assert_never(obj.type)
+
+    @staticmethod
+    def from_anthropic(
+        obj: ToolChoiceParam,
+    ) -> tuple[
+        Union[
+            PromptToolChoiceZeroOrMore,
+            PromptToolChoiceOneOrMore,
+            PromptToolChoiceSpecificFunctionTool,
+        ],
+        Optional[bool],
+    ]:
+        from phoenix.server.api.helpers.prompts.models import (
+            PromptToolChoiceOneOrMore,
+            PromptToolChoiceSpecificFunctionTool,
+            PromptToolChoiceZeroOrMore,
+        )
+
+        if obj["type"] == "auto":
+            disable_parallel_tool_use = (
+                obj["disable_parallel_tool_use"] if "disable_parallel_tool_use" in obj else None
+            )
+            choice_zero_or_more = PromptToolChoiceZeroOrMore(type="zero_or_more")
+            return choice_zero_or_more, disable_parallel_tool_use
+        if obj["type"] == "any":
+            disable_parallel_tool_use = (
+                obj["disable_parallel_tool_use"] if "disable_parallel_tool_use" in obj else None
+            )
+            choice_one_or_more = PromptToolChoiceOneOrMore(type="one_or_more")
+            return choice_one_or_more, disable_parallel_tool_use
+        if obj["type"] == "tool":
+            disable_parallel_tool_use = (
+                obj["disable_parallel_tool_use"] if "disable_parallel_tool_use" in obj else None
+            )
+            choice_function_tool = PromptToolChoiceSpecificFunctionTool(
+                type="specific_function",
+                function_name=obj["name"],
+            )
+            return choice_function_tool, disable_parallel_tool_use
+        assert_never(obj)

phoenix/server/api/helpers/prompts/conversions/openai.py

@@ -0,0 +1,78 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Union
+
+from typing_extensions import assert_never
+
+if TYPE_CHECKING:
+    from openai.types.chat import (
+        ChatCompletionNamedToolChoiceParam,
+        ChatCompletionToolChoiceOptionParam,
+    )
+    from openai.types.chat.chat_completion_named_tool_choice_param import Function
+
+    from phoenix.server.api.helpers.prompts.models import (
+        PromptToolChoiceNone,
+        PromptToolChoiceOneOrMore,
+        PromptToolChoiceSpecificFunctionTool,
+        PromptToolChoiceZeroOrMore,
+    )
+
+
+class OpenAIToolChoiceConversion:
+    @staticmethod
+    def to_openai(
+        obj: Union[
+            PromptToolChoiceNone,
+            PromptToolChoiceZeroOrMore,
+            PromptToolChoiceOneOrMore,
+            PromptToolChoiceSpecificFunctionTool,
+        ],
+    ) -> ChatCompletionToolChoiceOptionParam:
+        if obj.type == "none":
+            return "none"
+        if obj.type == "zero_or_more":
+            return "auto"
+        if obj.type == "one_or_more":
+            return "required"
+        if obj.type == "specific_function":
+            choice_tool: ChatCompletionNamedToolChoiceParam = {
+                "type": "function",
+                "function": {"name": obj.function_name},
+            }
+            return choice_tool
+        assert_never(obj)
+
+    @staticmethod
+    def from_openai(
+        obj: ChatCompletionToolChoiceOptionParam,
+    ) -> Union[
+        PromptToolChoiceNone,
+        PromptToolChoiceZeroOrMore,
+        PromptToolChoiceOneOrMore,
+        PromptToolChoiceSpecificFunctionTool,
+    ]:
+        from phoenix.server.api.helpers.prompts.models import (
+            PromptToolChoiceNone,
+            PromptToolChoiceOneOrMore,
+            PromptToolChoiceSpecificFunctionTool,
+            PromptToolChoiceZeroOrMore,
+        )
+
+        if obj == "none":
+            choice_none = PromptToolChoiceNone(type="none")
+            return choice_none
+        if obj == "auto":
+            choice_zero_or_more = PromptToolChoiceZeroOrMore(type="zero_or_more")
+            return choice_zero_or_more
+        if obj == "required":
+            choice_one_or_more = PromptToolChoiceOneOrMore(type="one_or_more")
+            return choice_one_or_more
+        if obj["type"] == "function":
+            function: Function = obj["function"]
+            choice_function_tool = PromptToolChoiceSpecificFunctionTool(
+                type="specific_function",
+                function_name=function["name"],
+            )
+            return choice_function_tool
+        assert_never(obj["type"])