add support to call model with a specific version

Author: Deepthi-Chand

api/views/aimodel_execution.py

@@ -65,11 +65,21 @@ def call_aimodel(request: Request, model_id: str) -> Response:
             )
 
         parameters = request.data.get("parameters", {})
+        version_id = request.data.get("version_id")
 
-        # Get the primary version and provider
-        primary_version = model.versions.filter(is_latest=True).first()
-        if not primary_version:
-            primary_version = model.versions.first()
+        # Get the version - either specific version or primary (latest)
+        if version_id:
+            primary_version = model.versions.filter(id=version_id).first()
+            if not primary_version:
+                return Response(
+                    {"error": f"Version with ID {version_id} not found for this model"},
+                    status=status.HTTP_400_BAD_REQUEST,
+                )
+        else:
+            # Fall back to primary (latest) version
+            primary_version = model.versions.filter(is_latest=True).first()
+            if not primary_version:
+                primary_version = model.versions.first()
 
         if not primary_version:
             return Response(

dataspace_sdk/resources/aimodels.py

@@ -337,7 +337,11 @@ def delete_model(self, model_id: str) -> Dict[str, Any]:
         return self.delete(f"/api/aimodels/{model_id}/")
 
     def call_model(
-        self, model_id: str, input_text: str, parameters: Optional[Dict[str, Any]] = None
+        self,
+        model_id: str,
+        input_text: str,
+        parameters: Optional[Dict[str, Any]] = None,
+        version_id: Optional[int] = None,
     ) -> Dict[str, Any]:
         """
         Call an AI model with input text using the appropriate client (API or HuggingFace).
@@ -346,6 +350,7 @@ def call_model(
             model_id: UUID of the AI model
             input_text: Input text to process
             parameters: Optional parameters for the model call (temperature, max_tokens, etc.)
+            version_id: Optional specific version ID to call (defaults to primary/latest version)
 
         Returns:
             Dictionary containing model response:
@@ -358,13 +363,24 @@ def call_model(
                 ...
             }
         """
+        payload: Dict[str, Any] = {
+            "input_text": input_text,
+            "parameters": parameters or {},
+        }
+        if version_id is not None:
+            payload["version_id"] = version_id
+
         return self.post(
             f"/api/aimodels/{model_id}/call/",
-            json_data={"input_text": input_text, "parameters": parameters or {}},
+            json_data=payload,
         )
 
     def call_model_async(
-        self, model_id: str, input_text: str, parameters: Optional[Dict[str, Any]] = None
+        self,
+        model_id: str,
+        input_text: str,
+        parameters: Optional[Dict[str, Any]] = None,
+        version_id: Optional[int] = None,
     ) -> Dict[str, Any]:
         """
         Call an AI model asynchronously (returns task ID for long-running operations).
@@ -373,6 +389,7 @@ def call_model_async(
             model_id: UUID of the AI model
             input_text: Input text to process
             parameters: Optional parameters for the model call
+            version_id: Optional specific version ID to call (defaults to primary/latest version)
 
         Returns:
             Dictionary containing task information:
@@ -382,9 +399,16 @@ def call_model_async(
                 "model_id": str
             }
         """
+        payload: Dict[str, Any] = {
+            "input_text": input_text,
+            "parameters": parameters or {},
+        }
+        if version_id is not None:
+            payload["version_id"] = version_id
+
         return self.post(
             f"/api/aimodels/{model_id}/call-async/",
-            json_data={"input_text": input_text, "parameters": parameters or {}},
+            json_data=payload,
         )
 
     # ==================== Version Management ====================