From 9971cac0cb9a5a7b9f2358ac0a863760f45846db Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Sat, 30 Dec 2023 23:49:54 +0100
Subject: [PATCH] Optimize the cognitive function a bit

---
 cognitive_architecture/fetch_secret.py | 8 ++++----
 main.py                                | 6 +++---
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/cognitive_architecture/fetch_secret.py b/cognitive_architecture/fetch_secret.py
index 3f551f596..b8635b009 100644
--- a/cognitive_architecture/fetch_secret.py
+++ b/cognitive_architecture/fetch_secret.py
@@ -52,7 +52,7 @@ def fetch_secret(secret_name, region_name, env_file_path):
     return "Success in loading env files"
 
 
-env_file = ".env"
+env_file = "../.env"
 if os.path.exists(env_file):
     # Load default environment variables (.env)
     load_dotenv()
@@ -63,7 +63,7 @@ else:
     secrets = fetch_secret(
         f"promethai-{environment}-backend-secretso-promethaijs-dotenv",
         "eu-west-1",
-        ".env",
+        "../.env",
     )
     if secrets:
         print(secrets)
@@ -76,7 +76,7 @@ if "dev" in environment:
     secrets = fetch_secret(
         f"promethai-dev-backend-secretso-promethaijs-dotenv",
         "eu-west-1",
-        ".env",
+        "../.env",
     )
     load_dotenv()
 elif "prd" in environment:
@@ -84,6 +84,6 @@ elif "prd" in environment:
     secrets = fetch_secret(
         f"promethai-prd-backend-secretso-promethaijs-dotenv",
         "eu-west-1",
-        ".env",
+        "../.env",
     )
     load_dotenv()
diff --git a/main.py b/main.py
index d1d35d736..3003c691d 100644
--- a/main.py
+++ b/main.py
@@ -38,7 +38,7 @@ import logging
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.future import select
 from cognitive_architecture.utils import get_document_names, generate_letter_uuid, get_memory_name_by_doc_id, get_unsumarized_vector_db_namespace, get_vectordb_namespace, get_vectordb_document_name
-
+from cognitive_architecture.shared.language_processing import translate_text, detect_language
 
 
 async def fetch_document_vectordb_namespace(session: AsyncSession, user_id: str, namespace_id:str, doc_id:str=None):
@@ -363,13 +363,13 @@ async def user_context_enrichment(session, user_id:str, query:str, generative_re
     neo4j_graph_db = Neo4jGraphDB(url=config.graph_database_url, username=config.graph_database_username,
                                   password=config.graph_database_password)
 
+    await user_query_to_graph_db(session, user_id, query)
+
     semantic_mem = neo4j_graph_db.retrieve_semantic_memory(user_id=user_id)
     episodic_mem = neo4j_graph_db.retrieve_episodic_memory(user_id=user_id)
     # public_mem = neo4j_graph_db.retrieve_public_memory(user_id=user_id)
 
 
-    from cognitive_architecture.shared.language_processing import translate_text, detect_language
-
 
     if detect_language(query) != "en":
         query = translate_text(query, "sr", "en")