PyPI - agent-coderag - Versions diffs - 1.3.4__tar.gz → 1.3.5__tar.gz - Mend

@@ -156,6 +156,34 @@ def test_bind_rejects_invalid_and_mismatch():
         embedder.bind_dimension(4)
+@pytest.mark.asyncio
+async def test_aembed_accepts_litellm_dict_items():
+    embedder = OpenAICompatEmbedder(api_base="http://e", model="m")
+    resp = MagicMock()
+    resp.data = [{"embedding": [3.0, 4.0, 0.0], "index": 0, "object": "embedding"}]
+    with patch(
+        "code_rag.intelligence.openai_embedder.litellm.aembedding",
+        new=AsyncMock(return_value=resp),
+    ):
+        rows = await embedder.aembed(["probe"])
+    assert len(rows[0]) == 3
+    np.testing.assert_allclose(np.linalg.norm(rows[0]), 1.0, rtol=1e-5)
+@pytest.mark.asyncio
+async def test_aembed_flattens_nested_embedding_vector():
+    embedder = OpenAICompatEmbedder(api_base="http://e", model="m")
+    embedder.bind_dimension(2)
+    resp = MagicMock()
+    resp.data = [{"embedding": [[1.0, 0.0]]}]
+    with patch(
+        "code_rag.intelligence.openai_embedder.litellm.aembedding",
+        new=AsyncMock(return_value=resp),
+    ):
+        rows = await embedder.aembed(["x"])
+    assert len(rows[0]) == 2
 @pytest.mark.asyncio
 async def test_aembed_rejects_unparseable_response():
     embedder = OpenAICompatEmbedder(api_base="http://e", model="m")

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-coderag
-Version: 1.3.4
+Version: 1.3.5
 Summary: Lightweight semantic code search and distillation utility for AI coding agents. It solves the API knowledge gap via real-time local signature extraction and intent analysis without PyTorch. Optimized for token efficiency, it compresses codebase context into compact semantic summaries stored in a local DuckDB vector similarity index.
 Author-email: Igor Boloban <naranor@gmail.com>
 License: MIT

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-coderag
-Version: 1.3.4
+Version: 1.3.5
 Summary: Lightweight semantic code search and distillation utility for AI coding agents. It solves the API knowledge gap via real-time local signature extraction and intent analysis without PyTorch. Optimized for token efficiency, it compresses codebase context into compact semantic summaries stored in a local DuckDB vector similarity index.
 Author-email: Igor Boloban <naranor@gmail.com>
 License: MIT

@@ -13,6 +13,18 @@ from .embedder import l2_normalize_rows
 logger = logging.getLogger(__name__)
+def _vector_from_item(item: object) -> list[float]:
+    if isinstance(item, dict):
+        vec = item.get("embedding")
+    else:
+        vec = getattr(item, "embedding", None)
+    if vec is None:
+        raise TypeError("embedding item has no embedding")
+    if isinstance(vec, (list, tuple)) and vec and isinstance(vec[0], (list, tuple)):
+        vec = vec[0]
+    return list(vec)
 class OpenAICompatEmbedder(IEmbedder):
     def __init__(
         self,
@@ -68,7 +80,7 @@ class OpenAICompatEmbedder(IEmbedder):
             except Exception as exc:
                 raise IntelligenceError(f"Embedding request failed: {exc}") from exc
         try:
-            raw = [list(item.embedding) for item in response.data]
+            raw = [_vector_from_item(item) for item in response.data]
         except Exception as exc:
             raise IntelligenceError(f"Embedding request failed: {exc}") from exc
         if len(raw) != len(texts):

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "agent-coderag"
-version = "1.3.4"
+version = "1.3.5"
 description = "Lightweight semantic code search and distillation utility for AI coding agents. It solves the API knowledge gap via real-time local signature extraction and intent analysis without PyTorch. Optimized for token efficiency, it compresses codebase context into compact semantic summaries stored in a local DuckDB vector similarity index."
 readme = "README.md"
 requires-python = ">=3.10"

agent-coderag 1.3.4__tar.gz → 1.3.5__tar.gz