From a810fbe0082a5b9f496ce4a931d164a360723181 Mon Sep 17 00:00:00 2001
From: lixinyue <2569494688@qq.com>
Date: Mon, 9 Feb 2026 21:02:29 +0800
Subject: [PATCH] bug/config_id

---
 api/app/tasks.py | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/api/app/tasks.py b/api/app/tasks.py
index c7303df6..676ce71d 100644
--- a/api/app/tasks.py
+++ b/api/app/tasks.py
@@ -983,13 +983,13 @@ def read_message_task(self, end_user_id: str, message: str, history: List[Dict[s
 
 
 @celery_app.task(name="app.core.memory.agent.write_message", bind=True)
-def write_message_task(self, end_user_id: str, message: list[dict], config_id: str, storage_type: str, user_rag_memory_id: str,
+def write_message_task(self, end_user_id: str, message: list[dict], config_id: str | int, storage_type: str, user_rag_memory_id: str,
                        language: str = "zh") -> Dict[str, Any]:
     """Celery task to process a write message via MemoryAgentService.
     Args:
         end_user_id: Group ID for the memory agent (also used as end_user_id)
         message: Message to write
-        config_id: Configuration ID as string (will be converted to UUID)
+        config_id: Configuration ID (can be UUID string, integer, or config_id_old)
         storage_type: Storage type (neo4j or rag)
         user_rag_memory_id: User RAG memory ID
         language: 语言类型 ("zh" 中文, "en" 英文)
@@ -1003,25 +1003,28 @@ def write_message_task(self, end_user_id: str, message: list[dict], config_id: s
     from app.core.logging_config import get_logger
     logger = get_logger(__name__)
 
-    logger.info(f"[CELERY WRITE] Starting write task - end_user_id={end_user_id}, config_id={config_id}, storage_type={storage_type}, language={language}")
+    logger.info(f"[CELERY WRITE] Starting write task - end_user_id={end_user_id}, config_id={config_id} (type: {type(config_id).__name__}), storage_type={storage_type}, language={language}")
     start_time = time.time()
 
-    # Convert config_id string to UUID
+    # Convert config_id to UUID
     actual_config_id = None
 
     if config_id:
         try:
             with get_db_context() as db:
                 actual_config_id = resolve_config_id(config_id, db)
+            print(100*'-')
+            print(actual_config_id)
+            print(100*'-')
             logger.info(
                 f"[CELERY WRITE] Converted config_id to UUID: {actual_config_id} (type: {type(actual_config_id).__name__})")
         except (ValueError, AttributeError) as e:
-            logger.error(f"[CELERY WRITE] Invalid config_id format: {config_id}, error: {e}")
+            logger.error(f"[CELERY WRITE] Invalid config_id format: {config_id} (type: {type(config_id).__name__}), error: {e}")
             return {
                 "status": "FAILURE",
-                "error": f"Invalid config_id format: {config_id}",
+                "error": f"Invalid config_id format: {config_id} - {str(e)}",
                 "end_user_id": end_user_id,
-                "config_id": config_id,
+                "config_id": str(config_id),
                 "elapsed_time": 0.0,
                 "task_id": self.request.id
             }