chore: [ML-1243]: Added Default Module in interface structure with the set of default tools (#68)

* chore: [ML-1243]: Updated branch

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

* Merge branch 'master' of https://git0.harness.io/l7B_kbSEQD2wjrM7PShm5w/PROD/Harness_Commons/mcp-server into ML-1243

* chore: [ML-1243]: Updated branch

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

* chore: [ML-1243]: Added Default Module in interface structure with the set of default tools

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

Author: Saranya-jena

README.md

@@ -466,7 +466,7 @@ func registerGenai(config *config.Config, tsg *toolsets.ToolsetGroup) error {
 	secret := <GENAI_SECRET>
 
 	// Create base client for genai with the default timeout
-	c, err := createClient(baseURL, config, secret, defaultGenaiTimeout)
+	c, err := CreateClient(baseURL, config, secret, defaultGenaiTimeout)
 	if err != nil {
 		return err
 	}

cmd/harness-mcp-server/config/config.go

@@ -2,11 +2,12 @@ package config
 
 type Config struct {
 	// Common fields for both modes
-	Version     string
-	ReadOnly    bool
-	Toolsets    []string
-	LogFilePath string
-	Debug       bool
+	Version        string
+	ReadOnly       bool
+	Toolsets       []string
+	EnabledModules []string
+	LogFilePath    string
+	Debug          bool
 
 	Internal bool
 

cmd/harness-mcp-server/main.go

@@ -3,6 +3,7 @@ package main
 import (
 	"context"
 	"fmt"
+	"github.com/harness/harness-mcp/pkg/harness/tools"
 	"io"
 	"log"
 	"log/slog"
@@ -78,6 +79,7 @@ var (
 				Toolsets:         toolsets,
 				LogFilePath:      viper.GetString("log_file"),
 				Debug:            viper.GetBool("debug"),
+				EnabledModules:   harness.EnabledModules,
 			}
 
 			if err := runStdioServer(ctx, cfg); err != nil {
@@ -124,13 +126,14 @@ var (
 
 			cfg := config.Config{
 				// Common fields
-				Version:     version,
-				ReadOnly:    true, // we keep it read-only for now
-				Toolsets:    toolsets,
-				LogFilePath: viper.GetString("log_file"),
-				Debug:       viper.GetBool("debug"),
-				Internal:    true,
-				AccountID:   session.Principal.AccountID,
+				Version:        version,
+				ReadOnly:       true, // we keep it read-only for now
+				Toolsets:       toolsets,
+				EnabledModules: harness.EnabledModules,
+				LogFilePath:    viper.GetString("log_file"),
+				Debug:          viper.GetBool("debug"),
+				Internal:       true,
+				AccountID:      session.Principal.AccountID,
 				// Internal mode specific fields
 				BearerToken:             viper.GetString("bearer_token"),
 				PipelineSvcBaseURL:      viper.GetString("pipeline_svc_base_url"),
@@ -342,7 +345,7 @@ func runStdioServer(ctx context.Context, config config.Config) error {
 
 	// Create server
 	// WithRecovery makes sure panics are logged and don't crash the server
-	harnessServer := harness.NewServer(version, server.WithHooks(hooks), server.WithRecovery())
+	harnessServer := tools.NewServer(version, server.WithHooks(hooks), server.WithRecovery())
 
 	// Initialize toolsets
 	toolsets, err := harness.InitToolsets(&config)
@@ -354,7 +357,7 @@ func runStdioServer(ctx context.Context, config config.Config) error {
 	toolsets.RegisterTools(harnessServer)
 
 	// Set the guidelines prompts
-	harness.RegisterPrompts(harnessServer)
+	tools.RegisterPrompts(harnessServer)
 
 	// Create stdio server
 	stdioServer := server.NewStdioServer(harnessServer)