JeffJiang
462752b462
* feat: add rag provider and retriever * feat: retriever tool * feat: add retriever tool to the researcher node * feat: add rag http apis * feat: new message input supports resource mentions * feat: new message input component support resource mentions * refactor: need_web_search to need_search * chore: RAG integration docs * chore: change example api host * fix: user message color in dark mode * fix: mentions style * feat: add local_search_tool to researcher prompt * chore: research prompt * fix: ragflow page size and reporter with * docs: ragflow integration and add acknowledgment projects * chore: format
27 lines
504 B
Python
27 lines
504 B
Python
# Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
import os
|
|
import enum
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
|
|
class SearchEngine(enum.Enum):
|
|
TAVILY = "tavily"
|
|
DUCKDUCKGO = "duckduckgo"
|
|
BRAVE_SEARCH = "brave_search"
|
|
ARXIV = "arxiv"
|
|
|
|
|
|
# Tool configuration
|
|
SELECTED_SEARCH_ENGINE = os.getenv("SEARCH_API", SearchEngine.TAVILY.value)
|
|
|
|
|
|
class RAGProvider(enum.Enum):
|
|
RAGFLOW = "ragflow"
|
|
|
|
|
|
SELECTED_RAG_PROVIDER = os.getenv("RAG_PROVIDER")
|