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:46:09 +08:00
parent 151058b89d
commit 1375b8eea7
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -109,13 +109,18 @@ def parse_query_mode(query: str) -> tuple[str, SearchMode, bool]:
mode_map = { mode_map = {
# 原有的前缀 # 原有的前缀
"/local ": (SearchMode.local, False), "/local ": (SearchMode.local, False),
"/global ": (SearchMode.global_, False), # global_ is used because 'global' is a Python keyword "/global ": (
SearchMode.global_,
False,
), # global_ is used because 'global' is a Python keyword
"/naive ": (SearchMode.naive, False), "/naive ": (SearchMode.naive, False),
"/hybrid ": (SearchMode.hybrid, False), "/hybrid ": (SearchMode.hybrid, False),
"/mix ": (SearchMode.mix, False), "/mix ": (SearchMode.mix, False),
"/bypass ": (SearchMode.bypass, False), "/bypass ": (SearchMode.bypass, False),
"/context": (SearchMode.hybrid, True), # context模式使用hybrid模式，并设置only_need_context为True "/context": (
SearchMode.hybrid,
True,
), # context模式使用hybrid模式，并设置only_need_context为True
# 新增的前缀 # 新增的前缀
"/localcontext": (SearchMode.local, True), "/localcontext": (SearchMode.local, True),
"/globalcontext": (SearchMode.global_, True), "/globalcontext": (SearchMode.global_, True),