From d8db1a05cb634b733e9fafec4e3d2e7515e2ca77 Mon Sep 17 00:00:00 2001
From: Dariusz L <dl122@wp.pl>
Date: Mon, 27 Jul 2026 17:39:35 +0200
Subject: [PATCH] Use shared workflow model inventory in model-hashes route

---
 __init__.py                       |  8 +++----
 tests/test_refactoring_targets.py | 36 +++++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+), 5 deletions(-)

diff --git a/__init__.py b/__init__.py
index c009dc9..1903271 100644
--- a/__init__.py
+++ b/__init__.py
@@ -428,7 +428,6 @@ class ModelResolverExtension:
                 from .core.workflow_analyzer import (
                     NODE_TYPE_MODEL_WIDGET_CATEGORIES,
                     URN_TYPE_MAP,
-                    analyze_workflow_models,
                     get_workflow_model_inventory,
                 )
             except ImportError as e:
@@ -1016,12 +1015,11 @@ class ModelResolverExtension:
                         }
                     )
 
-                available_models = await asyncio.to_thread(get_model_files, False)
-                refs = await asyncio.to_thread(
-                    analyze_workflow_models,
+                inventory = await asyncio.to_thread(
+                    get_workflow_model_inventory,
                     workflow_json,
-                    available_models,
                 )
+                refs = inventory["model_refs"]
 
                 by_node = {}
                 by_path = {}
diff --git a/tests/test_refactoring_targets.py b/tests/test_refactoring_targets.py
index 34fcee8..fede71c 100644
--- a/tests/test_refactoring_targets.py
+++ b/tests/test_refactoring_targets.py
@@ -146,6 +146,42 @@ class TestRefactoringTargets(unittest.IsolatedAsyncioTestCase):
             mock_json_res.assert_called_once_with({"status": "running"})
             mock_get_aria2_status.assert_called_once_with({"aria2c_path": "custom_path"})
 
+    async def test_workflow_model_hashes_reuses_shared_inventory(self):
+        post_handler = routes_registered.get(
+            ("POST", "/model_resolver/workflow-model-hashes")
+        )
+        self.assertIsNotNone(post_handler)
+
+        workflow = {"nodes": []}
+        mock_request = AsyncMock()
+        mock_request.json.return_value = {"workflow": workflow}
+        threaded_functions = []
+
+        async def fake_to_thread(func, *args, **kwargs):
+            function_name = getattr(func, "__name__", "")
+            threaded_functions.append(function_name or "<settings>")
+            if len(threaded_functions) == 1:
+                return {"workflow_hash_metadata_enabled": True}
+            if function_name == "get_workflow_model_inventory":
+                self.assertEqual((workflow,), args)
+                self.assertEqual({}, kwargs)
+                return {"available_models": [], "model_refs": []}
+            self.fail(f"Unexpected threaded function: {function_name}")
+
+        with (
+            patch("asyncio.to_thread", side_effect=fake_to_thread),
+            patch("aiohttp.web.json_response") as mock_json_res,
+        ):
+            await post_handler(mock_request)
+
+        self.assertEqual(
+            ["<settings>", "get_workflow_model_inventory"],
+            threaded_functions,
+        )
+        response_data = mock_json_res.call_args.args[0]
+        self.assertTrue(response_data["success"])
+        self.assertEqual(0, response_data["count"])
+
     async def test_search_progress_route(self):
         progress_id = "test_search_job"