apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-02 14:45:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add data type; fix instruction generation

Browse source
This commit is contained in:
Cyzus Chi 2024-10-28 14:11:53 +08:00
parent 4d1a6f4c2b
commit 61492d9ff5
6 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
metagpt/ext/sela/data.yaml
View file

@ -1,3 +1,3 @@
datasets_dir: "path/to/datasets" # path to the datasets directory
work_dir: ../workspace # path to the workspace directory
work_dir: ../../workspace # path to the workspace directory
role_dir: storage/SELA # path to the role directory

2
metagpt/ext/sela/evaluation/visualize_mcts.py
View file

@ -3,7 +3,7 @@ import textwrap
import matplotlib.pyplot as plt
import networkx as nx
from metagpt.ext.sela.MCTS import Node
from metagpt.ext.sela.search.tree_search import Node
NODE_TEMPLATE = """\
[Node {id}]

2
metagpt/ext/sela/experimenter/experimenter.py
View file

@ -6,7 +6,7 @@ import numpy as np
import pandas as pd
from metagpt.ext.sela.evaluation.evaluation import evaluate_score
from metagpt.ext.sela.MCTS import create_initial_state
from metagpt.ext.sela.search.tree_search import create_initial_state
from metagpt.ext.sela.research_assistant import ResearchAssistant
from metagpt.ext.sela.utils import DATA_CONFIG, save_notebook

2
metagpt/ext/sela/experimenter/random_search.py
View file

@ -17,7 +17,7 @@ class RandomSearchExperimenter(Experimenter):
# state = create_initial_state(self.args.task, start_task_id=1, data_config=self.data_config, low_is_better=self.args.low_is_better, name="")
user_requirement = self.state["requirement"]
exp_pool_path = get_exp_pool_path(self.args.task, self.data_config, pool_name="ds_analysis_pool")
exp_pool = InstructionGenerator.load_analysis_pool(
exp_pool = InstructionGenerator.load_insight_pool(
exp_pool_path, use_fixed_insights=self.args.use_fixed_insights
)
if self.args.rs_mode == "single":

2
metagpt/ext/sela/research_assistant.py
View file

@ -71,7 +71,7 @@ class ResearchAssistant(DataInterpreter):
return f"Node-{self.node_id}"
def get_next_instruction(self):
return self.planner.plan.tasks[self.start_task_id]
return self.planner.plan.tasks[self.start_task_id].instruction
def change_next_instruction(self, new_instruction):
if new_instruction is not None:

2
metagpt/ext/sela/search/tree_search.py
View file

@ -113,7 +113,7 @@ class Node:
normalized_reward: dict = {"train_score": 0, "dev_score": 0, "test_score": 0}
parent = None
def __init__(self, parent=None, state=None, action=None, value=0, max_depth=4, **kwargs):
def __init__(self, parent=None, state: dict = None, action: str = None, value: float = 0, max_depth: int = 4, **kwargs):
self.state = state
self.action = action
self.value = value