From 5ea1ac0a708f9bf22e0076069d191eef3c3c709f Mon Sep 17 00:00:00 2001
From: Dmitry Paramonov <asmatic075@gmail.com>
Date: Wed, 15 May 2024 16:08:13 +0300
Subject: [PATCH] fix: Set doc ids aliases

---
 agents-api/agents_api/models/entry/proc_mem_context.py | 4 ++--
 1 file changed, 2 insertions(+), 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
index 82b114e75..4b1d650d7 100644
--- 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 |