mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-03 06:12:14 +08:00
dd9af1eb50
* feat: Implement MilvusRetriever with embedding model and resource management * chore: Update configuration and loader files for consistency * chore: Clean up test_milvus.py for improved readability and organization * feat: Add tests for DashscopeEmbeddings query and document embedding methods * feat: Add tests for embedding model initialization and example file loading in MilvusProvider * chore: Remove unused imports and clean up test_milvus.py for better readability * chore: Clean up test_milvus.py for improved readability and organization * chore: Clean up test_milvus.py for improved readability and organization * fix: replace print statements with logging in recursion limit function * Implement feature X to enhance user experience and optimize performance * refactor: clean up unused imports and comments in AboutTab component * Implement feature X to enhance user experience and fix bug Y in module Z --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
31 lines
611 B
Python
31 lines
611 B
Python
# Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
import enum
|
|
import os
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
|
|
class SearchEngine(enum.Enum):
|
|
TAVILY = "tavily"
|
|
DUCKDUCKGO = "duckduckgo"
|
|
BRAVE_SEARCH = "brave_search"
|
|
ARXIV = "arxiv"
|
|
WIKIPEDIA = "wikipedia"
|
|
|
|
|
|
# Tool configuration
|
|
SELECTED_SEARCH_ENGINE = os.getenv("SEARCH_API", SearchEngine.TAVILY.value)
|
|
|
|
|
|
class RAGProvider(enum.Enum):
|
|
RAGFLOW = "ragflow"
|
|
VIKINGDB_KNOWLEDGE_BASE = "vikingdb_knowledge_base"
|
|
MILVUS = "milvus"
|
|
|
|
|
|
SELECTED_RAG_PROVIDER = os.getenv("RAG_PROVIDER")
|