后端添加cache;前端改动agents

This commit is contained in:
2024-12-30 19:45:48 +08:00
parent 34e25633ba
commit 99301d7b3d
19 changed files with 90 additions and 348 deletions

View File

@@ -8,13 +8,13 @@ import autogen
from .utils import load_agent_configs
import os
from .retrieval_group import init_retrieval_group
from .constant import config_list, STREAM, SILENT, WORK_DIR
from .constant import config_list, STREAM, SILENT, WORK_DIR, CACHE
agent_configs = load_agent_configs(os.path.join(os.path.dirname(os.path.abspath(__file__)), "config/generate_group.yaml"))
venv_context = create_virtual_env(WORK_DIR)
llm_config = {"config_list": config_list, "stream": STREAM}
llm_config = {"config_list": config_list, "stream": STREAM, "cache_seed": CACHE}
def init_generate_group(outer_retrieval_agent, inner_retrieval_admin):