fix: Set doc ids aliases

julep-ai · May 15, 2024 · 5ea1ac0 · 5ea1ac0
1 parent ff12a6b
commit 5ea1ac0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agents-api/agents_api/models/entry/proc_mem_context.py b/agents-api/agents_api/models/entry/proc_mem_context.py
@@ -183,7 +183,7 @@ def proc_mem_context_query(
                 created_at,
             }},
             ~information_snippets:embedding_space {{
-                doc_id,
+                doc_id: agent_doc_id,
                 snippet_idx,
                 title,
                 snippet |
@@ -210,7 +210,7 @@ def proc_mem_context_query(
                 created_at,
             }},
             ~information_snippets:embedding_space {{
-                doc_id,
+                doc_id: user_doc_id,
                 snippet_idx,
                 title,
                 snippet |