chore: [ML-1241]: updated the service chat struct with new field and fixed an exisitng bug with toll schema (#82)

* chore: [ML-1241]: updated the service chat struct with new field

Signed-off-by: Saranya-jena <[email protected]>

Author: Saranya-jena

client/dto/genai.go

@@ -46,6 +46,12 @@ const (
 	CreateProcess   RequestAction = "CREATE_PROCESS"
 )
 
+type Caller string
+
+const (
+	CallerUnifiedAgent Caller = "unified_agent"
+)
+
 type ServiceChatParameters struct {
 	Prompt          string          `json:"prompt"`
 	Provider        string          `json:"provider,omitempty"`
@@ -58,6 +64,7 @@ type ServiceChatParameters struct {
 	Action          RequestAction   `json:"action,omitempty"`
 	HarnessContext  *HarnessContext `json:"harness_context,omitempty"`
 	Stream          bool            `json:"stream,omitempty"`
+	Caller          Caller          `json:"caller,omitempty"`
 }
 
 type CapabilityToRun struct {

pkg/harness/tools/genai.go

@@ -44,6 +44,7 @@ func AIDevOpsAgentTool(config *config.Config, client *client.GenaiService) (tool
 						"payload": map[string]any{
 							"type":        []string{"object", "array", "string", "number", "boolean"},
 							"description": "The payload for this context item, accepts any valid JSON value. Example: {\"stage_type\": \"Custom\"}",
+							"items":       map[string]any{},
 						},
 					},
 					"required": []string{"type", "payload"},
@@ -164,6 +165,7 @@ func AIDevOpsAgentTool(config *config.Config, client *client.GenaiService) (tool
 				Action:          dto.RequestAction(strings.ToUpper(action)),
 				HarnessContext:  harnessContext,
 				Stream:          stream,
+				Caller:          dto.CallerUnifiedAgent,
 			}
 
 			shouldStream := stream && progressToken != nil