添加对 mix 查询模式的支持
This commit is contained in:
@@ -243,6 +243,7 @@ class SearchMode(str, Enum):
|
||||
local = "local"
|
||||
global_ = "global" # 使用 global_ 因为 global 是 Python 保留关键字,但枚举值会转换为字符串 "global"
|
||||
hybrid = "hybrid"
|
||||
mix = "mix"
|
||||
|
||||
# Ollama API compatible models
|
||||
class OllamaMessage(BaseModel):
|
||||
@@ -684,7 +685,8 @@ def create_app(args):
|
||||
"/local ": SearchMode.local,
|
||||
"/global ": SearchMode.global_, # global_ is used because 'global' is a Python keyword
|
||||
"/naive ": SearchMode.naive,
|
||||
"/hybrid ": SearchMode.hybrid
|
||||
"/hybrid ": SearchMode.hybrid,
|
||||
"/mix ": SearchMode.mix
|
||||
}
|
||||
|
||||
for prefix, mode in mode_map.items():
|
||||
|
@@ -337,7 +337,7 @@ def test_query_modes():
|
||||
每个模式都会返回相同格式的响应,但检索策略不同。
|
||||
"""
|
||||
url = get_base_url()
|
||||
modes = ["local", "global", "naive", "hybrid"] # 支持的查询模式
|
||||
modes = ["local", "global", "naive", "hybrid", "mix"] # 支持的查询模式
|
||||
|
||||
for mode in modes:
|
||||
if OutputControl.is_verbose():
|
||||
|
Reference in New Issue
Block a user