【AI ブロガー開発】Claude API を使った記事生成機能の実装

この記事はで読むことができます。

目次非表示

はじめに
Claude API クライアントの実装
記事生成ロジックの実装
main.go の更新
実行と動作確認
エラーハンドリングとログ記録
パフォーマンスの最適化
まとめ
次のステップ

はじめに

前回の記事では、AI ブロガー開発プロジェクトの概要と環境構築について解説しました。今回は、このプロジェクトの核心部分である、Claude API を使った記事生成機能の実装に踏み込んでいきます。

2024-07-01 【AI ブロガー開発】プロジェクト概要と環境構築

この記事を通じて、以下のスキルと知識を習得することができます：

Golang での HTTP クライアントの実装
Claude API の利用方法
非同期処理と error handling
構造化されたコードの書き方

それでは、実装に入っていきましょう。

Claude API クライアントの実装

まず、Claude API とやり取りするためのクライアントを実装します。internal/claude/client.go ファイルを以下のように作成します：

package claude

import (
    "bytes"
    "encoding/json"
    "fmt"
    "net/http"
    "io/ioutil"
)

type Client struct {
    APIKey  string
    BaseURL string
}

type APIResponse struct {
    Completion string `json:"completion"`
    StopReason string `json:"stop_reason"`
    Model      string `json:"model"`
}

func NewClient(apiKey string) *Client {
    return &Client{
        APIKey:  apiKey,
        BaseURL: "https://api.anthropic.com/v1/complete",
    }
}

func (c *Client) GenerateContent(prompt string) (string, error) {
    requestBody, err := json.Marshal(map[string]interface{}{
        "model": "claude-v1",
        "prompt": prompt,
        "max_tokens_to_sample": 1000,
        "temperature": 0.7,
    })
    if err != nil {
        return "", err
    }

    req, err := http.NewRequest("POST", c.BaseURL, bytes.NewBuffer(requestBody))
    if err != nil {
        return "", err
    }

    req.Header.Set("Content-Type", "application/json")
    req.Header.Set("X-API-Key", c.APIKey)

    client := &http.Client{}
    resp, err := client.Do(req)
    if err != nil {
        return "", err
    }
    defer resp.Body.Close()

    body, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        return "", err
    }

    if resp.StatusCode != http.StatusOK {
        return "", fmt.Errorf("API request failed with status code %d: %s", resp.StatusCode, string(body))
    }

    var apiResp APIResponse
    if err := json.Unmarshal(body, &apiResp); err != nil {
        return "", err
    }

    if apiResp.Completion == "" {
        return "", fmt.Errorf("no completion in API response")
    }

    return apiResp.Completion, nil
}

このコードでは、以下の主要な機能を実装しています：

Claude API クライアントの構造体定義
API レスポンスの構造体定義
クライアントの初期化関数
コンテンツ生成関数（HTTP リクエストの送信と応答の処理）

記事生成ロジックの実装

次に、記事生成のロジックを実装します。internal/content/generator.go ファイルを以下のように作成します：

package content

import (
    "fmt"
    "strings"

    "github.com/yourusername/ai-blogger-golang/internal/claude"
    "github.com/yourusername/ai-blogger-golang/internal/models"
)

type Generator struct {
    claudeClient *claude.Client
}

func NewGenerator(claudeClient *claude.Client) *Generator {
    return &Generator{
        claudeClient: claudeClient,
    }
}

func (g *Generator) GenerateArticle(keyword string) (*models.Article, error) {
    prompt := `以下の指示に従って、指定されたキーワードに関する記事を作成してください：

1. キーワード：%s


以上の指示に基づいて、%sに関する記事を作成してください。`

    formattedPrompt := fmt.Sprintf(prompt, keyword, keyword)

    content, err := g.claudeClient.GenerateContent(formattedPrompt)
    if err != nil {
        return nil, err
    }

    title := extractTitle(content)

    return &models.Article{
        Title:   title,
        Content: content,
        Keyword: keyword,
    }, nil
}

func extractTitle(content string) string {
    lines := strings.Split(content, "\n")
    for _, line := range lines {
        if strings.TrimSpace(line) != "" {
            return strings.TrimSpace(line)
        }
    }
    return "無題"
}

このコードでは、以下の主要な機能を実装しています：

記事生成器の構造体定義
記事生成関数（プロンプトの構築、Claude API の呼び出し、結果の処理）
タイトル抽出関数

main.go の更新

最後に、cmd/main.go ファイルを更新して、これらの新しい機能を利用できるようにします：

package main

import (
    "fmt"
    "log"

    "github.com/yourusername/ai-blogger-golang/config"
    "github.com/yourusername/ai-blogger-golang/internal/claude"
    "github.com/yourusername/ai-blogger-golang/internal/content"
)

func main() {
    cfg, err := config.Load()
    if err != nil {
        log.Fatalf("Error loading config: %v", err)
    }

    claudeClient := claude.NewClient(cfg.ClaudeAPIKey)
    generator := content.NewGenerator(claudeClient)

    keyword := "人工知能" // この部分は後でユーザー入力に変更できます

    article, err := generator.GenerateArticle(keyword)
    if err != nil {
        log.Fatalf("Error generating article: %v", err)
    }

    fmt.Printf("Generated Article:\nTitle: %s\nContent: %s\n", article.Title, article.Content)
}

実行と動作確認

これで、基本的な実装は完了です。以下のコマンドでプログラムを実行し、動作を確認できます：

bash

go run cmd/main.go

正常に動作すれば、指定されたキーワードに基づいて生成された記事のタイトルと内容が表示されるはずです。

エラーハンドリングとログ記録

実際の運用では、エラーハンドリングとログ記録を適切に行うことが重要です。以下のような改善を検討してください：

エラーの詳細をログに記録する
リトライメカニズムの実装（API呼び出しが失敗した場合）
タイムアウト設定の追加

例えば、internal/claude/client.go の GenerateContent 関数に以下のようなログ記録を追加できます：

func (c *Client) GenerateContent(prompt string) (string, error) {
    // ... (既存のコード)

    if resp.StatusCode != http.StatusOK {
        log.Printf("API request failed with status code %d: %s", resp.StatusCode, string(body))
        return "", fmt.Errorf("API request failed with status code %d", resp.StatusCode)
    }

    // ... (残りのコード)
}