changes before

agential-ai · Jan 28, 2025 · 0d13085 · 0d13085
1 parent f3e6053
commit 0d13085
Show file tree

Hide file tree

Showing 5 changed files with 720 additions and 286 deletions.
diff --git a/agential/training/agent_optimizer/agent.py b/agential/training/agent_optimizer/agent.py
@@ -6,7 +6,7 @@
 
 from typing import Any, Dict, Optional
 
-from agential.training.agent_optimizer.agent import BaseAgent
+from agential.agents.base.agent import BaseAgent
 from agential.training.agent_optimizer.output import PromptOptimizerOutput
 from agential.training.agent_optimizer.prompts import (
     PROMPT_OPTIMIZER_INSTRUCTION_AMBIGNQ,
@@ -19,7 +19,7 @@
     PROMPT_OPTIMIZER_INSTRUCTION_TABMWP,
     PROMPT_OPTIMIZER_INSTRUCTION_TRIVIAQA,
 )
-from agential.training.agent_optimizer.strategies.base import PromptOptimizerBaseStrategy, ReActBaseStrategy
+from agential.training.agent_optimizer.strategies.base import PromptOptimizerBaseStrategy
 from agential.training.agent_optimizer.strategies.code import PromptOptimizerHEvalStrategy, PromptOptimizerMBPPStrategy
 from agential.training.agent_optimizer.strategies.math import (
     PromptOptimizerGSM8KStrategy,

diff --git a/agential/training/agent_optimizer/strategies/base.py b/agential/training/agent_optimizer/strategies/base.py
@@ -4,14 +4,13 @@
 from typing import Any, Dict, Optional, Tuple
 
 from tiktoken import Encoding
-
 from agential.agents.base.strategies import BaseAgentStrategy
 from agential.core.llm import BaseLLM, Response
 from agential.training.agent_optimizer.output import PromptOptimizerOutput
 
 
 class PromptOptimizerBaseStrategy(BaseAgentStrategy):
-    """An abstract base class for defining strategies for the ReAct Agent.
+    """An abstract base class for defining strategies for the Prompt Optimizer Agent.
 
     Attributes:
         llm (BaseLLM): The language model used for generating answers and critiques.