fix: formatting style

Author: Mesh-ach

src/webapp/databricks.py

@@ -505,7 +505,11 @@ def _consume_chunk(chunk_obj: Any) -> int | None:
                 pass
         return records
 
-    def fetch_model_version(self, model_name: str):
+    def fetch_model_version(self, catalog_name: str, inst_name: str, model_name: str):
+
+        schema = databricksify_inst_name(inst_name)
+        model_name_path = f"{catalog_name}.{schema}_gold.{model_name}"
+
         try:
             w = WorkspaceClient(
                 host=databricks_vars["DATABRICKS_HOST_URL"],
@@ -520,7 +524,7 @@ def fetch_model_version(self, model_name: str):
             raise ValueError(f"setup_new_inst(): Workspace client creation failed: {e}")
 
         model_info = w.model_versions.list(
-            full_name=model_name,
+            full_name=model_name_path,
         )
 
         return model_info

src/webapp/routers/models.py

@@ -33,6 +33,7 @@
 import traceback
 import logging
 from ..gcsdbutils import update_db_from_bucket
+from ..config import env_vars
 
 from ..gcsutil import StorageControl
 
@@ -601,9 +602,30 @@ def get_model_versions(
     transformed_model_name = str(decode_url_piece(model_name)).strip()
     has_access_to_inst_or_err(inst_id, current_user)
 
+    local_session.set(sql_session)
+    query_result = (
+        local_session.get()
+        .execute(select(InstTable).where(InstTable.id == str_to_uuid(inst_id)))
+        .all()
+    )
+    if not query_result or len(query_result) == 0:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="Institution not found.",
+        )
+    if len(query_result) > 1:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail="Institution duplicates found.",
+        )
+
     print(f"Initial model name = {model_name}")
     print(f"Converted model name {transformed_model_name}")
 
-    model_version_info = databricks_control.fetch_model_version(transformed_model_name)
+    model_version_info = databricks_control.fetch_model_version(
+        catalog_name=env_vars["CATALOG_NAME"],
+        inst_name=f"{query_result[0][0].name}",
+        model_name=transformed_model_name,
+    )
 
     return model_version_info