Files
deer-flow/src/config/tools.py
2025-04-17 11:34:42 +08:00

21 lines
419 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)
SEARCH_MAX_RESULTS = 3