blaxel 0.1.14rc48__py3-none-any.whl → 0.1.14rc50__py3-none-any.whl

blaxel/common/internal.py

@@ -7,21 +7,25 @@ from typing import Optional
 
 logger = getLogger(__name__)
 
-def get_alphanumeric_limited_hash(input_str, max_size):
-    # Create SHA-256 hash of the input string
-    hash_obj = hashlib.sha256(input_str.encode('utf-8'))
-
-    # Get the hash digest in base64 format
-    hash_base64 = base64.b64encode(hash_obj.digest()).decode('utf-8')
-
-    # Remove non-alphanumeric characters and convert to lowercase
-    alphanumeric = re.sub(r'[^a-zA-Z0-9]', '', hash_base64).lower()
-
-    # Skip the first character to match the Node.js crypto output
-    alphanumeric = alphanumeric[1:]
-
-    # Limit to max_size characters
-    return alphanumeric[:max_size] if len(alphanumeric) > max_size else alphanumeric
+def get_alphanumeric_limited_hash(input_str, max_size=48):
+    """
+    Create an alphanumeric hash using MD5 that can be reproduced in Go, TypeScript, and Python.
+    
+    Args:
+        input_str (str): The input string to hash
+        max_size (int): The maximum length of the returned hash
+        
+    Returns:
+        str: An alphanumeric hash of the input string, limited to max_size
+    """
+    # Calculate MD5 hash and convert to hexadecimal
+    hash_hex = hashlib.md5(input_str.encode('utf-8')).hexdigest()
+    
+    # Limit to max_size
+    if len(hash_hex) > max_size:
+        return hash_hex[:max_size]
+    
+    return hash_hex
 
 
 def get_global_unique_hash(workspace: str, type: str, name: str) -> str:

blaxel/common/settings.py

@@ -62,9 +62,14 @@ class Settings:
     @property
     def run_internal_hostname(self) -> str:
         """Get the run internal hostname."""
-        return ''
-        # Not working in python
-        # return os.environ.get("BL_RUN_INTERNAL_HOSTNAME", "")
+        if self.generation == "":
+            return ""
+        return os.environ.get("BL_RUN_INTERNAL_HOSTNAME", "")
+    
+    @property
+    def generation(self) -> str:
+        """Get the generation."""
+        return os.environ.get("BL_GENERATION", "")
 
     @property
     def bl_cloud(self) -> bool:

blaxel/jobs/__init__.py

@@ -0,0 +1,55 @@
+import argparse
+import os
+import sys
+import asyncio
+from typing import Any, Dict, Callable
+import requests
+
+class BlJob:
+    def get_arguments(self) -> Dict[str, Any]:
+        if not os.getenv('BL_EXECUTION_DATA_URL'):
+            parser = argparse.ArgumentParser()
+            # Parse known args, ignore unknown
+            args, unknown = parser.parse_known_args()
+            # Convert to dict and include unknown args
+            args_dict = vars(args)
+            # Add unknown args to dict
+            for i in range(0, len(unknown), 2):
+                if i + 1 < len(unknown):
+                    key = unknown[i].lstrip('-')
+                    args_dict[key] = unknown[i + 1]
+            return args_dict
+
+        response = requests.get(os.getenv('BL_EXECUTION_DATA_URL'))
+        data = response.json()
+        tasks = data.get('tasks', [])
+        return tasks[self.index] if self.index < len(tasks) else {}
+
+    @property
+    def index_key(self) -> str:
+        return os.getenv('BL_EXECUTION_INDEX_KEY', 'TASK_INDEX')
+
+    @property
+    def index(self) -> int:
+        index_value = os.getenv(self.index_key)
+        return int(index_value) if index_value else 0
+
+    def start(self, func: Callable):
+        """
+        Run a job defined in a function, it's run in the current process.
+        Handles both async and sync functions.
+        Arguments are passed as keyword arguments to the function.
+        """
+        try:
+            parsed_args = self.get_arguments()
+            if asyncio.iscoroutinefunction(func):
+                asyncio.run(func(**parsed_args))
+            else:
+                func(**parsed_args)
+            sys.exit(0)
+        except Exception as error:
+            print('Job execution failed:', error, file=sys.stderr)
+            sys.exit(1)
+
+# Create a singleton instance
+bl_job = BlJob()

{blaxel-0.1.14rc48.dist-info → blaxel-0.1.14rc50.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.1.14rc48
+Version: 0.1.14rc50
 Summary: Add your description here
 Author-email: cploujoux <cploujoux@blaxel.ai>
 License-File: LICENSE

{blaxel-0.1.14rc48.dist-info → blaxel-0.1.14rc50.dist-info}/RECORD

@@ -262,14 +262,15 @@ blaxel/client/models/workspace_runtime.py,sha256=dxEpmwCFPOCRKHRKhY-iW7j6TbtL5qU
 blaxel/client/models/workspace_user.py,sha256=70CcifQWYbeWG7TDui4pblTzUe5sVK0AS19vNCzKE8g,3423
 blaxel/common/autoload.py,sha256=NFuK71-IHOY2JQyEBSjDCVfUaQ8D8PJsEUEryIdG4AU,263
 blaxel/common/env.py,sha256=wTbzPDdNgz4HMJiS2NCZmQlN0qpxy1PQEYBaZgtvhoc,1247
-blaxel/common/internal.py,sha256=J-etgnBzelb-ln8uBR9WIWzrEUyDds8rdT9FImjes9g,2390
+blaxel/common/internal.py,sha256=PExgeKfJEmjINKreNb3r2nB5GAfG7uJhbfqHxuxBED8,2395
 blaxel/common/logger.py,sha256=emqgonfZMBIaQPowpngWOOZxWQieKP-yj_OzGZT8Oe8,1918
-blaxel/common/settings.py,sha256=uVM6pjcovw_JPjiNZQb9WbtvLu3RdVGy7PNBcDGzoWY,2190
+blaxel/common/settings.py,sha256=S7Su64Tu9-7OL5qZIOMVfYOwQos9gs7B3XpRAvI50ho,2331
 blaxel/instrumentation/exporters.py,sha256=EoX3uaBVku1Rg49pSNXKFyHhgY5OV3Ih6UlqgjF5epw,1670
 blaxel/instrumentation/log.py,sha256=4tGyvLg6r4DbjqJfajYbbZ1toUzF4Q4H7kHVqYWFAEA,2537
 blaxel/instrumentation/manager.py,sha256=jVwVasyMI6tu0zv27DVYOaN57nXYuHFJ8QzJQKaNoK4,8880
 blaxel/instrumentation/map.py,sha256=zZoiUiQHmik5WQZ4VCWNARSa6ppMi0r7D6hlb41N-Mg,1589
 blaxel/instrumentation/span.py,sha256=X2lwfu_dyxwQTMQJT2vbXOrbVSChEhjRLc413QOxQJM,3244
+blaxel/jobs/__init__.py,sha256=r4p2tYz8fCKzDGynLTq55KFSjdSIxhxjnarQf4hdMWI,1838
 blaxel/mcp/__init__.py,sha256=KednMrtuc4Y0O3lv7u1Lla54FCk8UX9c1k0USjL3Ahk,69
 blaxel/mcp/client.py,sha256=cFFXfpKXoMu8qTUly2ejF0pX2iBQkSNAxqwvDV1V6xY,4979
 blaxel/mcp/server.py,sha256=GIldtA_NgIc2dzd7ZpPvpbhpIt_7AfKu5yS_YJ0bDGg,7310
@@ -340,7 +341,7 @@ blaxel/tools/llamaindex.py,sha256=-gQ-C9V_h9a11J4ItsbWjXrCJOg0lRKsb98v9rVsNak,71
 blaxel/tools/openai.py,sha256=GuFXkj6bXEwldyVr89jEsRAi5ihZUVEVe327QuWiGNs,653
 blaxel/tools/pydantic.py,sha256=CvnNbAG_J4yBtA-XFI4lQrq3FYKjNd39hu841vZT004,1801
 blaxel/tools/types.py,sha256=YPCGJ4vZDhqR0X2H_TWtc5chQScsC32nGTQdRKJlO8Y,707
-blaxel-0.1.14rc48.dist-info/METADATA,sha256=V8dnLg7IzMcr9dO9vLzcHEbUYLTBtr_zh8wV2eN3A7g,11776
-blaxel-0.1.14rc48.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-blaxel-0.1.14rc48.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-blaxel-0.1.14rc48.dist-info/RECORD,,
+blaxel-0.1.14rc50.dist-info/METADATA,sha256=ZCoNc3i7CshGtPr2XqfSFFEPwuJAfZfuopesn5G4Zno,11776
+blaxel-0.1.14rc50.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+blaxel-0.1.14rc50.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+blaxel-0.1.14rc50.dist-info/RECORD,,