blaxel 0.1.21rc69__py3-none-any.whl → 0.2.0__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/core/agents/__init__.py

@@ -0,0 +1,115 @@
+import json
+from logging import getLogger
+from typing import Any, Awaitable
+
+from ..cache import find_from_cache
+from ..client import client
+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
+
+logger = getLogger(__name__)
+
+
+class BlAgent:
+    def __init__(self, name: str):
+        self.name = name
+
+    @property
+    def internal_url(self):
+        """Get the internal URL for the agent using a hash of workspace and agent name."""
+        hash = get_global_unique_hash(settings.workspace, "agent", self.name)
+        return f"{settings.run_internal_protocol}://bl-{settings.env}-{hash}.{settings.run_internal_hostname}"
+
+    @property
+    def forced_url(self):
+        """Get the forced URL from environment variables if set."""
+        return get_forced_url("agent", self.name)
+
+    @property
+    def external_url(self):
+        return f"{settings.run_url}/{settings.workspace}/agents/{self.name}"
+
+    @property
+    def fallback_url(self):
+        if self.external_url != self.url:
+            return self.external_url
+        return None
+
+    @property
+    def url(self):
+        if self.forced_url:
+            return self.forced_url
+        if settings.run_internal_hostname:
+            return self.internal_url
+        return self.external_url
+
+    def call(self, url, input_data, headers: dict = {}, params: dict = {}):
+        body = input_data
+        if not isinstance(body, str):
+            body = json.dumps(body)
+
+        return client.get_httpx_client().post(
+            url, headers={"Content-Type": "application/json", **headers}, data=body, params=params
+        )
+
+    async def acall(self, url, input_data, headers: dict = {}, params: dict = {}):
+        logger.debug(f"Agent Calling: {self.name}")
+        body = input_data
+        if not isinstance(body, str):
+            body = json.dumps(body)
+
+        return await client.get_async_httpx_client().post(
+            url, headers={"Content-Type": "application/json", **headers}, data=body, params=params
+        )
+
+    def run(self, input: Any, headers: dict = {}, params: dict = {}) -> str:
+        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:
+                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]:
+        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:
+                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}"
+
+    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)
+    if cache_data:
+        return Agent.from_dict(cache_data)
+    try:
+        return await get_agent.asyncio(client=client, agent_name=name)
+    except Exception:
+        return None

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/__init__.py

@@ -4,6 +4,7 @@ from .acl import ACL
 from .agent import Agent
 from .agent_spec import AgentSpec
 from .api_key import ApiKey
+from .billable_time_metric import BillableTimeMetric
 from .check_workspace_availability_body import CheckWorkspaceAvailabilityBody
 from .configuration import Configuration
 from .continent import Continent
@@ -71,9 +72,12 @@ from .knowledgebase_spec_options import KnowledgebaseSpecOptions
 from .last_n_requests_metric import LastNRequestsMetric
 from .latency_metric import LatencyMetric
 from .location_response import LocationResponse
+from .logs_response import LogsResponse
+from .logs_response_data import LogsResponseData
 from .mcp_definition import MCPDefinition
 from .mcp_definition_entrypoint import MCPDefinitionEntrypoint
 from .mcp_definition_form import MCPDefinitionForm
+from .memory_allocation_by_name import MemoryAllocationByName
 from .memory_allocation_metric import MemoryAllocationMetric
 from .metadata import Metadata
 from .metadata_labels import MetadataLabels
@@ -123,7 +127,10 @@ from .request_total_metric import RequestTotalMetric
 from .request_total_metric_request_total_per_code import RequestTotalMetricRequestTotalPerCode
 from .request_total_metric_rps_per_code import RequestTotalMetricRpsPerCode
 from .request_total_response_data import RequestTotalResponseData
+from .resource import Resource
 from .resource_log import ResourceLog
+from .resource_log_chart import ResourceLogChart
+from .resource_log_response import ResourceLogResponse
 from .resource_metrics import ResourceMetrics
 from .resource_metrics_request_total_per_code import ResourceMetricsRequestTotalPerCode
 from .resource_metrics_request_total_per_code_previous import (
@@ -131,6 +138,7 @@ from .resource_metrics_request_total_per_code_previous import (
 )
 from .resource_metrics_rps_per_code import ResourceMetricsRpsPerCode
 from .resource_metrics_rps_per_code_previous import ResourceMetricsRpsPerCodePrevious
+from .resource_trace import ResourceTrace
 from .revision_configuration import RevisionConfiguration
 from .revision_metadata import RevisionMetadata
 from .runtime import Runtime
@@ -162,6 +170,7 @@ from .update_workspace_service_account_body import UpdateWorkspaceServiceAccount
 from .update_workspace_service_account_response_200 import UpdateWorkspaceServiceAccountResponse200
 from .update_workspace_user_role_body import UpdateWorkspaceUserRoleBody
 from .websocket_channel import WebsocketChannel
+from .websocket_message import WebsocketMessage
 from .workspace import Workspace
 from .workspace_labels import WorkspaceLabels
 from .workspace_runtime import WorkspaceRuntime
@@ -172,6 +181,7 @@ __all__ = (
     "Agent",
     "AgentSpec",
     "ApiKey",
+    "BillableTimeMetric",
     "CheckWorkspaceAvailabilityBody",
     "Configuration",
     "Continent",
@@ -237,9 +247,12 @@ __all__ = (
     "LastNRequestsMetric",
     "LatencyMetric",
     "LocationResponse",
+    "LogsResponse",
+    "LogsResponseData",
     "MCPDefinition",
     "MCPDefinitionEntrypoint",
     "MCPDefinitionForm",
+    "MemoryAllocationByName",
     "MemoryAllocationMetric",
     "Metadata",
     "MetadataLabels",
@@ -285,12 +298,16 @@ __all__ = (
     "RequestTotalMetricRequestTotalPerCode",
     "RequestTotalMetricRpsPerCode",
     "RequestTotalResponseData",
+    "Resource",
     "ResourceLog",
+    "ResourceLogChart",
+    "ResourceLogResponse",
     "ResourceMetrics",
     "ResourceMetricsRequestTotalPerCode",
     "ResourceMetricsRequestTotalPerCodePrevious",
     "ResourceMetricsRpsPerCode",
     "ResourceMetricsRpsPerCodePrevious",
+    "ResourceTrace",
     "RevisionConfiguration",
     "RevisionMetadata",
     "Runtime",
@@ -322,6 +339,7 @@ __all__ = (
     "UpdateWorkspaceServiceAccountResponse200",
     "UpdateWorkspaceUserRoleBody",
     "WebsocketChannel",
+    "WebsocketMessage",
     "Workspace",
     "WorkspaceLabels",
     "WorkspaceRuntime",

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()