Memory instance migration to library

Author: vmesel

src/dialog/llm/default.py

@@ -11,8 +11,9 @@
 
 from dialog.learn.idf import categorize_conversation_history
 from dialog_lib.agents.abstract import AbstractLLM
-from dialog.llm.embeddings import get_most_relevant_contents_from_message
-from dialog.llm.memory import generate_memory_instance
+from dialog_lib.embeddings.generate import get_most_relevant_contents_from_message
+from dialog_lib.db.memory import generate_memory_instance
+from dialog.llm.embeddings import EMBEDDINGS_LLM
 from dialog.settings import Settings
 from dialog.models.db import get_session
 
@@ -26,7 +27,9 @@ def memory(self) -> BaseChatMemory:
         if self.session_id:
             return generate_memory_instance(
                 session_id=self.session_id,
-                parent_session_id=self.parent_session_id
+                parent_session_id=self.parent_session_id,
+                dbsession=self.dbsession,
+                database_url=Settings().DATABASE_URL
             )
         return None
 
@@ -35,7 +38,9 @@ def generate_prompt(self, text):
             text,
             top=Settings().LLM_RELEVANT_CONTENTS,
             dataset=self.dataset,
-            session=self.dbsession
+            session=self.dbsession,
+            embeddings_llm=EMBEDDINGS_LLM,
+            cosine_similarity_threshold=Settings().COSINE_SIMILARITY_THRESHOLD
         )
         prompt_config = self.config.get("prompt")
         fallback = prompt_config.get("fallback") or \