fix: prefix key_path with field type in memory search fields

The LLMOps memory search API requires key_path to be prefixed with
the field type: ["agent-input", "fieldName"] for episodic memory,
["escalation-input", "fieldName"] for escalation memory. This matches
the Temporal backend's FieldBuilder (FieldBuilder.cs:15):
  keyPath = [fieldType.StringValue(), fieldName]

Without this prefix, the API returns 400 Bad Request.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: mjnovice

src/uipath_langchain/agent/react/memory_node.py

@@ -149,8 +149,15 @@ def _extract_user_inputs(state: Any) -> dict[str, Any]:
 def _build_search_fields(
     input_arguments: dict[str, Any],
     field_weights: dict[str, float] | None = None,
+    field_type: str = "agent-input",
 ) -> list[SearchField]:
-    """Convert agent input arguments to SearchField objects."""
+    """Convert agent input arguments to SearchField objects.
+
+    The key_path must be prefixed with the field type, matching the
+    Temporal backend's FieldBuilder (FieldBuilder.cs:15):
+      keyPath = [fieldType.StringValue(), fieldName]
+    e.g. ["agent-input", "a"] for episodic memory.
+    """
     fields: list[SearchField] = []
     for name, value in input_arguments.items():
         if value is None or name.startswith("uipath__"):
@@ -161,5 +168,9 @@ def _build_search_fields(
         settings = FieldSettings()
         if field_weights and name in field_weights:
             settings = FieldSettings(weight=field_weights[name])
-        fields.append(SearchField(key_path=[name], value=str(value), settings=settings))
+        fields.append(
+            SearchField(
+                key_path=[field_type, name], value=str(value), settings=settings
+            )
+        )
     return fields

src/uipath_langchain/agent/tools/escalation_tool.py

@@ -467,7 +467,14 @@ async def _check_escalation_memory_cache(
             if k in field_settings_lookup:
                 fs = field_settings_lookup[k]
                 settings = FieldSettings(weight=fs.get("weight", 1.0))
-            fields.append(SearchField(key_path=[k], value=str(v), settings=settings))
+            # key_path must be prefixed with field type (FieldBuilder.cs:15)
+            fields.append(
+                SearchField(
+                    key_path=["escalation-input", k],
+                    value=str(v),
+                    settings=settings,
+                )
+            )
         if not fields:
             return None
 

tests/agent/react/test_memory_node.py

@@ -38,15 +38,15 @@ def test_basic_fields(self) -> None:
         fields = _build_search_fields({"topic": "python", "level": "advanced"})
         assert len(fields) == 2
         key_paths = [f.key_path for f in fields]
-        assert ["topic"] in key_paths
-        assert ["level"] in key_paths
+        assert ["agent-input", "topic"] in key_paths
+        assert ["agent-input", "level"] in key_paths
 
     def test_filters_none_and_uipath_prefix(self) -> None:
         fields = _build_search_fields(
             {"topic": "py", "uipath__settings": {}, "empty": None}
         )
         assert len(fields) == 1
-        assert fields[0].key_path == ["topic"]
+        assert fields[0].key_path == ["agent-input", "topic"]
 
     def test_empty_input(self) -> None:
         assert _build_search_fields({}) == []