go-ai-client

README

go-ai-client

🎯 概要

go-ai-clientは、Go言語でGenerative AI（特にGoogle Gemini API）を簡単に利用するためのクライアントライブラリ、およびテンプレートベースのプロンプト生成ユーティリティを提供します。

✨ 特徴

• Gemini API クライアント: Google Gemini APIとの基本的なやり取りを行うためのシンプルで使いやすいクライアントを提供します。責務がAPI通信とリトライ処理に限定され、プロンプトの内容に依存しません。
• リトライ戦略: ネットワークエラーや一時的なAPIエラーに対応するため、自動リトライロジック（go-web-exact/pkg/retryを使用）を内蔵しています。
• 柔軟なプロンプトモード: 特定タスク向けのテンプレートベースのプロンプト構築機能（pkg/prompt）と、自由なテキストをそのまま送る汎用モードをサポートします。プロンプト構築のロジックはCLIツール側（cmd）に存在します。

---

🚀 インストール

Goモジュールとしてプロジェクトに追加します。

go get https://github.com/shouni/go-ai-client

🗝️ APIキーの設定

本ライブラリは、環境変数からGoogle Gemini APIキーを読み込むことを想定しています。

GEMINI_API_KEY または GOOGLE_API_KEY にAPIキーを設定してください。

export GEMINI_API_KEY="YOUR_API_KEY"

---

💡 使用方法

このライブラリのコア機能は、CLIツール (cmd/root.go) を通じて提供されています。

1. CLIツールとしての使用 (推奨)

CLIツールとしてビルドし、コマンドラインで直接使用します。

# ビルド
go build -o bin/ai-client

# 実行例 1: テンプレートを使用 (prompt コマンド)
# soloモードでナレーションスクリプトを生成
./bin/ai-client prompt "地球温暖化の主要な原因とその対策について、簡潔に説明してください。" -d solo 

# 実行例 2: テンプレートを使用 (dialogue コマンド)
# dialogueモードで対話スクリプトを生成（モデル名も指定）
cat input.txt | ./bin/ai-client prompt -d dialogue -m gemini-2.5-flash

# 実行例 3: テンプレートを使用しない (generic コマンド)
# 自由なテキストをそのままプロンプトとして送信
./bin/ai-client generic "AI技術が社会に与えるポジティブな影響と、リスクについて議論してください。"

# 主要なフラグ:
# -d, --mode (promptコマンド専用): 生成するスクリプトのモード (solo, dialogue)
# -m, --model: 使用するGeminiモデル名
# -t, --timeout: 全体のリクエストタイムアウト時間 (秒)

2. Goコード内でのクライアント使用

クライアントは最終的なプロンプト文字列を受け取ります。テンプレートを使用する場合は、呼び出し元で prompt.BuildFullPrompt を使ってプロンプトを構築する必要があります。

【テンプレートを使用する/しないにかかわらず共通】

クライアントには、テンプレート処理を経た最終的なプロンプト文字列を渡します。

package main

import (
    "context"
    "fmt"
    "log"
    "time"

    "go-ai-client/pkg/ai/gemini"
    "go-ai-client/pkg/prompt" // ★ テンプレートを使う場合はインポートが必要
)

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
    defer cancel()

    client, err := gemini.NewClientFromEnv(ctx)
    if err != nil {
       log.Fatalf("クライアント初期化エラー: %v", err)
    }
    
    // ----------------------------------------------------------------
    // 1. テンプレートを使用する場合 (例: solo モード)
    // ----------------------------------------------------------------
    // BuildFullPrompt のシグネチャ変更に伴い、string型で定義
    rawInput := "Go言語でAPIクライアントを作成する利点について教えてください。" 
    mode := "solo" 
    
    // ★ 呼び出し元で finalPrompt を構築 (stringを渡す)
    finalPrompt, err := prompt.BuildFullPrompt(rawInput, mode) 
    if err != nil {
        log.Fatalf("プロンプト構築エラー: %v", err)
    }
    
    model := "gemini-2.5-flash"
    
    // クライアントは最終プロンプト文字列のみを受け取る
    response, err := client.GenerateContent(ctx, finalPrompt, model)
    if err != nil {
       log.Fatalf("コンテンツ生成エラー: %v", err)
    }

    fmt.Println("--- 応答 ---")
    fmt.Println(response.Text)
    
    // ----------------------------------------------------------------
    // 2. テンプレートを使用しない場合
    // ----------------------------------------------------------------
    // client.GenerateContent(ctx, "自由なテキストプロンプト", "モデル名") で直接実行可能
}

---

📂 プロジェクト構造（pkgディレクトリ）

ディレクトリ/ファイル	概要
pkg/ai/gemini	Google Gemini API専用のクライアント実装。
pkg/ai/gemini/client.go	Gemini APIクライアントのコアロジック。API通信、リトライ、レスポンス処理のみを担う。
pkg/prompt	プロンプトのテンプレートおよび構築ロジック。
pkg/prompt/prompt.go	テンプレートベースのプロンプト定義と、入力内容を埋め込む BuildFullPrompt 関数を提供。テンプレートは事前解析され、キャッシュされる。

---

📜 ライセンス (License)

このプロジェクトは MIT License の下で公開されています。