blaxel 0.2.31__py3-none-any.whl → 0.2.31rc120__py3-none-any.whl

blaxel/core/tools/__init__.py

@@ -8,12 +8,7 @@ from typing import Any, cast
 import httpx
 from mcp import ClientSession
 from mcp.client.streamable_http import streamablehttp_client
-from mcp.types import (
-    CallToolRequest,
-    CallToolRequestParams,
-    CallToolResult,
-    ClientRequest,
-)
+from mcp.types import CallToolRequest, CallToolRequestParams, CallToolResult, ClientRequest
 from mcp.types import Tool as MCPTool
 
 from ..common.internal import get_forced_url, get_global_unique_hash
@@ -29,13 +24,7 @@ if os.getenv("BL_SERVER_PORT"):
 
 
 class PersistentMcpClient:
-    def __init__(
-        self,
-        name: str,
-        timeout: int = DEFAULT_TIMEOUT,
-        timeout_enabled: bool = True,
-        transport: str = None,
-    ):
+    def __init__(self, name: str, timeout: int = DEFAULT_TIMEOUT, timeout_enabled: bool = True, transport: str = None):
         self.name = name
         self.timeout = timeout
         self.type = "function"
@@ -108,9 +97,7 @@ class PersistentMcpClient:
             await self.initialize()
             if self.timeout_enabled:
                 self._remove_timer()
-            logger.debug(
-                f"Calling tool {tool_name} with arguments {arguments} and meta {self.metas}"
-            )
+            logger.debug(f"Calling tool {tool_name} with arguments {arguments} and meta {self.metas}")
 
             # Pass meta as a separate field instead of merging into arguments
             # This matches the TypeScript SDK pattern and MCP protocol specification
@@ -183,9 +170,7 @@ class PersistentMcpClient:
 
         except Exception as e:
             # Default to websocket if we can't determine the transport type
-            logger.warning(
-                f"Failed to detect transport type for {self.name}: {e}. Defaulting to websocket."
-            )
+            logger.warning(f"Failed to detect transport type for {self.name}: {e}. Defaulting to websocket.")
             self.transport_name = "websocket"
 
         return self.transport_name
@@ -341,10 +326,7 @@ class BlTools:
         if not toolPersistances.get(name):
             logger.debug(f"Creating new persistent connection for {name}")
             toolPersistances[name] = PersistentMcpClient(
-                name,
-                timeout=self.timeout,
-                timeout_enabled=self.timeout_enabled,
-                transport=self.transport,
+                name, timeout=self.timeout, timeout_enabled=self.timeout_enabled, transport=self.transport
             )
             await toolPersistances[name].list_tools()
         logger.debug(f"Loaded {len(toolPersistances[name].get_tools())} tools")
@@ -364,10 +346,4 @@ def bl_tools(
     timeout_enabled: bool = True,
     transport: str = None,
 ) -> BlTools:
-    return BlTools(
-        functions,
-        metas=metas,
-        timeout=timeout,
-        timeout_enabled=timeout_enabled,
-        transport=transport,
-    )
+    return BlTools(functions, metas=metas, timeout=timeout, timeout_enabled=timeout_enabled, transport=transport)

blaxel/core/tools/common.py

@@ -42,4 +42,4 @@ def create_model_from_json_schema(
             field_type,
             Field(default_value, description=field_schema.get("description", "")),
         )
-    return create_model(model_name, **fields)
+    return create_model(model_name, **fields)

blaxel/core/tools/types.py

@@ -12,10 +12,9 @@ class ToolException(Exception):  # noqa: N818
     to the agent as observation, and printed in red on the console.
     """
 
-
 class Tool(BaseModel):
     name: str
     description: str
     input_schema: Dict[str, Any]
     coroutine: Callable[..., Awaitable[Any]] | None = None
-    sync_coroutine: Callable[..., Any] | None = None
+    sync_coroutine: Callable[..., Any] | None = None

blaxel/crewai/model.py

@@ -18,24 +18,15 @@ async def bl_model(name: str, **kwargs):
     url, type, model = await bl_model_core(name).get_parameters()
     if type == "mistral":
         return AuthenticatedLLM(
-            model=f"mistral/{model}",
-            api_key="replaced",
-            base_url=f"{url}/v1",
-            **kwargs,
+            model=f"mistral/{model}", api_key="replaced", base_url=f"{url}/v1", **kwargs
         )
     elif type == "xai":
         return AuthenticatedLLM(
-            model=f"groq/{model}",
-            api_key="replaced",
-            base_url=f"{url}/v1",
-            **kwargs,
+            model=f"groq/{model}", api_key="replaced", base_url=f"{url}/v1", **kwargs
         )
     elif type == "deepseek":
         return AuthenticatedLLM(
-            model=f"openai/{model}",
-            api_key="replaced",
-            base_url=f"{url}/v1",
-            **kwargs,
+            model=f"openai/{model}", api_key="replaced", base_url=f"{url}/v1", **kwargs
         )
     elif type == "anthropic":
         return AuthenticatedLLM(
@@ -53,17 +44,11 @@ async def bl_model(name: str, **kwargs):
         )
     elif type == "cerebras":
         return AuthenticatedLLM(
-            model=f"cerebras/{model}",
-            api_key="replaced",
-            base_url=f"{url}/v1",
-            **kwargs,
+            model=f"cerebras/{model}", api_key="replaced", base_url=f"{url}/v1", **kwargs
         )
     else:
         if type != "openai":
             logger.warning(f"Model {model} is not supported by CrewAI, defaulting to OpenAI")
         return AuthenticatedLLM(
-            model=f"openai/{model}",
-            api_key="replaced",
-            base_url=f"{url}/v1",
-            **kwargs,
+            model=f"openai/{model}", api_key="replaced", base_url=f"{url}/v1", **kwargs
         )

blaxel/googleadk/__init__.py

@@ -6,4 +6,4 @@ from .model import *  # noqa: F403, F401
 from .tools import *  # noqa: F403, F401
 
 if "GEMINI_API_KEY" not in os.environ:
-    os.environ["GEMINI_API_KEY"] = "DUMMY_KEY"
+    os.environ["GEMINI_API_KEY"] = "DUMMY_KEY"

blaxel/googleadk/tools.py

@@ -55,11 +55,13 @@ class GoogleADKTool(BaseTool):
         return function_decl
 
     @override
-    async def run_async(self, *, args: dict[str, Any], tool_context: ToolContext) -> Any:
+    async def run_async(
+        self, *, args: dict[str, Any], tool_context: ToolContext
+    ) -> Any:
         args_to_call = args.copy()
         signature = inspect.signature(self._tool.coroutine)
-        if "tool_context" in signature.parameters:
-            args_to_call["tool_context"] = tool_context
+        if 'tool_context' in signature.parameters:
+            args_to_call['tool_context'] = tool_context
         return await self._tool.coroutine(**args_to_call) or {}