klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[bug-fix] fix issue related to bot memory when using multiple bots at the same time (#486)

Browse Source
This commit is contained in:
Deshraj Yadav
2023-08-25 21:59:39 -07:00
committed by GitHub
parent d0956a0dc1
commit 4388f6bfc2
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
embedchain/embedchain.py
View File

@@ -26,8 +26,6 @@ load_dotenv()
ABS_PATH = os.getcwd() ABS_PATH = os.getcwd()
DB_DIR = os.path.join(ABS_PATH, "db") DB_DIR = os.path.join(ABS_PATH, "db")
memory = ConversationBufferMemory()
class EmbedChain: class EmbedChain:
def __init__(self, config: BaseAppConfig): def __init__(self, config: BaseAppConfig):
@@ -44,6 +42,7 @@ class EmbedChain:
self.user_asks = [] self.user_asks = []
self.is_docs_site_instance = False self.is_docs_site_instance = False
self.online = False self.online = False
self.memory = ConversationBufferMemory()
# Send anonymous telemetry # Send anonymous telemetry
self.s_id = self.config.id if self.config.id else str(uuid.uuid4()) self.s_id = self.config.id if self.config.id else str(uuid.uuid4())
@@ -362,8 +361,7 @@ class EmbedChain:
k["web_search_result"] = self.access_search_and_get_results(input_query) k["web_search_result"] = self.access_search_and_get_results(input_query)
contexts = self.retrieve_from_database(input_query, config) contexts = self.retrieve_from_database(input_query, config)
global memory chat_history = self.memory.load_memory_variables({})["history"]
chat_history = memory.load_memory_variables({})["history"]
if chat_history: if chat_history:
config.set_history(chat_history) config.set_history(chat_history)
@@ -376,14 +374,14 @@ class EmbedChain:
answer = self.get_answer_from_llm(prompt, config) answer = self.get_answer_from_llm(prompt, config)
memory.chat_memory.add_user_message(input_query) self.memory.chat_memory.add_user_message(input_query)
# Send anonymous telemetry # Send anonymous telemetry
thread_telemetry = threading.Thread(target=self._send_telemetry_event, args=("chat",)) thread_telemetry = threading.Thread(target=self._send_telemetry_event, args=("chat",))
thread_telemetry.start() thread_telemetry.start()
if isinstance(answer, str): if isinstance(answer, str):
memory.chat_memory.add_ai_message(answer) self.memory.chat_memory.add_ai_message(answer)
logging.info(f"Answer: {answer}") logging.info(f"Answer: {answer}")
return answer return answer
else: else:
@@ -395,7 +393,7 @@ class EmbedChain:
for chunk in answer: for chunk in answer:
streamed_answer = streamed_answer + chunk streamed_answer = streamed_answer + chunk
yield chunk yield chunk
memory.chat_memory.add_ai_message(streamed_answer) self.memory.chat_memory.add_ai_message(streamed_answer)
logging.info(f"Answer: {streamed_answer}") logging.info(f"Answer: {streamed_answer}")
def set_collection(self, collection_name): def set_collection(self, collection_name):