From c4039f52bdd8300aee367c4611f7baa859a92004 Mon Sep 17 00:00:00 2001
From: lixinyue <2569494688@qq.com>
Date: Thu, 22 Jan 2026 12:12:41 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8A=8Agroup=5Fid=E6=9B=BF=E6=8D=A2end=5Fuser?=
 =?UTF-8?q?=5Fid=5F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/app/core/memory/src/search.py | 61 +------------------------------
 1 file changed, 1 insertion(+), 60 deletions(-)

diff --git a/api/app/core/memory/src/search.py b/api/app/core/memory/src/search.py
index 345cd69b..ef1e615c 100644
--- a/api/app/core/memory/src/search.py
+++ b/api/app/core/memory/src/search.py
@@ -1020,63 +1020,4 @@ async def search_chunk_by_chunk_id(
     )
     return {"chunks": chunks}
 
-if __name__ == '__main__':
-    # 测试混合检索功能
-    from app.schemas.memory_config_schema import MemoryConfig
-    from app.db import get_db
-    from app.services.memory_config_service import MemoryConfigService
-    
-    # 从数据库获取真实配置
-    db = next(get_db())
-    try:
-        config_service = MemoryConfigService(db)
-        
-        # 使用 config_id=17 获取配置
-        memory_config = config_service.load_memory_config(config_id=17)
-        
-        if not memory_config:
-            print("错误：找不到 config_id=17 的配置")
-            print("请先在数据库中创建配置，或修改 config_id")
-            exit(1)
-        
-        print(f"✓ 成功加载配置: {memory_config.config_name}")
-        print(f"  - Workspace: {memory_config.workspace_name}")
-        print(f"  - LLM Model: {memory_config.llm_model_name}")
-        print(f"  - Embedding Model: {memory_config.embedding_model_name}")
-        print(f"  - Storage Type: {memory_config.storage_type}")
-        print()
-        
-        # 修改这里的参数进行测试
-        test_end_user_id = "021886bc-fab9-4fd5-b607-497b262e0381"  # 修改为你的 end_user_id
-        test_query = "小明擅长什么？"  # 修改为你的查询
-        
-        print(f"开始测试检索...")
-        print(f"  - Query: {test_query}")
-        print(f"  - End User ID: {test_end_user_id}")
-        print(f"  - Search Type: hybrid")
-        print()
-        
-        results = asyncio.run(run_hybrid_search(
-            query_text=test_query,
-            search_type="hybrid",  # 可选: "keyword", "embedding", "hybrid"
-            end_user_id=test_end_user_id,
-            limit=10,
-            include=["statements", "entities", "chunks", "summaries"],
-            output_path=None,
-            memory_config=memory_config,
-            rerank_alpha=0.6,
-            use_forgetting_rerank=False,
-            use_llm_rerank=False
-        ))
-        
-        print("=" * 80)
-        print("检索结果:")
-        print("=" * 80)
-        print(results)
-        
-    except Exception as e:
-        print(f"错误: {e}")
-        import traceback
-        traceback.print_exc()
-    finally:
-        db.close()
+