feat: AI-анализ terraform plan (уровни 0-5, Google Gemini)
This commit is contained in:
Naeel
@@ -0,0 +1,85 @@
|
||||
// internal/ai/analyzer.go
|
||||
// Дата: 2026-03-12
|
||||
// Оркестратор AI-анализа: читает уровень, формирует промпт,
|
||||
// вызывает LLM-провайдер, форматирует вывод для пользователя.
|
||||
|
||||
package ai
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Analyzer — основной компонент AI-анализа.
|
||||
// Создаётся один раз при старте sless-plan, используется для одного анализа.
|
||||
type Analyzer struct {
|
||||
provider LLMProvider
|
||||
level int
|
||||
}
|
||||
|
||||
// NewAnalyzer создаёт Analyzer из конфига.
|
||||
// Возвращает ошибку если провайдер не может быть инициализирован.
|
||||
func NewAnalyzer(cfg Config) (*Analyzer, error) {
|
||||
if cfg.HintLevel == 0 {
|
||||
// Уровень 0 — AI выключен, провайдер не нужен
|
||||
return &Analyzer{level: 0}, nil
|
||||
}
|
||||
|
||||
provider, err := NewProvider(cfg)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("ошибка инициализации AI провайдера: %w", err)
|
||||
}
|
||||
|
||||
return &Analyzer{
|
||||
provider: provider,
|
||||
level: cfg.HintLevel,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// IsEnabled возвращает true если уровень > 0 и провайдер инициализирован.
|
||||
func (a *Analyzer) IsEnabled() bool {
|
||||
return a.level > 0 && a.provider != nil
|
||||
}
|
||||
|
||||
// Analyze отправляет вывод terraform plan в LLM и возвращает отформатированный результат.
|
||||
// planText — текстовый вывод terraform plan.
|
||||
func (a *Analyzer) Analyze(ctx context.Context, planText string) (string, error) {
|
||||
if !a.IsEnabled() {
|
||||
return "", nil
|
||||
}
|
||||
|
||||
// Ограничиваем размер payload — большие планы обрезаем
|
||||
// чтобы не превысить лимит токенов LLM и не отправлять лишнее
|
||||
planText = truncatePlan(planText, 8000)
|
||||
|
||||
prompt := BuildPrompt(a.level, planText)
|
||||
result, err := a.provider.Analyze(ctx, prompt)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("AI анализ не удался: %w", err)
|
||||
}
|
||||
|
||||
return FormatResult(a.level, a.provider.Name(), result), nil
|
||||
}
|
||||
|
||||
// FormatResult оборачивает ответ LLM в читаемый блок для вывода в терминал.
|
||||
func FormatResult(level int, providerName, result string) string {
|
||||
separator := strings.Repeat("─", 60)
|
||||
header := fmt.Sprintf("AI Analysis (level %d: %s) via %s", level, LevelName(level), providerName)
|
||||
return fmt.Sprintf("\n%s\n %s\n%s\n%s\n%s\n",
|
||||
separator,
|
||||
header,
|
||||
separator,
|
||||
result,
|
||||
separator,
|
||||
)
|
||||
}
|
||||
|
||||
// truncatePlan обрезает текст плана до maxChars символов с пометкой об обрезке.
|
||||
// Нужно чтобы не превышать лимиты токенов LLM и не отправлять лишние данные.
|
||||
func truncatePlan(text string, maxChars int) string {
|
||||
if len(text) <= maxChars {
|
||||
return text
|
||||
}
|
||||
return text[:maxChars] + fmt.Sprintf("\n\n[... обрезано, показаны первые %d символов из %d ...]", maxChars, len(text))
|
||||
}
|
||||
@@ -0,0 +1,40 @@
|
||||
// internal/ai/cloud_llm.go
|
||||
// Дата: 2026-03-12
|
||||
// Заглушка для будущего облачного LLM.
|
||||
// Когда облако предоставит свой LLM — реализовать этот файл
|
||||
// и переключить AI_PROVIDER=cloud. Остальной код не меняется.
|
||||
|
||||
package ai
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
// CloudLLMProvider — будущая реализация LLMProvider для облачного LLM.
|
||||
// Пока возвращает ошибку "не реализован".
|
||||
type CloudLLMProvider struct {
|
||||
endpoint string
|
||||
token string
|
||||
}
|
||||
|
||||
// NewCloudLLMProvider создаёт заглушку провайдера облачного LLM.
|
||||
func NewCloudLLMProvider(endpoint, token string) (*CloudLLMProvider, error) {
|
||||
if endpoint == "" {
|
||||
return nil, fmt.Errorf("CLOUD_LLM_ENDPOINT не задан")
|
||||
}
|
||||
return &CloudLLMProvider{
|
||||
endpoint: endpoint,
|
||||
token: token,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (c *CloudLLMProvider) Name() string {
|
||||
return fmt.Sprintf("cloud-llm/%s", c.endpoint)
|
||||
}
|
||||
|
||||
// Analyze — TODO: реализовать когда облако предоставит LLM API.
|
||||
// Документация по API будет в doc/api/cloud-llm.md.
|
||||
func (c *CloudLLMProvider) Analyze(_ context.Context, _ string) (string, error) {
|
||||
return "", fmt.Errorf("cloud LLM провайдер ещё не реализован (endpoint: %s)", c.endpoint)
|
||||
}
|
||||
@@ -0,0 +1,125 @@
|
||||
// internal/ai/google_gemini.go
|
||||
// Дата: 2026-03-12
|
||||
// Реализация LLMProvider для Google Gemini API.
|
||||
// Использует REST API напрямую через net/http — без внешних SDK,
|
||||
// чтобы не нарушать совместимость существующего go.mod.
|
||||
|
||||
package ai
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"time"
|
||||
)
|
||||
|
||||
const geminiBaseURL = "https://generativelanguage.googleapis.com/v1beta/models/%s:generateContent"
|
||||
|
||||
// GeminiProvider реализует LLMProvider через Google Gemini REST API.
|
||||
type GeminiProvider struct {
|
||||
apiKey string
|
||||
model string
|
||||
httpClient *http.Client
|
||||
}
|
||||
|
||||
// NewGeminiProvider создаёт провайдера для Google Gemini.
|
||||
// apiKey — значение env GOOGLE_AI_API_KEY.
|
||||
func NewGeminiProvider(apiKey, model string) (*GeminiProvider, error) {
|
||||
if apiKey == "" {
|
||||
return nil, fmt.Errorf("GOOGLE_AI_API_KEY не задан")
|
||||
}
|
||||
return &GeminiProvider{
|
||||
apiKey: apiKey,
|
||||
model: model,
|
||||
httpClient: &http.Client{
|
||||
Timeout: 60 * time.Second,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (g *GeminiProvider) Name() string {
|
||||
return fmt.Sprintf("google-gemini/%s", g.model)
|
||||
}
|
||||
|
||||
// geminiRequest — структура запроса к Gemini API.
|
||||
type geminiRequest struct {
|
||||
Contents []geminiContent `json:"contents"`
|
||||
}
|
||||
|
||||
type geminiContent struct {
|
||||
Parts []geminiPart `json:"parts"`
|
||||
}
|
||||
|
||||
type geminiPart struct {
|
||||
Text string `json:"text"`
|
||||
}
|
||||
|
||||
// geminiResponse — структура ответа от Gemini API.
|
||||
type geminiResponse struct {
|
||||
Candidates []struct {
|
||||
Content struct {
|
||||
Parts []struct {
|
||||
Text string `json:"text"`
|
||||
} `json:"parts"`
|
||||
} `json:"content"`
|
||||
} `json:"candidates"`
|
||||
Error *struct {
|
||||
Message string `json:"message"`
|
||||
Code int `json:"code"`
|
||||
} `json:"error,omitempty"`
|
||||
}
|
||||
|
||||
// Analyze отправляет промпт в Gemini и возвращает текстовый ответ.
|
||||
func (g *GeminiProvider) Analyze(ctx context.Context, prompt string) (string, error) {
|
||||
reqBody := geminiRequest{
|
||||
Contents: []geminiContent{
|
||||
{Parts: []geminiPart{{Text: prompt}}},
|
||||
},
|
||||
}
|
||||
|
||||
bodyBytes, err := json.Marshal(reqBody)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("ошибка маршалинга запроса: %w", err)
|
||||
}
|
||||
|
||||
url := fmt.Sprintf(geminiBaseURL+"?key=%s", g.model, g.apiKey)
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(bodyBytes))
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("ошибка создания HTTP запроса: %w", err)
|
||||
}
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
|
||||
resp, err := g.httpClient.Do(req)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("ошибка HTTP запроса к Gemini: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
respBytes, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("ошибка чтения ответа: %w", err)
|
||||
}
|
||||
|
||||
var gemResp geminiResponse
|
||||
if err := json.Unmarshal(respBytes, &gemResp); err != nil {
|
||||
return "", fmt.Errorf("ошибка парсинга ответа Gemini: %w", err)
|
||||
}
|
||||
|
||||
// Проверяем HTTP-ошибку и ошибку в теле ответа
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
msg := fmt.Sprintf("HTTP %d", resp.StatusCode)
|
||||
if gemResp.Error != nil {
|
||||
msg = gemResp.Error.Message
|
||||
}
|
||||
return "", fmt.Errorf("Gemini API ошибка: %s", msg)
|
||||
}
|
||||
|
||||
if len(gemResp.Candidates) == 0 || len(gemResp.Candidates[0].Content.Parts) == 0 {
|
||||
return "", fmt.Errorf("Gemini вернул пустой ответ")
|
||||
}
|
||||
|
||||
return gemResp.Candidates[0].Content.Parts[0].Text, nil
|
||||
}
|
||||
@@ -0,0 +1,76 @@
|
||||
// internal/ai/prompts.go
|
||||
// Дата: 2026-03-12
|
||||
// Шаблоны system-промптов для каждого уровня анализа (1–5).
|
||||
// Уровень 0 — AI не вызывается вообще (проверка в analyzer.go).
|
||||
|
||||
package ai
|
||||
|
||||
import "fmt"
|
||||
|
||||
// systemPrompts — system prompt для каждого уровня.
|
||||
// Каждый уровень строго ограничен своей областью — не расширять без явного указания.
|
||||
var systemPrompts = map[int]string{
|
||||
1: `Ты — анализатор Terraform конфигурации. Твоя задача: ТОЛЬКО синтаксический анализ.
|
||||
Найди синтаксические ошибки в HCL: опечатки в ключевых словах, незакрытые блоки, неверные типы значений, невалидные атрибуты.
|
||||
НЕ давай рекомендаций по улучшению, НЕ анализируй логику, НЕ комментируй архитектуру.
|
||||
Формат ответа: список найденных ошибок или "✓ Синтаксических ошибок не обнаружено."`,
|
||||
|
||||
2: `Ты — анализатор Terraform конфигурации. Твоя задача: синтаксис + базовые проблемы.
|
||||
1. Синтаксические ошибки в HCL.
|
||||
2. Базовые проблемы: неиспользуемые переменные/ресурсы, пустые обязательные значения, дублирующиеся ресурсы.
|
||||
НЕ анализируй зависимости, архитектуру, безопасность.
|
||||
Формат: список проблем с пометкой [SYNTAX] или [BASIC], или "✓ Проблем не обнаружено."`,
|
||||
|
||||
3: `Ты — анализатор Terraform конфигурации. Твоя задача: синтаксис + базовые проблемы + зависимости.
|
||||
1. Синтаксические ошибки.
|
||||
2. Базовые проблемы (неиспользуемые ресурсы, пустые значения).
|
||||
3. Анализ зависимостей: потенциальные race conditions при apply/destroy, неявные зависимости, проблемы с порядком создания ресурсов.
|
||||
НЕ давай рекомендаций по безопасности или рефакторингу.
|
||||
Формат: список проблем с пометкой [SYNTAX], [BASIC] или [DEPS], или "✓ Проблем не обнаружено."`,
|
||||
|
||||
4: `Ты — анализатор Terraform конфигурации. Твоя задача: полный анализ + безопасность.
|
||||
1. Синтаксические ошибки.
|
||||
2. Базовые проблемы.
|
||||
3. Зависимости и порядок операций.
|
||||
4. Безопасность и best practices: открытые порты (0.0.0.0/0), избыточные права доступа, незашифрованные ресурсы, жёстко заданные секреты.
|
||||
Формат: список проблем с пометкой [SYNTAX], [BASIC], [DEPS] или [SECURITY], или "✓ Проблем не обнаружено."`,
|
||||
|
||||
5: `Ты — старший DevOps-инженер, проводишь полный аудит Terraform конфигурации.
|
||||
Анализируй:
|
||||
1. Синтаксические ошибки.
|
||||
2. Базовые проблемы.
|
||||
3. Зависимости и порядок операций.
|
||||
4. Безопасность и best practices.
|
||||
5. Оптимизация: возможности для рефакторинга, использование модулей, устаревшие паттерны, альтернативные подходы.
|
||||
Давай конкретные рекомендации с примерами кода где уместно.
|
||||
Формат: список проблем и рекомендаций с пометкой [SYNTAX], [BASIC], [DEPS], [SECURITY] или [OPT].`,
|
||||
}
|
||||
|
||||
// levelNames — человекочитаемые названия уровней для вывода.
|
||||
var levelNames = map[int]string{
|
||||
0: "OFF",
|
||||
1: "SYNTAX",
|
||||
2: "BASIC",
|
||||
3: "MODERATE",
|
||||
4: "DETAILED",
|
||||
5: "FULL",
|
||||
}
|
||||
|
||||
// BuildPrompt формирует финальный промпт для отправки в LLM.
|
||||
// planText — вывод terraform plan (текст или JSON).
|
||||
func BuildPrompt(level int, planText string) string {
|
||||
sysPrompt, ok := systemPrompts[level]
|
||||
if !ok {
|
||||
sysPrompt = systemPrompts[1]
|
||||
}
|
||||
return fmt.Sprintf("%s\n\nTerraform plan output:\n```\n%s\n```", sysPrompt, planText)
|
||||
}
|
||||
|
||||
// LevelName возвращает название уровня для вывода пользователю.
|
||||
func LevelName(level int) string {
|
||||
name, ok := levelNames[level]
|
||||
if !ok {
|
||||
return "UNKNOWN"
|
||||
}
|
||||
return name
|
||||
}
|
||||
@@ -0,0 +1,53 @@
|
||||
// internal/ai/provider.go
|
||||
// Дата: 2026-03-12
|
||||
// Интерфейс LLMProvider — абстракция над любым LLM-бэкендом.
|
||||
// Сейчас реализован Google Gemini. В будущем — облачный LLM (drop-in замена).
|
||||
|
||||
package ai
|
||||
|
||||
import "context"
|
||||
|
||||
// LLMProvider — единственный интерфейс для всех LLM-провайдеров.
|
||||
// Менять провайдер = менять одну env-переменную AI_PROVIDER.
|
||||
type LLMProvider interface {
|
||||
// Analyze отправляет промпт в LLM и возвращает текстовый ответ.
|
||||
Analyze(ctx context.Context, prompt string) (string, error)
|
||||
// Name возвращает имя провайдера для логирования.
|
||||
Name() string
|
||||
}
|
||||
|
||||
// Config — конфигурация анализатора, читается из env-переменных.
|
||||
type Config struct {
|
||||
// Provider: "google" или "cloud" (в будущем)
|
||||
Provider string
|
||||
// HintLevel: 0–5, если 0 — AI не вызывается
|
||||
HintLevel int
|
||||
// Google Gemini
|
||||
GoogleAPIKey string
|
||||
GoogleModel string // по умолчанию "gemini-pro"
|
||||
// Будущий облачный LLM
|
||||
CloudEndpoint string
|
||||
CloudToken string
|
||||
}
|
||||
|
||||
// NewProvider — фабрика провайдеров. Выбор по Config.Provider.
|
||||
// При добавлении нового провайдера — только сюда добавить case.
|
||||
func NewProvider(cfg Config) (LLMProvider, error) {
|
||||
switch cfg.Provider {
|
||||
case "google":
|
||||
model := cfg.GoogleModel
|
||||
if model == "" {
|
||||
model = "gemini-2.0-flash"
|
||||
}
|
||||
return NewGeminiProvider(cfg.GoogleAPIKey, model)
|
||||
case "cloud":
|
||||
return NewCloudLLMProvider(cfg.CloudEndpoint, cfg.CloudToken)
|
||||
default:
|
||||
// По умолчанию google, если AI_PROVIDER не задан
|
||||
model := cfg.GoogleModel
|
||||
if model == "" {
|
||||
model = "gemini-2.0-flash"
|
||||
}
|
||||
return NewGeminiProvider(cfg.GoogleAPIKey, model)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user