synapse-sdk 1.0.0a72__tar.gz → 1.0.0a74__tar.gz

{synapse_sdk-1.0.0a72 → synapse_sdk-1.0.0a74}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-sdk
-Version: 1.0.0a72
+Version: 1.0.0a74
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT

{synapse_sdk-1.0.0a72 → synapse_sdk-1.0.0a74}/synapse_sdk/clients/backend/__init__.py

@@ -23,6 +23,7 @@ class BackendClient(
         authorization_token (str): The authorization token for the synapse backend API.
         tenant_token (str): The tenant token for the synapse backend API.
         agent_token (str): The agent token for the backend API.
+        timeout (Dict): Set reasonable default timeouts for better UX. It can receive keys called 'connect' and 'read'.
     """
 
     name = 'Backend'
@@ -31,8 +32,8 @@ class BackendClient(
     tenant_token = None
     agent_token = None
 
-    def __init__(self, base_url, access_token=None, token=None, tenant=None, agent_token=None, **kwargs):
-        super().__init__(base_url)
+    def __init__(self, base_url, access_token=None, token=None, tenant=None, agent_token=None, timeout=None, **kwargs):
+        super().__init__(base_url, timeout=timeout)
         self.access_token = access_token
         self.authorization_token = token
         self.tenant_token = tenant

{synapse_sdk-1.0.0a72 → synapse_sdk-1.0.0a74}/synapse_sdk/clients/backend/models.py

@@ -60,12 +60,14 @@ class UpdateJob(BaseModel):
     Attrs:
         status (str): The job status. (ex: pending, running, stopped, succeeded, failed)
         progress_record (Dict): The job progress record.
+        metrics_record (Dict): The job metrics record.
         console_logs (Dict): The job console logs.
         result (Dict): The job result.
     """
 
     status: Optional[JobStatus] = None
     progress_record: Optional[Dict] = None
+    metrics_record: Optional[Dict] = None
     console_logs: Optional[Dict | list] = None
     result: Optional[Dict | list] = None
 

{synapse_sdk-1.0.0a72 → synapse_sdk-1.0.0a74}/synapse_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-sdk
-Version: 1.0.0a72
+Version: 1.0.0a74
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT

{synapse_sdk-1.0.0a72 → synapse_sdk-1.0.0a74}/tests/clients/test_backend_models.py

@@ -1,7 +1,7 @@
 import pytest
 from pydantic import ValidationError
 
-from synapse_sdk.clients.backend.models import Storage, StorageCategory, StorageProvider, UpdateJob
+from synapse_sdk.clients.backend.models import JobStatus, Storage, StorageCategory, StorageProvider, UpdateJob
 
 
 def test_storage_model_validation_success():
@@ -76,13 +76,13 @@ def test_storage_model_enum_values():
 
 def test_update_job_model_validation_success():
     job_response = {
-        'status': 'running',
+        'status': JobStatus.RUNNING,
         'progress_record': {'key': 'value'},
         'console_logs': {'log': 'log content'},
         'result': {'result_key': 'result_value'},
     }
     update_job = UpdateJob(**job_response)
-    assert update_job.status == 'running'
+    assert update_job.status == JobStatus.RUNNING
     assert update_job.progress_record == {'key': 'value'}
     assert update_job.console_logs == {'log': 'log content'}
     assert update_job.result == {'result_key': 'result_value'}
@@ -90,10 +90,10 @@ def test_update_job_model_validation_success():
 
 def test_update_job_model_validation_with_single_field_success():
     job_response = {
-        'status': 'running',
+        'status': JobStatus.RUNNING,
     }
     update_job = UpdateJob(**job_response)
-    assert update_job.status == 'running'
+    assert update_job.status == JobStatus.RUNNING
     assert update_job.progress_record is None
     assert update_job.console_logs is None
     assert update_job.result is None
@@ -101,11 +101,11 @@ def test_update_job_model_validation_with_single_field_success():
 
 def test_update_job_model_validation_with_console_logs_as_list():
     job_response = {
-        'status': 'running',
+        'status': JobStatus.RUNNING,
         'console_logs': ['log line 1', 'log line 2', 'log line 3'],
     }
     update_job = UpdateJob(**job_response)
-    assert update_job.status == 'running'
+    assert update_job.status == JobStatus.RUNNING
     assert update_job.console_logs == ['log line 1', 'log line 2', 'log line 3']
     assert update_job.progress_record is None
     assert update_job.result is None
@@ -113,11 +113,67 @@ def test_update_job_model_validation_with_console_logs_as_list():
 
 def test_update_job_model_validation_with_result_as_list():
     job_response = {
-        'status': 'running',
+        'status': JobStatus.RUNNING,
         'result': ['result item 1', 'result item 2', 'result item 3'],
     }
     update_job = UpdateJob(**job_response)
-    assert update_job.status == 'running'
+    assert update_job.status == JobStatus.RUNNING
     assert update_job.result == ['result item 1', 'result item 2', 'result item 3']
     assert update_job.progress_record is None
     assert update_job.console_logs is None
+
+
+def test_update_job_model_validation_with_metrics_record():
+    job_response = {
+        'status': JobStatus.RUNNING,
+        'metrics_record': {'metric_key': 'metric_value'},
+    }
+    update_job = UpdateJob(**job_response)
+    assert update_job.status == JobStatus.RUNNING
+    assert update_job.metrics_record == {'metric_key': 'metric_value'}
+    assert update_job.progress_record is None
+    assert update_job.console_logs is None
+    assert update_job.result is None
+
+
+def test_update_job_model_validation_with_all_fields():
+    job_response = {
+        'status': JobStatus.RUNNING,
+        'progress_record': {'progress_key': 'progress_value'},
+        'metrics_record': {'metrics_key': 'metrics_value'},
+        'console_logs': {'log_key': 'log_value'},
+        'result': {'result_key': 'result_value'},
+    }
+    update_job = UpdateJob(**job_response)
+    assert update_job.status == JobStatus.RUNNING
+    assert update_job.progress_record == {'progress_key': 'progress_value'}
+    assert update_job.metrics_record == {'metrics_key': 'metrics_value'}
+    assert update_job.console_logs == {'log_key': 'log_value'}
+    assert update_job.result == {'result_key': 'result_value'}
+
+
+def test_update_job_model_validation_with_metrics_record_as_empty_dict():
+    job_response = {
+        'status': JobStatus.RUNNING,
+        'metrics_record': {},
+    }
+    update_job = UpdateJob(**job_response)
+    assert update_job.status == JobStatus.RUNNING
+    assert update_job.metrics_record == {}
+    assert update_job.progress_record is None
+    assert update_job.console_logs is None
+    assert update_job.result is None
+
+
+def test_update_job_model_validation_with_metrics_record_and_console_logs():
+    job_response = {
+        'status': JobStatus.RUNNING,
+        'metrics_record': {'accuracy': 0.95, 'loss': 0.1},
+        'console_logs': ['log line 1', 'log line 2'],
+    }
+    update_job = UpdateJob(**job_response)
+    assert update_job.status == JobStatus.RUNNING
+    assert update_job.metrics_record == {'accuracy': 0.95, 'loss': 0.1}
+    assert update_job.console_logs == ['log line 1', 'log line 2']
+    assert update_job.progress_record is None
+    assert update_job.result is None