gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-04-08 18:47:50 +08:00
parent 1375b8eea7
commit 25f2ecb5a9
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lightrag/api/routers/ollama_api.py
View File

@@ -105,9 +105,7 @@ def parse_query_mode(query: str) -> tuple[str, SearchMode, bool]:
"""Parse query prefix to determine search mode """Parse query prefix to determine search mode
Returns tuple of (cleaned_query, search_mode, only_need_context) Returns tuple of (cleaned_query, search_mode, only_need_context)
""" """
# 定义前缀映射，包含模式和是否只需要上下文
mode_map = { mode_map = {
# 原有的前缀
"/local ": (SearchMode.local, False), "/local ": (SearchMode.local, False),
"/global ": ( "/global ": (
SearchMode.global_, SearchMode.global_,
@@ -120,8 +118,7 @@ def parse_query_mode(query: str) -> tuple[str, SearchMode, bool]:
"/context": ( "/context": (
SearchMode.hybrid, SearchMode.hybrid,
True, True,
), # context模式使用hybrid模式，并设置only_need_context为True ),
# 新增的前缀
"/localcontext": (SearchMode.local, True), "/localcontext": (SearchMode.local, True),
"/globalcontext": (SearchMode.global_, True), "/globalcontext": (SearchMode.global_, True),
"/hybridcontext": (SearchMode.hybrid, True), "/hybridcontext": (SearchMode.hybrid, True),
@@ -135,7 +132,6 @@ def parse_query_mode(query: str) -> tuple[str, SearchMode, bool]:
cleaned_query = query[len(prefix) :].lstrip() cleaned_query = query[len(prefix) :].lstrip()
return cleaned_query, mode, only_need_context return cleaned_query, mode, only_need_context
# 默认使用hybrid模式，不需要上下文
return query, SearchMode.hybrid, False return query, SearchMode.hybrid, False