zyflow 0.6.4

package/server/adk/agents/pr-agent.ts

@@ -0,0 +1,264 @@
+/**
+ * PR Agent
+ *
+ * 수정 사항을 커밋하고 PR을 생성하는 에이전트
+ */
+
+import { LlmAgent } from '@google/adk'
+import { gitTools, gitCreateBranchTool, gitAddTool, gitCommitTool, gitPushTool } from '../tools/git-tools'
+import { githubTools, createPRTool, getPRStatusTool, mergePRTool, waitForCITool, deleteBranchTool } from '../tools/github-tools'
+import { loadConfig } from '../config'
+import type { CodeFix } from './fix-generator'
+import type { ErrorAnalysisResult } from './error-analyzer'
+
+const config = loadConfig()
+
+export interface PRCreationResult {
+  success: boolean
+  prNumber?: number
+  prUrl?: string
+  branch?: string
+  error?: string
+}
+
+export const prAgent = new LlmAgent({
+  name: 'pr-agent',
+  description: 'PR을 생성하고 관리합니다.',
+  model: config.model,
+  instruction: `당신은 Git과 GitHub 작업 전문가입니다.
+
+코드 수정 후:
+1. 새 브랜치를 생성합니다
+2. 변경사항을 커밋합니다
+3. 원격에 푸시합니다
+4. PR을 생성합니다
+
+PR 제목과 본문을 명확하게 작성하세요.`,
+  tools: [...gitTools, ...githubTools],
+})
+
+export async function createAutoFixPR(
+  analysis: ErrorAnalysisResult,
+  fixes: CodeFix[],
+  options?: {
+    repository?: string
+    baseBranch?: string
+    alertId?: string
+    dryRun?: boolean
+  }
+): Promise<PRCreationResult> {
+  const repository = options?.repository || process.env.GITHUB_REPOSITORY
+  if (!repository) {
+    return { success: false, error: 'Repository not specified' }
+  }
+
+  const alertId = options?.alertId || Date.now().toString()
+  const branchName = 'auto-fix/' + alertId
+
+  try {
+    // 1. 브랜치 생성
+    const branchResult = await gitCreateBranchTool.execute({
+      branchName,
+      baseBranch: options?.baseBranch || 'main',
+      checkout: true,
+    })
+
+    if (!branchResult.success) {
+      return { success: false, error: 'Branch creation failed: ' + branchResult.error }
+    }
+
+    // 2. 파일 스테이징
+    const filePaths = fixes.map(f => f.filePath)
+    const addResult = await gitAddTool.execute({
+      files: filePaths,
+    })
+
+    if (!addResult.success) {
+      return { success: false, error: 'Git add failed: ' + addResult.error }
+    }
+
+    // 3. 커밋
+    const commitMessage = generateCommitMessage(analysis)
+    const commitResult = await gitCommitTool.execute({
+      message: commitMessage,
+      body: generateCommitBody(analysis, fixes),
+    })
+
+    if (!commitResult.success) {
+      return { success: false, error: 'Commit failed: ' + commitResult.error }
+    }
+
+    if (options?.dryRun) {
+      return {
+        success: true,
+        branch: branchName,
+        prUrl: 'dry-run',
+      }
+    }
+
+    // 4. 푸시
+    const pushResult = await gitPushTool.execute({
+      branch: branchName,
+      setUpstream: true,
+    })
+
+    if (!pushResult.success) {
+      return { success: false, error: 'Push failed: ' + pushResult.error }
+    }
+
+    // 5. PR 생성
+    const prResult = await createPRTool.execute({
+      repository,
+      title: generatePRTitle(analysis),
+      body: generatePRBody(analysis, fixes),
+      head: branchName,
+      base: options?.baseBranch || 'main',
+      labels: ['auto-fix', 'gemini'],
+      draft: false,
+    })
+
+    if (!prResult.success) {
+      return { success: false, error: 'PR creation failed: ' + prResult.error }
+    }
+
+    return {
+      success: true,
+      prNumber: prResult.prNumber,
+      prUrl: prResult.prUrl,
+      branch: branchName,
+    }
+  } catch (err) {
+    return { success: false, error: String(err) }
+  }
+}
+
+export async function waitForCIAndMerge(
+  repository: string,
+  prNumber: number,
+  options?: {
+    timeout?: number
+    autoMerge?: boolean
+  }
+): Promise<{
+  ciPassed: boolean
+  merged: boolean
+  error?: string
+}> {
+  try {
+    // PR 상태 확인
+    const statusResult = await getPRStatusTool.execute({
+      repository,
+      prNumber,
+    })
+
+    if (!statusResult.success) {
+      return { ciPassed: false, merged: false, error: statusResult.error }
+    }
+
+    // CI 대기
+    const ciResult = await waitForCITool.execute({
+      repository,
+      ref: statusResult.headSha,
+      timeout: options?.timeout || 600,
+    })
+
+    if (!ciResult.completed) {
+      return { ciPassed: false, merged: false, error: 'CI timeout' }
+    }
+
+    if (!ciResult.passed) {
+      return { ciPassed: false, merged: false, error: 'CI failed' }
+    }
+
+    if (!options?.autoMerge) {
+      return { ciPassed: true, merged: false }
+    }
+
+    // 머지
+    const mergeResult = await mergePRTool.execute({
+      repository,
+      prNumber,
+      method: 'squash',
+    })
+
+    if (!mergeResult.success) {
+      return { ciPassed: true, merged: false, error: mergeResult.error }
+    }
+
+    // 브랜치 삭제
+    await deleteBranchTool.execute({
+      repository,
+      branch: 'auto-fix/' + prNumber,
+    })
+
+    return { ciPassed: true, merged: true }
+  } catch (err) {
+    return { ciPassed: false, merged: false, error: String(err) }
+  }
+}
+
+function generateCommitMessage(analysis: ErrorAnalysisResult): string {
+  const typePrefix = getCommitPrefix(analysis.errorType)
+  return typePrefix + ': ' + analysis.summary
+}
+
+function generateCommitBody(analysis: ErrorAnalysisResult, fixes: CodeFix[]): string {
+  const lines = [
+    'Auto-fix generated by Gemini ADK',
+    '',
+    'Root cause: ' + analysis.rootCause,
+    '',
+    'Modified files:',
+  ]
+
+  for (const fix of fixes) {
+    lines.push('- ' + fix.filePath + ': ' + fix.description)
+  }
+
+  return lines.join('\n')
+}
+
+function generatePRTitle(analysis: ErrorAnalysisResult): string {
+  return '[Auto-Fix] ' + analysis.summary
+}
+
+function generatePRBody(analysis: ErrorAnalysisResult, fixes: CodeFix[]): string {
+  const lines = [
+    '## Summary',
+    analysis.summary,
+    '',
+    '## Root Cause',
+    analysis.rootCause,
+    '',
+    '## Changes',
+  ]
+
+  for (const fix of fixes) {
+    lines.push('### ' + fix.filePath)
+    lines.push(fix.description)
+    lines.push('')
+  }
+
+  lines.push('---')
+  lines.push('*This PR was auto-generated by Gemini ADK.*')
+  lines.push('*Confidence: ' + (analysis.confidence * 100).toFixed(0) + '%*')
+
+  return lines.join('\n')
+}
+
+function getCommitPrefix(errorType: string): string {
+  switch (errorType) {
+    case 'type':
+      return 'fix(types)'
+    case 'lint':
+      return 'style'
+    case 'test':
+      return 'test'
+    case 'build':
+      return 'build'
+    case 'runtime':
+      return 'fix'
+    default:
+      return 'fix'
+  }
+}

package/server/adk/agents/validator.ts

@@ -0,0 +1,187 @@
+/**
+ * Validator Agent
+ *
+ * 코드 수정 결과를 검증하는 에이전트
+ */
+
+import { LlmAgent, InMemoryRunner, createPartFromText } from '@google/adk'
+import { buildTools, runTypecheckTool, runLintTool, runTestsTool } from '../tools/build-tools'
+import { loadConfig } from '../config'
+import type { CodeFix } from './fix-generator'
+
+const config = loadConfig()
+
+export interface ValidationResult {
+  passed: boolean
+  typecheck: {
+    passed: boolean
+    errors: Array<{ file: string; line: number; message: string }>
+  }
+  lint: {
+    passed: boolean
+    errors: Array<{ file: string; line: number; rule: string; message: string }>
+  }
+  tests: {
+    passed: boolean
+    failed: number
+    total: number
+    failedTests?: string[]
+  }
+  score: number
+  feedback?: string
+}
+
+export const validatorAgent = new LlmAgent({
+  name: 'validator',
+  description: '코드 수정 결과를 검증합니다.',
+  model: config.model,
+  instruction: `당신은 코드 품질 검증 전문가입니다.
+
+수정된 코드에 대해:
+1. TypeScript 타입 체크를 실행합니다
+2. ESLint 검사를 실행합니다
+3. 테스트를 실행합니다
+4. 결과를 종합하여 점수를 매깁니다
+
+검증 결과 피드백을 제공하세요.`,
+  tools: buildTools,
+})
+
+export async function validateFixes(
+  fixes: CodeFix[],
+  options?: {
+    skipTests?: boolean
+    directory?: string
+  }
+): Promise<ValidationResult> {
+  const result: ValidationResult = {
+    passed: false,
+    typecheck: { passed: false, errors: [] },
+    lint: { passed: false, errors: [] },
+    tests: { passed: false, failed: 0, total: 0 },
+    score: 0,
+  }
+
+  // TypeScript 체크
+  const typecheckResult = await runTypecheckTool.execute({
+    directory: options?.directory,
+    files: fixes.map(f => f.filePath),
+  })
+
+  result.typecheck.passed = typecheckResult.success
+  result.typecheck.errors = typecheckResult.errors || []
+
+  // ESLint 체크
+  const lintResult = await runLintTool.execute({
+    directory: options?.directory,
+    files: fixes.map(f => f.filePath),
+    fix: false,
+  })
+
+  result.lint.passed = lintResult.success
+  result.lint.errors = lintResult.errors || []
+
+  // 테스트 실행
+  if (!options?.skipTests) {
+    const testResult = await runTestsTool.execute({
+      directory: options?.directory,
+    })
+
+    result.tests.passed = testResult.success
+    result.tests.failed = testResult.failed || 0
+    result.tests.total = testResult.total || 0
+  } else {
+    result.tests.passed = true
+  }
+
+  // 점수 계산
+  let score = 0
+  if (result.typecheck.passed) score += 40
+  if (result.lint.passed) score += 30
+  if (result.tests.passed) score += 30
+
+  result.score = score
+  result.passed = score >= 70
+
+  // 피드백 생성
+  if (!result.passed) {
+    const issues: string[] = []
+    if (!result.typecheck.passed) {
+      issues.push('TypeScript: ' + result.typecheck.errors.length + ' errors')
+    }
+    if (!result.lint.passed) {
+      issues.push('ESLint: ' + result.lint.errors.length + ' errors')
+    }
+    if (!result.tests.passed) {
+      issues.push('Tests: ' + result.tests.failed + '/' + result.tests.total + ' failed')
+    }
+    result.feedback = 'Validation failed: ' + issues.join(', ')
+  }
+
+  return result
+}
+
+export async function analyzeValidationFailure(
+  validationResult: ValidationResult
+): Promise<string> {
+  const runner = new InMemoryRunner({
+    agent: validatorAgent,
+    appName: 'ZyFlowAutoFix',
+  })
+
+  const session = await runner.sessionService.getOrCreateSession({
+    appName: 'ZyFlowAutoFix',
+    userId: 'system',
+    sessionId: 'validation-analysis-' + Date.now(),
+  })
+
+  const issues: string[] = []
+
+  if (!validationResult.typecheck.passed) {
+    for (const err of validationResult.typecheck.errors.slice(0, 5)) {
+      issues.push('[TS] ' + err.file + ':' + err.line + ' - ' + err.message)
+    }
+  }
+
+  if (!validationResult.lint.passed) {
+    for (const err of validationResult.lint.errors.slice(0, 5)) {
+      issues.push('[Lint] ' + err.file + ':' + err.line + ' - ' + err.message + ' (' + err.rule + ')')
+    }
+  }
+
+  if (!validationResult.tests.passed && validationResult.tests.failedTests) {
+    for (const test of validationResult.tests.failedTests.slice(0, 3)) {
+      issues.push('[Test] ' + test)
+    }
+  }
+
+  const prompt = 'Analyze these validation failures and suggest how to fix them:\n' +
+    issues.join('\n') + '\nScore: ' + validationResult.score + '/100'
+
+  let responseText = ''
+
+  for await (const event of runner.runAsync({
+    userId: 'system',
+    sessionId: session.id,
+    newMessage: { parts: [createPartFromText(prompt)] },
+  })) {
+    if (event.content?.parts) {
+      for (const part of event.content.parts) {
+        if ('text' in part && part.text) {
+          responseText += part.text
+        }
+      }
+    }
+  }
+
+  return responseText || 'No analysis available'
+}
+
+export function shouldRetry(validationResult: ValidationResult, attemptCount: number): boolean {
+  if (attemptCount >= 3) return false
+  if (validationResult.score >= 50) return true
+  if (!validationResult.typecheck.passed && validationResult.lint.passed && validationResult.tests.passed) {
+    return true
+  }
+  return false
+}

package/server/adk/config.ts

@@ -0,0 +1,43 @@
+/**
+ * ADK Configuration
+ *
+ * Google ADK 기반 멀티 에이전트 시스템 설정
+ */
+
+export interface ADKConfig {
+  apiKey: string
+  model: string
+  maxRetries: number
+  timeout: number
+}
+
+// 기본 설정
+export const defaultConfig: ADKConfig = {
+  apiKey: process.env.GEMINI_API_KEY || '',
+  model: 'gemini-2.0-flash-exp',
+  maxRetries: 3,
+  timeout: 120000, // 2분
+}
+
+/**
+ * 설정 검증
+ */
+export function validateConfig(config: ADKConfig): void {
+  if (!config.apiKey) {
+    throw new Error('GEMINI_API_KEY is required')
+  }
+}
+
+/**
+ * 환경별 설정 로드
+ */
+export function loadConfig(): ADKConfig {
+  const config: ADKConfig = {
+    apiKey: process.env.GEMINI_API_KEY || '',
+    model: process.env.GEMINI_MODEL || defaultConfig.model,
+    maxRetries: parseInt(process.env.ADK_MAX_RETRIES || '3', 10),
+    timeout: parseInt(process.env.ADK_TIMEOUT || '120000', 10),
+  }
+
+  return config
+}

package/server/adk/index.ts

@@ -0,0 +1,69 @@
+/**
+ * ADK (Agent Development Kit) Module
+ *
+ * Gemini ADK 기반 멀티 에이전트 시스템
+ *
+ * 사용법:
+ * ```typescript
+ * import { runAutoFix, analyzeOnly } from './adk'
+ *
+ * // 전체 자동 수정 워크플로우
+ * const result = await runAutoFix(errorLog, {
+ *   repository: 'owner/repo',
+ *   autoMerge: true,
+ * })
+ *
+ * // 분석만
+ * const analysis = await analyzeOnly(errorLog)
+ * ```
+ */
+
+// Configuration
+export { loadConfig, validateConfig, defaultConfig } from './config'
+export type { ADKConfig } from './config'
+
+// Orchestrator (main entry point)
+export {
+  runAutoFix,
+  analyzeOnly,
+  generateOnly,
+  validateOnly,
+} from './orchestrator'
+export type {
+  AutoFixOptions,
+  AutoFixStep,
+  AutoFixResult,
+} from './orchestrator'
+
+// Agents
+export { errorAnalyzerAgent, analyzeError } from './agents/error-analyzer'
+export type { ErrorAnalysisResult } from './agents/error-analyzer'
+
+export { fixGeneratorAgent, generateFix, applyFixes } from './agents/fix-generator'
+export type { CodeFix, FixGenerationResult } from './agents/fix-generator'
+
+export { validatorAgent, validateFixes } from './agents/validator'
+export type { ValidationResult } from './agents/validator'
+
+export { prAgent, createAutoFixPR, waitForCIAndMerge } from './agents/pr-agent'
+export type { PRCreationResult } from './agents/pr-agent'
+
+// Tools
+export { fileTools } from './tools/file-tools'
+export { gitTools } from './tools/git-tools'
+export { githubTools } from './tools/github-tools'
+export { buildTools } from './tools/build-tools'
+
+// Utilities
+export {
+  extractErrorsFromCILog,
+  parseTypeScriptErrors,
+  parseESLintErrors,
+} from './agents/error-analyzer'
+
+export { generateDiff } from './agents/fix-generator'
+
+export {
+  analyzeValidationFailure,
+  shouldRetry,
+} from './agents/validator'