@pentatonic-ai/ai-agent-sdk 0.7.2 → 0.7.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pentatonic-ai/ai-agent-sdk",
-  "version": "0.7.2",
+  "version": "0.7.3",
   "description": "TES SDK — LLM observability and lifecycle tracking via Pentatonic Thing Event System. Track token usage, tool calls, and conversations. Manage things through event-sourced lifecycle stages with AI enrichment and vector search.",
   "type": "module",
   "main": "./dist/index.cjs",

package/packages/memory-engine/compat/server.py

@@ -607,10 +607,21 @@ async def search(req: SearchRequest):
                         if m:
                             attached_meta = m
                             break
+            # Some layers surface arena/kind/layer_type as top-level
+            # fields on the raw item rather than nested under metadata.
+            # Pull them through into the formatted metadata so the
+            # post-filter can see them even on cache miss (e.g., after
+            # a compat container restart wipes _META_CACHE).
+            raw_top_level = {
+                k: item[k]
+                for k in ("arena", "kind", "layer_type", "doc_type", "source_file")
+                if item.get(k)
+            }
+            merged_meta = {**raw_top_level, **(attached_meta or item.get("metadata") or {})}
             out_results.append({
                 "id": key,
                 "content": item.get("text") or item.get("content") or item.get("snippet") or "",
-                "metadata": attached_meta or item.get("metadata") or {},
+                "metadata": merged_meta,
                 "similarity": float(rrf_scores[key]),
                 "layer_id": f"ml_{CLIENT_ID}_episodic",
                 "client_id": CLIENT_ID,
@@ -692,10 +703,18 @@ async def search(req: SearchRequest):
         if not chosen_id:
             chosen_id = (item.get("id") or item.get("doc_id")
                          or item.get("path") or item.get("source_file") or "")
+        # Pull arena/kind/layer_type from raw item top-level if present
+        # (L6 returns these as top-level columns; cache may be empty).
+        raw_top_level = {
+            k: item[k]
+            for k in ("arena", "kind", "layer_type", "doc_type", "source_file")
+            if item.get(k)
+        }
+        merged_meta = {**raw_top_level, **(attached_meta or item.get("metadata") or {})}
         out_results.append({
             "id": chosen_id,
             "content": item.get("text") or item.get("content") or item.get("snippet") or "",
-            "metadata": attached_meta or item.get("metadata") or {},
+            "metadata": merged_meta,
             "similarity": float(item.get("score") or item.get("similarity") or 0.0),
             "layer_id": f"ml_{CLIENT_ID}_episodic",
             "client_id": CLIENT_ID,