gllm-inference-binary 0.5.42__cp313-cp313-win_amd64.whl → 0.5.44__cp313-cp313-win_amd64.whl

gllm_inference/builder/build_em_invoker.pyi

@@ -1,7 +1,7 @@
 from _typeshed import Incomplete
-from gllm_inference.em_invoker import AzureOpenAIEMInvoker as AzureOpenAIEMInvoker, BedrockEMInvoker as BedrockEMInvoker, 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 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 gllm_inference.schema.model_id import ModelId as ModelId, ModelProvider as ModelProvider, OPTIONAL_PATH_PROVIDERS as OPTIONAL_PATH_PROVIDERS
 from typing import Any
 
 PROVIDER_TO_EM_INVOKER_MAP: dict[str, type[BaseEMInvoker]]

gllm_inference/em_invoker/__init__.pyi

@@ -1,5 +1,6 @@
 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
@@ -8,4 +9,4 @@ from gllm_inference.em_invoker.openai_em_invoker import OpenAIEMInvoker as OpenA
 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', 'JinaEMInvoker', 'LangChainEMInvoker', 'OpenAIEMInvoker', 'OpenAICompatibleEMInvoker', 'TwelveLabsEMInvoker', 'VoyageEMInvoker']
+__all__ = ['AzureOpenAIEMInvoker', 'BedrockEMInvoker', 'CohereEMInvoker', 'GoogleEMInvoker', 'JinaEMInvoker', 'LangChainEMInvoker', 'OpenAIEMInvoker', 'OpenAICompatibleEMInvoker', 'TwelveLabsEMInvoker', 'VoyageEMInvoker']

gllm_inference/em_invoker/cohere_em_invoker.pyi

@@ -0,0 +1,128 @@
+from _typeshed import Incomplete
+from gllm_core.utils.retry 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 gllm_inference.utils import validate_string_enum as validate_string_enum
+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/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/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/lm_invoker.pyi

@@ -7,7 +7,7 @@ from gllm_core.utils import RetryConfig
 from gllm_inference.constants import DOCUMENT_MIME_TYPES as DOCUMENT_MIME_TYPES, INVOKER_DEFAULT_TIMEOUT as INVOKER_DEFAULT_TIMEOUT
 from gllm_inference.exceptions import BaseInvokerError as BaseInvokerError, convert_to_base_invoker_error as convert_to_base_invoker_error
 from gllm_inference.lm_invoker.batch import BatchOperations as BatchOperations
-from gllm_inference.schema import Activity as Activity, Attachment as Attachment, AttachmentType as AttachmentType, BatchStatus as BatchStatus, LMInput as LMInput, LMOutput as LMOutput, Message as Message, MessageContent as MessageContent, MessageRole as MessageRole, ModelId as ModelId, Reasoning as Reasoning, ResponseSchema as ResponseSchema, ToolCall as ToolCall, ToolResult as ToolResult
+from gllm_inference.schema import Attachment as Attachment, AttachmentType as AttachmentType, BatchStatus as BatchStatus, LMEventType as LMEventType, LMInput as LMInput, LMOutput as LMOutput, Message as Message, MessageContent as MessageContent, MessageRole as MessageRole, ModelId as ModelId, Reasoning as Reasoning, ResponseSchema as ResponseSchema, ToolCall as ToolCall, ToolResult as ToolResult
 from langchain_core.tools import Tool as LangChainTool
 from typing import Any
 

gllm_inference/schema/__init__.pyi

@@ -2,7 +2,7 @@ from gllm_inference.schema.activity import Activity as Activity, MCPCallActivity
 from gllm_inference.schema.attachment import Attachment as Attachment
 from gllm_inference.schema.code_exec_result import CodeExecResult as CodeExecResult
 from gllm_inference.schema.config import TruncationConfig as TruncationConfig
-from gllm_inference.schema.enums import AttachmentType as AttachmentType, BatchStatus as BatchStatus, EmitDataType as EmitDataType, MessageRole as MessageRole, TruncateSide as TruncateSide
+from gllm_inference.schema.enums import AttachmentType as AttachmentType, BatchStatus as BatchStatus, EmitDataType as EmitDataType, LMEventType as LMEventType, MessageRole as MessageRole, TruncateSide as TruncateSide
 from gllm_inference.schema.events import ActivityEvent as ActivityEvent, CodeEvent as CodeEvent, ThinkingEvent as ThinkingEvent
 from gllm_inference.schema.lm_input import LMInput as LMInput
 from gllm_inference.schema.lm_output import LMOutput as LMOutput
@@ -15,4 +15,4 @@ from gllm_inference.schema.tool_call import ToolCall as ToolCall
 from gllm_inference.schema.tool_result import ToolResult as ToolResult
 from gllm_inference.schema.type_alias import EMContent as EMContent, MessageContent as MessageContent, ResponseSchema as ResponseSchema, Vector as Vector
 
-__all__ = ['Activity', 'ActivityEvent', 'Attachment', 'AttachmentType', 'BatchStatus', 'CodeEvent', 'CodeExecResult', 'EMContent', 'EmitDataType', 'InputTokenDetails', 'LMInput', 'LMOutput', 'MCPCall', 'MCPCallActivity', 'MCPListToolsActivity', 'MCPServer', 'Message', 'MessageContent', 'MessageRole', 'ModelId', 'ModelProvider', 'OutputTokenDetails', 'Reasoning', 'ThinkingEvent', 'ResponseSchema', 'TokenUsage', 'ToolCall', 'ToolResult', 'TruncateSide', 'TruncationConfig', 'Vector', 'WebSearchActivity']
+__all__ = ['Activity', 'ActivityEvent', 'Attachment', 'AttachmentType', 'BatchStatus', 'CodeEvent', 'CodeExecResult', 'EMContent', 'EmitDataType', 'LMEventType', 'InputTokenDetails', 'LMInput', 'LMOutput', 'MCPCall', 'MCPCallActivity', 'MCPListToolsActivity', 'MCPServer', 'Message', 'MessageContent', 'MessageRole', 'ModelId', 'ModelProvider', 'OutputTokenDetails', 'Reasoning', 'ThinkingEvent', 'ResponseSchema', 'TokenUsage', 'ToolCall', 'ToolResult', 'TruncateSide', 'TruncationConfig', 'Vector', 'WebSearchActivity']

gllm_inference/schema/activity.pyi

@@ -4,12 +4,13 @@ from pydantic import BaseModel
 from typing import Literal
 
 WEB_SEARCH_VISIBLE_FIELDS: Incomplete
+WebSearchActivityTypes: Incomplete
 
 class Activity(BaseModel):
     """Base schema for any activity.
 
     Attributes:
-        type (str): The type of activity being performed.
+        type (str): The type of activity being performed. Defaults to an empty string.
     """
     type: str
 
@@ -17,9 +18,10 @@ class MCPListToolsActivity(Activity):
     """Schema for listing tools in MCP.
 
     Attributes:
-        server_name (str): The name of the MCP server.
-        tools (list[dict[str, str]] | None): The tools in the MCP server.
-        type (str): The type of activity being performed.
+        type (Literal[ActivityType.MCP_LIST_TOOLS]): The type of activity being performed.
+            Defaults to ActivityType.MCP_LIST_TOOLS.
+        server_name (str): The name of the MCP server. Defaults to an empty string.
+        tools (list[dict[str, str]] | None): The tools in the MCP server. Defaults to None.
     """
     type: Literal[ActivityType.MCP_LIST_TOOLS]
     server_name: str
@@ -29,10 +31,10 @@ class MCPCallActivity(Activity):
     """Schema for MCP tool call.
 
     Attributes:
+        type (Literal[ActivityType.MCP_CALL]): The type of activity being performed. Defaults to ActivityType.MCP_CALL.
         server_name (str): The name of the MCP server.
         tool_name (str): The name of the tool.
         args (dict[str, str]): The arguments of the tool.
-        type (str): The type of activity being performed.
     """
     type: Literal[ActivityType.MCP_CALL]
     server_name: str
@@ -43,16 +45,16 @@ class WebSearchActivity(Activity):
     """Schema for web search tool call.
 
     Attributes:
-        type (str): The type of activity being performed.
-        pattern (str): The pattern of the web search.
-        url (str): The URL of the page.
-        query (str): The query of the web search.
+        type (WebSearchActivityTypes): The type of activity being performed. Defaults to ActivityType.SEARCH.
+        query (str | None): The query of the web search. Defaults to None.
+        url (str | None): The URL of the page. Defaults to None.
+        pattern (str | None): The pattern of the web search. Defaults to None.
         sources (list[dict[str, str]] | None): The sources of the web search.
     """
-    type: Literal[ActivityType.FIND_IN_PAGE, ActivityType.OPEN_PAGE, ActivityType.SEARCH]
+    type: WebSearchActivityTypes
+    query: str | None
     url: str | None
     pattern: str | None
-    query: str | None
     sources: list[dict[str, str]] | None
     def model_dump(self, *args, **kwargs) -> dict[str, str]:
         """Serialize the activity for display.

gllm_inference/schema/enums.pyi

@@ -14,6 +14,16 @@ class BatchStatus(StrEnum):
     FINISHED = 'finished'
     UNKNOWN = 'unknown'
 
+class LMEventType(StrEnum):
+    """Defines event types to be emitted by the LM invoker."""
+    ACTIVITY = 'activity'
+    CODE_START = 'code_start'
+    CODE = 'code'
+    CODE_END = 'code_end'
+    THINKING_START = 'thinking_start'
+    THINKING = 'thinking'
+    THINKING_END = 'thinking_end'
+
 class EmitDataType(StrEnum):
     """Defines valid data types for emitting events."""
     ACTIVITY = 'activity'

gllm_inference/schema/events.pyi

@@ -1,108 +1,123 @@
-from gllm_core.constants import EventLevel
+from _typeshed import Incomplete
 from gllm_core.schema import Event
 from gllm_inference.schema.activity import Activity as Activity
-from gllm_inference.schema.enums import EmitDataType as EmitDataType
-from typing import Literal
+from gllm_inference.schema.enums import LMEventType as LMEventType
+from typing import Any, Literal
+
+CodeEventType: Incomplete
+ThinkingEventType: Incomplete
 
 class ActivityEvent(Event):
-    """Event schema for model-triggered activities (e.g. web search, MCP).
+    """Event schema for model-triggered activities (e.g. web search, MCP call, etc.).
 
     Attributes:
-        id (str): The unique identifier for the activity event. Defaults to an empty string.
-        type (Literal): The type of event, always 'activity'.
-        value (Activity): The activity data containing message and type.
-        level (EventLevel): The severity level of the event. Defined through the EventLevel constants.
+        id (str): The ID of the activity event. Defaults to None.
+        value (dict[str, Any]): The value of the activity event.
+        level (EventLevel): The severity level of the activity event. Defaults to EventLevel.INFO.
+        type (Literal[EventType.ACTIVITY]): The type of the activity event. Defaults to EventType.ACTIVITY.
+        timestamp (datetime): The timestamp of the activity event. Defaults to the current timestamp.
+        metadata (dict[str, Any]): The metadata of the activity event. Defaults to an empty dictionary.
     """
-    id: str
-    type: Literal[EmitDataType.ACTIVITY]
-    value: Activity
-    level: EventLevel
+    value: dict[str, Any]
+    type: Literal[LMEventType.ACTIVITY]
+    @classmethod
+    def from_activity(cls, id_: str | None = None, activity: Activity | None = None) -> ActivityEvent:
+        """Create an activity event from an Activity object.
+
+        Args:
+            id_ (str | None, optional): The ID of the activity event. Defaults to None.
+            activity (Activity | None, optional): The activity object to create the event from.
+                Defaults to None, in which case the value will be an empty dictionary.
+
+        Returns:
+            ActivityEvent: The activity event.
+        """
 
 class CodeEvent(Event):
-    """Event schema for model-triggered code execution.
+    """Event schema for model-generated code to be executed.
 
     Attributes:
-        id (str): The unique identifier for the code event. Defaults to an empty string.
-        type (Literal): The type of event (code, code_start, or code_end).
-        value (str): The code content.
-        level (EventLevel): The severity level of the event. Defined through the EventLevel constants.
+        id (str): The ID of the code event. Defaults to None.
+        value (str): The value of the code event. Defaults to an empty string.
+        level (EventLevel): The severity level of the code event. Defaults to EventLevel.INFO.
+        type (CodeEventType): The type of the code event. Defaults to EventType.CODE.
+        timestamp (datetime): The timestamp of the code event. Defaults to the current timestamp.
+        metadata (dict[str, Any]): The metadata of the code event. Defaults to an empty dictionary.
     """
-    id: str
-    type: Literal[EmitDataType.CODE, EmitDataType.CODE_START, EmitDataType.CODE_END]
     value: str
-    level: EventLevel
+    type: CodeEventType
     @classmethod
-    def start(cls, id_: str | None = '') -> CodeEvent:
+    def start(cls, id_: str | None = None) -> CodeEvent:
         """Create a code start event.
 
         Args:
-            id_ (str | None): The unique identifier for the code event. Defaults to an empty string.
+            id_ (str | None, optional): The ID of the code event. Defaults to None.
 
         Returns:
             CodeEvent: The code start event.
         """
     @classmethod
-    def content(cls, id_: str | None = '', value: str = '') -> CodeEvent:
+    def content(cls, id_: str | None = None, value: str = '') -> CodeEvent:
         """Create a code content event.
 
         Args:
-            id_ (str | None): The unique identifier for the code event. Defaults to an empty string.
-            value (str): The code content.
+            id_ (str | None, optional): The ID of the code event. Defaults to None.
+            value (str, optional): The code content. Defaults to an empty string.
 
         Returns:
             CodeEvent: The code value event.
         """
     @classmethod
-    def end(cls, id_: str | None = '') -> CodeEvent:
+    def end(cls, id_: str | None = None) -> CodeEvent:
         """Create a code end event.
 
         Args:
-            id_ (str | None): The unique identifier for the code event. Defaults to an empty string.
+            id_ (str | None, optional): The ID of the code event. Defaults to None.
 
         Returns:
             CodeEvent: The code end event.
         """
 
 class ThinkingEvent(Event):
-    """Event schema for model thinking.
+    """Event schema for model-generated thinking.
 
     Attributes:
-        id (str): The unique identifier for the thinking event. Defaults to an empty string.
-        type (Literal): The type of thinking event (thinking, thinking_start, or thinking_end).
-        value (str): The thinking content or message.
-        level (EventLevel): The severity level of the event. Defined through the EventLevel constants.
+        id (str): The ID of the thinking event. Defaults to None.
+        value (str): The value of the thinking event. Defaults to an empty string.
+        level (EventLevel): The severity level of the thinking event. Defaults to EventLevel.INFO.
+        type (ThinkingEventType): The type of the thinking event. Defaults to EventType.THINKING.
+        timestamp (datetime): The timestamp of the thinking event. Defaults to the current timestamp.
+        metadata (dict[str, Any]): The metadata of the thinking event. Defaults to an empty dictionary.
     """
-    id: str
-    type: Literal[EmitDataType.THINKING, EmitDataType.THINKING_START, EmitDataType.THINKING_END]
     value: str
-    level: EventLevel
+    type: ThinkingEventType
     @classmethod
-    def start(cls, id_: str | None = '') -> ThinkingEvent:
+    def start(cls, id_: str | None = None) -> ThinkingEvent:
         """Create a thinking start event.
 
         Args:
-            id_ (str | None): The unique identifier for the thinking event. Defaults to an empty string.
+            id_ (str | None, optional): The ID of the thinking event. Defaults to None.
 
         Returns:
             ThinkingEvent: The thinking start event.
         """
     @classmethod
-    def content(cls, id_: str | None = '', value: str = '') -> ThinkingEvent:
+    def content(cls, id_: str | None = None, value: str = '') -> ThinkingEvent:
         """Create a thinking value event.
 
         Args:
-            id_ (str | None): The unique identifier for the thinking event. Defaults to an empty string.
-            value (str): The thinking content or message.
+            id_ (str | None, optional): The ID of the thinking event. Defaults to None.
+            value (str, optional): The thinking content or message. Defaults to an empty string.
 
         Returns:
             ThinkingEvent: The thinking value event.
         """
     @classmethod
-    def end(cls, id_: str | None = '') -> ThinkingEvent:
+    def end(cls, id_: str | None = None) -> ThinkingEvent:
         """Create a thinking end event.
 
         Args:
-            id_ (str | None): The unique identifier for the thinking event. Defaults to an empty string.
+            id_ (str | None, optional): The ID of the thinking event. Defaults to None.
 
         Returns:
             ThinkingEvent: The thinking end event.

gllm_inference/schema/model_id.pyi

@@ -12,6 +12,7 @@ class ModelProvider(StrEnum):
     ANTHROPIC = 'anthropic'
     AZURE_OPENAI = 'azure-openai'
     BEDROCK = 'bedrock'
+    COHERE = 'cohere'
     DATASAUR = 'datasaur'
     GOOGLE = 'google'
     JINA = 'jina'
@@ -46,6 +47,11 @@ class ModelId(BaseModel):
         model_id = ModelId.from_string("bedrock/us.anthropic.claude-sonnet-4-20250514-v1:0")
         ```
 
+        # Using Cohere
+        ```python
+        model_id = ModelId.from_string("cohere/embed-english-v3.0")
+        ```
+
         # Using Datasaur
         ```python
         model_id = ModelId.from_string("datasaur/https://deployment.datasaur.ai/api/deployment/teamId/deploymentId/")
@@ -102,6 +108,7 @@ class ModelId(BaseModel):
         ```python
         model_id = ModelId.from_string("langchain/langchain_openai.ChatOpenAI:gpt-4o-mini")
         ```
+
         For the list of supported providers, please refer to the following table:
         https://python.langchain.com/docs/integrations/chat/#featured-providers
 

gllm_inference.pyi

@@ -14,6 +14,7 @@ import gllm_core
 import gllm_core.utils
 import gllm_inference.em_invoker.AzureOpenAIEMInvoker
 import gllm_inference.em_invoker.BedrockEMInvoker
+import gllm_inference.em_invoker.CohereEMInvoker
 import gllm_inference.em_invoker.GoogleEMInvoker
 import gllm_inference.em_invoker.JinaEMInvoker
 import gllm_inference.em_invoker.LangChainEMInvoker
@@ -51,17 +52,19 @@ import gllm_inference.exceptions.BaseInvokerError
 import gllm_inference.exceptions.convert_http_status_to_base_invoker_error
 import gllm_inference.schema.Vector
 import aioboto3
-import asyncio.CancelledError
-import gllm_inference.exceptions.convert_to_base_invoker_error
+import base64
 import gllm_inference.schema.Attachment
 import gllm_inference.schema.AttachmentType
 import gllm_inference.schema.EMContent
+import gllm_inference.utils.validate_string_enum
+import cohere
+import asyncio.CancelledError
+import gllm_inference.exceptions.convert_to_base_invoker_error
 import gllm_inference.schema.TruncateSide
 import google
 import google.auth
 import google.genai
 import google.genai.types
-import base64
 import httpx
 import gllm_inference.exceptions.ProviderInternalError
 import concurrent
@@ -111,9 +114,8 @@ import inspect
 import time
 import jsonschema
 import gllm_inference.lm_invoker.batch.BatchOperations
-import gllm_inference.schema.Activity
+import gllm_inference.schema.LMEventType
 import gllm_inference.schema.MessageContent
-import gllm_inference.utils.validate_string_enum
 import __future__
 import gllm_inference.schema.ActivityEvent
 import gllm_inference.schema.CodeEvent

{gllm_inference_binary-0.5.42.dist-info → gllm_inference_binary-0.5.44.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.2
 Name: gllm-inference-binary
-Version: 0.5.42
+Version: 0.5.44
 Summary: A library containing components related to model inferences in Gen AI applications.
 Author-email: Henry Wicaksono <henry.wicaksono@gdplabs.id>, Resti Febrina <resti.febrina@gdplabs.id>
 Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
 Requires-Dist: poetry<3.0.0,>=2.1.3
-Requires-Dist: gllm-core-binary<0.4.0,>=0.3.0
+Requires-Dist: gllm-core-binary<0.4.0,>=0.3.21
 Requires-Dist: aiohttp<4.0.0,>=3.12.14
 Requires-Dist: filetype<2.0.0,>=1.2.0
 Requires-Dist: httpx<0.29.0,>=0.28.0
@@ -30,6 +30,8 @@ Provides-Extra: anthropic
 Requires-Dist: anthropic<0.61.0,>=0.60.0; extra == "anthropic"
 Provides-Extra: bedrock
 Requires-Dist: aioboto3<16.0.0,>=15.0.0; extra == "bedrock"
+Provides-Extra: cohere
+Requires-Dist: cohere<6.0.0,>=5.18.0; extra == "cohere"
 Provides-Extra: datasaur
 Requires-Dist: openai<2.0.0,>=1.98.0; extra == "datasaur"
 Provides-Extra: google

{gllm_inference_binary-0.5.42.dist-info → gllm_inference_binary-0.5.44.dist-info}/RECORD

@@ -1,9 +1,9 @@
-gllm_inference.cp313-win_amd64.pyd,sha256=FYH5KTQcK7_qnRISOhH_EgUHRR5OuNaCaJrqsP7zsmc,3619840
-gllm_inference.pyi,sha256=ifuvcH5p0J0Q2zXwfqXT4Sh5kcodA7O8x6SmQJ-_M0w,4852
+gllm_inference.cp313-win_amd64.pyd,sha256=3Xnjm5F7tiaEIKRL19GPUPLSnwzUmugqGBtivByuSOs,3671552
+gllm_inference.pyi,sha256=PN9a6F49IGwGrPXny7b-knSKeGsO_ER8e9DaBqiOD2k,4918
 gllm_inference/__init__.pyi,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 gllm_inference/constants.pyi,sha256=PncjVw-mkzcJ3ln1ohvVZGdJ-TD-VZy1Ygn4Va8Z7i0,350
 gllm_inference/builder/__init__.pyi,sha256=-bw1uDx7CAM7pkvjvb1ZXku9zXlQ7aEAyC83KIn3bz8,506
-gllm_inference/builder/build_em_invoker.pyi,sha256=ij24IMOYIYjia_cONdVhfW0QpulqyqG6nQhTdo5fYs0,6195
+gllm_inference/builder/build_em_invoker.pyi,sha256=3vO_pLokR4BAZflOMu6qzXoKx6vibT16uwJETH5Y_yc,6283
 gllm_inference/builder/build_lm_invoker.pyi,sha256=HvQICF-qvOTzfXZUqhi7rlwcpkMZpxaC-8QZmhnXKzI,7466
 gllm_inference/builder/build_lm_request_processor.pyi,sha256=H7Rg88e7PTTCtuyY64r333moTmh4-ypOwgnG10gkEdY,4232
 gllm_inference/builder/build_output_parser.pyi,sha256=sgSTrzUmSRxPzUUum0fDU7A3NXYoYhpi6bEx4Q2XMnA,965
@@ -11,9 +11,10 @@ gllm_inference/catalog/__init__.pyi,sha256=HWgPKWIzprpMHRKe_qN9BZSIQhVhrqiyjLjIX
 gllm_inference/catalog/catalog.pyi,sha256=eWPqgQKi-SJGHabi_XOTEKpAj96OSRypKsb5ZEC1VWU,4911
 gllm_inference/catalog/lm_request_processor_catalog.pyi,sha256=FiveqPDkV58XbDO2znXL-Ix5tFbZwNiVnitlEa90YOY,5536
 gllm_inference/catalog/prompt_builder_catalog.pyi,sha256=iViWB4SaezzjQY4UY1YxeoXUNxqxa2cTJGaD9JSx4Q8,3279
-gllm_inference/em_invoker/__init__.pyi,sha256=ClcAHvB87iqUZ-clWxKqT0dD3v6Tas43pQiRWOgYnog,1100
+gllm_inference/em_invoker/__init__.pyi,sha256=uCWfCjh5a5DciRFcUdbHndewokM3J5hp3mbhmM5wQC8,1211
 gllm_inference/em_invoker/azure_openai_em_invoker.pyi,sha256=TXC5Kgf1eZqK2FHKAyeG3LB1SEsSEStnbk9bI1mjC5k,5049
 gllm_inference/em_invoker/bedrock_em_invoker.pyi,sha256=kQETh2r-WR_H3APtt4QavmfwGOR3KB4k6USNYvFateY,5831
+gllm_inference/em_invoker/cohere_em_invoker.pyi,sha256=4eLqeKLoK8vJB61bGdttfWUUvNDBToBqNA6KQYBMT8s,6793
 gllm_inference/em_invoker/em_invoker.pyi,sha256=YDYJ8TGScsz5Gg-OBnEENN1tI1RYvwoddypxUr6SAWw,5191
 gllm_inference/em_invoker/google_em_invoker.pyi,sha256=zZYjeLp9ncwIVM4UHqDJSVOFn1eXiaz9Ba24-_fCF2c,6953
 gllm_inference/em_invoker/jina_em_invoker.pyi,sha256=wrivFiPFRWBSw-TEAkaQKBvBRhlzN6CBaZkc0HOslRw,5754
@@ -26,6 +27,7 @@ gllm_inference/em_invoker/langchain/__init__.pyi,sha256=aOTlRvS9aG1tBErjsmhe75s4
 gllm_inference/em_invoker/langchain/em_invoker_embeddings.pyi,sha256=BBSDazMOckO9Aw17tC3LGUTPqLb01my1xUZLtKZlwJY,3388
 gllm_inference/em_invoker/schema/__init__.pyi,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 gllm_inference/em_invoker/schema/bedrock.pyi,sha256=HoNgVi0T21aFd1JrCnSLu4yryv8k8RnYdR3-tIdHFgA,498
+gllm_inference/em_invoker/schema/cohere.pyi,sha256=Wio6h0sbY93GygqETtflRaaucFzYSeLZRg7jyxMDK0s,567
 gllm_inference/em_invoker/schema/google.pyi,sha256=bzdtu4DFH2kATLybIeNl_Lznj99H-6u2Fvx3Zx52oZg,190
 gllm_inference/em_invoker/schema/jina.pyi,sha256=EFo4d8HuzPEZDV5F0PwUIkYPrCTEiCqHq17ICzYxKeg,739
 gllm_inference/em_invoker/schema/langchain.pyi,sha256=SZ13HDcvAOGmDTi2b72H6Y1J5GePR21JdnM6gYrwcGs,117
@@ -36,7 +38,7 @@ gllm_inference/em_invoker/schema/voyage.pyi,sha256=Aqvu6mhFkNb01aXAI5mChLKIgEnFn
 gllm_inference/exceptions/__init__.pyi,sha256=nXOqwsuwUgsnBcJEANVuxbZ1nDfcJ6-pKUfKeZwltkk,1218
 gllm_inference/exceptions/error_parser.pyi,sha256=4aiJZhBzBOqlhdmpvaCvildGy7_XxlJzQpe3PzGt8eE,2040
 gllm_inference/exceptions/exceptions.pyi,sha256=6y3ECgHAStqMGgQv8Dv-Ui-5PDD07mSj6qaRZeSWea4,5857
-gllm_inference/exceptions/provider_error_map.pyi,sha256=4AsAgbXAh91mxEW2YiomEuhBoeSNeAIo9WbT9WK8gQk,1233
+gllm_inference/exceptions/provider_error_map.pyi,sha256=vWa4ZIHn7qIghECGvO-dS2KzOmf3c10GRWKZ4YDPnSQ,1267
 gllm_inference/lm_invoker/__init__.pyi,sha256=jG1xc5fTOeIgeKKVYSnsMzQThKk9kTW38yO_MYtv540,1387
 gllm_inference/lm_invoker/anthropic_lm_invoker.pyi,sha256=JSgKUk9d1ZHlitv_ZjHlAk2hIW-J7u6yslVHflIeUro,16726
 gllm_inference/lm_invoker/azure_openai_lm_invoker.pyi,sha256=FYfRNPG-oD4wIfitjTHnGib1uMZL7Pid0gbrRsymAHU,14601
@@ -45,7 +47,7 @@ gllm_inference/lm_invoker/datasaur_lm_invoker.pyi,sha256=LR0EM4vTfufq9OWk8JVIwLy
 gllm_inference/lm_invoker/google_lm_invoker.pyi,sha256=aSmEgoYj_V72Nb6erDResphw9RaHfbE5C6PhqpMfEeQ,17674
 gllm_inference/lm_invoker/langchain_lm_invoker.pyi,sha256=tJIxkFUKjLF-yz0niaDjN3L0QNCbn4sT8hmPKtERpog,12742
 gllm_inference/lm_invoker/litellm_lm_invoker.pyi,sha256=IJxRUkmgXY8oQwS7tJoskO8fiESB7M4pyvpE64pyXDo,12648
-gllm_inference/lm_invoker/lm_invoker.pyi,sha256=vUmMNEl7F__PavQJ42scoYGyWdEvZOw2Bwxhoqv_gKE,8659
+gllm_inference/lm_invoker/lm_invoker.pyi,sha256=PS4cJ5VLNfHqeTgCerY-c1Xaa7ktdWAiT-f3Lnc0wvE,8665
 gllm_inference/lm_invoker/openai_chat_completions_lm_invoker.pyi,sha256=uYJFgi4tJGab77232IC1gdoU9h9AqoClIUj6tM6O47s,15177
 gllm_inference/lm_invoker/openai_compatible_lm_invoker.pyi,sha256=T9sShA_9fgEuaaAuT2gJZq_EYNbEhf3IkWwMCwfszY8,4244
 gllm_inference/lm_invoker/openai_lm_invoker.pyi,sha256=10iKCyleqHNbJc8M1rj3ogRcNlNxcVgyk0v6TcS6gf4,23452
@@ -97,18 +99,18 @@ gllm_inference/realtime_chat/output_streamer/output_streamer.pyi,sha256=5P9NQ0aJ
 gllm_inference/request_processor/__init__.pyi,sha256=giEme2WFQhgyKiBZHhSet0_nKSCHwGy-_2p6NRzg0Zc,231
 gllm_inference/request_processor/lm_request_processor.pyi,sha256=0fy1HyILCVDw6y46E-7tLnQTRYx4ppeRMe0QP6t9Jyw,5990
 gllm_inference/request_processor/uses_lm_mixin.pyi,sha256=LYHq-zLoXEMel1LfVdYv7W3BZ8WtBLo_WWFjRf10Yto,6512
-gllm_inference/schema/__init__.pyi,sha256=Cu626ItICSPEArngfqm78RFz6xq2_Z4YXdyFxQHb-V4,2134
-gllm_inference/schema/activity.pyi,sha256=JsM2EwvaDicM6OOmVMBSPSdizbQgq-IEoD881w0RMyg,2036
+gllm_inference/schema/__init__.pyi,sha256=cGEeGc5xDyfepbCi5r_c61VmS6VnWyvoRjjK7K13lLg,2177
+gllm_inference/schema/activity.pyi,sha256=atrU4OwLesA9FEt1H7K3gsUWYNdOqpI5i2VdWkmo6cs,2367
 gllm_inference/schema/attachment.pyi,sha256=9zgAjGXBjLfzPGaKi68FMW6b5mXdEA352nDe-ynOSvY,3385
 gllm_inference/schema/code_exec_result.pyi,sha256=WQ-ARoGM9r6nyRX-A0Ro1XKiqrc9R3jRYXZpu_xo5S4,573
 gllm_inference/schema/config.pyi,sha256=NVmjQK6HipIE0dKSfx12hgIC0O-S1HEcAc-TWlXAF5A,689
-gllm_inference/schema/enums.pyi,sha256=U30RGvNFcNNJxTZZPt8vK7SFp3W4KSPVFxTZaiF1eLU,1375
-gllm_inference/schema/events.pyi,sha256=HMkGU1XoLDR6h9-L13LEn_27Z9jkmKk-kBTD-0qKzJY,4029
+gllm_inference/schema/enums.pyi,sha256=zuKgBxcm0GXvyEZag3FAETFTG-uikgjGJ5QRoXluSZg,1675
+gllm_inference/schema/events.pyi,sha256=iG3sFAhvek-fSJgmUE6nJ5M0XzSpRKKpJJiXyuB4Wq0,5058
 gllm_inference/schema/lm_input.pyi,sha256=HxQiZgY7zcXh_Dw8nK8LSeBTZEHMPZVwmPmnfgSsAbs,197
 gllm_inference/schema/lm_output.pyi,sha256=DIV8BiIOPaSnMKxzKzH_Mp7j7-MScWCvmllegJDLqFg,2479
 gllm_inference/schema/mcp.pyi,sha256=4SgQ83pEowfWm2p-w9lupV4NayqqVBOy7SuYxIFeWRs,1045
 gllm_inference/schema/message.pyi,sha256=jJV6A0ihEcun2OhzyMtNkiHnf7d6v5R-GdpTBGfJ0AQ,2272
-gllm_inference/schema/model_id.pyi,sha256=eFNCNAmDSk0M6SqjDq8JRs0I1UcVzx2JhYWaHSFDGTQ,5667
+gllm_inference/schema/model_id.pyi,sha256=3PeAHrCyMAZJPCa7CfaArmOHdh2oDdUu6lkyx9mKG9g,5819
 gllm_inference/schema/reasoning.pyi,sha256=jbPxkDRHt0Vt-zdcc8lTT1l2hIE1Jm3HIHeNd0hfXGo,577
 gllm_inference/schema/token_usage.pyi,sha256=WJiGQyz5qatzBK2b-sABLCyTRLCBbAvxCRcqSJOzu-8,3025
 gllm_inference/schema/tool_call.pyi,sha256=OWT9LUqs_xfUcOkPG0aokAAqzLYYDkfnjTa0zOWvugk,403
@@ -119,7 +121,7 @@ gllm_inference/utils/io_utils.pyi,sha256=Eg7dvHWdXslTKdjh1j3dG50i7r35XG2zTmJ9XXv
 gllm_inference/utils/langchain.pyi,sha256=4AwFiVAO0ZpdgmqeC4Pb5NJwBt8vVr0MSUqLeCdTscc,1194
 gllm_inference/utils/validation.pyi,sha256=-RdMmb8afH7F7q4Ao7x6FbwaDfxUHn3hA3WiOgzB-3s,397
 gllm_inference.build/.gitignore,sha256=aEiIwOuxfzdCmLZe4oB1JsBmCUxwG8x-u-HBCV9JT8E,1
-gllm_inference_binary-0.5.42.dist-info/METADATA,sha256=tpuAFFqRUaKa8IeR8xqcX5b0j4e_BkVO9xA7AnIfNGY,5770
-gllm_inference_binary-0.5.42.dist-info/WHEEL,sha256=O_u6PJIQ2pIcyIInxVQ9r-yArMuUZbBIaF1kpYVkYxA,96
-gllm_inference_binary-0.5.42.dist-info/top_level.txt,sha256=FpOjtN80F-qVNgbScXSEyqa0w09FYn6301iq6qt69IQ,15
-gllm_inference_binary-0.5.42.dist-info/RECORD,,
+gllm_inference_binary-0.5.44.dist-info/METADATA,sha256=k03CIyjygon71qyD5--ATKsmuR5bi1IZyMotR-Gyqes,5852
+gllm_inference_binary-0.5.44.dist-info/WHEEL,sha256=O_u6PJIQ2pIcyIInxVQ9r-yArMuUZbBIaF1kpYVkYxA,96
+gllm_inference_binary-0.5.44.dist-info/top_level.txt,sha256=FpOjtN80F-qVNgbScXSEyqa0w09FYn6301iq6qt69IQ,15
+gllm_inference_binary-0.5.44.dist-info/RECORD,,