style:Change mem to mem0

Undertone0809 · Sep 6, 2024 · 218155b · 218155b
1 parent 77cec19
commit 218155b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/example/mem0/app.py b/example/mem0/app.py
@@ -8,7 +8,7 @@ def main():
     # todo llm default answer with cn
     with st.sidebar:
         mem0_user_id = st.text_input("mem0 user id", type="password")
-        mem_api_key = st.text_input(
+        mem0_api_key = st.text_input(
             "mem0 API Key", key="provider_mem0_api_key", type="password"
         )
 
@@ -30,11 +30,11 @@ def main():
             st.info("Please add your model API key to continue.")
             st.stop()
 
-        if not mem_api_key:
+        if not mem0_api_key:
             st.error("Please provide your mem0 API Key to continue.")
             st.stop()
 
-        ai_assistant.set_mem_api_key(mem_api_key)
+        ai_assistant.set_mem0_api_key(mem0_api_key)
 
         answer = ai_assistant.ask_question(
             question=prompt, user_id=mem0_user_id, config=config

diff --git a/example/mem0/core.py b/example/mem0/core.py
@@ -9,8 +9,8 @@ def __init__(self):
             {"role": "system", "content": "You are a personal healing AI Assistant."}
         ]
 
-    def set_mem_api_key(self, mem_api_key: str):
-        self.memory = MemoryClient(api_key=mem_api_key)
+    def set_mem0_api_key(self, mem0_api_key: str):
+        self.memory = MemoryClient(api_key=mem0_api_key)
 
     def ask_question(self, question: str, user_id: str, config) -> str:
         # Fetch previous related memories