fix: [ML-1380]: add template_ref field (#180)

* 34cdb3 fix lint error

* 4bf225 Revert "merge master"

* 436305 merge master

* 6210c7 merge with master

* c6932d merge master

* 18e4d1 run make format

* 59bcde modify intelligent_template_search response

* 268a4e fix: [ML-1380]: add template_ref field

* b18613 fix: [ML-1380]: add template_ref field

Author: prachi-shah-harness

client/dto/intelligence.go

@@ -23,6 +23,8 @@ type SimilaritySearchResponse struct {
 	Results []SimilaritySearchResult `json:"results"`
 }
 
+type TemplateData string
+
 // ContextType defines the type of context for AI DevOps agent
 type ContextType string
 

client/intelligence.go

@@ -28,7 +28,7 @@ func (ts *IntelligenceService) buildPath(basePath string) string {
 }
 
 // SimilaritySearch searches for similar templates based on the provided request
-func (ts *IntelligenceService) SimilaritySearch(ctx context.Context, request *dto.SimilaritySearchRequest) (*dto.SimilaritySearchResponse, error) {
+func (ts *IntelligenceService) SimilaritySearch(ctx context.Context, request *dto.SimilaritySearchRequest) (*dto.TemplateData, error) {
 	endpoint := ts.buildPath(similaritySearchPath)
 
 	// Validate required parameters
@@ -67,7 +67,60 @@ func (ts *IntelligenceService) SimilaritySearch(ctx context.Context, request *dt
 		return nil, fmt.Errorf("failed to perform similarity search: %w", err)
 	}
 
-	return &result, nil
+	var templateRef string
+
+	res := result.Results[0]
+	// Check if metadata is an array of key-value pairs
+	metadataArray, ok := res.Metadata.([]interface{})
+	if !ok {
+		return nil, fmt.Errorf("metadata is not an array of key-value pairs")
+	}
+
+	// Extract template_id, org_id, and project_id from metadata
+	var templateID, orgID, projectID string
+	for _, item := range metadataArray {
+		kvPair, ok := item.(map[string]interface{})
+		if !ok {
+			continue
+		}
+
+		key, keyOk := kvPair["Key"].(string)
+		value, valueOk := kvPair["Value"].(string)
+
+		if !keyOk || !valueOk {
+			continue
+		}
+
+		switch key {
+		case "template_id":
+			templateID = value
+		case "org_id":
+			orgID = value
+		case "project_id":
+			projectID = value
+		}
+	}
+
+	// Skip if template_id is missing
+	if templateID == "" {
+		return nil, fmt.Errorf("template_id is missing")
+	}
+
+	// Build template reference based on scope
+	templateRef = buildTemplateRef(orgID, projectID, templateID)
+	templateData := dto.TemplateData("template_ref: " + templateRef + " type: " + request.TemplateType)
+
+	return &templateData, nil
+}
+
+func buildTemplateRef(orgID, projectID, templateID string) string {
+	if projectID != "" {
+		return templateID
+	} else if orgID != "" {
+		return fmt.Sprintf("org.%s", templateID)
+	} else {
+		return fmt.Sprintf("account.%s", templateID)
+	}
 }
 
 // sendAIDevOpsChatRequest handles sending AI DevOps chat requests with support for both streaming and non-streaming modes

pkg/harness/tools/intelligence.go

@@ -19,7 +19,8 @@ import (
 // FindSimilarTemplates creates a tool that allows finding similar templates based on provided description.
 func FindSimilarTemplates(config *config.Config, client *client.IntelligenceService) (tool mcp.Tool, handler server.ToolHandlerFunc) {
 	return mcp.NewTool("intelligent_template_search",
-			mcp.WithDescription("Finds the most relevant templates based on a natural language description. "+
+			mcp.WithDescription("The tool is used to find most relevant entity template for any entity everytime a template is explicitly requested for any of pipeline/stage/step"+
+				"Finds the most relevant templates based on a natural language description. "+
 				"Searches across template identifiers, names, types, capabilities, and use cases to find the best matches. "+
 				"Returns templates ranked by similarity score with metadata including template IDs and organizational context. "+
 				"Ideal for discovering templates that fulfill specific requirements without knowing exact identifiers."),
@@ -32,6 +33,7 @@ func FindSimilarTemplates(config *config.Config, client *client.IntelligenceServ
 			),
 			mcp.WithNumber("count",
 				mcp.Description("Maximum number of similar templates to return"),
+				mcp.DefaultNumber(1),
 			),
 			common.WithScope(config, false),
 		),