2025-04-17 11:34:42 +08:00
|
|
|
# Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
|
|
|
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
|
|
2025-04-10 11:45:04 +08:00
|
|
|
import os
|
|
|
|
|
import enum
|
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
|
|
|
|
|
|
load_dotenv()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class SearchEngine(enum.Enum):
|
|
|
|
|
TAVILY = "tavily"
|
|
|
|
|
DUCKDUCKGO = "duckduckgo"
|
2025-04-11 15:37:55 +08:00
|
|
|
BRAVE_SEARCH = "brave_search"
|
|
|
|
|
ARXIV = "arxiv"
|
2025-04-10 11:45:04 +08:00
|
|
|
|
|
|
|
|
|
2025-04-07 16:25:55 +08:00
|
|
|
# Tool configuration
|
2025-04-10 11:45:04 +08:00
|
|
|
SELECTED_SEARCH_ENGINE = os.getenv("SEARCH_API", SearchEngine.TAVILY.value)
|
|
|
|
|
SEARCH_MAX_RESULTS = 3
|
