feat: model version retrieval testing

Author: Mesh-ach

src/webapp/databricks.py

@@ -505,6 +505,29 @@ def _consume_chunk(chunk_obj: Any) -> int | None:
                 pass
         return records
 
+    def fetch_model_version(
+            self,
+            model_name: str):
+        try:
+            w = WorkspaceClient(
+                host=databricks_vars["DATABRICKS_HOST_URL"],
+                google_service_account=gcs_vars["GCP_SERVICE_ACCOUNT_EMAIL"],
+            )
+        except Exception as e:
+            LOGGER.exception(
+                "Failed to create Databricks WorkspaceClient with host: %s and service account: %s",
+                databricks_vars["DATABRICKS_HOST_URL"],
+                gcs_vars["GCP_SERVICE_ACCOUNT_EMAIL"],
+            )
+            raise ValueError(f"setup_new_inst(): Workspace client creation failed: {e}")
+
+        model_info = w.registered_models.list(
+            full_name=model_name,
+            include_aliases = True,
+            )
+
+        return model_info
+    
     def get_key_for_file(
         self, mapping: Dict[str, Any], file_name: str
     ) -> Optional[str]:

src/webapp/routers/models.py

@@ -587,3 +587,21 @@ def trigger_inference_run(
         "output_valid": False,
         "framework": query_result[0][0].framework,
     }
+
+@router.get(
+    "/{inst_id}/models/{model_name}/get-model-versions"
+)
+def get_model_versions(
+    model_name: str,
+    inst_id: str,
+    current_user: Annotated[BaseUser, Depends(get_current_active_user)],
+    sql_session: Annotated[Session, Depends(get_session)],
+    databricks_control: Annotated[DatabricksControl, Depends(DatabricksControl)],
+) -> Any:
+    
+    model_name = decode_url_piece(model_name)
+    has_access_to_inst_or_err(inst_id, current_user)
+    
+    model_version_info = databricks_control.fetch_model_version(model_name)
+    
+    return model_version_info