yangdx
commited on
Commit
·
bd1264b
1
Parent(s):
088f812
Fix linting
Browse files
lightrag/api/routers/ollama_api.py
CHANGED
@@ -105,9 +105,7 @@ def parse_query_mode(query: str) -> tuple[str, SearchMode, bool]:
|
|
105 |
"""Parse query prefix to determine search mode
|
106 |
Returns tuple of (cleaned_query, search_mode, only_need_context)
|
107 |
"""
|
108 |
-
# 定义前缀映射,包含模式和是否只需要上下文
|
109 |
mode_map = {
|
110 |
-
# 原有的前缀
|
111 |
"/local ": (SearchMode.local, False),
|
112 |
"/global ": (
|
113 |
SearchMode.global_,
|
@@ -120,8 +118,7 @@ def parse_query_mode(query: str) -> tuple[str, SearchMode, bool]:
|
|
120 |
"/context": (
|
121 |
SearchMode.hybrid,
|
122 |
True,
|
123 |
-
),
|
124 |
-
# 新增的前缀
|
125 |
"/localcontext": (SearchMode.local, True),
|
126 |
"/globalcontext": (SearchMode.global_, True),
|
127 |
"/hybridcontext": (SearchMode.hybrid, True),
|
@@ -135,7 +132,6 @@ def parse_query_mode(query: str) -> tuple[str, SearchMode, bool]:
|
|
135 |
cleaned_query = query[len(prefix) :].lstrip()
|
136 |
return cleaned_query, mode, only_need_context
|
137 |
|
138 |
-
# 默认使用hybrid模式,不需要上下文
|
139 |
return query, SearchMode.hybrid, False
|
140 |
|
141 |
|
|
|
105 |
"""Parse query prefix to determine search mode
|
106 |
Returns tuple of (cleaned_query, search_mode, only_need_context)
|
107 |
"""
|
|
|
108 |
mode_map = {
|
|
|
109 |
"/local ": (SearchMode.local, False),
|
110 |
"/global ": (
|
111 |
SearchMode.global_,
|
|
|
118 |
"/context": (
|
119 |
SearchMode.hybrid,
|
120 |
True,
|
121 |
+
),
|
|
|
122 |
"/localcontext": (SearchMode.local, True),
|
123 |
"/globalcontext": (SearchMode.global_, True),
|
124 |
"/hybridcontext": (SearchMode.hybrid, True),
|
|
|
132 |
cleaned_query = query[len(prefix) :].lstrip()
|
133 |
return cleaned_query, mode, only_need_context
|
134 |
|
|
|
135 |
return query, SearchMode.hybrid, False
|
136 |
|
137 |
|