blaxel 0.2.0rc6__py3-none-any.whl → 0.2.1rc71__py3-none-any.whl

blaxel/core/sandbox/sandbox.py

@@ -1,6 +1,7 @@
 import asyncio
 import logging
 import time
+import uuid
 from typing import Any, Dict, List, Union
 
 from ..client.api.compute.create_sandbox import asyncio as create_sandbox
@@ -8,7 +9,7 @@ from ..client.api.compute.delete_sandbox import asyncio as delete_sandbox
 from ..client.api.compute.get_sandbox import asyncio as get_sandbox
 from ..client.api.compute.list_sandboxes import asyncio as list_sandboxes
 from ..client.client import client
-from ..client.models import Metadata, Sandbox
+from ..client.models import Metadata, Runtime, Sandbox, SandboxSpec
 from .filesystem import SandboxFileSystem
 from .network import SandboxNetwork
 from .preview import SandboxPreviews
@@ -67,14 +68,20 @@ class SandboxInstance:
                 raise Exception("Sandbox did not deploy in time")
 
     @classmethod
-    async def create(cls, sandbox: Union[Sandbox, Dict[str, Any]]) -> "SandboxInstance":
+    async def create(
+        cls, sandbox: Union[Sandbox, Dict[str, Any], None] = None
+    ) -> "SandboxInstance":
+        if sandbox is None:
+            sandbox = Sandbox()
         if isinstance(sandbox, dict):
             sandbox = Sandbox.from_dict(sandbox)
-
+        if not sandbox.metadata:
+            sandbox.metadata = Metadata(name=uuid.uuid4().hex)
         if not sandbox.spec:
-            raise Exception("Sandbox spec is required")
+            sandbox.spec = SandboxSpec(runtime=Runtime(image="blaxel/prod-base:latest"))
         if not sandbox.spec.runtime:
-            raise Exception("Sandbox runtime is required")
+            sandbox.spec.runtime = Runtime(image="blaxel/prod-base:latest")
+
         sandbox.spec.runtime.generation = sandbox.spec.runtime.generation or "mk3"
 
         response = await create_sandbox(

blaxel/llamaindex/custom/cohere.py

@@ -47,7 +47,6 @@ class Cohere(FunctionCallingLLM):
 
         llm = Cohere(model="command", api_key=api_key)
         resp = llm.complete("Paul Graham is ")
-        print(resp)
         ```
     """
 
@@ -55,9 +54,7 @@ class Cohere(FunctionCallingLLM):
     temperature: Optional[float] = Field(
         description="The temperature to use for sampling.", default=None
     )
-    max_retries: int = Field(
-        default=10, description="The maximum number of API retries."
-    )
+    max_retries: int = Field(default=10, description="The maximum number of API retries.")
     additional_kwargs: Dict[str, Any] = Field(
         default_factory=dict, description="Additional kwargs for the Cohere API."
     )
@@ -167,9 +164,7 @@ class Cohere(FunctionCallingLLM):
         error_on_no_tool_call: bool = False,
     ) -> List[ToolSelection]:
         """Predict and call the tool."""
-        tool_calls: List[ToolCall] = (
-            response.message.additional_kwargs.get("tool_calls", []) or []
-        )
+        tool_calls: List[ToolCall] = response.message.additional_kwargs.get("tool_calls", []) or []
 
         if len(tool_calls) < 1 and error_on_no_tool_call:
             raise ValueError(
@@ -218,10 +213,8 @@ class Cohere(FunctionCallingLLM):
 
         messages, documents = remove_documents_from_messages(messages)
 
-        tool_results: Optional[
-            List[Dict[str, Any]]
-        ] = _messages_to_cohere_tool_results_curr_chat_turn(messages) or kwargs.get(
-            "tool_results"
+        tool_results: Optional[List[Dict[str, Any]]] = (
+            _messages_to_cohere_tool_results_curr_chat_turn(messages) or kwargs.get("tool_results")
         )
         if not tool_results:
             tool_results = None
@@ -235,12 +228,8 @@ class Cohere(FunctionCallingLLM):
                 if message.role == MessageRole.TOOL:
                     temp_tool_results += _message_to_cohere_tool_results(messages, i)
 
-                    if (i == len(messages) - 1) or messages[
-                        i + 1
-                    ].role != MessageRole.TOOL:
-                        cohere_message = _get_message_cohere_format(
-                            message, temp_tool_results
-                        )
+                    if (i == len(messages) - 1) or messages[i + 1].role != MessageRole.TOOL:
+                        cohere_message = _get_message_cohere_format(message, temp_tool_results)
                         chat_history.append(cohere_message)
                         temp_tool_results = []
                 else:
@@ -262,12 +251,8 @@ class Cohere(FunctionCallingLLM):
                 if message.role == MessageRole.TOOL:
                     temp_tool_results += _message_to_cohere_tool_results(messages, i)
 
-                    if (i == len(messages) - 1) or messages[
-                        i + 1
-                    ].role != MessageRole.TOOL:
-                        cohere_message = _get_message_cohere_format(
-                            message, temp_tool_results
-                        )
+                    if (i == len(messages) - 1) or messages[i + 1].role != MessageRole.TOOL:
+                        cohere_message = _get_message_cohere_format(message, temp_tool_results)
                         chat_history.append(cohere_message)
                         temp_tool_results = []
                 else:
@@ -327,9 +312,7 @@ class Cohere(FunctionCallingLLM):
         )
 
     @llm_completion_callback()
-    def complete(
-        self, prompt: str, formatted: bool = False, **kwargs: Any
-    ) -> CompletionResponse:
+    def complete(self, prompt: str, formatted: bool = False, **kwargs: Any) -> CompletionResponse:
         all_kwargs = self._get_all_kwargs(**kwargs)
         if "stream" in all_kwargs:
             warnings.warn(
@@ -351,9 +334,7 @@ class Cohere(FunctionCallingLLM):
         )
 
     @llm_chat_callback()
-    def stream_chat(
-        self, messages: Sequence[ChatMessage], **kwargs: Any
-    ) -> ChatResponseGen:
+    def stream_chat(self, messages: Sequence[ChatMessage], **kwargs: Any) -> ChatResponseGen:
         all_kwargs = self._get_all_kwargs(**kwargs)
         all_kwargs["stream"] = True
         if all_kwargs["model"] not in CHAT_MODELS:
@@ -402,16 +383,12 @@ class Cohere(FunctionCallingLLM):
             for r in response:
                 content_delta = r.text
                 content += content_delta
-                yield CompletionResponse(
-                    text=content, delta=content_delta, raw=r._asdict()
-                )
+                yield CompletionResponse(text=content, delta=content_delta, raw=r._asdict())
 
         return gen()
 
     @llm_chat_callback()
-    async def achat(
-        self, messages: Sequence[ChatMessage], **kwargs: Any
-    ) -> ChatResponse:
+    async def achat(self, messages: Sequence[ChatMessage], **kwargs: Any) -> ChatResponse:
         all_kwargs = self._get_all_kwargs(**kwargs)
         if all_kwargs["model"] not in CHAT_MODELS:
             raise ValueError(f"{all_kwargs['model']} not supported for chat")
@@ -535,8 +512,6 @@ class Cohere(FunctionCallingLLM):
             async for r in response:
                 content_delta = r.text
                 content += content_delta
-                yield CompletionResponse(
-                    text=content, delta=content_delta, raw=r._asdict()
-                )
+                yield CompletionResponse(text=content, delta=content_delta, raw=r._asdict())
 
         return gen()

blaxel/telemetry/instrumentation/blaxel_core.py

@@ -10,9 +10,10 @@ from blaxel.telemetry.span import SpanManager
 
 logger = logging.getLogger(__name__)
 
+
 class BlaxelCoreInstrumentor(BaseInstrumentor):
     def instrumentation_dependencies(self):
-        return ["blaxel.core"]
+        return []
 
     def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")
@@ -121,4 +122,4 @@ class BlaxelCoreInstrumentor(BaseInstrumentor):
             Tool.sync_coroutine = traced_sync_coroutine
             return Tool
 
-        blaxel.core.tools.convert_mcp_tool_to_blaxel_tool = traced_convert_mcp_tool_to_blaxel_tool
+        blaxel.core.tools.convert_mcp_tool_to_blaxel_tool = traced_convert_mcp_tool_to_blaxel_tool

blaxel/telemetry/instrumentation/blaxel_langgraph.py

@@ -11,7 +11,7 @@ logger = logging.getLogger(__name__)
 
 class BlaxelLanggraphInstrumentor(BaseInstrumentor):
     def instrumentation_dependencies(self):
-        return ["blaxel.langgraph", "blaxel.core"]
+        return ["langgraph"]
 
     def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")

blaxel/telemetry/instrumentation/blaxel_langgraph_gemini.py

@@ -316,7 +316,7 @@ class BlaxelLanggraphGeminiInstrumentor(BaseInstrumentor):
         Config.exception_logger = exception_logger
 
     def instrumentation_dependencies(self) -> Collection[str]:
-        return ["blaxel.langgraph"]
+        return ["langgraph"]
 
     def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")

blaxel/telemetry/instrumentation/blaxel_llamaindex.py

@@ -30,7 +30,7 @@ TO_INSTRUMENT = [
 
 class BlaxelLlamaIndexInstrumentor(BaseInstrumentor):
     def instrumentation_dependencies(self):
-        return ["blaxel.llamaindex", "blaxel.core"]
+        return ["llama_index"]
 
     def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")

blaxel/telemetry/instrumentation/map.py

@@ -35,28 +35,28 @@ MAPPINGS: Dict[str, InstrumentationMapping] = {
         required_packages=["google-generativeai"],
         ignore_if_packages=[],
     ),
-    "blaxel.core": InstrumentationMapping(
+    "blaxel_core": InstrumentationMapping(
         module_path="blaxel.telemetry.instrumentation.blaxel_core",
         class_name="BlaxelCoreInstrumentor",
-        required_packages=["blaxel.core"],
+        required_packages=[],
         ignore_if_packages=[],
     ),
-    "blaxel.langgraph": InstrumentationMapping(
+    "blaxel_langgraph": InstrumentationMapping(
         module_path="blaxel.telemetry.instrumentation.blaxel_langgraph",
         class_name="BlaxelLanggraphInstrumentor",
-        required_packages=["blaxel.langgraph"],
+        required_packages=["langgraph"],
         ignore_if_packages=[],
     ),
-    "blaxel.langgraph_gemini": InstrumentationMapping(
+    "blaxel_langgraph_gemini": InstrumentationMapping(
         module_path="blaxel.telemetry.instrumentation.blaxel_langgraph_gemini",
         class_name="BlaxelLanggraphGeminiInstrumentor",
-        required_packages=["blaxel.langgraph"],
+        required_packages=["langgraph"],
         ignore_if_packages=[],
     ),
-    "blaxel.llamaindex": InstrumentationMapping(
+    "blaxel_llamaindex": InstrumentationMapping(
         module_path="blaxel.telemetry.instrumentation.blaxel_llamaindex",
         class_name="BlaxelLlamaIndexInstrumentor",
-        required_packages=["blaxel_llamaindex"],
+        required_packages=["llama_index"],
         ignore_if_packages=[],
     ),
 }

{blaxel-0.2.0rc6.dist-info → blaxel-0.2.1rc71.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.2.0rc6
+Version: 0.2.1rc71
 Summary: Blaxel - AI development platform SDK
 Project-URL: Homepage, https://blaxel.ai
 Project-URL: Documentation, https://docs.blaxel.ai

{blaxel-0.2.0rc6.dist-info → blaxel-0.2.1rc71.dist-info}/RECORD

@@ -310,7 +310,7 @@ blaxel/core/sandbox/filesystem.py,sha256=dyIvDdlPZO0ijD6mXXX8Yl0t75VijQ6_uMz_9rJ
 blaxel/core/sandbox/network.py,sha256=P5jLd4AAg1zgyIK4qGWvZaDZ5BzIcxRx2ffz_JLsLMI,357
 blaxel/core/sandbox/preview.py,sha256=M6FulOxPghUBpb5fLxu1Rd3ekLeCbZ_dgt4s1X2Cneo,5354
 blaxel/core/sandbox/process.py,sha256=FahDx-FOqXPV3ajVI6aKdXqx4Oi4fK3egojHjTGy8ro,6705
-blaxel/core/sandbox/sandbox.py,sha256=an06JrTl2kfS5guYo0ZT2A-Uu5Zr7BQAu-fWXcCZnhY,5368
+blaxel/core/sandbox/sandbox.py,sha256=BRMl74XTY5JQFcwZdEJoCgqcN3tTdRMD4gL7J-1Go1Q,5635
 blaxel/core/sandbox/session.py,sha256=4SH1tyXcQ9UqJx4lMwxAlp7x9Te_anDrdSEG6AlNkvU,4496
 blaxel/core/sandbox/types.py,sha256=gmNAt8x7PrJZHpw2_2aWWDE5uG6H-uEgdIKTYIw4G0g,2981
 blaxel/core/sandbox/client/__init__.py,sha256=N26bD5o1jsTb48oExow6Rgivd8ylaU9jaWZfZsVilP8,128
@@ -379,7 +379,7 @@ blaxel/llamaindex/__init__.py,sha256=iZ3QbZhlwKvP91ChcqSXVkpRrzurMxJoQfKdZFzE2AA
 blaxel/llamaindex/model.py,sha256=FYUCTA80LKC4-UfESh43htoHrsoZjwwVaxXD51T0IuE,2557
 blaxel/llamaindex/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 blaxel/llamaindex/tools.py,sha256=LgrKe-o0I8_JE4DxcM2YYlHN-15LKmXuBKm08NYLV9w,907
-blaxel/llamaindex/custom/cohere.py,sha256=Igj5Y1ozf1V4feIXfBHDdaTFU7od_wuOhm0yChZNxMY,19109
+blaxel/llamaindex/custom/cohere.py,sha256=zoUv4NWwMZLZEynYTC3JuOvtinguRgtuRwS31wIm3rI,18713
 blaxel/openai/__init__.py,sha256=YkizVtcYL2m9v-z5B1EReYVu9n9V-DCxJhSB2mvqOs0,123
 blaxel/openai/model.py,sha256=lGz4zrV4sURPb6aLtroMRV5-CgfQUq15PeTgjc7QkTI,600
 blaxel/openai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -394,15 +394,15 @@ blaxel/telemetry/exporters.py,sha256=EoX3uaBVku1Rg49pSNXKFyHhgY5OV3Ih6UlqgjF5epw
 blaxel/telemetry/manager.py,sha256=3yYBxxqQKl1rCKrn0GVz9jR5jouC1nsElbAaH8tTtgA,9075
 blaxel/telemetry/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 blaxel/telemetry/span.py,sha256=b8lpWe5nqek4w9YrBLJCIYz4Z3n9Z_49H_AkqpAGOiM,3660
-blaxel/telemetry/instrumentation/blaxel_core.py,sha256=gY2QoAr2UgnMabLyoxR1X79SVUaFM_hQJIzDdy80SuY,4892
-blaxel/telemetry/instrumentation/blaxel_langgraph.py,sha256=5C6mM0aKa8rbaAmhe7Z3HWHhZ-A_BZ0iio9Hn-GvIVM,4257
-blaxel/telemetry/instrumentation/blaxel_langgraph_gemini.py,sha256=2_5vcaNU7fS7ZisSLadNpFfiuy4zeRUDONJXhlQ9L1M,11960
-blaxel/telemetry/instrumentation/blaxel_llamaindex.py,sha256=EvgFhLUv5QViz7rAMJs5w_Oks-5NOe1v7A3B9b0NxeM,3122
-blaxel/telemetry/instrumentation/map.py,sha256=D7htA88-H8r7jHutBWnr_KEL8OZqLvvnRbU9jWiAA8g,2207
+blaxel/telemetry/instrumentation/blaxel_core.py,sha256=7PNseKq7bOQ4Z6a8jW7AGm0ojcdcusjoiH_5iQsR5ic,4881
+blaxel/telemetry/instrumentation/blaxel_langgraph.py,sha256=sBxt8kzRjsFepm86vaCBbGvpYI4mWPK_PB3wiLxSuOk,4235
+blaxel/telemetry/instrumentation/blaxel_langgraph_gemini.py,sha256=hQbOaaMsHte_Igq_vhPgNXYd40IwuqPB0TUP-5VAtME,11953
+blaxel/telemetry/instrumentation/blaxel_llamaindex.py,sha256=HA4YUXrRYvG5U1J8MAJnymmmj4ZCDydv-pXXhrftHOA,3101
+blaxel/telemetry/instrumentation/map.py,sha256=PCzZJj39yiYVYJrxLBNP-NW-tjjYyTijwEDeI9njuDY,2174
 blaxel/telemetry/instrumentation/utils.py,sha256=KInMYZH-mu9_wvetmf0EmgrfN3Sw8IWk2Y95v2u90_U,1901
 blaxel/telemetry/log/log.py,sha256=RvQByRjZMoP_dRaAZu8oK6DTegsHs-xV4W-UIqis6CA,2461
 blaxel/telemetry/log/logger.py,sha256=NPAS3g82ryROjvc_DEZaTIfrcehoLEZoP-JkLxADxc0,4113
-blaxel-0.2.0rc6.dist-info/METADATA,sha256=aI1uHOODyBfwDgJAbMDwmZlI__mZnGgzX9u-tY1PO8g,9878
-blaxel-0.2.0rc6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-blaxel-0.2.0rc6.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-blaxel-0.2.0rc6.dist-info/RECORD,,
+blaxel-0.2.1rc71.dist-info/METADATA,sha256=v0CXAwoZmLommJuFSHXUTdXHRMakSGnzuwrjvYZcKV8,9879
+blaxel-0.2.1rc71.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+blaxel-0.2.1rc71.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+blaxel-0.2.1rc71.dist-info/RECORD,,