From d8f621f05682d30b87484b1cd98302278d72c2b1 Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Wed, 17 Jan 2024 12:49:43 +0100
Subject: [PATCH] Reduce size of context, hack for now

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index be52bd556..54801c31f 100644
--- a/main.py
+++ b/main.py
@@ -485,7 +485,7 @@ async def user_context_enrichment(session, user_id:str, query:str, generative_re
         return context
     else:
         generative_result = generate_graph(context)
-        return generative_result
+        return generative_result.response
 
 
 async def create_public_memory(user_id: str=None, labels:list=None, topic:str=None) -> Optional[int]: