Merge branch 'test' into feat/mem-search-rank

Author: fridayL

.github/workflows/python-tests.yml

@@ -12,11 +12,13 @@ on:
       - "main"
       - "dev"
       - "feat/*"
+      - "test"
   pull_request:
     branches:
       - "main"
       - "dev"
       - "feat/*"
+      - "test"
 
 jobs:
   build:

poetry.lock

@@ -42,6 +42,7 @@ dependencies = [
     "tenacity (>=9.1.2,<10.0.0)",  # Error handling and retrying library
     "fastapi[all] (>=0.115.12,<0.116.0)",  # Web framework for building APIs
     "sqlalchemy (>=2.0.41,<3.0.0)",  # SQL toolkit
+    "pymysql (>=1.1.0,<2.0.0)",  # MySQL Python driver
     "scikit-learn (>=1.7.0,<2.0.0)",  # Machine learning
     "fastmcp (>=2.10.5,<3.0.0)",
     "python-dateutil (>=2.9.0.post0,<3.0.0)",
@@ -76,6 +77,11 @@ mem-scheduler = [
     "pika (>=1.3.2,<2.0.0)",  # RabbitMQ client
 ]
 
+# MemUser (MySQL support)
+mem-user = [
+    "pymysql (>=1.1.0,<2.0.0)",  # MySQL client for SQLAlchemy
+]
+
 # MemReader
 mem-reader = [
     "chonkie (>=1.0.7,<2.0.0)",  # Sentence chunking library
@@ -90,6 +96,7 @@ all = [
     "schedule (>=1.2.2,<2.0.0)",
     "redis (>=6.2.0,<7.0.0)",
     "pika (>=1.3.2,<2.0.0)",
+    "pymysql (>=1.1.0,<2.0.0)",
     "chonkie (>=1.0.7,<2.0.0)",
     "markitdown[docx,pdf,pptx,xls,xlsx] (>=0.1.1,<0.2.0)",
 
@@ -158,6 +165,10 @@ python-dotenv = "^1.1.1"
 langgraph = "^0.5.1"
 langmem = "^0.0.27"
 
+
+[tool.poetry.group.mem-user.dependencies]
+pymysql = "^1.1.2"
+
 [[tool.poetry.source]]
 name = "mirrors"
 url = "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple/"

pyproject.toml

@@ -21,7 +21,7 @@ class APIConfig:
     def get_openai_config() -> dict[str, Any]:
         """Get OpenAI configuration."""
         return {
-            "model_name_or_path": os.getenv("MOS_OPENAI_MODEL", "gpt-4o-mini"),
+            "model_name_or_path": os.getenv("MOS_CHAT_MODEL", "gpt-4o-mini"),
             "temperature": float(os.getenv("MOS_CHAT_TEMPERATURE", "0.8")),
             "max_tokens": int(os.getenv("MOS_MAX_TOKENS", "1024")),
             "top_p": float(os.getenv("MOS_TOP_P", "0.9")),