From 8b27f2c47bac323edb74b8d36067f674e1320798 Mon Sep 17 00:00:00 2001
From: Reinier van der Leer <pwuts@agpt.co>
Date: Fri, 8 Dec 2023 00:54:06 +0100
Subject: [PATCH] fix(forge/db): Fix persistence of Step.additional_input and
 Step.additional_output

- Fixed the persistence issue of additional_input and additional_output in the Step class in `forge.sdk`. The additional_input and additional_output attributes were not typed and initialized properly.
---
 autogpts/forge/forge/sdk/model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/autogpts/forge/forge/sdk/model.py b/autogpts/forge/forge/sdk/model.py
index 1dc2e25e..238caae8 100644
--- a/autogpts/forge/forge/sdk/model.py
+++ b/autogpts/forge/forge/sdk/model.py
@@ -113,7 +113,7 @@ class StepRequestBody(BaseModel):
         description="Input prompt for the step.",
         example="Washington",
     )
-    additional_input: Optional[dict] = {}
+    additional_input: Optional[dict] = None
 
 
 class Status(Enum):
@@ -156,7 +156,7 @@ class Step(StepRequestBody):
         description="Output of the task step.",
         example="I am going to use the write_to_file command and write Washington to a file called output.txt <write_to_file('output.txt', 'Washington')",
     )
-    additional_output: Optional[StepOutput] = Field(default_factory=dict)
+    additional_output: Optional[dict] = None
     artifacts: Optional[List[Artifact]] = Field(
         [], description="A list of artifacts that the step has produced."
     )