cocoindex-code 0.2.40__tar.gz → 0.2.41__tar.gz

{cocoindex_code-0.2.40 → cocoindex_code-0.2.41}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cocoindex-code
-Version: 0.2.40
+Version: 0.2.41
 Summary: MCP server for indexing and querying codebases using CocoIndex
 Project-URL: Homepage, https://github.com/cocoindex-io/cocoindex-code
 Project-URL: Repository, https://github.com/cocoindex-io/cocoindex-code

{cocoindex_code-0.2.40 → cocoindex_code-0.2.41}/src/cocoindex_code/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.2.40'
-__version_tuple__ = version_tuple = (0, 2, 40)
+__version__ = version = '0.2.41'
+__version_tuple__ = version_tuple = (0, 2, 41)
 
 __commit_id__ = commit_id = None

{cocoindex_code-0.2.40 → cocoindex_code-0.2.41}/src/cocoindex_code/client.py

@@ -228,6 +228,19 @@ def _handle_vanished_daemon() -> None:
     )
 
 
+def _daemon_error(resp: ErrorResponse) -> RuntimeError:
+    """Build the exception for an ``ErrorResponse`` from the daemon.
+
+    The daemon-side traceback is appended when present: without it the caller
+    only ever sees the client's own re-raise frames, which says nothing about
+    where the failure actually happened (issue #270).
+    """
+    detail = f"Daemon error: {resp.message}"
+    if resp.traceback:
+        detail += f"\n{resp.traceback}"
+    return RuntimeError(detail)
+
+
 class _HandshakeResult(NamedTuple):
     conn: Connection
     resp: HandshakeResponse
@@ -261,7 +274,7 @@ def _raw_connect_and_handshake() -> _HandshakeResult:
         raise DaemonProtocolError(f"Undecodable handshake reply from daemon: {e}") from e
     if isinstance(resp, ErrorResponse):
         conn.close()
-        raise RuntimeError(f"Daemon error: {resp.message}")
+        raise _daemon_error(resp)
     if not isinstance(resp, HandshakeResponse):
         conn.close()
         raise RuntimeError(f"Unexpected handshake response: {type(resp).__name__}")
@@ -340,7 +353,7 @@ def _send(req: Request) -> Response:
         conn.close()
     resp = decode_response(data)
     if isinstance(resp, ErrorResponse):
-        raise RuntimeError(f"Daemon error: {resp.message}")
+        raise _daemon_error(resp)
     return resp
 
 
@@ -366,7 +379,7 @@ def index(
                 raise RuntimeError("Connection to daemon lost during indexing")
             resp = decode_response(data)
             if isinstance(resp, ErrorResponse):
-                raise RuntimeError(f"Daemon error: {resp.message}")
+                raise _daemon_error(resp)
             if isinstance(resp, IndexWaitingNotice):
                 if on_waiting is not None:
                     on_waiting()
@@ -419,7 +432,7 @@ def search(
                 raise RuntimeError("Connection to daemon lost during search")
             resp = decode_response(data)
             if isinstance(resp, ErrorResponse):
-                raise RuntimeError(f"Daemon error: {resp.message}")
+                raise _daemon_error(resp)
             if isinstance(resp, IndexWaitingNotice):
                 if on_waiting is not None:
                     on_waiting()
@@ -496,10 +509,7 @@ def doctor(
                 raise RuntimeError("Connection to daemon lost during doctor checks")
             resp = decode_response(data)
             if isinstance(resp, ErrorResponse):
-                detail = f"Daemon error: {resp.message}"
-                if resp.traceback:
-                    detail += f"\n{resp.traceback}"
-                raise RuntimeError(detail)
+                raise _daemon_error(resp)
             if isinstance(resp, DoctorResponse):
                 results.append(resp.result)
                 if on_result is not None:

{cocoindex_code-0.2.40 → cocoindex_code-0.2.41}/src/cocoindex_code/daemon.py

@@ -346,7 +346,8 @@ async def _search_with_wait(
             offset=req.offset,
         )
     except Exception as e:
-        yield ErrorResponse(message=str(e))
+        logger.exception("Error handling search request")
+        yield ErrorResponse(message=str(e), traceback=traceback.format_exc())
 
 
 async def _handle_doctor(
@@ -605,7 +606,7 @@ async def _dispatch(
         return ErrorResponse(message=f"Unknown request type: {type(req).__name__}")
     except Exception as e:
         logger.exception("Error dispatching request")
-        return ErrorResponse(message=str(e))
+        return ErrorResponse(message=str(e), traceback=traceback.format_exc())
 
 
 # ---------------------------------------------------------------------------

{cocoindex_code-0.2.40 → cocoindex_code-0.2.41}/src/cocoindex_code/query.py

@@ -16,6 +16,33 @@ def _l2_to_score(distance: float) -> float:
     return 1.0 - distance * distance / 2.0
 
 
+def _checked(rows: list[tuple[Any, ...]], query_shape: str) -> list[tuple[Any, ...]]:
+    """Return *rows*, failing loudly if any of them carries a NULL distance.
+
+    ``code_chunks_vec`` is a vec0 virtual table whose ``distance`` is a *hidden*
+    column: sqlite-vec populates it only under the KNN query plan and yields
+    NULL for it on a plain full scan.  A NULL therefore means the plan we asked
+    for is not the plan we got, which is a bug worth reporting rather than a
+    result worth ranking.  Left unchecked, the NULL flows into ``_l2_to_score``
+    and surfaces as an unrelated-looking ``TypeError: unsupported operand
+    type(s) for *: 'NoneType' and 'NoneType'`` (issue #270).
+
+    Only KNN queries need this guard: ``_full_scan_query`` computes its own
+    ``vec_distance_L2(...)``, which works under any plan and raises (never
+    returns NULL) on bad input.
+    """
+    bad = next((row for row in rows if row[5] is None), None)
+    if bad is not None:
+        raise RuntimeError(
+            f"Vector index returned a row with no distance ({query_shape}, "
+            f"file_path={bad[0]!r}) — the sqlite-vec KNN query plan was not "
+            "used. Please report this at "
+            "https://github.com/cocoindex-io/cocoindex-code/issues along with "
+            "the output of `ccc doctor`."
+        )
+    return rows
+
+
 def _knn_query(
     conn: sqlite3.Connection,
     embedding_bytes: bytes,
@@ -24,24 +51,30 @@ def _knn_query(
 ) -> list[tuple[Any, ...]]:
     """Run a vec0 KNN query, optionally constrained to a language partition."""
     if language is not None:
-        return conn.execute(
+        return _checked(
+            conn.execute(
+                """
+                SELECT file_path, language, content, start_line, end_line, distance
+                FROM code_chunks_vec
+                WHERE embedding MATCH ? AND k = ? AND language = ?
+                ORDER BY distance
+                """,
+                (embedding_bytes, k, language),
+            ).fetchall(),
+            f"knn language={language!r}",
+        )
+    return _checked(
+        conn.execute(
             """
             SELECT file_path, language, content, start_line, end_line, distance
             FROM code_chunks_vec
-            WHERE embedding MATCH ? AND k = ? AND language = ?
+            WHERE embedding MATCH ? AND k = ?
             ORDER BY distance
             """,
-            (embedding_bytes, k, language),
-        ).fetchall()
-    return conn.execute(
-        """
-        SELECT file_path, language, content, start_line, end_line, distance
-        FROM code_chunks_vec
-        WHERE embedding MATCH ? AND k = ?
-        ORDER BY distance
-        """,
-        (embedding_bytes, k),
-    ).fetchall()
+            (embedding_bytes, k),
+        ).fetchall(),
+        "knn unfiltered",
+    )
 
 
 def _full_scan_query(