fix: revert model and make SOTA model available to COSTEER (#351)

* revert model and make SOTA model available to COSTEER

* fix CI

* fix CI

* fix CI
This commit is contained in:
Xu Yang
2024-09-26 17:54:45 +08:00
committed by GitHub
parent 66db69e104
commit f3210f3baa
12 changed files with 84 additions and 44 deletions
@@ -88,7 +88,7 @@ class FactorCoSTEER(Developer[FactorExperiment]):
self.rag = FactorGraphRAGStrategy(factor_knowledge_base)
# init intermediate items
factor_experiment = FactorEvolvingItem(sub_tasks=exp.sub_tasks)
factor_experiment = FactorEvolvingItem.from_experiment(exp)
self.evolve_agent = FactorRAGEvoAgent(
max_loop=self.max_loop,
@@ -28,3 +28,10 @@ class FactorEvolvingItem(FactorExperiment, EvolvableSubjects):
)
else:
self.sub_gt_implementations = sub_gt_implementations
@classmethod
def from_experiment(cls, exp: FactorExperiment) -> "FactorExperiment":
ei = cls(sub_tasks=exp.sub_tasks)
ei.based_experiments = exp.based_experiments
ei.experiment_workspace = exp.experiment_workspace
return ei