zgjja
3b4e993531
2. use tavily from`langchain-tavily` rather than the older one from `langchain-community` Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
# Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
from .loader import load_yaml_config
|
|
from .questions import BUILT_IN_QUESTIONS, BUILT_IN_QUESTIONS_ZH_CN
|
|
from .tools import SELECTED_SEARCH_ENGINE, SearchEngine
|
|
|
|
# Load environment variables
|
|
load_dotenv()
|
|
|
|
# Team configuration
|
|
TEAM_MEMBER_CONFIGURATIONS = {
|
|
"researcher": {
|
|
"name": "researcher",
|
|
"desc": (
|
|
"Responsible for searching and collecting relevant information, understanding user needs and conducting research analysis"
|
|
),
|
|
"desc_for_llm": (
|
|
"Uses search engines and web crawlers to gather information from the internet. "
|
|
"Outputs a Markdown report summarizing findings. Researcher can not do math or programming."
|
|
),
|
|
"is_optional": False,
|
|
},
|
|
"coder": {
|
|
"name": "coder",
|
|
"desc": (
|
|
"Responsible for code implementation, debugging and optimization, handling technical programming tasks"
|
|
),
|
|
"desc_for_llm": (
|
|
"Executes Python or Bash commands, performs mathematical calculations, and outputs a Markdown report. "
|
|
"Must be used for all mathematical computations."
|
|
),
|
|
"is_optional": True,
|
|
},
|
|
}
|
|
|
|
TEAM_MEMBERS = list(TEAM_MEMBER_CONFIGURATIONS.keys())
|
|
|
|
__all__ = [
|
|
# Other configurations
|
|
"TEAM_MEMBERS",
|
|
"TEAM_MEMBER_CONFIGURATIONS",
|
|
"SELECTED_SEARCH_ENGINE",
|
|
"SearchEngine",
|
|
"BUILT_IN_QUESTIONS",
|
|
"BUILT_IN_QUESTIONS_ZH_CN",
|
|
load_yaml_config,
|
|
]
|