featrixsphere 0.2.1439__py3-none-any.whl → 0.2.1462__py3-none-any.whl

featrixsphere/__init__.py

@@ -38,7 +38,7 @@ Example:
     ...                                labels=['Experiment A', 'Experiment B'])
 """
 
-__version__ = "0.2.1439"
+__version__ = "0.2.1462"
 __author__ = "Featrix"
 __email__ = "support@featrix.com"
 __license__ = "MIT"

featrixsphere/client.py

@@ -1635,7 +1635,7 @@ class FeatrixSphereClient:
             ax.text(0.5, 0.5, f'Error plotting embedding: {e}', 
                    transform=ax.transAxes, ha='center', va='center')
 
-    def create_embedding_space(self, name: str, s3_training_dataset: str, s3_validation_dataset: str, webhooks: Dict[str, str] = None) -> SessionInfo:
+    def create_embedding_space(self, name: str, s3_training_dataset: str, s3_validation_dataset: str, webhooks: Dict[str, str] = None, user_metadata: Dict[str, Any] = None) -> SessionInfo:
         """
         Create a new embedding space from S3 training and validation datasets.
         
@@ -1644,6 +1644,7 @@ class FeatrixSphereClient:
             s3_training_dataset: S3 URL for training dataset (must start with 's3://')
             s3_validation_dataset: S3 URL for validation dataset (must start with 's3://')
             webhooks: Optional dict with webhook configuration keys (webhook_callback_secret, s3_backup_url, model_id_update_url)
+            user_metadata: Optional user metadata for ES/SP identification (max 32KB)
             
         Returns:
             SessionInfo for the newly created embedding space session
@@ -1668,6 +1669,10 @@ class FeatrixSphereClient:
         }
         if webhooks:
             data['webhooks'] = webhooks
+        if user_metadata:
+            import json
+            data['user_metadata'] = json.dumps(user_metadata)
+            print(f"User metadata: {user_metadata}")
         
         response_data = self._post_json("/compute/create-embedding-space", data)
         
@@ -1689,7 +1694,8 @@ class FeatrixSphereClient:
         parent_embedding_space_path: str = None,
         s3_training_dataset: str = None,
         s3_validation_dataset: str = None,
-        webhooks: Dict[str, str] = None
+        webhooks: Dict[str, str] = None,
+        user_metadata: Dict[str, Any] = None
     ) -> SessionInfo:
         """
         Fine-tune an existing embedding space on new data.
@@ -1785,6 +1791,10 @@ class FeatrixSphereClient:
         
         if webhooks:
             data['webhooks'] = webhooks
+        if user_metadata:
+            import json
+            data['user_metadata'] = json.dumps(user_metadata)
+            print(f"User metadata: {user_metadata}")
         
         response_data = self._post_json("/compute/fine-tune-embedding-space", data)
         
@@ -4203,7 +4213,8 @@ class FeatrixSphereClient:
                                     class_imbalance: dict = None,
                                     poll_interval: int = 30, max_poll_time: int = 3600,
                                     verbose: bool = True,
-                                    webhooks: Dict[str, str] = None) -> SessionInfo:
+                                    webhooks: Dict[str, str] = None,
+                                    user_metadata: Dict[str, Any] = None) -> SessionInfo:
         """
         Train a single predictor on a foundation model (existing embedding space).
         
@@ -4227,6 +4238,7 @@ class FeatrixSphereClient:
             max_poll_time: Maximum time to poll in seconds (default: 3600 = 1 hour)
             verbose: Whether to print status updates during polling (default: True)
             webhooks: Optional dict with webhook configuration keys (webhook_callback_secret, s3_backup_url, model_id_update_url)
+            user_metadata: Optional user metadata for ES/SP identification (max 32KB)
             
         Returns:
             SessionInfo for the newly created predictor training session
@@ -4236,9 +4248,28 @@ class FeatrixSphereClient:
         
         # Get the compute cluster from the foundation model session
         # This ensures we upload files to the same node where the foundation model lives
-        foundation_session = self.get_session_status(foundation_model_id)
-        foundation_compute_cluster = self.get_last_server_metadata()
-        foundation_compute_cluster = foundation_compute_cluster.get('compute_cluster') if foundation_compute_cluster else None
+        # If the foundation session doesn't exist (404), we'll proceed with current compute cluster
+        foundation_compute_cluster = None
+        try:
+            foundation_session = self.get_session_status(foundation_model_id)
+            foundation_compute_cluster = self.get_last_server_metadata()
+            foundation_compute_cluster = foundation_compute_cluster.get('compute_cluster') if foundation_compute_cluster else None
+        except Exception as e:
+            # Foundation session might not exist or be accessible - that's okay
+            # The server will validate it when we submit the training request
+            if verbose:
+                # Check if it's a 404 HTTP error
+                is_404 = False
+                if isinstance(e, requests.exceptions.HTTPError):
+                    if hasattr(e, 'response') and e.response.status_code == 404:
+                        is_404 = True
+                
+                if is_404:
+                    print(f"  ⚠️  Foundation session not found (404) - will use current compute cluster")
+                    print(f"     Server will validate foundation model when training starts")
+                else:
+                    print(f"  ⚠️  Could not fetch foundation session: {e}")
+                    print(f"     Will proceed with current compute cluster")
         
         # Temporarily set compute cluster for file uploads if we found one
         original_compute_cluster = self.compute_cluster
@@ -4247,6 +4278,8 @@ class FeatrixSphereClient:
             self.set_compute_cluster(foundation_compute_cluster)
             if verbose:
                 print(f"  Using compute cluster: {foundation_compute_cluster}")
+        elif verbose and self.compute_cluster:
+            print(f"  Using current compute cluster: {self.compute_cluster}")
         
         try:
             # Validate that only one data source is provided
@@ -4318,6 +4351,11 @@ class FeatrixSphereClient:
             if webhooks:
                 import json
                 data["webhooks"] = json.dumps(webhooks)
+            if user_metadata:
+                import json
+                data["user_metadata"] = json.dumps(user_metadata)
+                if verbose:
+                    print(f"User metadata: {user_metadata}")
             
             # Send request with file if provided
             try:

{featrixsphere-0.2.1439.dist-info → featrixsphere-0.2.1462.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: featrixsphere
-Version: 0.2.1439
+Version: 0.2.1462
 Summary: Transform any CSV into a production-ready ML model in minutes, not months.
 Home-page: https://github.com/Featrix/sphere
 Author: Featrix

featrixsphere-0.2.1462.dist-info/RECORD

@@ -0,0 +1,9 @@
+featrixsphere/__init__.py,sha256=oJsLRgA3q8pSL3rPso0kfVmgPoKdivl2gt4Gj3OZW_M,1888
+featrixsphere/cli.py,sha256=AW9O3vCvCNJ2UxVGN66eRmeN7XLSiHJlvK6JLZ9UJXc,13358
+featrixsphere/client.py,sha256=HWbVkvvZ3MZae_hfa0ChurzrSHMnxTWDW59tOcLERag,386715
+featrixsphere/test_client.py,sha256=4SiRbib0ms3poK0UpnUv4G0HFQSzidF3Iswo_J2cjLk,11981
+featrixsphere-0.2.1462.dist-info/METADATA,sha256=odh92kvhANd28hRmWToVCsW8hv-jb2YLpBdFegOHNkM,16232
+featrixsphere-0.2.1462.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+featrixsphere-0.2.1462.dist-info/entry_points.txt,sha256=QreJeYfD_VWvbEqPmMXZ3pqqlFlJ1qZb-NtqnyhEldc,51
+featrixsphere-0.2.1462.dist-info/top_level.txt,sha256=AyN4wjfzlD0hWnDieuEHX0KckphIk_aC73XCG4df5uU,14
+featrixsphere-0.2.1462.dist-info/RECORD,,

featrixsphere-0.2.1439.dist-info/RECORD

@@ -1,9 +0,0 @@
-featrixsphere/__init__.py,sha256=7qEuBqtpDfJZYvEJQAVVrtdL-nZvI74oPx7tcVJN9Jo,1888
-featrixsphere/cli.py,sha256=AW9O3vCvCNJ2UxVGN66eRmeN7XLSiHJlvK6JLZ9UJXc,13358
-featrixsphere/client.py,sha256=NM-dBRANvN0GpcLf0BPqUxpqwMhzHP6hJktD8FndyCM,384646
-featrixsphere/test_client.py,sha256=4SiRbib0ms3poK0UpnUv4G0HFQSzidF3Iswo_J2cjLk,11981
-featrixsphere-0.2.1439.dist-info/METADATA,sha256=_KJ6nguFKv3xMG1kbWWS5_7hN2DjQEa-z3sloLIh5z4,16232
-featrixsphere-0.2.1439.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-featrixsphere-0.2.1439.dist-info/entry_points.txt,sha256=QreJeYfD_VWvbEqPmMXZ3pqqlFlJ1qZb-NtqnyhEldc,51
-featrixsphere-0.2.1439.dist-info/top_level.txt,sha256=AyN4wjfzlD0hWnDieuEHX0KckphIk_aC73XCG4df5uU,14
-featrixsphere-0.2.1439.dist-info/RECORD,,