mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 07:57:44 +00:00
you-n-g
ffc85936f1
* refactor: Update type annotations and remove unused class in evolving modules * refactor: Simplify evolving agent and feedback handling in CoSTEER module * lint & CI * mypy * ruff for core * mypy * refactor: remove unnecessary comments and update feedback handling logic * refactor: Add prev_task_feedback parameter to evolving strategies * feat: Clear folder before extracting zip file in DockerEnv * fix: Correct retrieval of last experiment from history
105 lines
2.6 KiB
Python
105 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import copy
|
|
from abc import ABC, abstractmethod
|
|
from dataclasses import dataclass
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
from rdagent.core.evaluation import EvaluableObj
|
|
from rdagent.core.knowledge_base import KnowledgeBase
|
|
|
|
if TYPE_CHECKING:
|
|
from rdagent.core.evaluation import Feedback
|
|
from rdagent.core.scenario import Scenario
|
|
|
|
|
|
class Knowledge:
|
|
pass
|
|
|
|
|
|
class QueriedKnowledge:
|
|
pass
|
|
|
|
|
|
class EvolvingKnowledgeBase(KnowledgeBase):
|
|
@abstractmethod
|
|
def query(
|
|
self,
|
|
) -> QueriedKnowledge | None:
|
|
raise NotImplementedError
|
|
|
|
|
|
class EvolvableSubjects(EvaluableObj):
|
|
"""The target object to be evolved"""
|
|
|
|
def clone(self) -> EvolvableSubjects:
|
|
return copy.deepcopy(self)
|
|
|
|
|
|
@dataclass
|
|
class EvoStep:
|
|
"""At a specific step,
|
|
based on
|
|
- previous trace
|
|
- newly RAG knowledge `QueriedKnowledge`
|
|
|
|
the EvolvableSubjects is evolved to a new one `EvolvableSubjects`.
|
|
|
|
(optional) After evaluation, we get feedback `feedback`.
|
|
"""
|
|
|
|
evolvable_subjects: EvolvableSubjects
|
|
queried_knowledge: QueriedKnowledge | None = None
|
|
feedback: Feedback | None = None
|
|
|
|
|
|
class EvolvingStrategy(ABC):
|
|
def __init__(self, scen: Scenario) -> None:
|
|
self.scen = scen
|
|
|
|
@abstractmethod
|
|
def evolve(
|
|
self,
|
|
*evo: EvolvableSubjects,
|
|
evolving_trace: list[EvoStep] | None = None,
|
|
queried_knowledge: QueriedKnowledge | None = None,
|
|
**kwargs: Any,
|
|
) -> EvolvableSubjects:
|
|
"""The evolving trace is a list of (evolvable_subjects, feedback) ordered
|
|
according to the time.
|
|
|
|
The reason why the parameter is important for the evolving.
|
|
- evolving_trace: the historical feedback is important.
|
|
- queried_knowledge: queried knowledge
|
|
"""
|
|
|
|
|
|
class RAGStrategy(ABC):
|
|
"""Retrieval Augmentation Generation Strategy"""
|
|
|
|
def __init__(self, knowledgebase: EvolvingKnowledgeBase) -> None:
|
|
self.knowledgebase: EvolvingKnowledgeBase = knowledgebase
|
|
|
|
@abstractmethod
|
|
def query(
|
|
self,
|
|
evo: EvolvableSubjects,
|
|
evolving_trace: list[EvoStep],
|
|
**kwargs: Any,
|
|
) -> QueriedKnowledge | None:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def generate_knowledge(
|
|
self,
|
|
evolving_trace: list[EvoStep],
|
|
*,
|
|
return_knowledge: bool = False,
|
|
**kwargs: Any,
|
|
) -> Knowledge | None:
|
|
"""Generating new knowledge based on the evolving trace.
|
|
- It is encouraged to query related knowledge before generating new knowledge.
|
|
|
|
RAGStrategy should maintain the new knowledge all by itself.
|
|
"""
|