Optimize the cognitive function a bit

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()

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")