Fix Windows call graph traversal for flows helper modules

Resolve PyCG graph keys such as utils.xx to flows\utils.xx on Windows
so extract/build pick up OneCode elements defined in imported helper
modules under flows/.

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: theweaklink

onecode/cli/utils.py

@@ -130,6 +130,31 @@ def run():
         json.dump(flows, f, indent=4)
 
 
+def _resolve_graph_key(name: str, graph: Dict) -> str:
+    """
+    Resolve a PyCG graph key across platform naming differences.
+
+    On Windows, relative imports inside ``flows/`` are often keyed as ``utils.xx``
+    while the analyzed module lives under ``flows\\utils.xx``.
+    """
+    if name in graph:
+        return name
+
+    candidates = []
+    module_name = name[len('flows.'):] if name.startswith('flows.') else name
+
+    if not name.startswith('flows.'):
+        candidates.append(f'flows.{module_name}')
+    candidates.append(name)
+    candidates.append(f'flows\\{module_name}')
+
+    for candidate in candidates:
+        if candidate in graph:
+            return candidate
+
+    return name
+
+
 # check_type decorator not compatible with recursive calls
 def extract_calls(
     entry_point: str,
@@ -164,7 +189,9 @@ def extract_calls(
 
     # PyCG is not exactly equivalent on Windows vs Linux wrt to graph keys
     if os.name == 'nt' and not entry_point.startswith('flows\\'):
-        entry_point = f'flows\\{entry_point}'
+        entry_point = f'flows\\{entry_point.replace(".", "\\")}'
+
+    entry_point = _resolve_graph_key(entry_point, graph)
 
     if entry_point in graph:
         for fn in graph[entry_point]:
@@ -183,7 +210,8 @@ def extract_calls(
                 if verbose:
                     print(f" >> ({entry_point}) function {fn['normed']} ⏩")
 
-                extract_calls(fn['normed'], graph, calls)
+                next_point = _resolve_graph_key(fn['normed'], graph)
+                extract_calls(next_point, graph, calls)
 
 
 @check_type

tests/unit/cli/test_utils.py

@@ -5,6 +5,46 @@
 
 from onecode import Project, register_ext_module
 from onecode.cli import process_call_graph
+from onecode.cli.utils import _resolve_graph_key, extract_calls
+
+
+def test_resolve_graph_key_windows_style():
+    graph = {
+        'flows\\step2.run': [],
+        'flows\\utils.xx': [
+            {
+                'normed': 'onecode.slider',
+                'code': "onecode.slider('My slider\"1', 0.5, max=6)",
+            }
+        ],
+    }
+
+    assert _resolve_graph_key('utils.xx', graph) == 'flows\\utils.xx'
+    assert _resolve_graph_key('flows\\step2.run', graph) == 'flows\\step2.run'
+
+
+def test_extract_calls_resolves_windows_helper_modules():
+    graph = {
+        'flows\\step2.run': [
+            {
+                'normed': 'utils.xx',
+                'code': 'xx()',
+            }
+        ],
+        'flows\\utils.xx': [
+            {
+                'normed': 'onecode.slider',
+                'code': "onecode.slider('My slider\"1', 0.5, max=6)",
+            }
+        ],
+    }
+
+    Project().reset()
+    calls = []
+    extract_calls('flows\\step2.run', graph, calls)
+
+    assert len(calls) == 1
+    assert calls[0]['func'] == 'onecode.slider'
 
 
 def test_invalid_call_graph():