mirascope 1.20.0__py3-none-any.whl → 1.21.0__py3-none-any.whl

mirascope/llm/llm_override.py

@@ -1,233 +0,0 @@
-"""Overrides the provider-specific call with the specified provider."""
-
-from __future__ import annotations
-
-from collections.abc import Callable
-from typing import TYPE_CHECKING, Any, Literal, ParamSpec, TypeVar, overload
-
-from ..core.base import CommonCallParams
-from ..core.base.types import LocalProvider, Provider
-from .llm_call import _call
-
-if TYPE_CHECKING:
-    from ..core.anthropic import AnthropicCallParams
-    from ..core.azure import AzureCallParams
-    from ..core.bedrock import BedrockCallParams
-    from ..core.cohere import CohereCallParams
-    from ..core.gemini import GeminiCallParams
-    from ..core.google import GoogleCallParams
-    from ..core.groq import GroqCallParams
-    from ..core.litellm import LiteLLMCallParams
-    from ..core.mistral import MistralCallParams
-    from ..core.openai import OpenAICallParams
-    from ..core.vertex import VertexCallParams
-else:
-    AnthropicCallParams = AzureCallParams = BedrockCallParams = CohereCallParams = (
-        GeminiCallParams
-    ) = GoogleCallParams = GroqCallParams = LiteLLMCallParams = MistralCallParams = (
-        OpenAICallParams
-    ) = VertexCallParams = None
-
-
-_P = ParamSpec("_P")
-_R = TypeVar("_R")
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Literal["anthropic"],
-    model: str,
-    call_params: CommonCallParams | AnthropicCallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Literal["azure"],
-    model: str,
-    call_params: CommonCallParams | AzureCallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Literal["bedrock"],
-    model: str,
-    call_params: CommonCallParams | BedrockCallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Literal["cohere"],
-    model: str,
-    call_params: CommonCallParams | CohereCallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Literal["gemini"],
-    model: str,
-    call_params: CommonCallParams | GeminiCallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Literal["google"],
-    model: str,
-    call_params: CommonCallParams | GoogleCallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Literal["groq"],
-    model: str,
-    call_params: CommonCallParams | GroqCallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Literal["mistral"],
-    model: str,
-    call_params: CommonCallParams | MistralCallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Literal["openai"],
-    model: str,
-    call_params: CommonCallParams | OpenAICallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Literal["litellm"],
-    model: str,
-    call_params: CommonCallParams | LiteLLMCallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Literal["vertex"],
-    model: str,
-    call_params: CommonCallParams | VertexCallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: None = None,
-    model: None = None,
-    call_params: CommonCallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-@overload
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Provider,
-    model: str,
-    call_params: CommonCallParams | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]: ...
-
-
-def override(
-    provider_agnostic_call: Callable[_P, _R],
-    *,
-    provider: Provider | LocalProvider | None = None,
-    model: str | None = None,
-    call_params: CommonCallParams
-    | AnthropicCallParams
-    | GeminiCallParams
-    | GoogleCallParams
-    | AzureCallParams
-    | BedrockCallParams
-    | CohereCallParams
-    | GroqCallParams
-    | MistralCallParams
-    | OpenAICallParams
-    | VertexCallParams
-    | None = None,
-    client: Any = None,  # noqa: ANN401
-) -> Callable[_P, _R]:
-    """Overrides the provider-specific call with the specified provider.
-
-    Args:
-        provider_agnostic_call: The provider-agnostic call to override.
-        provider: The provider to override with.
-        model: The model to override with.
-        call_params: The call params to override with.
-        client: The client to override with.
-
-    Returns:
-        The overridden function.
-    """
-    if (provider and not model) or (model and not provider):
-        raise ValueError(
-            "Provider and model must both be overridden if either is overridden."
-        )
-
-    original_provider = provider_agnostic_call._original_provider  # pyright: ignore [reportFunctionMemberAccess]
-    original_args = provider_agnostic_call._original_args  # pyright: ignore [reportFunctionMemberAccess]
-
-    # Note: if switching providers, we will always use `client` since `original_client`
-    # would be from a different provider and fail.
-    if provider and provider == original_provider:
-        client = client or original_args["client"]
-
-    return _call(  # pyright: ignore [reportReturnType]
-        provider=provider or original_provider,
-        model=model or original_args["model"],
-        stream=original_args["stream"],
-        tools=original_args["tools"],
-        response_model=original_args["response_model"],
-        output_parser=original_args["output_parser"],
-        json_mode=original_args["json_mode"],
-        client=client,
-        call_params=call_params or original_args["call_params"],
-    )(provider_agnostic_call._original_fn)  # pyright: ignore [reportFunctionMemberAccess]