ai

Documentation

Index

• Constants
• type AIClient
  • func (a *AIClient) AskAISummaryFromFinding(ctx context.Context, f *model.Finding, r *model.Recommend, lang string) (string, error)
• type AIService
  • func NewAIClient(token string, logger logging.Logger) AIService

Constants

const (
	CACHE_SIZE       = 200 * 1024 * 1024 // 200MB
	CACHE_EXPIRE_SEC = 3600
	CACHE_KEY_FORMAT = "OpenAICache/%d/%s"

	LANG_JP               = "ja"
	PROMPT_SYSTEM_MSG_EN  = "You are a helpful security assistant."
	PROMPT_SYSTEM_MSG_JP  = "あなたは役に立つセキュリティアシスタントです。"
	PROMPT_SUMMARY_EN     = "" /* 205-byte string literal not displayed */
	PROMPT_SUMMARY_JP     = "" /* 288-byte string literal not displayed */
	FINDING_FORMAT_FOR_AI = `The RISKEN tool detected the following issue related to cloud security.
Type: %s
Description: %s
`
	RECOMMEND_FORMAT_FOR_AI = `Detail: %s
Recommendation: %s
`
)

Types

type AIClient

type AIClient struct {
	// contains filtered or unexported fields
}

func (*AIClient) AskAISummaryFromFinding

func (a *AIClient) AskAISummaryFromFinding(ctx context.Context, f *model.Finding, r *model.Recommend, lang string) (string, error)

type AIService

type AIService interface {
	AskAISummaryFromFinding(ctx context.Context, f *model.Finding, r *model.Recommend, lang string) (string, error)
}

func NewAIClient

func NewAIClient(token string, logger logging.Logger) AIService