Skip to content

Adds Playbook ID as an argument for Devin Provider #139

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
justushar wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Adds Playbook ID as an argument for Devin Provider #139

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 12 additions & 3 deletions pkg/ai/devin.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ import (
// Devin API structures for creating sessions
type CreateSessionRequest struct {
Prompt string `json:"prompt"` // Required: The task description for Devin
PlaybookID []string `json:"playbook_id,omitempty"` // Optional: ID of a playbook to guide Devin's behavior
SnapshotID *string `json:"snapshot_id,omitempty"` // Optional: ID of a machine snapshot to use
Unlisted *bool `json:"unlisted,omitempty"` // Optional: Whether the session should be unlisted
Idempotent *bool `json:"idempotent,omitempty"` // Optional: Enable idempotent session creation
Expand Down Expand Up @@ -90,14 +91,19 @@ func (s *Summarizer) callDevin(ctx context.Context, sessionID string, prompt str

// createDevinSessionAndSendMessage creates a new session and sends the initial message
func (s *Summarizer) createDevinSessionAndSendMessage(ctx context.Context, prompt string) (string, error) {
session, err := s.CreateSession(ctx, prompt, nil)
options := &CreateSessionRequest{
PlaybookID: s.config.PlaybookIDs,
}

session, err := s.CreateSession(ctx, prompt, options)
if err != nil {
return "", fmt.Errorf("create Devin session: %w", err)
}

s.logger.WithFields(logrus.Fields{
"session_id": session.SessionID,
"url": session.URL,
"session_id": session.SessionID,
"url": session.URL,
"playbook_id": s.config.PlaybookIDs,
}).Info("new Devin session created")

// Return information about the session creation
Expand Down Expand Up @@ -250,6 +256,9 @@ func (s *Summarizer) CreateSession(ctx context.Context, prompt string, options *

// Merge optional parameters if provided
if options != nil {
if options.PlaybookID != nil {
reqBody.PlaybookID = options.PlaybookID
}
if options.SnapshotID != nil {
reqBody.SnapshotID = options.SnapshotID
}
Expand Down
7 changes: 4 additions & 3 deletions pkg/ai/summary.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,10 @@ type SummaryConfig struct {
Template string
TemplateFile string
MaxTokens int
OperationType string // "plan" or "apply"
IsSuccess bool // true for success, false for failure
SessionID string // Optional: existing Devin session ID to use
OperationType string // "plan" or "apply"
IsSuccess bool // true for success, false for failure
SessionID string // Optional: existing Devin session ID to use
PlaybookIDs []string // Optional: Devin playbook IDs to guide behavior (for Devin provider only)
}

// PlanData contains the terraform plan data to be summarized
Expand Down
10 changes: 10 additions & 0 deletions pkg/cli/app.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -109,6 +109,11 @@ $ tfnotify [<global options>] plan [-patch] [-skip-no-changes] -- terraform plan
Usage: "AI model to use for summary generation (not used for Devin provider)",
Sources: cli.EnvVars("TFNOTIFY_AI_MODEL"),
},
&cli.StringSliceFlag{
Name: "playbook-id",
Usage: "Devin playbook IDs to guide behavior (for Devin provider only, can be specified multiple times)",
Sources: cli.EnvVars("TFNOTIFY_PLAYBOOK_IDS", "DEVIN_PLAYBOOK_IDS"),
},
&cli.StringFlag{
Name: "summary-template",
Usage: "Path to custom AI prompt template file",
Expand Down Expand Up @@ -146,6 +151,11 @@ $ tfnotify [<global options>] apply -- terraform apply [<terraform apply options
Usage: "AI model to use for summary generation (not used for Devin provider)",
Sources: cli.EnvVars("TFNOTIFY_AI_MODEL"),
},
&cli.StringSliceFlag{
Name: "playbook-id",
Usage: "Devin playbook IDs to guide behavior (for Devin provider only, can be specified multiple times)",
Sources: cli.EnvVars("TFNOTIFY_PLAYBOOK_IDS", "DEVIN_PLAYBOOK_IDS"),
},
&cli.StringFlag{
Name: "summary-template",
Usage: "Custom template file for AI summary prompt",
Expand Down
10 changes: 8 additions & 2 deletions pkg/cli/apply.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,10 @@ func cmdApply(ctx context.Context, cmd *cli.Command) error {
if template := cmd.String("summary-template"); template != "" {
cfg.AISummary.TemplateFile = template
}
// Playbook IDs for Devin provider
if playbookIDs := cmd.StringSlice("playbook-id"); len(playbookIDs) > 0 {
cfg.AISummary.PlaybookIDs = playbookIDs
}
}

// Get session ID if provided
Expand Down Expand Up @@ -75,12 +79,14 @@ func cmdApply(ctx context.Context, cmd *cli.Command) error {
TemplateFile: cfg.AISummary.TemplateFile,
MaxTokens: cfg.AISummary.MaxTokens,
SessionID: sessionID,
PlaybookIDs: cfg.AISummary.PlaybookIDs,
OperationType: "apply",
})
t.AISummarizer = summarizer
logrus.WithFields(logrus.Fields{
"provider": cfg.AISummary.Provider,
"model": cfg.AISummary.Model,
"provider": cfg.AISummary.Provider,
"model": cfg.AISummary.Model,
"playbook_id": cfg.AISummary.PlaybookIDs,
}).Info("AI summary enabled for apply")
}

Expand Down
10 changes: 8 additions & 2 deletions pkg/cli/plan.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,10 @@ func cmdPlan(ctx context.Context, cmd *cli.Command) error {
if template := cmd.String("summary-template"); template != "" {
cfg.AISummary.TemplateFile = template
}
// Playbook IDs for Devin provider
if playbookIDs := cmd.StringSlice("playbook-id"); len(playbookIDs) > 0 {
cfg.AISummary.PlaybookIDs = playbookIDs
}
}

// Get session ID if provided
Expand Down Expand Up @@ -74,11 +78,13 @@ func cmdPlan(ctx context.Context, cmd *cli.Command) error {
TemplateFile: cfg.AISummary.TemplateFile,
MaxTokens: cfg.AISummary.MaxTokens,
SessionID: sessionID,
PlaybookIDs: cfg.AISummary.PlaybookIDs,
})
t.AISummarizer = summarizer
logrus.WithFields(logrus.Fields{
"provider": cfg.AISummary.Provider,
"model": cfg.AISummary.Model,
"provider": cfg.AISummary.Provider,
"model": cfg.AISummary.Model,
"playbook_id": cfg.AISummary.PlaybookIDs,
}).Info("AI summary enabled")
}

Expand Down
13 changes: 7 additions & 6 deletions pkg/config/config.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,12 +36,13 @@ type Mask struct {

// AISummary configuration for AI-powered summary generation
type AISummary struct {
Enabled bool `json:"enabled,omitempty" yaml:"enabled"`
Provider string `json:"provider,omitempty" yaml:"provider"`
Model string `json:"model,omitempty" yaml:"model"`
Template string `json:"template,omitempty" yaml:"template"`
TemplateFile string `json:"template_file,omitempty" yaml:"template_file"`
MaxTokens int `json:"max_tokens,omitempty" yaml:"max_tokens"`
Enabled bool `json:"enabled,omitempty" yaml:"enabled"`
Provider string `json:"provider,omitempty" yaml:"provider"`
Model string `json:"model,omitempty" yaml:"model"`
Template string `json:"template,omitempty" yaml:"template"`
TemplateFile string `json:"template_file,omitempty" yaml:"template_file"`
MaxTokens int `json:"max_tokens,omitempty" yaml:"max_tokens"`
PlaybookIDs []string `json:"playbook_ids,omitempty" yaml:"playbook_ids"`
}

type CI struct {
Expand Down