inputlayer-client-dev 0.1.0.dev912__py3-none-any.whl → 0.1.0.dev913__py3-none-any.whl

inputlayer/integrations/langchain/retriever.py

@@ -21,6 +21,21 @@ from inputlayer.integrations.langchain.params import bind_params
 
 logger = logging.getLogger(__name__)
 
+
+def _extract_error_message(result: Any) -> str:
+    """Extract the error text from an ``columns == ["error"]`` result.
+
+    Guards against missing rows and empty rows so the retriever never
+    crashes with IndexError when the engine sends a malformed error
+    envelope.
+    """
+    rows = getattr(result, "rows", None) or []
+    if not rows or not rows[0]:
+        return "unknown error"
+    msg = rows[0][0]
+    return str(msg) if msg is not None else "unknown error"
+
+
 class InputLayerRetriever(BaseRetriever):
     """Retrieve documents from an InputLayer KnowledgeGraph.
 
@@ -59,6 +74,11 @@ class InputLayerRetriever(BaseRetriever):
             )
 
     Both sync and async paths are supported natively.
+
+    When the engine rejects a query (parse error, unknown relation,
+    etc.) the retriever raises ``RuntimeError`` with the engine message
+    rather than returning empty results, so a misconfigured chain fails
+    loudly instead of silently producing zero documents.
     """
 
     kg: Any  # KnowledgeGraph - typed as Any for Pydantic compatibility
@@ -145,8 +165,9 @@ class InputLayerRetriever(BaseRetriever):
         logger.debug("IQL retriever query: %s", compiled)
         result = await self.kg.execute(compiled)
         if result.columns == ["error"]:
-            msg = result.rows[0][0] if result.rows else "unknown error"
-            raise RuntimeError(f"InputLayer rejected query: {msg}")
+            raise RuntimeError(
+                f"InputLayer rejected query: {_extract_error_message(result)}"
+            )
         return self._to_documents(result.columns, result.rows, hidden_columns=set())
 
     def _resolve_params(self, user_query: str) -> dict[str, Any]:
@@ -244,7 +265,9 @@ class InputLayerRetriever(BaseRetriever):
 
         resolved_content: list[str] = []
         content_cols = (
-            self.page_content_columns if self.page_content_columns is not None else ["content"]
+            self.page_content_columns
+            if self.page_content_columns is not None
+            else ["content"]
         )
         explicit_content = self.page_content_columns is not None
         for c in content_cols:

inputlayer/integrations/langchain/tool.py

@@ -82,6 +82,10 @@ class InputLayerIQLTool(BaseTool):
        placeholder; the agent's input is safely bound (escaped, quoted)
        rather than spliced as raw text.
 
+    Engine errors are returned to the agent as ``"Error: <message>"`` so
+    the tool-calling LLM can observe the failure and adjust its next
+    action instead of the coroutine raising an exception mid-chain.
+
     Prefer ``tools_from_relations`` for normal agent use.
     """
 
@@ -137,6 +141,8 @@ class InputLayerIQLTool(BaseTool):
 
         logger.debug("IQL tool query: %s", compiled)
         result = await self.kg.execute(compiled)
+        if result.columns == ["error"]:
+            return f"Error: {_extract_error_message(result)}"
         return _format_result(result, self.max_rows)
 
 
@@ -171,7 +177,9 @@ def tools_from_relations(
         agent = create_tool_calling_agent(llm, tools, prompt)
 
     Returned tools emit JSON arrays of row dicts so tool-calling LLMs
-    can parse them directly.
+    can parse them directly. Engine errors are returned as a JSON
+    ``{"error": "<message>"}`` object rather than raising, so the agent
+    can recover from a bad filter without aborting the chain.
     """
     return [
         _relation_to_tool(kg, r, max_rows=max_rows, name_prefix=name_prefix)
@@ -343,9 +351,7 @@ def _make_relation_runner(
             logger.debug("Structured tool query: %s", q)
             result = await kg.execute(q)
             if result.columns == ["error"]:
-                return json.dumps(
-                    {"error": result.rows[0][0] if result.rows else "unknown error"}
-                )
+                return json.dumps({"error": _extract_error_message(result)})
             if not merged_columns:
                 merged_columns = result.columns
             for row in result.rows:
@@ -359,8 +365,12 @@ def _make_relation_runner(
             if len(merged_rows) >= max_rows + 1:
                 break
 
-        # Wrap merged rows in a synthetic ResultSet-like object.
-        return _format_result(_FakeResult(merged_columns, merged_rows), max_rows)
+        was_truncated = len(merged_rows) > max_rows
+        capped = merged_rows[:max_rows]
+        fake = _FakeResult(merged_columns, capped)
+        if was_truncated:
+            fake.row_count = max_rows + 1
+        return _format_result(fake, max_rows)
 
     run.__name__ = f"search_{rel_name}"
     run.parse_clauses = parse_clauses  # type: ignore[attr-defined]
@@ -393,6 +403,24 @@ def _hashable(v: Any) -> Any:
 # ── Result formatting ────────────────────────────────────────────────
 
 
+def _extract_error_message(result: Any) -> str:
+    """Extract a single error message from an ``error`` result set.
+
+    The engine signals an error with ``columns == ["error"]`` and a single
+    row whose first cell is the message. Guard against both the no-row and
+    empty-row shapes so a malformed response never masquerades as a
+    successful result or crashes the caller.
+    """
+    rows = getattr(result, "rows", None) or []
+    if not rows:
+        return "unknown error"
+    first = rows[0]
+    if not first:
+        return "unknown error"
+    msg = first[0]
+    return str(msg) if msg is not None else "unknown error"
+
+
 def _format_result(result: Any, max_rows: int) -> str:
     """Format a ResultSet as a JSON array of row dicts.
 
@@ -403,18 +431,10 @@ def _format_result(result: Any, max_rows: int) -> str:
         return "[]"
 
     rows = result.rows[:max_rows]
-    columns = result.columns
-    payload = []
-    for row in rows:
-        if len(row) != len(columns):
-            logger.warning(
-                "Row length (%d) does not match column count (%d); "
-                "truncating to shorter. Columns: %r",
-                len(row), len(columns), columns,
-            )
-        payload.append(
-            {col: _jsonify(val) for col, val in zip(columns, row, strict=False)}
-        )
+    payload = [
+        {col: _jsonify(val) for col, val in zip(result.columns, row, strict=True)}
+        for row in rows
+    ]
 
     total = getattr(result, "row_count", len(result.rows)) or len(result.rows)
     if total > max_rows:

inputlayer/integrations/langchain/vector_store.py

@@ -72,6 +72,13 @@ class InputLayerVectorStore(VectorStore):
         on the provider itself, e.g.
         ``OpenAIEmbeddings(timeout=30, max_retries=2)``.
 
+    .. note::
+
+        ``add_texts`` validates that ``texts``, ``metadatas``, and ``ids``
+        all have the same length and raises ``ValueError`` up front.
+        Previously a mismatch would surface as an opaque ``zip`` error
+        deep inside the embedding call.
+
     Usage::
 
         class Chunk(Relation):
@@ -190,6 +197,17 @@ class InputLayerVectorStore(VectorStore):
         if not texts_list:
             return []
 
+        if metadatas is not None and len(metadatas) != len(texts_list):
+            raise ValueError(
+                f"Length mismatch: {len(texts_list)} texts but "
+                f"{len(metadatas)} metadata dicts"
+            )
+        if ids is not None and len(ids) != len(texts_list):
+            raise ValueError(
+                f"Length mismatch: {len(texts_list)} texts but "
+                f"{len(ids)} ids"
+            )
+
         vectors = await self._embeddings.aembed_documents(texts_list)
         ids_out = ids or [str(uuid.uuid4()) for _ in texts_list]
         metas = metadatas or [{} for _ in texts_list]
@@ -448,13 +466,14 @@ class InputLayerVectorStore(VectorStore):
             ),
         )
 
-        # Find the vector column index in the result.
-        vec_col_idx = None
-        for i, c in enumerate(result.columns):
-            if c.lower() == self._vector_field.lower():
-                vec_col_idx = i
-                break
-
+        vec_col_idx = next(
+            (
+                i
+                for i, c in enumerate(result.columns)
+                if c.lower() == self._vector_field.lower()
+            ),
+            None,
+        )
         if vec_col_idx is None:
             raise ValueError(
                 f"Could not find vector column {self._vector_field!r} in "
@@ -467,7 +486,7 @@ class InputLayerVectorStore(VectorStore):
         out: list[tuple[Document, float, list[float]]] = []
         docs_and_scores = self._rows_to_documents(result.columns, result.rows)
         for (doc, score), row in zip(docs_and_scores, result.rows, strict=True):
-            raw = row[vec_col_idx] if vec_col_idx is not None else None
+            raw = row[vec_col_idx]
             vec = list(raw) if raw is not None else []
             out.append((doc, score, vec))
         return out
@@ -584,7 +603,8 @@ class InputLayerVectorStore(VectorStore):
                 if k.lower() == self._vector_field.lower():
                     continue
                 metadata[canonical(k)] = v
-            score = float(row_dict[score_col]) if score_col else 0.0
+            raw_score = row_dict.get(score_col) if score_col else None
+            score = float(raw_score) if raw_score is not None else 0.0
             out.append((Document(page_content=content, metadata=metadata), score))
         return out
 

{inputlayer_client_dev-0.1.0.dev912.dist-info → inputlayer_client_dev-0.1.0.dev913.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inputlayer-client-dev
-Version: 0.1.0.dev912
+Version: 0.1.0.dev913
 Summary: Python Object-Logic Mapper for InputLayer knowledge graph engine
 License-Expression: Apache-2.0
 Requires-Python: >=3.10

{inputlayer_client_dev-0.1.0.dev912.dist-info → inputlayer_client_dev-0.1.0.dev913.dist-info}/RECORD

@@ -24,9 +24,9 @@ inputlayer/types.py,sha256=EfL2oqjxvfdNzZCKoJyBaobge4Z4cT1AJYEIzJIsPQo,6546
 inputlayer/integrations/__init__.py,sha256=4feiWcNvqCShcgjsYid1AcrPxJiLwREQ3OMgnIaahxg,276
 inputlayer/integrations/langchain/__init__.py,sha256=JHirtEbpniYtjmXzux7py1miAuIAwhGQX3PBwMylY3Y,1252
 inputlayer/integrations/langchain/params.py,sha256=9VBcTcr0A4oKkyVDvAwPYNL2xraUa_COAFU60DZKjFA,4437
-inputlayer/integrations/langchain/retriever.py,sha256=h59yjroOHXzzEUM1Gjl-AzHYSa8I9nsx6JpsE2of3wU,12946
-inputlayer/integrations/langchain/tool.py,sha256=6OJkMol2tQ101HL5XlIYV-vHqI4mhGTOjKCSTzVr1AU,15616
-inputlayer/integrations/langchain/vector_store.py,sha256=mqpDgFI99xXlLQBQzo5giAZ3UKktqm-L7mewzr7tb6w,22664
+inputlayer/integrations/langchain/retriever.py,sha256=Q0Llppmb7pmN_it2ePm1wRC19UyfcIr7h15Mds0G6G8,13699
+inputlayer/integrations/langchain/tool.py,sha256=-8t6w7r14nEFbymZ_iqE6V25HOXyn4ajw6HV-Uy34aI,16478
+inputlayer/integrations/langchain/vector_store.py,sha256=7Db1sRgkHJvgLiC_6avbca0NusieIkmWE_8IqjVkuMA,23384
 inputlayer/integrations/langgraph/__init__.py,sha256=nbZx0_mJlsHixdqcfQMmUSN8dLYkePou2k2LqwfIfqU,2345
 inputlayer/integrations/langgraph/_checkpoint_serde.py,sha256=DWu7zTx4jLfBROMjzEFbFrEWK12LweYQ0WcxY5yDhsk,3294
 inputlayer/integrations/langgraph/_checkpointer_mixin.py,sha256=N15WjHFJ89q_H9GGjWHsW2xtKJ_NIwbb0tZapV5c5Ac,12210
@@ -47,7 +47,7 @@ inputlayer/migrations/operations.py,sha256=_qiguq0GFcXVZgIJa4vHu7eeKSa5BUh2maDce
 inputlayer/migrations/recorder.py,sha256=2LZyhifIyultiNLLKmUnqmi9voKtDfmOYVDmjC5hNxs,1499
 inputlayer/migrations/state.py,sha256=BH5_ly9I9bVPn1lvhqS_UWS7NvYg2iCLJrbG50-mwlQ,3695
 inputlayer/migrations/writer.py,sha256=XbQN5t5H538AASStAQ9jkrveYZwf2aFEFiXZsifUWI0,6465
-inputlayer_client_dev-0.1.0.dev912.dist-info/METADATA,sha256=UpW-PnvvAfAZEa8DIzs21YyKL_kxkpCc13boHHz80Wc,19726
-inputlayer_client_dev-0.1.0.dev912.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-inputlayer_client_dev-0.1.0.dev912.dist-info/entry_points.txt,sha256=Y90cH3HHrT10BztOigvvD1WdNIt_RYR6k6IsVbs_A8U,54
-inputlayer_client_dev-0.1.0.dev912.dist-info/RECORD,,
+inputlayer_client_dev-0.1.0.dev913.dist-info/METADATA,sha256=w78ZmmnmDRXK9i9sIxeZ6LbYAZAW2qO4EDtxx1rqqlk,19726
+inputlayer_client_dev-0.1.0.dev913.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+inputlayer_client_dev-0.1.0.dev913.dist-info/entry_points.txt,sha256=Y90cH3HHrT10BztOigvvD1WdNIt_RYR6k6IsVbs_A8U,54
+inputlayer_client_dev-0.1.0.dev913.dist-info/RECORD,,