response changes

Author: mohit-singlestore

singlestoredb/fusion/handlers/models.py

@@ -285,12 +285,10 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         operation_result = inference_api.start()
 
         res = FusionSQLResult()
-        res.add_field('Model', result.STRING)
         res.add_field('Status', result.STRING)
         res.add_field('Message', result.STRING)
         res.set_rows([
             (
-                operation_result.name,
                 operation_result.status,
                 operation_result.get_message(),
             ),
@@ -335,12 +333,10 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         operation_result = inference_api.stop()
 
         res = FusionSQLResult()
-        res.add_field('Model', result.STRING)
         res.add_field('Status', result.STRING)
         res.add_field('Message', result.STRING)
         res.set_rows([
             (
-                operation_result.name,
                 operation_result.status,
                 operation_result.get_message(),
             ),

singlestoredb/management/inference_api.py

@@ -50,7 +50,7 @@ def from_start_response(cls, response: Dict[str, Any]) -> 'ModelOperationResult'
 
         """
         return cls(
-            name=response.get('modelID', ''),
+            name=response.get('modelName', ''),
             status='Initializing',
             hosting_platform=response.get('hostingPlatform', ''),
         )
@@ -242,11 +242,7 @@ def start(self, model_name: str) -> ModelOperationResult:
         if self._manager is None:
             raise ManagementError(msg='Manager not initialized')
         res = self._manager._post(f'inferenceapis/{model_name}/start')
-        response_data = res.json()
-        # Debug: print the actual response to see what we're getting
-        import json
-        print(f"DEBUG - Start response: {json.dumps(response_data, indent=2, default=str)}")
-        return ModelOperationResult.from_start_response(response_data)
+        return ModelOperationResult.from_start_response(res.json())
 
     def stop(self, model_name: str) -> ModelOperationResult:
         """
@@ -266,8 +262,4 @@ def stop(self, model_name: str) -> ModelOperationResult:
         if self._manager is None:
             raise ManagementError(msg='Manager not initialized')
         res = self._manager._post(f'inferenceapis/{model_name}/stop')
-        response_data = res.json()
-        # Debug: print the actual response to see what we're getting
-        import json
-        print(f"DEBUG - Stop response: {json.dumps(response_data, indent=2, default=str)}")
-        return ModelOperationResult.from_stop_response(response_data)
+        return ModelOperationResult.from_stop_response(res.json())