后端添加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

@@ -1,13 +1,13 @@
from autogen.coding import LocalCommandLineCodeExecutor
import autogen
import os
from .constant import config_list, STREAM, SILENT, WORK_DIR
from .constant import config_list, STREAM, SILENT, WORK_DIR, CACHE
from .utils import load_agent_configs,retrieval_from_knowledge_base, web_searcher, retrieval_from_graphrag
agent_configs = load_agent_configs(os.path.join(os.path.dirname(os.path.abspath(__file__)), "config/retrieval_group.yaml"))
llm_config = {"config_list": config_list, "stream": STREAM}
llm_config = {"config_list": config_list, "stream": STREAM, "cache_seed": CACHE}
def init_retrieval_group(work_dir, venv_context):