From 5033bbebc8f079adcafa6ded9e83cc0e21a23724 Mon Sep 17 00:00:00 2001
From: vignesh_arivazhagan <vignesh.arivazhagan@solitontech.com>
Date: Thu, 21 Dec 2023 11:50:34 +0530
Subject: [PATCH 1/4] Update models.py with additional input support

---
 packages/sdk/python/agent_protocol/models.py | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/packages/sdk/python/agent_protocol/models.py b/packages/sdk/python/agent_protocol/models.py
index a8145ed..00ee368 100644
--- a/packages/sdk/python/agent_protocol/models.py
+++ b/packages/sdk/python/agent_protocol/models.py
@@ -5,7 +5,7 @@
 from __future__ import annotations
 
 from enum import Enum
-from typing import List, Optional
+from typing import List, Optional, Dict, Any
 
 from pydantic import BaseModel, Field
 
@@ -25,10 +25,6 @@ class Pagination(BaseModel):
     page_size: int = Field(..., description="Number of items per page.", example=25)
 
 
-class TaskInput(BaseModel):
-    pass
-
-
 class Artifact(BaseModel):
     artifact_id: str = Field(
         ...,
@@ -61,10 +57,6 @@ class ArtifactUpload(BaseModel):
     )
 
 
-class StepInput(BaseModel):
-    pass
-
-
 class StepOutput(BaseModel):
     pass
 
@@ -75,7 +67,7 @@ class TaskRequestBody(BaseModel):
         description="Input prompt for the task.",
         example="Write 'Washington' to the file 'output.txt'.",
     )
-    additional_input: Optional[TaskInput] = None
+    additional_input: Optional[Dict[str, Any]] = None
 
 
 class Task(TaskRequestBody):
@@ -100,7 +92,7 @@ class StepRequestBody(BaseModel):
         description="Input prompt for the step.",
         example="Write the words you receive to the file 'output.txt'.",
     )
-    additional_input: Optional[StepInput] = None
+    additional_input: Optional[Dict[str, Any]] = None
 
 
 class Status(Enum):

From cfca056610e789b420f4eaa76a987f278a498ebc Mon Sep 17 00:00:00 2001
From: vignesh_arivazhagan <vignesh.arivazhagan@solitontech.com>
Date: Thu, 21 Dec 2023 11:56:32 +0530
Subject: [PATCH 2/4] Update additional_properties type to allow any value

---
 packages/sdk/python/agent_protocol/db.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/sdk/python/agent_protocol/db.py b/packages/sdk/python/agent_protocol/db.py
index 0c52c15..24efb73 100644
--- a/packages/sdk/python/agent_protocol/db.py
+++ b/packages/sdk/python/agent_protocol/db.py
@@ -5,7 +5,7 @@
 
 
 class Step(APIStep):
-    additional_properties: Optional[Dict[str, str]] = None
+    additional_properties: Optional[Dict[str, Any]] = None
 
 
 class Task(APITask):

From d385ce1779432244c46ebc7b864c1958f9121ca9 Mon Sep 17 00:00:00 2001
From: vignesh_arivazhagan <vignesh.arivazhagan@solitontech.com>
Date: Thu, 21 Dec 2023 12:08:38 +0530
Subject: [PATCH 3/4] js SDK : Add additional_input property to Step and Task
 interfaces

---
 packages/sdk/js/src/models.ts | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/packages/sdk/js/src/models.ts b/packages/sdk/js/src/models.ts
index 740f06b..0405401 100644
--- a/packages/sdk/js/src/models.ts
+++ b/packages/sdk/js/src/models.ts
@@ -60,6 +60,10 @@ export interface Step {
    * Current status of step
    */
   status: StepStatus
+   /**
+   * Additional input that the step has.
+   */
+  additional_input?: {[key: string]: any}
 }
 
 export interface StepRequestBody {
@@ -95,6 +99,10 @@ export interface Task {
    * A list of artifacts that the task has produced.
    */
   artifacts?: Artifact[]
+  /**
+   * Additional input that the task has.
+   */
+  additional_input?: {[key: string]: any}
 }
 
 export interface TaskRequestBody {

From d66f652544af80689b4146298df4e7ad342ed21d Mon Sep 17 00:00:00 2001
From: vignesh_arivazhagan <vignesh.arivazhagan@solitontech.com>
Date: Tue, 23 Jan 2024 10:10:19 +0530
Subject: [PATCH 4/4] js-sdk : Revert Adding additional_input

---
 packages/sdk/js/src/models.ts | 8 --------
 1 file changed, 8 deletions(-)

diff --git a/packages/sdk/js/src/models.ts b/packages/sdk/js/src/models.ts
index 0405401..740f06b 100644
--- a/packages/sdk/js/src/models.ts
+++ b/packages/sdk/js/src/models.ts
@@ -60,10 +60,6 @@ export interface Step {
    * Current status of step
    */
   status: StepStatus
-   /**
-   * Additional input that the step has.
-   */
-  additional_input?: {[key: string]: any}
 }
 
 export interface StepRequestBody {
@@ -99,10 +95,6 @@ export interface Task {
    * A list of artifacts that the task has produced.
    */
   artifacts?: Artifact[]
-  /**
-   * Additional input that the task has.
-   */
-  additional_input?: {[key: string]: any}
 }
 
 export interface TaskRequestBody {