codegraphcontext 0.6.1__py3-none-any.whl → 0.6.2__py3-none-any.whl

codegraphcontext/core/database_embedded_kuzu.py

@@ -798,21 +798,32 @@ class EmbeddedSessionWrapper:
         translated_query, translated_params = self._translate_query(query, parameters)
         debug_log(f"Translated Query: {translated_query[:200]}")
         try:
-            # Force loop fallback for relationship writes inside UNWIND to avoid Kuzu query planner bugs
-            # which can incorrectly bind/corrupt relationship endpoints across rows in the batch.
-            # Only force it for `UNWIND $param AS row`, the single shape the
-            # recovery path below can rewrite into a per-row loop. Read queries
-            # that unwind a bound list (`WITH relationships(p) AS rels UNWIND
-            # rels AS r`, used by find_all_callers/find_all_callees and the
-            # visualizer) matched this guard too, but no `$`-UNWIND for the
-            # fallback to latch onto — so the fabricated exception fell through
-            # every handler and surfaced to the caller as a hard failure.
+            # Force the per-row loop fallback ONLY for UNWIND batches containing a
+            # node-MERGE. Kùzu's MERGE pipeline mis-binds when a merged node's key
+            # value repeats NON-adjacently across the batch: in
+            # [(A,P), (B,X), (C,P)] the C row binds to X's node instead of
+            # re-matching P (reproduced on Kùzu 0.11.3; see #1605). Relationship-only
+            # MERGEs (MATCH … MATCH … MERGE (a)-[r]->(b)) do not exhibit the bug —
+            # verified with interleaved duplicate endpoint keys and duplicate pairs —
+            # so they now run batched, which removes the bulk of the per-row planner
+            # overhead (~2.4x on a full index of the Python sample project).
+            # Only `UNWIND $param AS row` shapes are guarded — that is the single
+            # shape the recovery path below can rewrite into a per-row loop. Read
+            # queries that unwind a bound list (`WITH relationships(p) AS rels
+            # UNWIND rels AS r`) must not match, or the fabricated exception would
+            # surface to the caller as a hard failure.
+            # Node-only UNWIND writes (`MERGE (n:Label {…}) SET n += row`, no
+            # relationship pattern) have always batched through the SET-expansion
+            # path and the per-row rewriter cannot handle their bare `SET n += row`,
+            # so the guard additionally requires a relationship pattern.
+            _has_node_merge = re.search(r"MERGE\s*\(\s*\w+\s*:", query)
             if (
                 re.search(r"UNWIND\s+\$\w+\s+AS\s+\w+", query)
                 and ("-[" in query or "]->" in query)
+                and _has_node_merge
                 and not getattr(self, "_skip_unwind_fallback", False)
             ):
-                raise Exception("unordered_map::at (forced fallback to avoid relationship UNWIND planner bugs)")
+                raise Exception("unordered_map::at (forced fallback: node-MERGE inside UNWIND mis-binds repeated keys)")
 
             # 2. Execute under the lock. _write_lock (name kept for backward
             # compat) now serializes ALL access, reads included: kuzu.Connection

codegraphcontext/tools/indexing/persistence/writer.py

@@ -1138,8 +1138,23 @@ class GraphWriter:
         batch_size = 500
         backend = get_backend_type(self.driver, self._db_manager)
         def _work(session):
-            internal_batch = [r for r in inheritance_batch if r.get("resolved_parent_file_path") != "__external__"]
-            external_batch = [r for r in inheritance_batch if r.get("resolved_parent_file_path") == "__external__"]
+            # Dedupe identical rows first: parsers can emit the same
+            # (child, parent) record more than once, and while Neo4j's MERGE
+            # absorbs the repeat, the embedded backends' rewritten
+            # node-MERGE+rel-MERGE shape has been observed writing a duplicate
+            # INHERITS edge for it. Deduping is correct on every backend.
+            seen_rows: set = set()
+            deduped_batch: List[Dict[str, Any]] = []
+            for r in inheritance_batch:
+                key = (r.get("child_name"), r.get("path"), r.get("parent_name"),
+                       r.get("resolved_parent_file_path"))
+                if key in seen_rows:
+                    continue
+                seen_rows.add(key)
+                deduped_batch.append(r)
+
+            internal_batch = [r for r in deduped_batch if r.get("resolved_parent_file_path") != "__external__"]
+            external_batch = [r for r in deduped_batch if r.get("resolved_parent_file_path") == "__external__"]
 
             labels = ("Class", "Trait", "Interface", "Struct", "Enum", "Union", "Record", "Mixin", "Extension", "Module", "Object", "Variable")
 

codegraphcontext/tools/indexing/pipeline.py

@@ -201,10 +201,32 @@ async def run_tree_sitter_index_async(
             # it to this file's path string turned a single recoverable write
             # failure into an AttributeError that aborted the whole job.
             failed_path = file_data.get("path")
-            write_failures.append({"path": failed_path, "error": str(exc)})
-            error_logger(f"Failed to write {failed_path} to the graph: {exc}")
-            file_data["error"] = str(exc)
-            file_data["parse_failed"] = True
+            # Retry once before giving up: writes are MERGE-idempotent, and a
+            # transient failure under runner load silently costs the graph the
+            # whole file's edges (observed as LadybugDB intermittently writing
+            # ~51 fewer CONTAINS edges in the parity run, #1612).
+            retried_ok = False
+            if "error" not in file_data:
+                try:
+                    await asyncio.sleep(0.2)
+                    await asyncio.to_thread(
+                        writer.add_file_to_graph,
+                        file_data,
+                        repo_name,
+                        imports_map,
+                        repo_path_str=resolved_repo_path_str,
+                    )
+                    retried_ok = True
+                    warning_logger(
+                        f"Write for {failed_path} succeeded on retry after: {exc}"
+                    )
+                except Exception as retry_exc:  # noqa: BLE001
+                    exc = retry_exc
+            if not retried_ok:
+                write_failures.append({"path": failed_path, "error": str(exc)})
+                error_logger(f"Failed to write {failed_path} to the graph: {exc}")
+                file_data["error"] = str(exc)
+                file_data["parse_failed"] = True
 
     if write_failures:
         warning_logger(

codegraphcontext/tools/languages/cpp.py

@@ -519,11 +519,7 @@ class CppTreeSitterParser:
                 if lambda_node is None or lambda_node.type != 'lambda_expression':
                     continue
 
-                params_node = lambda_node.child_by_field_name('declarator')
-                if params_node:
-                    params_node = params_node.child_by_field_name('parameters')
                 name = self._get_node_text(node)
-                params_node = lambda_node.child_by_field_name('parameters')
 
                 context, context_type, _ = self._get_parent_context(assignment_node)
                 class_context, _, _ = self._get_parent_context(assignment_node, types=('class_specifier',))
@@ -532,7 +528,15 @@ class CppTreeSitterParser:
                     "name": name,
                     "line_number": node.start_point[0] + 1,
                     "end_line": assignment_node.end_point[0] + 1,
-                    "args": [p for p in [self._get_node_text(p) for p in params_node.children if p.type == 'identifier'] if p] if params_node else [],
+                    # lambda_expression carries the same declarator→parameters
+                    # field chain as a function_definition, so the shared
+                    # extractor handles parameter_declaration unwrapping
+                    # (pointers/refs included). The old inline version first
+                    # clobbered its correctly-walked parameter_list with a
+                    # nonexistent field and then filtered for bare identifiers
+                    # over parameter_declaration nodes — every lambda got
+                    # args: [] (#1527, case 5).
+                    "args": self._extract_function_params(lambda_node),
                     
                     "docstring": None,
                     "cyclomatic_complexity": 1,

{codegraphcontext-0.6.1.dist-info → codegraphcontext-0.6.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.6.1
+Version: 0.6.2
 Summary: An MCP server that indexes local code into a graph database to provide context to AI assistants.
 Author-email: Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 License: MIT License

{codegraphcontext-0.6.1.dist-info → codegraphcontext-0.6.2.dist-info}/RECORD

@@ -26,7 +26,7 @@ codegraphcontext/core/bundle_registry.py,sha256=xlVBe-fmvEOLY9xDq7w0vXFlTPzeFVIw
 codegraphcontext/core/cgc_bundle.py,sha256=X2aOji754UyjvCEJc3hOQCP6KF_H4OJNQMfUunrYY5I,74714
 codegraphcontext/core/cgcignore.py,sha256=x50FOG04kmR12LqixTlfiHW4ka05jf6_jaRA-5yiqis,10847
 codegraphcontext/core/database.py,sha256=ZLOHXv46hVxWiPYQala0Co7_Z8vqj5juaenudwq6DTU,14893
-codegraphcontext/core/database_embedded_kuzu.py,sha256=xa3hpCJcV4mcigmJf0RkafZULOzOxhkZtZD4Y0h0CcQ,81677
+codegraphcontext/core/database_embedded_kuzu.py,sha256=85J43enxfPgEAR9lu2HflaD-Pmfcx4Mc5vCka_Vho5o,82482
 codegraphcontext/core/database_falkordb.py,sha256=8TSFfgKoaNdTzSx7P4qXMMs2kTNBzrcW5tZxR8JV4bY,35834
 codegraphcontext/core/database_falkordb_remote.py,sha256=HdkcmwgUxPy1bwiFqqcU8yMQYib7B3JarYQ1LlTwfJ8,7957
 codegraphcontext/core/database_kuzu.py,sha256=5I31dGm7wRo3YjIWeWGRVcX1eYRXYc7fPWnT7zHFjf0,1123
@@ -61,7 +61,7 @@ codegraphcontext/tools/indexing/__init__.py,sha256=exCNeB4UYhKhNSoEBNta_EZ51Kdze
 codegraphcontext/tools/indexing/constants.py,sha256=-ZiD5tdX95efNo2gOFrTNlmDnY4TALMN4GT0jOGPbcQ,742
 codegraphcontext/tools/indexing/discovery.py,sha256=l0uoCCPRWClXPbHzbExrkC2RRjhN8u5KX6gOhQZliII,5326
 codegraphcontext/tools/indexing/embeddings.py,sha256=Iny1K1HZaHtgApWpw88RLy1T10Rk4yJ8G6LNt5OcKfg,11406
-codegraphcontext/tools/indexing/pipeline.py,sha256=oDx3l65OC5lwqtkJ5ImaquaCfnpyZ4DNNEAXQhvAvto,18602
+codegraphcontext/tools/indexing/pipeline.py,sha256=1JFgIj5VYjtzPQOERdFJqunh4CFjH7QJNiqbsYPspjo,19653
 codegraphcontext/tools/indexing/pre_scan.py,sha256=St7Hc3-NFTMYF39vWneJL6iIRIc78vyKEWABJstFM3c,6956
 codegraphcontext/tools/indexing/sanitize.py,sha256=lh78GDdYHxsd5fXlhS-bdOaOlMGD0Z2t8xALy9gX9fQ,3870
 codegraphcontext/tools/indexing/schema.py,sha256=3iIX6ToqGyZ41hkDmxJnETLSl98FXOqBM5dz8tRkQmg,8910
@@ -70,13 +70,13 @@ codegraphcontext/tools/indexing/scip_pipeline.py,sha256=OLbf0rhs1XU4LTQ6B3mS1-ng
 codegraphcontext/tools/indexing/vector_resolver.py,sha256=0D0QClro7nx9Wx6hbbFlOsO4dqgcTrwPMqE0CsJK01o,5477
 codegraphcontext/tools/indexing/persistence/__init__.py,sha256=hFRlNgiYS7uN9X4e4xfP3c1BvcYSxpmq0k_uaDRn2Fg,121
 codegraphcontext/tools/indexing/persistence/utils.py,sha256=OtndHpE_R14mPz5i-Bo9mA7NsoIuYnCNER-oIwxDOYA,2028
-codegraphcontext/tools/indexing/persistence/writer.py,sha256=j55lOTg422A7MRsxX1bf3GtILFw3axYcIrDfhWU-p2Q,111514
+codegraphcontext/tools/indexing/persistence/writer.py,sha256=qh8-QHiUtIKUBbzNqzbEqmiHESmJ8FfCCrDmSxFVHoY,112282
 codegraphcontext/tools/indexing/resolution/__init__.py,sha256=BR57mVydpzjOnFSbGbwIIrPEWp1eHjLPAq6oUq_jutw,567
 codegraphcontext/tools/indexing/resolution/calls.py,sha256=q2YFGc0PACvCJn-Vr4pUwM4_IZwWm9Spof_4IpUpvo4,131656
 codegraphcontext/tools/indexing/resolution/inheritance.py,sha256=wS3fczEPcUkCDc-H7ON7YXm32EaDteuukK_Quq1I8dg,35772
 codegraphcontext/tools/indexing/resolution/post_resolution.py,sha256=OzmbbmLwIbKb5wm1RodoyvDST3d35A2e35ruDKBQDF4,10859
 codegraphcontext/tools/languages/c.py,sha256=quo2j0PCTlSOJctqqF_5zKeisERvTDyKuPtz_qvFgQA,31618
-codegraphcontext/tools/languages/cpp.py,sha256=lEErpGUSOlQzRn7NxK58JUF41OFj12sYdWMoLeUISxE,34521
+codegraphcontext/tools/languages/cpp.py,sha256=s6dZsb2uiW2Spe0TVs37A8GzDeiLQG-yROUZCg4k1tM,34761
 codegraphcontext/tools/languages/csharp.py,sha256=_Kmpz1gqqkBCzhuHVyDJ5MsD2yI1Vh6PMGp0Ser_xjA,27170
 codegraphcontext/tools/languages/css.py,sha256=aN1Nm8vHmfV0J0jQ-_q5x0YDTOtLV2IW-5lp7WtGG5o,3789
 codegraphcontext/tools/languages/dart.py,sha256=yjqVA6XwoADrZab95p6wva0jT-PAGw_x90LR-cp2BDk,27160
@@ -174,9 +174,9 @@ codegraphcontext/viz/dist/wasm/tree-sitter-typescript.wasm,sha256=hRVATc7tOOHthq
 codegraphcontext/viz/dist/wasm/tree-sitter.wasm,sha256=CCeVuI_hXktkBD-DW01xYPv5XoAYVRIqzJUJI5te-RY,196763
 codegraphcontext/viz/dist/wasm/web-tree-sitter.js,sha256=DIaCNqRylrT_PBVw8g4ImeSnhP9uXNe_ycOlUiVGPko,153666
 codegraphcontext/viz/dist/wasm/web-tree-sitter.wasm,sha256=CCeVuI_hXktkBD-DW01xYPv5XoAYVRIqzJUJI5te-RY,196763
-codegraphcontext-0.6.1.dist-info/licenses/LICENSE,sha256=rh8M-bJpQYJnw2vtRVgt0t7piMZXh5QzaKeNEI0vqqA,1061
-codegraphcontext-0.6.1.dist-info/METADATA,sha256=nBcEmRNjpr0p4qu4-umRhZlF79XICLP4VzvwQ1FQO0E,33422
-codegraphcontext-0.6.1.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
-codegraphcontext-0.6.1.dist-info/entry_points.txt,sha256=LCxWCWMshdvYGoHBPuQZ8C-e4CiNSHCLXofrNSGHkoE,103
-codegraphcontext-0.6.1.dist-info/top_level.txt,sha256=CBgc6LAPZIO5FS0nSYYkylDifHsZTIqw3Gf5UwDxeGI,17
-codegraphcontext-0.6.1.dist-info/RECORD,,
+codegraphcontext-0.6.2.dist-info/licenses/LICENSE,sha256=rh8M-bJpQYJnw2vtRVgt0t7piMZXh5QzaKeNEI0vqqA,1061
+codegraphcontext-0.6.2.dist-info/METADATA,sha256=PDwab-ix3AJJeGpZY6MBVK5TBc-clvKlMPGblbwnrVE,33422
+codegraphcontext-0.6.2.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+codegraphcontext-0.6.2.dist-info/entry_points.txt,sha256=LCxWCWMshdvYGoHBPuQZ8C-e4CiNSHCLXofrNSGHkoE,103
+codegraphcontext-0.6.2.dist-info/top_level.txt,sha256=CBgc6LAPZIO5FS0nSYYkylDifHsZTIqw3Gf5UwDxeGI,17
+codegraphcontext-0.6.2.dist-info/RECORD,,