检查项目,修复group_id的遗留问题
This commit is contained in:
@@ -1017,60 +1017,3 @@ 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
|
||||
))
|
||||
|
||||
except Exception as e:
|
||||
print(f"错误: {e}")
|
||||
import traceback
|
||||
|
||||
traceback.print_exc()
|
||||
finally:
|
||||
db.close()
|
||||
Reference in New Issue
Block a user