Update Operator Optimize Method.

This commit is contained in:
didi 2024-09-02 16:47:03 +08:00
parent d97f90f9c7
commit c3903412b4
14 changed files with 1161 additions and 1040 deletions

View file

@ -3,7 +3,6 @@
# @Author : didi
# @Desc : Experiment of graph optimization
from examples.ags.w_action_node.optimizer import Optimizer
from metagpt.configs.models_config import ModelsConfig
@ -15,10 +14,11 @@ optimized_path = "examples/ags/w_action_node/optimized" # 优化结果保存路
# 初始化LLM模型
deepseek_llm_config = ModelsConfig.default().get("deepseek-coder")
claude_llm_config = ModelsConfig.default().get("claude-3.5-sonnet")
claude_llm_config = ModelsConfig.default().get("claude-3-5-sonnet-20240620")
# claude_llm_config = ModelsConfig.default().get("deepseek-coder")
# 初始化操作符列表
gsm8k_operators = [
"Custom",
"Generate",
"ContextualGenerate",
"Format",
@ -42,4 +42,4 @@ optimizer = Optimizer(
)
# 运行优化器
optimizer.optimize("Graph")
optimizer.optimize("Operator")