diff options
Diffstat (limited to 'autogpts/forge/forge/agent.py')
-rw-r--r-- | autogpts/forge/forge/agent.py | 19 |
1 files changed, 9 insertions, 10 deletions
diff --git a/autogpts/forge/forge/agent.py b/autogpts/forge/forge/agent.py index f96a6a91d..7757c6680 100644 --- a/autogpts/forge/forge/agent.py +++ b/autogpts/forge/forge/agent.py @@ -1,3 +1,4 @@ +from forge.actions import ActionRegister from forge.sdk import ( Agent, AgentDB, @@ -6,13 +7,8 @@ from forge.sdk import ( StepRequestBody, Task, TaskRequestBody, - Workspace, - PromptEngine, - chat_completion_request, - ChromaMemStore + Workspace, ) -import json -import pprint LOG = ForgeLogger(__name__) @@ -78,6 +74,7 @@ class ForgeAgent(Agent): Feel free to create subclasses of the database and workspace to implement your own storage """ super().__init__(database, workspace) + self.abilities = ActionRegister(self) async def create_task(self, task_request: TaskRequestBody) -> Task: """ @@ -139,9 +136,11 @@ class ForgeAgent(Agent): step.output = "Washington D.C" - LOG.info(f"\t✅ Final Step completed: {step.step_id}. \n" + - f"Output should be placeholder text Washington D.C. You'll need to \n" + - f"modify execute_step to include LLM behavior. Follow the tutorial " + - f"if confused. ") + LOG.info( + f"\t✅ Final Step completed: {step.step_id}. \n" + + f"Output should be placeholder text Washington D.C. You'll need to \n" + + f"modify execute_step to include LLM behavior. Follow the tutorial " + + f"if confused. " + ) return step |