gllm-inference-binary 0.5.40__cp311-cp311-win_amd64.whl → 0.5.66__cp311-cp311-win_amd64.whl

gllm_inference/builder/_build_invoker.pyi

@@ -0,0 +1,28 @@
+from _typeshed import Incomplete
+from gllm_inference.em_invoker.em_invoker import BaseEMInvoker as BaseEMInvoker
+from gllm_inference.lm_invoker.lm_invoker import BaseLMInvoker as BaseLMInvoker
+from gllm_inference.schema.model_id import ModelId as ModelId, ModelProvider as ModelProvider, PROVIDERS_OPTIONAL_PATH as PROVIDERS_OPTIONAL_PATH
+
+logger: Incomplete
+
+class Key:
+    """Defines valid keys in the config."""
+    ACCESS_KEY_ID: str
+    API_KEY: str
+    AZURE_DEPLOYMENT: str
+    AZURE_ENDPOINT: str
+    BASE_URL: str
+    CONFIG: str
+    CUSTOM_HOST: str
+    CREDENTIALS_PATH: str
+    MODEL_ID: str
+    MODEL_KWARGS: str
+    MODEL_NAME: str
+    MODEL_CLASS_PATH: str
+    PORTKEY_API_KEY: str
+    PROVIDER: str
+    SECRET_ACCESS_KEY: str
+
+PROVIDERS_REQUIRE_BASE_URL: Incomplete
+MODEL_NAME_KEY_MAP: Incomplete
+DEFAULT_MODEL_NAME_KEY: Incomplete

gllm_inference/builder/build_em_invoker.pyi

@@ -1,24 +1,9 @@
-from _typeshed import Incomplete
-from gllm_inference.em_invoker import AzureOpenAIEMInvoker as AzureOpenAIEMInvoker, BedrockEMInvoker as BedrockEMInvoker, GoogleEMInvoker as GoogleEMInvoker, LangChainEMInvoker as LangChainEMInvoker, OpenAICompatibleEMInvoker as OpenAICompatibleEMInvoker, OpenAIEMInvoker as OpenAIEMInvoker, TwelveLabsEMInvoker as TwelveLabsEMInvoker, VoyageEMInvoker as VoyageEMInvoker
+from gllm_inference.em_invoker import AzureOpenAIEMInvoker as AzureOpenAIEMInvoker, BedrockEMInvoker as BedrockEMInvoker, CohereEMInvoker as CohereEMInvoker, GoogleEMInvoker as GoogleEMInvoker, JinaEMInvoker as JinaEMInvoker, LangChainEMInvoker as LangChainEMInvoker, OpenAICompatibleEMInvoker as OpenAICompatibleEMInvoker, OpenAIEMInvoker as OpenAIEMInvoker, TwelveLabsEMInvoker as TwelveLabsEMInvoker, VoyageEMInvoker as VoyageEMInvoker
 from gllm_inference.em_invoker.em_invoker import BaseEMInvoker as BaseEMInvoker
 from gllm_inference.schema.model_id import ModelId as ModelId, ModelProvider as ModelProvider
 from typing import Any
 
 PROVIDER_TO_EM_INVOKER_MAP: dict[str, type[BaseEMInvoker]]
-logger: Incomplete
-
-class Key:
-    """Defines valid keys in the config."""
-    ACCESS_KEY_ID: str
-    API_KEY: str
-    AZURE_DEPLOYMENT: str
-    AZURE_ENDPOINT: str
-    BASE_URL: str
-    CREDENTIALS_PATH: str
-    MODEL_KWARGS: str
-    MODEL_NAME: str
-    MODEL_CLASS_PATH: str
-    SECRET_ACCESS_KEY: str
 
 def build_em_invoker(model_id: str | ModelId, credentials: str | dict[str, Any] | None = None, config: dict[str, Any] | None = None) -> BaseEMInvoker:
     '''Build an embedding model invoker based on the provided configurations.
@@ -71,6 +56,16 @@ def build_em_invoker(model_id: str | ModelId, credentials: str | dict[str, Any]
         ```
         Providing credentials through environment variable is not supported for Google Vertex AI.
 
+        # Using Jina
+        ```python
+        em_invoker = build_em_invoker(
+            model_id="jina/jina-embeddings-v2-large",
+            credentials="jina-api-key"
+        )
+        ```
+        The credentials can also be provided through the `JINA_API_KEY` environment variable. For the list of supported
+        models, please refer to the following page: https://jina.ai/models
+
         # Using OpenAI
         ```python
         em_invoker = build_em_invoker(
@@ -128,6 +123,7 @@ def build_em_invoker(model_id: str | ModelId, credentials: str | dict[str, Any]
         variables credentials, please refer to the following page:
         https://python.langchain.com/docs/integrations/text_embedding/
 
+
     Security warning:
         Please provide the EM invoker credentials ONLY to the `credentials` parameter. Do not put any kind of
         credentials in the `config` parameter as the content of the `config` parameter will be logged.

gllm_inference/builder/build_lm_invoker.pyi

@@ -1,25 +1,9 @@
-from _typeshed import Incomplete
-from gllm_inference.lm_invoker import AnthropicLMInvoker as AnthropicLMInvoker, AzureOpenAILMInvoker as AzureOpenAILMInvoker, BedrockLMInvoker as BedrockLMInvoker, DatasaurLMInvoker as DatasaurLMInvoker, GoogleLMInvoker as GoogleLMInvoker, LangChainLMInvoker as LangChainLMInvoker, LiteLLMLMInvoker as LiteLLMLMInvoker, OpenAIChatCompletionsLMInvoker as OpenAIChatCompletionsLMInvoker, OpenAICompatibleLMInvoker as OpenAICompatibleLMInvoker, OpenAILMInvoker as OpenAILMInvoker, XAILMInvoker as XAILMInvoker
+from gllm_inference.lm_invoker import AnthropicLMInvoker as AnthropicLMInvoker, AzureOpenAILMInvoker as AzureOpenAILMInvoker, BedrockLMInvoker as BedrockLMInvoker, DatasaurLMInvoker as DatasaurLMInvoker, GoogleLMInvoker as GoogleLMInvoker, LangChainLMInvoker as LangChainLMInvoker, LiteLLMLMInvoker as LiteLLMLMInvoker, OpenAIChatCompletionsLMInvoker as OpenAIChatCompletionsLMInvoker, OpenAICompatibleLMInvoker as OpenAICompatibleLMInvoker, OpenAILMInvoker as OpenAILMInvoker, PortkeyLMInvoker as PortkeyLMInvoker, SeaLionLMInvoker as SeaLionLMInvoker, XAILMInvoker as XAILMInvoker
 from gllm_inference.lm_invoker.lm_invoker import BaseLMInvoker as BaseLMInvoker
 from gllm_inference.schema.model_id import ModelId as ModelId, ModelProvider as ModelProvider
 from typing import Any
 
 PROVIDER_TO_LM_INVOKER_MAP: dict[str, type[BaseLMInvoker]]
-logger: Incomplete
-
-class Key:
-    """Defines valid keys in the config."""
-    ACCESS_KEY_ID: str
-    API_KEY: str
-    AZURE_DEPLOYMENT: str
-    AZURE_ENDPOINT: str
-    BASE_URL: str
-    CREDENTIALS_PATH: str
-    MODEL_ID: str
-    MODEL_KWARGS: str
-    MODEL_NAME: str
-    MODEL_CLASS_PATH: str
-    SECRET_ACCESS_KEY: str
 
 def build_lm_invoker(model_id: str | ModelId, credentials: str | dict[str, Any] | None = None, config: dict[str, Any] | None = None) -> BaseLMInvoker:
     '''Build a language model invoker based on the provided configurations.
@@ -135,6 +119,15 @@ def build_lm_invoker(model_id: str | ModelId, credentials: str | dict[str, Any]
         ```
         The credentials can also be provided through the `AZURE_OPENAI_API_KEY` environment variable.
 
+        # Using SEA-LION
+        ```python
+        lm_invoker = build_lm_invoker(
+            model_id="sea-lion/aisingapore/Qwen-SEA-LION-v4-32B-IT",
+            credentials="sk-..."
+        )
+        ```
+        The credentials can also be provided through the `SEA_LION_API_KEY` environment variable.
+
         # Using LangChain
         ```python
         lm_invoker = build_lm_invoker(
@@ -157,6 +150,61 @@ def build_lm_invoker(model_id: str | ModelId, credentials: str | dict[str, Any]
         For the list of supported providers, please refer to the following page:
         https://docs.litellm.ai/docs/providers/
 
+        # Using Portkey
+        Portkey supports multiple authentication methods with strict precedence order.
+        Authentication methods are mutually exclusive and cannot be combined.
+
+        ## Config ID Authentication (Highest Precedence)
+        ```python
+        lm_invoker = build_lm_invoker(
+            model_id="portkey/any-model",
+            credentials="portkey-api-key",
+            config={"config": "pc-openai-4f6905"}
+        )
+        ```
+
+        ## Model Catalog Authentication (Combined Format)
+        ```python
+        lm_invoker = build_lm_invoker(
+            model_id="portkey/@openai-custom/gpt-4o",
+            credentials="portkey-api-key"
+        )
+        ```
+
+        ## Model Catalog Authentication (Separate Parameters)
+        ```python
+        lm_invoker = build_lm_invoker(
+            model_id="portkey/gpt-4o",
+            credentials="portkey-api-key",
+            config={"provider": "@openai-custom"}
+        )
+        ```
+
+        ## Direct Provider Authentication
+        ```python
+        lm_invoker = build_lm_invoker(
+            model_id="portkey/gpt-4o",
+            credentials={
+                "portkey_api_key": "portkey-api-key",
+                "api_key": "sk-...",  # Provider\'s API key
+                "provider": "openai"  # Direct provider (no \'@\' prefix)
+            }
+        )
+        ```
+
+        ## Custom Host Override
+        ```python
+        lm_invoker = build_lm_invoker(
+            model_id="portkey/@custom-provider/gpt-4o",
+            credentials="portkey-api-key",
+            config={"custom_host": "https://your-custom-endpoint.com"}
+        )
+        ```
+
+        The Portkey API key can also be provided through the `PORTKEY_API_KEY` environment variable.
+        For more details on authentication methods, please refer to:
+        https://portkey.ai/docs/product/ai-gateway/universal-api
+
         # Using xAI
         ```python
         lm_invoker = build_lm_invoker(

gllm_inference/constants.pyi

@@ -2,10 +2,11 @@ from _typeshed import Incomplete
 
 AZURE_OPENAI_URL_SUFFIX: str
 DOCUMENT_MIME_TYPES: Incomplete
+EMBEDDING_ENDPOINT: str
 GOOGLE_SCOPES: Incomplete
 GRPC_ENABLE_RETRIES_KEY: str
-INVOKER_PROPAGATED_MAX_RETRIES: int
 INVOKER_DEFAULT_TIMEOUT: float
-HEX_REPR_LENGTH: int
+INVOKER_PROPAGATED_MAX_RETRIES: int
+JINA_DEFAULT_URL: str
 OPENAI_DEFAULT_URL: str
 SECONDS_TO_MILLISECONDS: int

gllm_inference/em_invoker/__init__.pyi

@@ -1,10 +1,12 @@
 from gllm_inference.em_invoker.azure_openai_em_invoker import AzureOpenAIEMInvoker as AzureOpenAIEMInvoker
 from gllm_inference.em_invoker.bedrock_em_invoker import BedrockEMInvoker as BedrockEMInvoker
+from gllm_inference.em_invoker.cohere_em_invoker import CohereEMInvoker as CohereEMInvoker
 from gllm_inference.em_invoker.google_em_invoker import GoogleEMInvoker as GoogleEMInvoker
+from gllm_inference.em_invoker.jina_em_invoker import JinaEMInvoker as JinaEMInvoker
 from gllm_inference.em_invoker.langchain_em_invoker import LangChainEMInvoker as LangChainEMInvoker
 from gllm_inference.em_invoker.openai_compatible_em_invoker import OpenAICompatibleEMInvoker as OpenAICompatibleEMInvoker
 from gllm_inference.em_invoker.openai_em_invoker import OpenAIEMInvoker as OpenAIEMInvoker
 from gllm_inference.em_invoker.twelevelabs_em_invoker import TwelveLabsEMInvoker as TwelveLabsEMInvoker
 from gllm_inference.em_invoker.voyage_em_invoker import VoyageEMInvoker as VoyageEMInvoker
 
-__all__ = ['AzureOpenAIEMInvoker', 'BedrockEMInvoker', 'GoogleEMInvoker', 'LangChainEMInvoker', 'OpenAIEMInvoker', 'OpenAICompatibleEMInvoker', 'TwelveLabsEMInvoker', 'VoyageEMInvoker']
+__all__ = ['AzureOpenAIEMInvoker', 'BedrockEMInvoker', 'CohereEMInvoker', 'GoogleEMInvoker', 'JinaEMInvoker', 'LangChainEMInvoker', 'OpenAIEMInvoker', 'OpenAICompatibleEMInvoker', 'TwelveLabsEMInvoker', 'VoyageEMInvoker']

gllm_inference/em_invoker/bedrock_em_invoker.pyi

@@ -5,12 +5,13 @@ from gllm_inference.em_invoker.em_invoker import BaseEMInvoker as BaseEMInvoker
 from gllm_inference.em_invoker.schema.bedrock import InputType as InputType, Key as Key, OutputType as OutputType
 from gllm_inference.exceptions import BaseInvokerError as BaseInvokerError, convert_http_status_to_base_invoker_error as convert_http_status_to_base_invoker_error
 from gllm_inference.exceptions.provider_error_map import BEDROCK_ERROR_MAPPING as BEDROCK_ERROR_MAPPING
-from gllm_inference.schema import ModelId as ModelId, ModelProvider as ModelProvider, TruncationConfig as TruncationConfig, Vector as Vector
+from gllm_inference.schema import Attachment as Attachment, AttachmentType as AttachmentType, EMContent as EMContent, ModelId as ModelId, ModelProvider as ModelProvider, TruncationConfig as TruncationConfig, Vector as Vector
 from typing import Any
 
 class ModelType(StrEnum):
     """Defines the type of the Bedrock embedding model."""
     COHERE = 'cohere'
+    MARENGO = 'marengo'
     TITAN = 'titan'
 
 SUPPORTED_ATTACHMENTS: Incomplete
@@ -29,12 +30,14 @@ class BedrockEMInvoker(BaseEMInvoker):
         truncation_config (TruncationConfig | None): The truncation configuration for the embedding model.
 
     Input types:
-        The `BedrockEMInvoker` only supports text inputs.
+        The `BedrockEMInvoker` supports:
+        1. Text inputs for Cohere, Titan, and Marengo models
+        2. Image inputs for Marengo models through Attachment objects
 
     Output format:
         The `BedrockEMInvoker` can embed either:
         1. A single content.
-           1. A single content is a single text.
+           1. A single content is a single text or single image (image only supported for Marengo).
            2. The output will be a `Vector`, representing the embedding of the content.
 
            # Example 1: Embedding a text content.
@@ -43,10 +46,19 @@ class BedrockEMInvoker(BaseEMInvoker):
            result = await em_invoker.invoke(text)
            ```
 
+           # Example 2: Embedding an image with Marengo.
+           ```python
+           em_invoker = BedrockEMInvoker(
+               model_name="us.twelvelabs.marengo-2.7"
+           )
+           image = Attachment.from_path("path/to/local/image.png")
+           result = await em_invoker.invoke(image)
+           ```
+
            The above examples will return a `Vector` with a size of (embedding_size,).
 
         2. A list of contents.
-           1. A list of contents is a list of texts.
+           1. A list of contents is a list of texts or images (images only supported for Marengo).
            2. The output will be a `list[Vector]`, where each element is a `Vector` representing the
               embedding of each single content.
 

gllm_inference/em_invoker/cohere_em_invoker.pyi

@@ -0,0 +1,127 @@
+from _typeshed import Incomplete
+from gllm_core.utils import RetryConfig as RetryConfig
+from gllm_inference.em_invoker.em_invoker import BaseEMInvoker as BaseEMInvoker
+from gllm_inference.em_invoker.schema.cohere import CohereInputType as CohereInputType, Key as Key
+from gllm_inference.schema import Attachment as Attachment, AttachmentType as AttachmentType, EMContent as EMContent, ModelId as ModelId, ModelProvider as ModelProvider, TruncationConfig as TruncationConfig, Vector as Vector
+from typing import Any
+
+SUPPORTED_ATTACHMENTS: Incomplete
+MULTIMODAL_MODEL_VERSION: Incomplete
+
+class CohereEMInvoker(BaseEMInvoker):
+    '''An embedding model invoker to interact with Cohere embedding models.
+
+    Attributes:
+        model_id (str): The model ID of the embedding model.
+        model_provider (str): The provider of the embedding model (Cohere).
+        model_name (str): The name of the Cohere embedding model.
+        client (AsyncClient): The asynchronous client for the Cohere API.
+        default_hyperparameters (dict[str, Any]): Default hyperparameters for invoking the embedding model.
+        retry_config (RetryConfig): The retry configuration for the embedding model.
+        truncation_config (TruncationConfig | None): The truncation configuration for the embedding model.
+        input_type (CohereInputType): The input type for the embedding model. Supported values include:
+            1. `CohereInputType.SEARCH_DOCUMENT`,
+            2. `CohereInputType.SEARCH_QUERY`,
+            3. `CohereInputType.CLASSIFICATION`,
+            4. `CohereInputType.CLUSTERING`,
+            5. `CohereInputType.IMAGE`.
+
+    Initialization:
+        You can initialize the `CohereEMInvoker` as follows:
+        ```python
+        em_invoker = CohereEMInvoker(
+            model_name="embed-english-v4.0",
+            input_type="search_document"
+        )
+        ```
+
+        Note: The `input_type` parameter can be one of the following:
+        1. "search_document"
+        2. "search_query"
+        3. "classification"
+        4. "clustering"
+        5. "image"
+
+        This parameter is optional and defaults to "search_document". For more information about
+        input_type, please refer to https://docs.cohere.com/docs/embeddings#the-input_type-parameter.
+
+    Input types:
+        The `CohereEMInvoker` supports the following input types: text and image.
+        Non-text inputs must be passed as an `Attachment` object.
+
+    Output format:
+        The `CohereEMInvoker` can embed either:
+        1. A single content.
+           1. A single content is either a text or an image.
+           2. The output will be a `Vector`, representing the embedding of the content.
+
+           # Example 1: Embedding a text content.
+           ```python
+           text = "What animal is in this image?"
+           result = await em_invoker.invoke(text)
+           ```
+
+           # Example 2: Embedding an image content.
+           ```python
+           image = Attachment.from_path("path/to/local/image.png")
+           result = await em_invoker.invoke(image)
+           ```
+
+           The above examples will return a `Vector` with a size of (embedding_size,).
+
+        2. A list of contents.
+           1. A list of contents is a list that consists of any of the above single contents.
+           2. The output will be a `list[Vector]`, where each element is a `Vector` representing the
+              embedding of each single content.
+
+           # Example: Embedding a list of contents.
+           ```python
+           text = "What animal is in this image?"
+           image = Attachment.from_path("path/to/local/image.png")
+           result = await em_invoker.invoke([text, image])
+           ```
+
+           The above examples will return a `list[Vector]` with a size of (2, embedding_size).
+
+    Retry and timeout:
+        The `CohereEMInvoker` supports retry and timeout configuration.
+        By default, the max retries is set to 0 and the timeout is set to 30.0 seconds.
+        They can be customized by providing a custom `RetryConfig` object to the `retry_config` parameter.
+
+        Retry config examples:
+        ```python
+        retry_config = RetryConfig(max_retries=0, timeout=None)  # No retry, no timeout
+        retry_config = RetryConfig(max_retries=0, timeout=10.0)  # No retry, 10.0 seconds timeout
+        retry_config = RetryConfig(max_retries=5, timeout=None)  # 5 max retries, no timeout
+        retry_config = RetryConfig(max_retries=5, timeout=10.0)  # 5 max retries, 10.0 seconds timeout
+        ```
+
+        Usage example:
+        ```python
+        em_invoker = CohereEMInvoker(..., retry_config=retry_config)
+        ```
+
+    '''
+    input_type: Incomplete
+    client: Incomplete
+    def __init__(self, model_name: str, api_key: str | None = None, base_url: str | None = None, model_kwargs: dict[str, Any] | None = None, default_hyperparameters: dict[str, Any] | None = None, retry_config: RetryConfig | None = None, truncation_config: TruncationConfig | None = None, input_type: CohereInputType = ...) -> None:
+        '''Initializes a new instance of the CohereEMInvoker class.
+
+        Args:
+            model_name (str): The name of the Cohere embedding model to be used.
+            api_key (str | None, optional): The API key for authenticating with Cohere. Defaults to None, in which
+                case the `COHERE_API_KEY` environment variable will be used.
+            base_url (str | None, optional): The base URL for a custom Cohere-compatible endpoint.
+                Defaults to None, in which case Cohere\'s default URL will be used.
+            model_kwargs (dict[str, Any] | None, optional): Additional keyword arguments for the Cohere client.
+                Defaults to None.
+            default_hyperparameters (dict[str, Any] | None, optional): Default hyperparameters for invoking the model.
+                Defaults to None.
+            retry_config (RetryConfig | None, optional): The retry configuration for the embedding model.
+                Defaults to None, in which case a default config with no retry and 30.0 seconds timeout will be used.
+            truncation_config (TruncationConfig | None, optional): Configuration for text truncation behavior.
+                Defaults to None, in which case no truncation is applied.
+            input_type (CohereInputType, optional): The input type for the embedding model.
+                Defaults to `CohereInputType.SEARCH_DOCUMENT`. Valid values are: "search_document", "search_query",
+                "classification", "clustering", and "image".
+        '''

gllm_inference/em_invoker/jina_em_invoker.pyi

@@ -0,0 +1,103 @@
+from _typeshed import Incomplete
+from gllm_core.utils.retry import RetryConfig as RetryConfig
+from gllm_inference.constants import EMBEDDING_ENDPOINT as EMBEDDING_ENDPOINT, JINA_DEFAULT_URL as JINA_DEFAULT_URL
+from gllm_inference.em_invoker.em_invoker import BaseEMInvoker as BaseEMInvoker
+from gllm_inference.em_invoker.schema.jina import InputType as InputType, Key as Key
+from gllm_inference.exceptions import BaseInvokerError as BaseInvokerError, ProviderInternalError as ProviderInternalError
+from gllm_inference.exceptions.error_parser import convert_http_status_to_base_invoker_error as convert_http_status_to_base_invoker_error
+from gllm_inference.schema import Attachment as Attachment, AttachmentType as AttachmentType, EMContent as EMContent, ModelId as ModelId, ModelProvider as ModelProvider, TruncationConfig as TruncationConfig, Vector as Vector
+from typing import Any
+
+SUPPORTED_ATTACHMENTS: Incomplete
+MULTIMODAL_MODELS: Incomplete
+
+class JinaEMInvoker(BaseEMInvoker):
+    '''An embedding model invoker to interact with Jina AI embedding models.
+
+    Attributes:
+        model_id (str): The model ID of the embedding model.
+        model_provider (str): The provider of the embedding model.
+        model_name (str): The name of the embedding model.
+        client (AsyncClient): The client for the Jina AI API.
+        default_hyperparameters (dict[str, Any]): Default hyperparameters for invoking the embedding model.
+        retry_config (RetryConfig): The retry configuration for the embedding model.
+        truncation_config (TruncationConfig | None): The truncation configuration for the embedding model.
+
+    Input types:
+        The `JinaEMInvoker` supports the following input types: text and image.
+        Non-text inputs must be passed as a `Attachment` object.
+
+    Output format:
+        The `JinaEMInvoker` can embed either:
+        1. A single content.
+           1. A single content is either a text or an image.
+           2. The output will be a `Vector`, representing the embedding of the content.
+
+           # Example 1: Embedding a text content.
+           ```python
+           text = "What animal is in this image?"
+           result = await em_invoker.invoke(text)
+           ```
+
+           # Example 2: Embedding an image content.
+           ```python
+           image = Attachment.from_path("path/to/local/image.png")
+           result = await em_invoker.invoke(image)
+           ```
+
+           The above examples will return a `Vector` with a size of (embedding_size,).
+
+        2. A list of contents.
+           1. A list of contents is a list that consists of any of the above single contents.
+           2. The output will be a `list[Vector]`, where each element is a `Vector` representing the
+              embedding of each single content.
+
+           # Example: Embedding a list of contents.
+           ```python
+           text = "What animal is in this image?"
+           image = Attachment.from_path("path/to/local/image.png")
+           result = await em_invoker.invoke([text, image])
+           ```
+
+           The above examples will return a `list[Vector]` with a size of (2, embedding_size).
+
+    Retry and timeout:
+        The `JinaEMInvoker` supports retry and timeout configuration.
+        By default, the max retries is set to 0 and the timeout is set to 30.0 seconds.
+        They can be customized by providing a custom `RetryConfig` object to the `retry_config` parameter.
+
+        Retry config examples:
+        ```python
+        retry_config = RetryConfig(max_retries=0, timeout=None)  # No retry, no timeout
+        retry_config = RetryConfig(max_retries=0, timeout=10.0)  # No retry, 10.0 seconds timeout
+        retry_config = RetryConfig(max_retries=5, timeout=None)  # 5 max retries, no timeout
+        retry_config = RetryConfig(max_retries=5, timeout=10.0)  # 5 max retries, 10.0 seconds timeout
+        ```
+
+        Usage example:
+        ```python
+        em_invoker = JinaEMInvoker(..., retry_config=retry_config)
+        ```
+    '''
+    client: Incomplete
+    model_kwargs: Incomplete
+    def __init__(self, model_name: str, api_key: str | None = None, base_url: str = ..., model_kwargs: dict[str, Any] | None = None, default_hyperparameters: dict[str, Any] | None = None, retry_config: RetryConfig | None = None, truncation_config: TruncationConfig | None = None) -> None:
+        '''Initializes a new instance of the JinaEMInvoker class.
+
+        Args:
+            model_name (str): The name of the Jina embedding model to be used.
+            api_key (str | None, optional): The API key for authenticating with Jina AI.
+                Defaults to None, in which case the `JINA_API_KEY` environment variable will be used.
+            base_url (str, optional): The base URL for the Jina AI API. Defaults to "https://api.jina.ai/v1".
+            model_kwargs (dict[str, Any] | None, optional): Additional keyword arguments for the HTTP client.
+                Defaults to None.
+            default_hyperparameters (dict[str, Any] | None, optional): Default hyperparameters for invoking the model.
+                Defaults to None.
+            retry_config (RetryConfig | None, optional): The retry configuration for the embedding model.
+                Defaults to None, in which case a default config with no retry and 30.0 seconds timeout will be used.
+            truncation_config (TruncationConfig | None, optional): Configuration for text truncation behavior.
+                Defaults to None, in which case no truncation is applied.
+
+        Raises:
+            ValueError: If neither `api_key` nor `JINA_API_KEY` environment variable is provided.
+        '''

gllm_inference/em_invoker/schema/bedrock.pyi

@@ -1,22 +1,29 @@
 class Key:
     """Defines valid keys in Bedrock."""
     ACCEPT: str
+    BASE64_STRING: str
     CONTENT_TYPE: str
     HTTP_STATUS_CODE: str
     INPUT_TEXT: str
     INPUT_TYPE: str
+    INPUT_TYPE_MARENGO: str
+    MEDIA_SOURCE: str
     MODEL_ID: str
     RESPONSE_METADATA: str
+    TEXT_TRUNCATE: str
     TEXTS: str
 
 class InputType:
     """Defines valid input types in Bedrock."""
     APPLICATION_JSON: str
+    IMAGE: str
     SEARCH_DOCUMENT: str
     SEARCH_QUERY: str
+    TEXT: str
 
 class OutputType:
     """Defines valid output types in Bedrock."""
     BODY: str
+    DATA: str
     EMBEDDING: str
     EMBEDDINGS: str

gllm_inference/em_invoker/schema/cohere.pyi

@@ -0,0 +1,20 @@
+from enum import StrEnum
+
+class Key(StrEnum):
+    """Defines valid keys in Cohere."""
+    BASE_URL = 'base_url'
+    IMAGE_URL = 'image_url'
+    INPUT_TYPE = 'input_type'
+    MAX_RETRIES = 'max_retries'
+    MODEL = 'model'
+    TIMEOUT = 'timeout'
+    TYPE = 'type'
+    URL = 'url'
+
+class CohereInputType(StrEnum):
+    """Defines valid embedding input types for Cohere embedding API."""
+    CLASSIFICATION = 'classification'
+    CLUSTERING = 'clustering'
+    IMAGE = 'image'
+    SEARCH_DOCUMENT = 'search_document'
+    SEARCH_QUERY = 'search_query'

gllm_inference/em_invoker/schema/jina.pyi

@@ -0,0 +1,29 @@
+from enum import StrEnum
+
+class InputType(StrEnum):
+    """Defines the supported input types for the Jina AI embedding API."""
+    IMAGE_URL = 'image_url'
+    TEXT = 'text'
+
+class Key(StrEnum):
+    """Defines key constants used in the Jina AI API payloads."""
+    DATA = 'data'
+    EMBEDDING = 'embedding'
+    EMBEDDINGS = 'embeddings'
+    ERROR = 'error'
+    IMAGE_URL = 'image_url'
+    INPUT = 'input'
+    JSON = 'json'
+    MESSAGE = 'message'
+    MODEL = 'model'
+    RESPONSE = 'response'
+    STATUS = 'status'
+    TASK = 'task'
+    TEXT = 'text'
+    TYPE = 'type'
+    URL = 'url'
+
+class OutputType(StrEnum):
+    """Defines the expected output types returned by the Jina AI embedding API."""
+    DATA = 'data'
+    EMBEDDING = 'embedding'

gllm_inference/exceptions/provider_error_map.pyi

@@ -13,6 +13,7 @@ class ExtendedHTTPStatus(IntEnum):
 HTTP_STATUS_TO_EXCEPTION_MAP: dict[int, type[BaseInvokerError]]
 ANTHROPIC_ERROR_MAPPING: Incomplete
 BEDROCK_ERROR_MAPPING: Incomplete
+COHERE_ERROR_MAPPING: Incomplete
 GOOGLE_ERROR_MAPPING: Incomplete
 LANGCHAIN_ERROR_CODE_MAPPING: Incomplete
 LITELLM_ERROR_MAPPING: Incomplete

gllm_inference/lm_invoker/__init__.pyi

@@ -8,6 +8,8 @@ from gllm_inference.lm_invoker.litellm_lm_invoker import LiteLLMLMInvoker as Lit
 from gllm_inference.lm_invoker.openai_chat_completions_lm_invoker import OpenAIChatCompletionsLMInvoker as OpenAIChatCompletionsLMInvoker
 from gllm_inference.lm_invoker.openai_compatible_lm_invoker import OpenAICompatibleLMInvoker as OpenAICompatibleLMInvoker
 from gllm_inference.lm_invoker.openai_lm_invoker import OpenAILMInvoker as OpenAILMInvoker
+from gllm_inference.lm_invoker.portkey_lm_invoker import PortkeyLMInvoker as PortkeyLMInvoker
+from gllm_inference.lm_invoker.sea_lion_lm_invoker import SeaLionLMInvoker as SeaLionLMInvoker
 from gllm_inference.lm_invoker.xai_lm_invoker import XAILMInvoker as XAILMInvoker
 
-__all__ = ['AnthropicLMInvoker', 'AzureOpenAILMInvoker', 'BedrockLMInvoker', 'DatasaurLMInvoker', 'GoogleLMInvoker', 'LangChainLMInvoker', 'LiteLLMLMInvoker', 'OpenAIChatCompletionsLMInvoker', 'OpenAICompatibleLMInvoker', 'OpenAILMInvoker', 'XAILMInvoker']
+__all__ = ['AnthropicLMInvoker', 'AzureOpenAILMInvoker', 'BedrockLMInvoker', 'DatasaurLMInvoker', 'GoogleLMInvoker', 'LangChainLMInvoker', 'LiteLLMLMInvoker', 'OpenAIChatCompletionsLMInvoker', 'OpenAICompatibleLMInvoker', 'OpenAILMInvoker', 'PortkeyLMInvoker', 'SeaLionLMInvoker', 'XAILMInvoker']