blaxel 0.1.22rc70__py3-none-any.whl → 0.2.0rc2__py3-none-any.whl

blaxel/__init__.py

@@ -1,7 +1,10 @@
-from .common.autoload import autoload
-from .common.env import env
-from .common.settings import settings
+"""Blaxel - AI development platform SDK."""
+
+from .core.common.autoload import autoload
+from .core.common.env import env
+from .core.common.settings import settings
 
 autoload()
 
+__version__ = "0.2.0"
 __all__ = ["autoload", "settings", "env"]

blaxel/core/__init__.py

@@ -0,0 +1,44 @@
+"""Blaxel core module."""
+
+from .agents import BlAgent, bl_agent
+from .authentication import BlaxelAuth, auth, get_credentials
+from .cache import find_from_cache
+from .client.client import client
+from .common import autoload, env, settings
+from .jobs import BlJobWrapper
+from .mcp import BlaxelMcpServerTransport, websocket_client
+from .models import BLModel, bl_model
+from .sandbox import (
+    Sandbox,
+    SandboxFileSystem,
+    SandboxInstance,
+    SandboxPreviews,
+    SandboxProcess,
+)
+from .tools import BlTools, bl_tools, convert_mcp_tool_to_blaxel_tool
+
+__all__ = [
+    "BlAgent",
+    "bl_agent",
+    "BlaxelAuth",
+    "auth",
+    "get_credentials",
+    "find_from_cache",
+    "client",
+    "settings",
+    "env",
+    "autoload",
+    "BlJobWrapper",
+    "BlaxelMcpServerTransport",
+    "BLModel",
+    "bl_model",
+    "Sandbox",
+    "SandboxFileSystem",
+    "SandboxInstance",
+    "SandboxPreviews",
+    "SandboxProcess",
+    "BlTools",
+    "bl_tools",
+    "convert_mcp_tool_to_blaxel_tool",
+    "websocket_client",
+]

blaxel/{agents → core/agents}/__init__.py

@@ -8,10 +8,10 @@ from ..client.api.agents import get_agent
 from ..client.models import Agent
 from ..common.internal import get_forced_url, get_global_unique_hash
 from ..common.settings import settings
-from ..instrumentation.span import SpanManager
 
 logger = getLogger(__name__)
 
+
 class BlAgent:
     def __init__(self, name: str):
         self.name = name
@@ -51,13 +51,7 @@ class BlAgent:
             body = json.dumps(body)
 
         return client.get_httpx_client().post(
-            url,
-            headers={
-                'Content-Type': 'application/json',
-                **headers
-            },
-            data=body,
-            params=params
+            url, headers={"Content-Type": "application/json", **headers}, data=body, params=params
         )
 
     async def acall(self, url, input_data, headers: dict = {}, params: dict = {}):
@@ -67,54 +61,38 @@ class BlAgent:
             body = json.dumps(body)
 
         return await client.get_async_httpx_client().post(
-            url,
-            headers={
-                'Content-Type': 'application/json',
-                **headers
-            },
-            data=body,
-            params=params
+            url, headers={"Content-Type": "application/json", **headers}, data=body, params=params
         )
 
     def run(self, input: Any, headers: dict = {}, params: dict = {}) -> str:
-        attributes = {
-            "agent.name": self.name,
-            "agent.args": json.dumps(input),
-            "span.type": "agent.run",
-        }
-        with SpanManager("blaxel-tracer").create_active_span(self.name, attributes) as span:
-            logger.debug(f"Agent Calling: {self.name}")
-            response = self.call(self.url, input, headers, params)
+        logger.debug(f"Agent Calling: {self.name}")
+        response = self.call(self.url, input, headers, params)
+        if response.status_code >= 400:
+            if not self.fallback_url:
+                raise Exception(
+                    f"Agent {self.name} returned status code {response.status_code} with body {response.text}"
+                )
+            response = self.call(self.fallback_url, input, headers, params)
             if response.status_code >= 400:
-                if not self.fallback_url:
-                    span.set_attribute("agent.run.error", response.text)
-                    raise Exception(f"Agent {self.name} returned status code {response.status_code} with body {response.text}")
-                response = self.call(self.fallback_url, input, headers, params)
-                if response.status_code >= 400:
-                    span.set_attribute("agent.run.error", response.text)
-                    raise Exception(f"Agent {self.name} returned status code {response.status_code} with body {response.text}")
-            span.set_attribute("agent.run.result", response.text)
-            return response.text
+                raise Exception(
+                    f"Agent {self.name} returned status code {response.status_code} with body {response.text}"
+                )
+        return response.text
 
     async def arun(self, input: Any, headers: dict = {}, params: dict = {}) -> Awaitable[str]:
-        attributes = {
-            "agent.name": self.name,
-            "agent.args": json.dumps(input),
-            "span.type": "agent.run",
-        }
-        with SpanManager("blaxel-tracer").create_active_span(self.name, attributes) as span:
-            logger.debug(f"Agent Calling: {self.name}")
-            response = await self.acall(self.url, input, headers, params)
+        logger.debug(f"Agent Calling: {self.name}")
+        response = await self.acall(self.url, input, headers, params)
+        if response.status_code >= 400:
+            if not self.fallback_url:
+                raise Exception(
+                    f"Agent {self.name} returned status code {response.status_code} with body {response.text}"
+                )
+            response = await self.acall(self.fallback_url, input, headers, params)
             if response.status_code >= 400:
-                if not self.fallback_url:
-                    span.set_attribute("agent.run.error", response.text)
-                    raise Exception(f"Agent {self.name} returned status code {response.status_code} with body {response.text}")
-                response = await self.acall(self.fallback_url, input, headers, params)
-                if response.status_code >= 400:
-                    span.set_attribute("agent.run.error", response.text)
-                    raise Exception(f"Agent {self.name} returned status code {response.status_code} with body {response.text}")
-            span.set_attribute("agent.run.result", response.text)
-            return response.text
+                raise Exception(
+                    f"Agent {self.name} returned status code {response.status_code} with body {response.text}"
+                )
+        return response.text
 
     def __str__(self):
         return f"Agent {self.name}"
@@ -122,11 +100,13 @@ class BlAgent:
     def __repr__(self):
         return self.__str__()
 
+
 def bl_agent(name: str):
     return BlAgent(name)
 
+
 async def get_agent_metadata(name):
-    cache_data = await find_from_cache('Agent', name)
+    cache_data = await find_from_cache("Agent", name)
     if cache_data:
         return Agent.from_dict(cache_data)
     try:

blaxel/{authentication → core/authentication}/apikey.py

@@ -13,6 +13,7 @@ class ApiKey(BlaxelAuth):
     """
     ApiKey auth that authenticates requests using an API key.
     """
+
     def get_headers(self):
         """
         Retrieves the authentication headers containing the API key and workspace information.

blaxel/{authentication → core/authentication}/clientcredentials.py

@@ -27,6 +27,7 @@ class ClientCredentials(BlaxelAuth):
     """
     ClientCredentials auth that authenticates requests using client credentials.
     """
+
     expires_at: Optional[datetime] = None
 
     def get_headers(self):
@@ -58,7 +59,10 @@ class ClientCredentials(BlaxelAuth):
             Optional[Exception]: An exception if refreshing fails after all retries, otherwise None.
         """
         try:
-            headers = {"Authorization": f"Basic {self.credentials.client_credentials}", "Content-Type": "application/json"}
+            headers = {
+                "Authorization": f"Basic {self.credentials.client_credentials}",
+                "Content-Type": "application/json",
+            }
             body = {"grant_type": "client_credentials"}
             response = requests.post(f"{self.base_url}/oauth/token", headers=headers, json=body)
             response.raise_for_status()
@@ -113,4 +117,4 @@ class ClientCredentials(BlaxelAuth):
     def token(self):
         if not self.credentials.access_token:
             self.get_token()
-        return self.credentials.access_token
+        return self.credentials.access_token

blaxel/{client → core/client}/api/agents/create_agent.py

@@ -20,7 +20,7 @@ def _get_kwargs(
         "url": "/agents",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/agents/update_agent.py

@@ -21,7 +21,7 @@ def _get_kwargs(
         "url": f"/agents/{agent_name}",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/compute/create_sandbox.py

@@ -20,7 +20,7 @@ def _get_kwargs(
         "url": "/sandboxes",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/compute/create_sandbox_preview.py

@@ -21,7 +21,7 @@ def _get_kwargs(
         "url": f"/sandboxes/{sandbox_name}/previews",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/compute/create_sandbox_preview_token.py

@@ -22,7 +22,7 @@ def _get_kwargs(
         "url": f"/sandboxes/{sandbox_name}/previews/{preview_name}/tokens",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/compute/update_sandbox.py

@@ -21,7 +21,7 @@ def _get_kwargs(
         "url": f"/sandboxes/{sandbox_name}",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/compute/update_sandbox_preview.py

@@ -22,7 +22,7 @@ def _get_kwargs(
         "url": f"/sandboxes/{sandbox_name}/previews/{preview_name}",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/functions/create_function.py

@@ -20,7 +20,7 @@ def _get_kwargs(
         "url": "/functions",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/functions/update_function.py

@@ -21,7 +21,7 @@ def _get_kwargs(
         "url": f"/functions/{function_name}",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/integrations/create_integration_connection.py

@@ -20,7 +20,7 @@ def _get_kwargs(
         "url": "/integrations/connections",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/integrations/update_integration_connection.py

@@ -21,7 +21,7 @@ def _get_kwargs(
         "url": f"/integrations/connections/{connection_name}",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/jobs/create_job.py

@@ -20,7 +20,7 @@ def _get_kwargs(
         "url": "/jobs",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/jobs/update_job.py

@@ -21,7 +21,7 @@ def _get_kwargs(
         "url": f"/jobs/{job_id}",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/knowledgebases/create_knowledgebase.py

@@ -20,7 +20,7 @@ def _get_kwargs(
         "url": "/knowledgebases",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/knowledgebases/update_knowledgebase.py

@@ -21,7 +21,7 @@ def _get_kwargs(
         "url": f"/knowledgebases/{knowledgebase_name}",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/models/create_model.py

@@ -20,7 +20,7 @@ def _get_kwargs(
         "url": "/models",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/models/update_model.py

@@ -21,7 +21,7 @@ def _get_kwargs(
         "url": f"/models/{model_name}",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/policies/create_policy.py

@@ -20,7 +20,7 @@ def _get_kwargs(
         "url": "/policies",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/policies/update_policy.py

@@ -21,7 +21,7 @@ def _get_kwargs(
         "url": f"/policies/{policy_name}",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/service_accounts/create_api_key_for_service_account.py

@@ -22,7 +22,7 @@ def _get_kwargs(
         "url": f"/service_accounts/{client_id}/api_keys",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/service_accounts/create_workspace_service_account.py

@@ -23,7 +23,7 @@ def _get_kwargs(
         "url": "/service_accounts",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/service_accounts/update_workspace_service_account.py

@@ -24,7 +24,7 @@ def _get_kwargs(
         "url": f"/service_accounts/{client_id}",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/workspaces/check_workspace_availability.py

@@ -20,7 +20,7 @@ def _get_kwargs(
         "url": "/workspaces/availability",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/workspaces/create_worspace.py

@@ -20,7 +20,7 @@ def _get_kwargs(
         "url": "/workspaces",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/workspaces/invite_workspace_user.py

@@ -21,7 +21,7 @@ def _get_kwargs(
         "url": "/users",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/workspaces/update_workspace.py

@@ -21,7 +21,7 @@ def _get_kwargs(
         "url": f"/workspaces/{workspace_name}",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/api/workspaces/update_workspace_user_role.py

@@ -22,7 +22,7 @@ def _get_kwargs(
         "url": f"/users/{sub_or_email}",
     }
 
-    if type(body) == dict:
+    if type(body) is dict:
         _body = body
     else:
         _body = body.to_dict()

blaxel/{client → core/client}/models/agent.py

@@ -36,7 +36,7 @@ class Agent:
         if not isinstance(self.events, Unset):
             events = []
             for componentsschemas_core_events_item_data in self.events:
-                if type(componentsschemas_core_events_item_data) == dict:
+                if type(componentsschemas_core_events_item_data) is dict:
                     componentsschemas_core_events_item = componentsschemas_core_events_item_data
                 else:
                     componentsschemas_core_events_item = componentsschemas_core_events_item_data.to_dict()

blaxel/{client → core/client}/models/agent_spec.py

@@ -78,7 +78,7 @@ class AgentSpec:
         if not isinstance(self.flavors, Unset):
             flavors = []
             for componentsschemas_flavors_item_data in self.flavors:
-                if type(componentsschemas_flavors_item_data) == dict:
+                if type(componentsschemas_flavors_item_data) is dict:
                     componentsschemas_flavors_item = componentsschemas_flavors_item_data
                 else:
                     componentsschemas_flavors_item = componentsschemas_flavors_item_data.to_dict()
@@ -140,7 +140,7 @@ class AgentSpec:
         if not isinstance(self.triggers, Unset):
             triggers = []
             for componentsschemas_triggers_item_data in self.triggers:
-                if type(componentsschemas_triggers_item_data) == dict:
+                if type(componentsschemas_triggers_item_data) is dict:
                     componentsschemas_triggers_item = componentsschemas_triggers_item_data
                 else:
                     componentsschemas_triggers_item = componentsschemas_triggers_item_data.to_dict()

blaxel/{client → core/client}/models/core_spec.py

@@ -60,7 +60,7 @@ class CoreSpec:
         if not isinstance(self.flavors, Unset):
             flavors = []
             for componentsschemas_flavors_item_data in self.flavors:
-                if type(componentsschemas_flavors_item_data) == dict:
+                if type(componentsschemas_flavors_item_data) is dict:
                     componentsschemas_flavors_item = componentsschemas_flavors_item_data
                 else:
                     componentsschemas_flavors_item = componentsschemas_flavors_item_data.to_dict()

blaxel/{client → core/client}/models/function.py

@@ -36,7 +36,7 @@ class Function:
         if not isinstance(self.events, Unset):
             events = []
             for componentsschemas_core_events_item_data in self.events:
-                if type(componentsschemas_core_events_item_data) == dict:
+                if type(componentsschemas_core_events_item_data) is dict:
                     componentsschemas_core_events_item = componentsschemas_core_events_item_data
                 else:
                     componentsschemas_core_events_item = componentsschemas_core_events_item_data.to_dict()

blaxel/{client → core/client}/models/function_schema_properties.py

@@ -19,7 +19,7 @@ class FunctionSchemaProperties:
     def to_dict(self) -> dict[str, Any]:
         field_dict: dict[str, Any] = {}
         for prop_name, prop in self.additional_properties.items():
-            if type(prop) == dict:
+            if type(prop) is dict:
                 field_dict[prop_name] = prop
             else:
                 field_dict[prop_name] = prop.to_dict()

blaxel/{client → core/client}/models/function_spec.py

@@ -68,7 +68,7 @@ class FunctionSpec:
         if not isinstance(self.flavors, Unset):
             flavors = []
             for componentsschemas_flavors_item_data in self.flavors:
-                if type(componentsschemas_flavors_item_data) == dict:
+                if type(componentsschemas_flavors_item_data) is dict:
                     componentsschemas_flavors_item = componentsschemas_flavors_item_data
                 else:
                     componentsschemas_flavors_item = componentsschemas_flavors_item_data.to_dict()
@@ -112,7 +112,7 @@ class FunctionSpec:
         if not isinstance(self.kit, Unset):
             kit = []
             for kit_item_data in self.kit:
-                if type(kit_item_data) == dict:
+                if type(kit_item_data) is dict:
                     kit_item = kit_item_data
                 else:
                     kit_item = kit_item_data.to_dict()

blaxel/{client → core/client}/models/integration.py

@@ -69,7 +69,7 @@ class Integration:
         if not isinstance(self.organizations, Unset):
             organizations = []
             for organizations_item_data in self.organizations:
-                if type(organizations_item_data) == dict:
+                if type(organizations_item_data) is dict:
                     organizations_item = organizations_item_data
                 else:
                     organizations_item = organizations_item_data.to_dict()
@@ -85,7 +85,7 @@ class Integration:
         if not isinstance(self.repositories, Unset):
             repositories = []
             for repositories_item_data in self.repositories:
-                if type(repositories_item_data) == dict:
+                if type(repositories_item_data) is dict:
                     repositories_item = repositories_item_data
                 else:
                     repositories_item = repositories_item_data.to_dict()

blaxel/{client → core/client}/models/integration_endpoints.py

@@ -19,7 +19,7 @@ class IntegrationEndpoints:
     def to_dict(self) -> dict[str, Any]:
         field_dict: dict[str, Any] = {}
         for prop_name, prop in self.additional_properties.items():
-            if type(prop) == dict:
+            if type(prop) is dict:
                 field_dict[prop_name] = prop
             else:
                 field_dict[prop_name] = prop.to_dict()

blaxel/{client → core/client}/models/job.py

@@ -36,7 +36,7 @@ class Job:
         if not isinstance(self.events, Unset):
             events = []
             for componentsschemas_core_events_item_data in self.events:
-                if type(componentsschemas_core_events_item_data) == dict:
+                if type(componentsschemas_core_events_item_data) is dict:
                     componentsschemas_core_events_item = componentsschemas_core_events_item_data
                 else:
                     componentsschemas_core_events_item = componentsschemas_core_events_item_data.to_dict()

blaxel/{client → core/client}/models/job_spec.py

@@ -60,7 +60,7 @@ class JobSpec:
         if not isinstance(self.flavors, Unset):
             flavors = []
             for componentsschemas_flavors_item_data in self.flavors:
-                if type(componentsschemas_flavors_item_data) == dict:
+                if type(componentsschemas_flavors_item_data) is dict:
                     componentsschemas_flavors_item = componentsschemas_flavors_item_data
                 else:
                     componentsschemas_flavors_item = componentsschemas_flavors_item_data.to_dict()

blaxel/{client → core/client}/models/knowledgebase.py

@@ -36,7 +36,7 @@ class Knowledgebase:
         if not isinstance(self.events, Unset):
             events = []
             for componentsschemas_core_events_item_data in self.events:
-                if type(componentsschemas_core_events_item_data) == dict:
+                if type(componentsschemas_core_events_item_data) is dict:
                     componentsschemas_core_events_item = componentsschemas_core_events_item_data
                 else:
                     componentsschemas_core_events_item = componentsschemas_core_events_item_data.to_dict()

blaxel/{client → core/client}/models/location_response.py

@@ -40,7 +40,7 @@ class LocationResponse:
         if not isinstance(self.flavors, Unset):
             flavors = []
             for flavors_item_data in self.flavors:
-                if type(flavors_item_data) == dict:
+                if type(flavors_item_data) is dict:
                     flavors_item = flavors_item_data
                 else:
                     flavors_item = flavors_item_data.to_dict()