Merge pull request #344 from ca-risken/fix-ai

fix: finding details
ca-risken · Oct 24, 2023 · 788165a · 788165a
2 parents db8feea + 5de0f23
commit 788165a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pkg/ai/openai.go b/pkg/ai/openai.go
@@ -54,6 +54,9 @@ Type:
 
 Description: 
 %s
+
+ScanResult(json):
+%s
 `
 	RECOMMEND_FORMAT_FOR_AI = `Detail: %s
 Recommendation: %s
@@ -209,7 +212,7 @@ func (a *AIClient) AskAISummaryStreamFromFinding(
 }
 
 func generateFindingDataForAI(f *model.Finding, r *model.Recommend) string {
-	text := fmt.Sprintf(FINDING_FORMAT_FOR_AI, f.DataSource, f.Description)
+	text := fmt.Sprintf(FINDING_FORMAT_FOR_AI, f.DataSource, f.Description, f.Data)
 	if r != nil {
 		text += fmt.Sprintf(RECOMMEND_FORMAT_FOR_AI, r.Risk, r.Recommendation)
 	}