featrixsphere 0.2.6708__py3-none-any.whl → 0.2.6710__py3-none-any.whl

featrixsphere/__init__.py

@@ -57,7 +57,7 @@ TWO API OPTIONS:
     >>> print(result['prediction'])
 """
 
-__version__ = "0.2.6708"
+__version__ = "0.2.6710"
 __author__ = "Featrix"
 __email__ = "support@featrix.com"
 __license__ = "MIT"

featrixsphere/api/client.py

@@ -342,6 +342,34 @@ class FeatrixSphere(HTTPClientMixin):
             ground_truth=ground_truth
         )
 
+    def list_sessions(
+        self,
+        name_prefix: str = "",
+    ) -> List[str]:
+        """
+        List sessions matching a name prefix/search term.
+
+        Searches session directory names on the compute cluster for
+        partial matches (not just prefix).
+
+        Args:
+            name_prefix: Term to match in session names
+
+        Returns:
+            List of matching session ID strings
+
+        Example:
+            sessions = featrix.list_sessions(name_prefix="customer")
+            for sid in sessions:
+                fm = featrix.foundational_model(sid)
+                print(f"{sid}: {fm.status}")
+        """
+        params = {}
+        if name_prefix:
+            params['name_prefix'] = name_prefix
+        response = self._get_json("/compute/sessions-for-org", params=params)
+        return response.get('sessions', [])
+
     def health_check(self) -> Dict[str, Any]:
         """
         Check if the API server is healthy.

featrixsphere/api/foundational_model.py

@@ -22,6 +22,21 @@ from .reference_record import ReferenceRecord
 logger = logging.getLogger(__name__)
 
 
+def _parse_datetime(value) -> Optional[datetime]:
+    """Parse a datetime from ISO string or return as-is if already datetime."""
+    if value is None:
+        return None
+    if isinstance(value, datetime):
+        return value
+    if isinstance(value, str):
+        try:
+            # Handle ISO format with or without timezone
+            return datetime.fromisoformat(value.replace('Z', '+00:00'))
+        except (ValueError, AttributeError):
+            return None
+    return None
+
+
 @dataclass
 class FoundationalModel:
     """
@@ -68,6 +83,11 @@ class FoundationalModel:
     epochs: Optional[int] = None
     final_loss: Optional[float] = None
     created_at: Optional[datetime] = None
+    updated_at: Optional[datetime] = None
+    session_type: Optional[str] = None
+    compute_cluster: Optional[str] = None
+    error_message: Optional[str] = None
+    training_progress: Optional[Dict[str, Any]] = None
 
     # Internal
     _ctx: Optional['ClientContext'] = field(default=None, repr=False)
@@ -88,7 +108,9 @@ class FoundationalModel:
             dimensions=response.get('d_model') or response.get('dimensions'),
             epochs=response.get('epochs') or response.get('final_epoch'),
             final_loss=response.get('final_loss'),
-            created_at=datetime.now(),
+            created_at=_parse_datetime(response.get('created_at')),
+            session_type=response.get('session_type'),
+            compute_cluster=response.get('compute_cluster'),
             _ctx=ctx,
         )
 
@@ -99,19 +121,22 @@ class FoundationalModel:
         ctx: 'ClientContext'
     ) -> 'FoundationalModel':
         """Load FoundationalModel from session ID."""
-        # Get session info
-        session_data = ctx.get_json(f"/compute/session/{session_id}")
+        # Get session info - response has {"session": {...}, "jobs": {...}}
+        response_data = ctx.get_json(f"/compute/session/{session_id}")
+        session = response_data.get('session', response_data)
 
         fm = cls(
             id=session_id,
-            name=session_data.get('name'),
-            status=session_data.get('status'),
-            created_at=datetime.now(),
+            name=session.get('name'),
+            status=session.get('status'),
+            created_at=_parse_datetime(session.get('created_at')),
+            session_type=session.get('session_type'),
+            compute_cluster=session.get('compute_cluster'),
             _ctx=ctx,
         )
 
-        # Try to get model info
-        fm._update_from_session(session_data)
+        # Extract model info, training stats, jobs, error_message
+        fm._update_from_session(response_data)
 
         return fm
 
@@ -439,10 +464,11 @@ class FoundationalModel:
         last_status = None
 
         while time.time() - start_time < max_wait_time:
-            # Get session status
-            session_data = self._ctx.get_json(f"/compute/session/{self.id}")
+            # Get session status - response has {"session": {...}, "jobs": {...}}
+            response_data = self._ctx.get_json(f"/compute/session/{self.id}")
+            session_data = response_data.get('session', response_data)
             status = session_data.get('status', 'unknown')
-            jobs = session_data.get('jobs', {})
+            jobs = response_data.get('jobs', {})
 
             # Look for ES training job
             es_job = None
@@ -475,7 +501,7 @@ class FoundationalModel:
             # Check completion
             if job_status == 'done' or status == 'done':
                 self.status = 'done'
-                self._update_from_session(session_data)
+                self._update_from_session(response_data)
                 if show_progress:
                     print(f"Training complete!")
                     if self.dimensions:
@@ -627,27 +653,73 @@ class FoundationalModel:
 
         return predictors
 
-    def _update_from_session(self, session_data: Dict[str, Any]) -> None:
-        """Update fields from session data."""
-        # Try to get model info from various places
-        model_info = session_data.get('model_info', {})
-        training_stats = session_data.get('training_stats', {})
+    def _update_from_session(self, response_data: Dict[str, Any]) -> None:
+        """Update fields from session API response.
+
+        The response from GET /session/{id} has structure:
+            {"session": {...}, "jobs": {...}, ...}
+        """
+        # Handle both nested and flat response formats
+        session = response_data.get('session', response_data)
+        jobs = response_data.get('jobs', {})
+
+        # Core session fields
+        if session.get('name') and not self.name:
+            self.name = session['name']
+        if session.get('status'):
+            self.status = session['status']
+        if session.get('session_type'):
+            self.session_type = session['session_type']
+        if session.get('compute_cluster'):
+            self.compute_cluster = session['compute_cluster']
+        if session.get('created_at') and not self.created_at:
+            self.created_at = _parse_datetime(session['created_at'])
+        if session.get('finished_at'):
+            self.updated_at = _parse_datetime(session['finished_at'])
+        elif session.get('started_at'):
+            self.updated_at = _parse_datetime(session['started_at'])
+
+        # Model info from session
+        model_info = session.get('model_info', {})
+        training_stats = session.get('training_stats', {})
 
         self.dimensions = (
             model_info.get('d_model') or
             model_info.get('embedding_dim') or
-            session_data.get('d_model')
+            session.get('d_model')
         )
         self.epochs = (
             training_stats.get('final_epoch') or
             training_stats.get('epochs_trained') or
-            session_data.get('epochs')
+            session.get('epochs')
         )
         self.final_loss = (
             training_stats.get('final_loss') or
-            session_data.get('final_loss')
+            session.get('final_loss')
         )
 
+        # Extract error_message and training_progress from jobs
+        for job_id, job in jobs.items():
+            job_type = job.get('job_type', '')
+            job_status = job.get('status', '')
+
+            # Training progress from ES training job
+            if job_type in ('train_embedding_space', 'train_es', 'training'):
+                current_epoch = job.get('current_epoch') or job.get('epoch')
+                total_epochs = job.get('total_epochs') or job.get('epochs')
+                if current_epoch or total_epochs:
+                    self.training_progress = {
+                        'current_epoch': current_epoch,
+                        'total_epochs': total_epochs,
+                        'job_status': job_status,
+                    }
+
+            # Error message from any failed job
+            if job_status in ('failed', 'error'):
+                err = job.get('error') or job.get('error_message')
+                if err:
+                    self.error_message = err
+
     def _clean_record(self, record: Dict[str, Any]) -> Dict[str, Any]:
         """Clean a record for API submission."""
         import math
@@ -684,6 +756,17 @@ class FoundationalModel:
         """Column names in this foundational model's embedding space."""
         return self.get_columns()
 
+    @property
+    def schema_metadata(self) -> Dict[str, Any]:
+        """Get schema metadata including column names and types.
+
+        Returns:
+            Dict with 'column_names', 'column_types', and 'num_columns'
+        """
+        if not self._ctx:
+            raise ValueError("FoundationalModel not connected to client")
+        return self._ctx.get_json(f"/compute/session/{self.id}/columns")
+
     def clone(
         self,
         target_compute_cluster: Optional[str] = None,
@@ -749,7 +832,6 @@ class FoundationalModel:
 
         data = self._ctx.get_json(f"/compute/session/{self.id}")
         self._update_from_session(data)
-        self.status = data.get('status', self.status)
         return data
 
     def is_ready(self) -> bool:
@@ -767,9 +849,8 @@ class FoundationalModel:
             raise ValueError("FoundationalModel not connected to client")
 
         data = self._ctx.get_json(f"/compute/session/{self.id}")
-        status = data.get('status', 'unknown')
-        self.status = status
-        return status == 'done'
+        self._update_from_session(data)
+        return self.status == 'done'
 
     def publish(
         self,
@@ -938,6 +1019,11 @@ class FoundationalModel:
             'epochs': self.epochs,
             'final_loss': self.final_loss,
             'created_at': self.created_at.isoformat() if self.created_at else None,
+            'updated_at': self.updated_at.isoformat() if self.updated_at else None,
+            'session_type': self.session_type,
+            'compute_cluster': self.compute_cluster,
+            'error_message': self.error_message,
+            'training_progress': self.training_progress,
         }
 
     def __repr__(self) -> str:

{featrixsphere-0.2.6708.dist-info → featrixsphere-0.2.6710.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: featrixsphere
-Version: 0.2.6708
+Version: 0.2.6710
 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.6708.dist-info → featrixsphere-0.2.6710.dist-info}/RECORD

@@ -1,17 +1,17 @@
-featrixsphere/__init__.py,sha256=NPVY3_tG74OqiLz1MYbrb9R4Wff68xFCI6zr3gVRTyQ,2190
+featrixsphere/__init__.py,sha256=QQglKOYv0bjonuO0-wOkeyyXMWhv3yK0_s5Uaap-GVk,2190
 featrixsphere/client.py,sha256=JNYAHDFtxmhQVuclO3dWEphJuxNFLi-PfvWylZWKF_4,452929
 featrixsphere/api/__init__.py,sha256=quyvuPphVj9wb6v8Dio0SMG9iHgJAmY3asHk3f_zF10,1269
 featrixsphere/api/api_endpoint.py,sha256=i3eCWuaUXftnH1Ai6MFZ7md7pC2FcRAIRO87CBZhyEQ,9000
-featrixsphere/api/client.py,sha256=TdpujNsJxO4GfPMI_KoemQWV89go3KuK6OPAo9jX6Bs,12574
-featrixsphere/api/foundational_model.py,sha256=wf5-VvVUXYoiyKr3y4Ok8OnwMhtaUuYLXwdgCwFuM-k,31065
+featrixsphere/api/client.py,sha256=TvNqrzSPQdw0A4kW48M0S3SDrBRmkc6kTY8UkzO4eRs,13426
+featrixsphere/api/foundational_model.py,sha256=0ZFO-mJs66nVRXQbM0o1fB4HmhzLBXUqbTCF46LVH1k,34925
 featrixsphere/api/http_client.py,sha256=q59-41fHua_7AwtPFCvshlSUKJ-fS0X337L9Ooyn0DI,8440
 featrixsphere/api/notebook_helper.py,sha256=xY9jsao26eaNiFh2s0_TlRZnR8xZ4P_e0EOKr2PtoVs,20060
 featrixsphere/api/prediction_result.py,sha256=HQsJdr89zWxdRx395nevN3aP7ZXZuZxB4UGX5Ykhkfk,12235
 featrixsphere/api/predictor.py,sha256=1v0ffkEjmrO3BP0PNWAXtiAU-AlOQJSiDICmW1bQbGU,20300
 featrixsphere/api/reference_record.py,sha256=-XOTF6ynznB3ouz06w3AF8X9SVId0g_dO20VvGNesUQ,7095
 featrixsphere/api/vector_database.py,sha256=BplxKkPnAbcBX1A4KxFBJVb3qkQ-FH9zi9v2dWG5CgY,7976
-featrixsphere-0.2.6708.dist-info/METADATA,sha256=IpNh40w32ZlavhKaRkBFRLvpXyXmoNddYMjbc2NpBGI,16232
-featrixsphere-0.2.6708.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-featrixsphere-0.2.6708.dist-info/entry_points.txt,sha256=QreJeYfD_VWvbEqPmMXZ3pqqlFlJ1qZb-NtqnyhEldc,51
-featrixsphere-0.2.6708.dist-info/top_level.txt,sha256=AyN4wjfzlD0hWnDieuEHX0KckphIk_aC73XCG4df5uU,14
-featrixsphere-0.2.6708.dist-info/RECORD,,
+featrixsphere-0.2.6710.dist-info/METADATA,sha256=_gDwyRsSfEa0thWd6IjhMeCEZF5dMc8BfLBL4J2b5HQ,16232
+featrixsphere-0.2.6710.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+featrixsphere-0.2.6710.dist-info/entry_points.txt,sha256=QreJeYfD_VWvbEqPmMXZ3pqqlFlJ1qZb-NtqnyhEldc,51
+featrixsphere-0.2.6710.dist-info/top_level.txt,sha256=AyN4wjfzlD0hWnDieuEHX0KckphIk_aC73XCG4df5uU,14
+featrixsphere-0.2.6710.dist-info/RECORD,,