【Gemini Pro API】ChatGPTより80倍安い！？ ・Go言語での実装方法

go get github.com/google/generative-ai-go

package clocky_be

import (
	"context"
	"fmt"
	"log"

	"github.com/google/generative-ai-go/genai"
	"google.golang.org/api/option"
)

func GenerateText(config *Config) {
	log.Println("GenerateText function called")
	ctx := context.Background()

	GeminiAPIKey := config.GeminiAPIKey
	// Access your API key as an environment variable (see "Set up your API key" above)
	client, err := genai.NewClient(ctx, option.WithAPIKey(GeminiAPIKey))
	if err != nil {
		log.Fatal(err)
	}
	defer client.Close()

	// For text-only input, use the gemini-pro model
	model := client.GenerativeModel("gemini-pro")
	prompt := genai.Text("Write a story about a magic backpack.")
	resp, err := model.GenerateContent(ctx, prompt)
	if err != nil {
		log.Fatal(err)
	}
	printResponse(resp)
}

func GenerateText(config *Config) {
	log.Println("GenerateText function called")
	ctx := context.Background()

	GeminiAPIKey := config.GeminiAPIKey
	// Access your API key as an environment variable (see "Set up your API key" above)
	client, err := genai.NewClient(ctx, option.WithAPIKey(GeminiAPIKey))
	if err != nil {
		log.Fatal(err)
	}
	defer client.Close()

	// For text-only input, use the gemini-pro model
	model := client.GenerativeModel("gemini-pro")
	prompt := genai.Text("今日の天気を教えて")
	resp, err := model.GenerateContent(ctx, prompt)
	if err != nil {
		log.Fatal(err)
	}
	respJSON, err := json.MarshalIndent(resp, "", "  ")
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println(string(respJSON))
}

{
  "Candidates": [
    {
      "Index": 0,
      "Content": {
        "Parts": [
          "申し訳ございませんが、私はリアルタイムの情報を提供することができません。私は2023年2月までに得た知識のみを基に回答を生成しています。最新の天気予報については、気象庁のウェブサイトやテレビ、ラジオなどの天気予報番組をご覧いただくことをお勧めします。"
        ],
        "Role": "model"
      },
      "FinishReason": 1,
      "SafetyRatings": [
        {
          "Category": 9,
          "Probability": 1,
          "Blocked": false
        },
        {
          "Category": 8,
          "Probability": 1,
          "Blocked": false
        },
        {
          "Category": 7,
          "Probability": 1,
          "Blocked": false
        },
        {
          "Category": 10,
          "Probability": 1,
          "Blocked": false
        }
      ],
      "CitationMetadata": null,
      "TokenCount": 0
    }
  ],
  "PromptFeedback": {
    "BlockReason": 0,
    "SafetyRatings": [
      {
        "Category": 9,
        "Probability": 1,
        "Blocked": false
      },
      {
        "Category": 8,
        "Probability": 1,
        "Blocked": false
      },
      {
        "Category": 7,
        "Probability": 1,
        "Blocked": false
      },
      {
        "Category": 10,
        "Probability": 1,
        "Blocked": false
      }
    ]
  }
}

func printResponse(resp *genai.GenerateContentResponse) {
    for _, candidate := range resp.Candidates {
        // Content が nil でないことを確認
        if candidate.Content != nil {
            // Parts に含まれる各テキスト部分をループで処理
            for _, part := range candidate.Content.Parts {
                fmt.Println(part)
            }
        }
    }
}

package main

func main() {
	config := clocky.NewConfig()
	clocky.GenerateText(config)
}