Fix linting

This commit is contained in:
yangdx
2025-02-05 22:29:07 +08:00
parent f703334ce4
commit 1a61d9ee7f
2 changed files with 26 additions and 3 deletions

View File

@@ -41,12 +41,14 @@ from .ollama_api import ollama_server_infos
# Load environment variables
load_dotenv()
class RAGStorageConfig:
KV_STORAGE = "JsonKVStorage"
DOC_STATUS_STORAGE = "JsonDocStatusStorage"
GRAPH_STORAGE = "NetworkXStorage"
VECTOR_STORAGE = "NanoVectorDBStorage"
# Initialize rag storage config
rag_storage_config = RAGStorageConfig()
@@ -592,6 +594,7 @@ class SearchMode(str, Enum):
hybrid = "hybrid"
mix = "mix"
class QueryRequest(BaseModel):
query: str
mode: SearchMode = SearchMode.hybrid