chore: [ML-1194]: Fixed Issue with harness-context fields not getting passed as argument to devops agent (#38)

* chore: [ML-1194]: Fixed  Issue with harness-context fields not getting passed as argument to devops agent

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

* chore: [ML-1194]: Fixed  Issue with harness-context fields not getting passed as argument to devops agent

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

Author: Saranya-jena

README.md

@@ -385,6 +385,35 @@ Environment variables are prefixed with `HARNESS_`:
 
 The server uses a Harness API key for authentication. This can be set via the `HARNESS_API_KEY` environment variable.
 
+## Notes for Local Testing
+
+There might be certain tools that are not added in external mode or vice-versa, for local testing, following changes in `pkg/harness/tools.go` need to be done in mcp-server code to enable that certain tool in both internal and external modes.
+
+```
+baseURL := "https://localhost:8000"
+secret := <SERVICE_SECRET>
+```
+
+### Example
+
+```
+// registerGenai registers the genai toolset
+func registerGenai(config *config.Config, tsg *toolsets.ToolsetGroup) error {
+
+	// Determine the base URL and secret for genai service
+	baseURL := "http://localhost:8000"
+	secret := <GENAI_SECRET>
+
+	// Create base client for genai with the default timeout
+	c, err := createClient(baseURL, config, secret, defaultGenaiTimeout)
+	if err != nil {
+		return err
+	}
+  .
+  .
+  .
+```
+
 ## Debugging
 
 Since MCP servers run over stdio, debugging can be challenging. For the best debugging experience, we strongly recommend using the [MCP Inspector](https://github.com/modelcontextprotocol/inspector).

pkg/harness/genai.go

@@ -88,6 +88,7 @@ func AIDevOpsAgentTool(config *config.Config, client *client.GenaiService) (tool
 			interactionID, _ := OptionalParam[string](request, "interaction_id")
 			contextRaw, _ := OptionalParam[[]any](request, "context")
 			conversationRaw, _ := OptionalParam[[]any](request, "conversation_raw")
+			harnessContextRaw, _ := OptionalParam[map[string]interface{}](request, "harness_context")
 			stream := true // default value
 			if streamArg, ok := request.Params.Arguments["stream"].(bool); ok {
 				stream = streamArg
@@ -124,6 +125,19 @@ func AIDevOpsAgentTool(config *config.Config, client *client.GenaiService) (tool
 				ProjectID: scope.ProjectID,
 			}
 
+			// Override with values from request if provided
+			if harnessContextRaw != nil {
+				if accountID, ok := harnessContextRaw["accountID"].(string); ok && accountID != "" {
+					harnessContext.AccountID = accountID
+				}
+				if orgID, ok := harnessContextRaw["orgID"].(string); ok && orgID != "" {
+					harnessContext.OrgID = orgID
+				}
+				if projectID, ok := harnessContextRaw["projectID"].(string); ok && projectID != "" {
+					harnessContext.ProjectID = projectID
+				}
+			}
+
 			// Generate or use provided IDs
 			var finalConversationID, finalInteractionID string