blaxel 0.1.9rc35__py3-none-any.whl → 0.1.9rc36__py3-none-any.whl

blaxel/client/models/trigger.py

@@ -0,0 +1,98 @@
+from typing import TYPE_CHECKING, Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.trigger_configuration import TriggerConfiguration
+
+
+T = TypeVar("T", bound="Trigger")
+
+
+@_attrs_define
+class Trigger:
+    """Trigger configuration
+
+    Attributes:
+        configuration (Union[Unset, TriggerConfiguration]): The configuration of the trigger
+        id (Union[Unset, str]): The id of the trigger
+        type_ (Union[Unset, str]): The type of trigger, can be http or http-async
+    """
+
+    configuration: Union[Unset, "TriggerConfiguration"] = UNSET
+    id: Union[Unset, str] = UNSET
+    type_: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        configuration: Union[Unset, dict[str, Any]] = UNSET
+        if (
+            self.configuration
+            and not isinstance(self.configuration, Unset)
+            and not isinstance(self.configuration, dict)
+        ):
+            configuration = self.configuration.to_dict()
+        elif self.configuration and isinstance(self.configuration, dict):
+            configuration = self.configuration
+
+        id = self.id
+
+        type_ = self.type_
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if configuration is not UNSET:
+            field_dict["configuration"] = configuration
+        if id is not UNSET:
+            field_dict["id"] = id
+        if type_ is not UNSET:
+            field_dict["type"] = type_
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.trigger_configuration import TriggerConfiguration
+
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        _configuration = d.pop("configuration", UNSET)
+        configuration: Union[Unset, TriggerConfiguration]
+        if isinstance(_configuration, Unset):
+            configuration = UNSET
+        else:
+            configuration = TriggerConfiguration.from_dict(_configuration)
+
+        id = d.pop("id", UNSET)
+
+        type_ = d.pop("type", UNSET)
+
+        trigger = cls(
+            configuration=configuration,
+            id=id,
+            type_=type_,
+        )
+
+        trigger.additional_properties = d
+        return trigger
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/client/models/trigger_configuration.py

@@ -0,0 +1,45 @@
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="TriggerConfiguration")
+
+
+@_attrs_define
+class TriggerConfiguration:
+    """The configuration of the trigger"""
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        trigger_configuration = cls()
+
+        trigger_configuration.additional_properties = d
+        return trigger_configuration
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/client/models/workspace.py

@@ -7,6 +7,7 @@ from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
     from ..models.workspace_labels import WorkspaceLabels
+    from ..models.workspace_runtime import WorkspaceRuntime
 
 
 T = TypeVar("T", bound="Workspace")
@@ -26,6 +27,7 @@ class Workspace:
         labels (Union[Unset, WorkspaceLabels]): Workspace labels
         name (Union[Unset, str]): Workspace name
         region (Union[Unset, str]): Workspace write region
+        runtime (Union[Unset, WorkspaceRuntime]): Workspace runtime
     """
 
     created_at: Union[Unset, str] = UNSET
@@ -37,6 +39,7 @@ class Workspace:
     labels: Union[Unset, "WorkspaceLabels"] = UNSET
     name: Union[Unset, str] = UNSET
     region: Union[Unset, str] = UNSET
+    runtime: Union[Unset, "WorkspaceRuntime"] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -62,6 +65,12 @@ class Workspace:
 
         region = self.region
 
+        runtime: Union[Unset, dict[str, Any]] = UNSET
+        if self.runtime and not isinstance(self.runtime, Unset) and not isinstance(self.runtime, dict):
+            runtime = self.runtime.to_dict()
+        elif self.runtime and isinstance(self.runtime, dict):
+            runtime = self.runtime
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
@@ -83,12 +92,15 @@ class Workspace:
             field_dict["name"] = name
         if region is not UNSET:
             field_dict["region"] = region
+        if runtime is not UNSET:
+            field_dict["runtime"] = runtime
 
         return field_dict
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.workspace_labels import WorkspaceLabels
+        from ..models.workspace_runtime import WorkspaceRuntime
 
         if not src_dict:
             return None
@@ -116,6 +128,13 @@ class Workspace:
 
         region = d.pop("region", UNSET)
 
+        _runtime = d.pop("runtime", UNSET)
+        runtime: Union[Unset, WorkspaceRuntime]
+        if isinstance(_runtime, Unset):
+            runtime = UNSET
+        else:
+            runtime = WorkspaceRuntime.from_dict(_runtime)
+
         workspace = cls(
             created_at=created_at,
             updated_at=updated_at,
@@ -126,6 +145,7 @@ class Workspace:
             labels=labels,
             name=name,
             region=region,
+            runtime=runtime,
         )
 
         workspace.additional_properties = d

blaxel/client/models/workspace_runtime.py

@@ -0,0 +1,61 @@
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="WorkspaceRuntime")
+
+
+@_attrs_define
+class WorkspaceRuntime:
+    """Workspace runtime
+
+    Attributes:
+        generation (Union[Unset, str]): Workspace generation
+    """
+
+    generation: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        generation = self.generation
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if generation is not UNSET:
+            field_dict["generation"] = generation
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        generation = d.pop("generation", UNSET)
+
+        workspace_runtime = cls(
+            generation=generation,
+        )
+
+        workspace_runtime.additional_properties = d
+        return workspace_runtime
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/common/autoload.py

@@ -1,5 +1,4 @@
 
-import logging
 
 from ..client import client
 from ..instrumentation.manager import telemetry_manager

blaxel/instrumentation/exporters.py

@@ -1,11 +1,8 @@
 from typing import Callable, Dict, Sequence
 
-from opentelemetry.exporter.otlp.proto.http._log_exporter import \
-    OTLPLogExporter
-from opentelemetry.exporter.otlp.proto.http.metric_exporter import \
-    OTLPMetricExporter
-from opentelemetry.exporter.otlp.proto.http.trace_exporter import \
-    OTLPSpanExporter
+from opentelemetry.exporter.otlp.proto.http._log_exporter import OTLPLogExporter
+from opentelemetry.exporter.otlp.proto.http.metric_exporter import OTLPMetricExporter
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
 from opentelemetry.sdk._logs import LogData
 from opentelemetry.sdk.metrics.export import MetricExportResult, MetricsData
 

blaxel/instrumentation/manager.py

@@ -20,9 +20,11 @@ from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor
 from opentelemetry.trace import NoOpTracerProvider
 
-from blaxel.instrumentation.exporters import (DynamicHeadersLogExporter,
-                                              DynamicHeadersMetricExporter,
-                                              DynamicHeadersSpanExporter)
+from blaxel.instrumentation.exporters import (
+    DynamicHeadersLogExporter,
+    DynamicHeadersMetricExporter,
+    DynamicHeadersSpanExporter,
+)
 from blaxel.instrumentation.span import DefaultAttributesSpanProcessor
 
 from ..common.settings import Settings

blaxel/mcp/client.py

@@ -1,5 +1,4 @@
 import logging
-import os
 from contextlib import asynccontextmanager
 from typing import Any
 from urllib.parse import urljoin, urlparse
@@ -7,8 +6,7 @@ from urllib.parse import urljoin, urlparse
 import anyio
 import mcp.types as types
 from anyio.abc import TaskStatus
-from anyio.streams.memory import (MemoryObjectReceiveStream,
-                                  MemoryObjectSendStream)
+from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
 from websockets.client import WebSocketClientProtocol
 from websockets.client import connect as ws_connect
 

blaxel/mcp/server.py

@@ -1,19 +1,18 @@
 import logging
+import traceback
 import uuid
 from contextlib import asynccontextmanager
 from typing import Dict, Literal
 
 import anyio
 import mcp.types as types
-from anyio.streams.memory import (MemoryObjectReceiveStream,
-                                  MemoryObjectSendStream)
+from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
 from mcp.server.fastmcp import FastMCP as FastMCPBase
 from opentelemetry.trace import Span, StatusCode
 from websockets.server import WebSocketServerProtocol, serve
 
 from ..common.env import env
 from ..instrumentation.span import SpanManager
-import traceback
 
 logger = logging.getLogger(__name__)
 

blaxel/models/__init__.py

@@ -4,6 +4,7 @@ from ..client import client
 from ..client.api.models import get_model
 from ..client.models import Model
 from ..common.settings import settings
+
 # This has to be here because livekit plugins must be registered on the main thread
 from .livekit import get_livekit_model
 
@@ -94,7 +95,7 @@ class BLModel:
 
         try:
             return await get_model.asyncio(client=client, model_name=self.model_name)
-        except Exception as e:
+        except Exception:
             return None
 
 def bl_model(model_name, **kwargs):

blaxel/models/custom/langchain/gemini.py

@@ -6,33 +6,56 @@ import logging
 import uuid
 import warnings
 from operator import itemgetter
-from typing import (Any, AsyncIterator, Callable, Dict, Iterator, List,
-                    Mapping, Optional, Sequence, Tuple, Type, Union, cast)
+from typing import (
+    Any,
+    AsyncIterator,
+    Callable,
+    Dict,
+    Iterator,
+    List,
+    Mapping,
+    Optional,
+    Sequence,
+    Tuple,
+    Type,
+    Union,
+    cast,
+)
 
 import httpx
 import requests
-from langchain_core.callbacks.manager import (AsyncCallbackManagerForLLMRun,
-                                              CallbackManagerForLLMRun)
+from langchain_core.callbacks.manager import AsyncCallbackManagerForLLMRun, CallbackManagerForLLMRun
 from langchain_core.language_models import LanguageModelInput
-from langchain_core.language_models.chat_models import (BaseChatModel,
-                                                        LangSmithParams)
-from langchain_core.messages import (AIMessage, AIMessageChunk, BaseMessage,
-                                     FunctionMessage, HumanMessage,
-                                     SystemMessage, ToolMessage)
+from langchain_core.language_models.chat_models import BaseChatModel, LangSmithParams
+from langchain_core.messages import (
+    AIMessage,
+    AIMessageChunk,
+    BaseMessage,
+    FunctionMessage,
+    HumanMessage,
+    SystemMessage,
+    ToolMessage,
+)
 from langchain_core.messages.ai import UsageMetadata
-from langchain_core.messages.tool import (invalid_tool_call, tool_call,
-                                          tool_call_chunk)
+from langchain_core.messages.tool import invalid_tool_call, tool_call, tool_call_chunk
 from langchain_core.output_parsers.openai_tools import (
-    JsonOutputKeyToolsParser, PydanticToolsParser, parse_tool_calls)
-from langchain_core.outputs import (ChatGeneration, ChatGenerationChunk,
-                                    ChatResult)
+    JsonOutputKeyToolsParser,
+    PydanticToolsParser,
+    parse_tool_calls,
+)
+from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult
 from langchain_core.runnables import Runnable, RunnablePassthrough
 from langchain_core.tools import BaseTool
 from langchain_core.utils.function_calling import convert_to_openai_tool
 from PIL import Image
 from pydantic import BaseModel, ConfigDict, Field, SecretStr, model_validator
-from tenacity import (before_sleep_log, retry, retry_if_exception_type,
-                      stop_after_attempt, wait_exponential)
+from tenacity import (
+    before_sleep_log,
+    retry,
+    retry_if_exception_type,
+    stop_after_attempt,
+    wait_exponential,
+)
 from typing_extensions import Self, is_typeddict
 
 WARNED_STRUCTURED_OUTPUT_JSON_MODE = False
@@ -189,7 +212,7 @@ def _chat_with_retry(generation_method: Callable, **kwargs: Any) -> Any:
         try:
             # Extract request parameters and other kwargs
             request = kwargs.pop('request', {})
-            metadata = kwargs.pop('metadata', None)
+            kwargs.pop('metadata', None)
 
             # Unpack request parameters into kwargs
             kwargs.update(request)
@@ -238,7 +261,7 @@ async def _achat_with_retry(generation_method: Callable, **kwargs: Any) -> Any:
         try:
             # Extract request parameters and other kwargs
             request = kwargs.pop('request', {})
-            metadata = kwargs.pop('metadata', None)
+            kwargs.pop('metadata', None)
 
             # Unpack request parameters into kwargs
             kwargs.update(request)

blaxel/models/custom/llamaindex/cohere.py

@@ -4,27 +4,36 @@ from typing import Any, Callable, Dict, List, Optional, Sequence, Union
 
 import cohere
 from cohere.types import ToolCall
-from llama_index.core.base.llms.types import (ChatMessage, ChatResponse,
-                                              ChatResponseAsyncGen,
-                                              ChatResponseGen,
-                                              CompletionResponse,
-                                              CompletionResponseAsyncGen,
-                                              CompletionResponseGen,
-                                              LLMMetadata, MessageRole)
+from llama_index.core.base.llms.types import (
+    ChatMessage,
+    ChatResponse,
+    ChatResponseAsyncGen,
+    ChatResponseGen,
+    CompletionResponse,
+    CompletionResponseAsyncGen,
+    CompletionResponseGen,
+    LLMMetadata,
+    MessageRole,
+)
 from llama_index.core.bridge.pydantic import Field, PrivateAttr
 from llama_index.core.callbacks import CallbackManager
-from llama_index.core.llms.callbacks import (llm_chat_callback,
-                                             llm_completion_callback)
+from llama_index.core.llms.callbacks import llm_chat_callback, llm_completion_callback
 from llama_index.core.llms.function_calling import FunctionCallingLLM
 from llama_index.core.llms.llm import ToolSelection
 from llama_index.core.tools.types import BaseTool
 from llama_index.core.types import BaseOutputParser, PydanticProgramMode
 from llama_index.llms.cohere.utils import (
-    CHAT_MODELS, _get_message_cohere_format, _message_to_cohere_tool_results,
-    _messages_to_cohere_tool_results_curr_chat_turn, acompletion_with_retry,
-    cohere_modelname_to_contextsize, completion_with_retry,
-    format_to_cohere_tools, is_cohere_function_calling_model,
-    remove_documents_from_messages)
+    CHAT_MODELS,
+    _get_message_cohere_format,
+    _message_to_cohere_tool_results,
+    _messages_to_cohere_tool_results_curr_chat_turn,
+    acompletion_with_retry,
+    cohere_modelname_to_contextsize,
+    completion_with_retry,
+    format_to_cohere_tools,
+    is_cohere_function_calling_model,
+    remove_documents_from_messages,
+)
 
 
 class Cohere(FunctionCallingLLM):
@@ -200,9 +209,9 @@ class Cohere(FunctionCallingLLM):
             The request for the Cohere chat API.
         """
         additional_kwargs = messages[-1].additional_kwargs
-
+        documents = additional_kwargs.pop("documents", [])
         # cohere SDK will fail loudly if both connectors and documents are provided
-        if additional_kwargs.get("documents", []) and documents and len(documents) > 0:
+        if documents and len(documents) > 0:
             raise ValueError(
                 "Received documents both as a keyword argument and as an prompt additional keyword argument. Please choose only one option."
             )

blaxel/models/custom/pydantic/gemini.py

@@ -1,4 +1,3 @@
-import os
 
 import httpx
 from httpx import AsyncClient

blaxel/models/livekit.py

@@ -1,6 +1,6 @@
 from logging import getLogger
 
-from livekit.plugins import anthropic, groq, openai
+from livekit.plugins import groq, openai
 
 from ..common.settings import settings
 

blaxel/tools/__init__.py

@@ -126,7 +126,7 @@ def convert_mcp_tool_to_blaxel_tool(
             "tool.server": url,
             "tool.server_name": name,
         }
-        with SpanManager("blaxel-tracer").create_active_span("blaxel-tool-call", span_attributes) as span:
+        with SpanManager("blaxel-tracer").create_active_span("blaxel-tool-call", span_attributes):
             logger.debug(f"Calling tool {tool.name} with arguments {arguments}")
             call_tool_result = await websocket_client.call_tool(tool.name, arguments)
             logger.debug(f"Tool {tool.name} returned {call_tool_result}")

blaxel/tools/langchain.py

@@ -1,8 +1,7 @@
 from typing import Any
 
 from langchain_core.tools import StructuredTool
-from mcp.types import (CallToolResult, EmbeddedResource, ImageContent,
-                       TextContent)
+from mcp.types import CallToolResult, EmbeddedResource, ImageContent, TextContent
 
 from .types import Tool, ToolException
 

{blaxel-0.1.9rc35.dist-info → blaxel-0.1.9rc36.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.1.9rc35
+Version: 0.1.9rc36
 Summary: Add your description here
 Author-email: cploujoux <cploujoux@blaxel.ai>
 License-File: LICENSE
@@ -25,9 +25,6 @@ Requires-Dist: pyyaml<6.1.0,>=6.0.2
 Requires-Dist: requests<2.33.0,>=2.32.3
 Requires-Dist: tomli>=2.2.1
 Requires-Dist: websockets<15.0.0
-Provides-Extra: crewai
-Requires-Dist: crewai>=0.102.0; extra == 'crewai'
-Requires-Dist: opentelemetry-instrumentation-crewai>=0.39.0; extra == 'crewai'
 Provides-Extra: google-adk
 Requires-Dist: google-adk>=0.2.0; extra == 'google-adk'
 Requires-Dist: litellm>=1.63.11; extra == 'google-adk'