blaxel 0.2.0rc7__py3-none-any.whl → 0.2.1rc72__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,22 @@ 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.image = sandbox.spec.runtime.image or "blaxel/prod-base:latest"
+        sandbox.spec.runtime.memory = sandbox.spec.runtime.memory or 4096
         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-0.2.0rc7.dist-info → blaxel-0.2.1rc72.dist-info}/METADATA

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

{blaxel-0.2.0rc7.dist-info → blaxel-0.2.1rc72.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=tsRGvkb2dnGM5pmOiIxWvOmT42P9BDXWtW3h6_CIffI,5802
 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
@@ -402,7 +402,7 @@ blaxel/telemetry/instrumentation/map.py,sha256=PCzZJj39yiYVYJrxLBNP-NW-tjjYyTijw
 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.0rc7.dist-info/METADATA,sha256=kaL-NN6NG1sQz6DrmEPhFoXo5Z5Sb70yrPOQ7R7rgNk,9878
-blaxel-0.2.0rc7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-blaxel-0.2.0rc7.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-blaxel-0.2.0rc7.dist-info/RECORD,,
+blaxel-0.2.1rc72.dist-info/METADATA,sha256=_s4a8QbC9FoAHIggS4-uyWKl6YFP9GbEDmTuaWV4Yks,9879
+blaxel-0.2.1rc72.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+blaxel-0.2.1rc72.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+blaxel-0.2.1rc72.dist-info/RECORD,,