He Tao
18 lines
329 B
Python
18 lines
329 B
Python
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)
|
|
SEARCH_MAX_RESULTS = 3
|