yangdx commited on
Commit
bd1264b
·
1 Parent(s): 088f812

Fix linting

Browse files
Files changed (1) hide show
  1. lightrag/api/routers/ollama_api.py +1 -5
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
- ), # context模式使用hybrid模式,并设置only_need_context为True
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