actions/ai-inference を使って GitHub Actions のワークフローから AI モデルを呼び出す

name: 'AI inference'
# 手動でワークフローをトリガーするための設定
on: 
  workflow_dispatch:
  # ワークフローの入力としてプロンプトを受け取る
  # これは GitHub Actions の UI から入力できる
    inputs:
      prompt:
        description: 'Prompt for the AI model'
        required: true
 
jobs:
  inference:
    # GitHub Models を読み取るための権限を付与
    permissions:
      models: read
    runs-on: ubuntu-latest
    steps:
      - name: Test Local Action
        id: inference
        # actions/ai-inference アクションを使用して AI モデルを呼び出す
        uses: actions/ai-inference@v1
        with:
          # プロンプトには GitHub Actions の入力を指定
          prompt: "${{ github.event.inputs.prompt }}"
          # GitHub Models で使用可能な AI モデルを指定
          # デフォルトは gpt-4o
          model: "gpt-4o"
 
      - name: Print Output
        id: output
        # AI モデルの出力を表示
        run: echo "${{ steps.inference.outputs.response }}"

name: 'Review with AI'
on: 
  pull_request:
    paths:
      # 記事を保存しているディレクトリが変更されたときのみワークフローを実行する
      - contents/blogPost/**/*.md
 
jobs:
  review:
    permissions:
      models: read
      # ファイルをプロンプトとして渡すために必要な権限
      contents: read
      pull-requests: write
    runs-on: ubuntu-latest
    steps:
      - name: Checkout
        uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
        with:
          fetch-depth: 0
 
      # 変更されたファイルを取得する
      - name: Get changed files in the contents folder
        id: changed-files-specific
        uses: tj-actions/changed-files@4cd184a1dd542b79cca1d4d7938e4154a6520ca7 # v46.0.0
        with:
          files: |
            contents/**
 
      - name: Test Local Action
        id: inference
        uses: actions/ai-inference@f8ee4c952b7dca7b8a4529edd04dc5cc3d49c435 # v1.0.0
        if: steps.changed-files-specific.outputs.all_changed_files != ''
        env:
          ALL_CHANGED_FILES: ${{ steps.changed-files-specific.outputs.all_changed_files }}
        with:
          system-prompt: |
            あなたはプロフェッショナルなテックブログ・レビューアー AIです。
            あなたの役割は、開発者や技術者が書いたブログ記事を技術的正確性・表現の明確さ・構成の論理性・読者への価値提供という観点でレビューし、フィードバックを提供することです。
          # 変更されたファイルの内容をプロンプトとして渡す
          prompt-file: "${{ env.ALL_CHANGED_FILES }}"
 
      - name: Comment on PR
        id: comment
        if: steps.changed-files-specific.outputs.all_changed_files != ''
        env:
          RESPONSE: ${{ steps.inference.outputs.response }}
          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
        # gh CLI を使用してプルリクエストにコメントを投稿する
        run: |
          echo "${{ env.RESPONSE }}" > comment.txt
          gh pr comment ${{ github.event.pull_request.number }} --body-file comment.txt
          echo "Commented on PR #${{ github.event.pull_request.number }} with response: ${{ env.RESPONSE }}"