mem1 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

mem1/__init__.py

@@ -1,16 +1,17 @@
 """
-Mem1 - 基于 Elasticsearch 的用户记忆系统
+Mem1 - 用户记忆系统（支持可插拔存储后端）
 """
 import logging
 
-__version__ = "0.0.5"
+__version__ = "0.0.9"
 
 # 屏蔽第三方库的详细日志（必须在导入前设置）
 logging.getLogger("elastic_transport").setLevel(logging.WARNING)
 logging.getLogger("elastic_transport.transport").setLevel(logging.WARNING)
 logging.getLogger("httpx").setLevel(logging.WARNING)
 
-from mem1.memory_es import Mem1Memory
+from mem1.memory import Mem1Memory
 from mem1.config import Mem1Config, LLMConfig
+from mem1.storage import StorageBackend, ESStorage
 
-__all__ = ["Mem1Memory", "Mem1Config", "LLMConfig"]
+__all__ = ["Mem1Memory", "Mem1Config", "LLMConfig", "StorageBackend", "ESStorage"]

mem1/config.py

@@ -10,6 +10,7 @@ class LLMConfig(BaseModel):
     model: str
     api_key: str
     base_url: str
+    temperature: float
 
 
 class VLConfig(BaseModel):
@@ -95,6 +96,8 @@ class Mem1Config(BaseModel):
             "MEM1_UPDATE_INTERVAL_MINUTES": os.getenv("MEM1_UPDATE_INTERVAL_MINUTES"),
             "MEM1_SAVE_ASSISTANT_MESSAGES": os.getenv("MEM1_SAVE_ASSISTANT_MESSAGES"),
             "MEM1_MAX_ASSISTANT_CHARS": os.getenv("MEM1_MAX_ASSISTANT_CHARS"),
+            "MEM1_LLM_TEMPERATURE": os.getenv("MEM1_LLM_TEMPERATURE"),
+            "MEM1_CONTEXT_DAYS_LIMIT": os.getenv("MEM1_CONTEXT_DAYS_LIMIT"),
         }
         
         missing = [k for k, v in required_vars.items() if not v]
@@ -121,7 +124,8 @@ class Mem1Config(BaseModel):
                 provider="openai",
                 model=required_vars["MEM1_LLM_MODEL"],
                 api_key=required_vars["MEM1_LLM_API_KEY"],
-                base_url=required_vars["MEM1_LLM_BASE_URL"]
+                base_url=required_vars["MEM1_LLM_BASE_URL"],
+                temperature=float(required_vars["MEM1_LLM_TEMPERATURE"])
             ),
             vl=vl_config,
             memory=MemoryConfig(
@@ -132,7 +136,7 @@ class Mem1Config(BaseModel):
                 update_interval_minutes=int(required_vars["MEM1_UPDATE_INTERVAL_MINUTES"]),
                 save_assistant_messages=required_vars["MEM1_SAVE_ASSISTANT_MESSAGES"].lower() == "true",
                 max_assistant_chars=int(required_vars["MEM1_MAX_ASSISTANT_CHARS"]),
-                context_days_limit=int(os.getenv("MEM1_CONTEXT_DAYS_LIMIT", "31"))
+                context_days_limit=int(required_vars["MEM1_CONTEXT_DAYS_LIMIT"])
             ),
             es=ESConfig(
                 hosts=es_hosts,

mem1/llm.py

@@ -32,7 +32,7 @@ class LLMClient:
         kwargs = {
             "model": self.config.model,
             "messages": messages,
-            "temperature": 0.7,
+            "temperature": self.config.temperature,
         }
         
         if response_format == "json":