Erste Schritte mit der Gemini API in Go-Anwendungen

In dieser Anleitung wird gezeigt, wie Sie mit dem Google AI Go SDK auf die Gemini API für Ihre Go-Anwendung zugreifen.

In dieser Anleitung erfahren Sie, wie Sie Folgendes tun:

Darüber hinaus enthält diese Anleitung Abschnitte zu erweiterten Anwendungsfällen (z. B. Einbettungen und Zählmethoden) sowie Optionen zur Steuerung der Inhaltsgenerierung.

Voraussetzungen

In dieser Anleitung wird davon ausgegangen, dass Sie mit dem Erstellen von Anwendungen mit Go vertraut sind.

Achten Sie beim Durcharbeiten dieser Anleitung darauf, dass Ihre Entwicklungsumgebung die folgenden Anforderungen erfüllt:

  • Go 1.20 oder höher

Projekt einrichten

Bevor Sie die Gemini API aufrufen, müssen Sie Ihr Projekt einrichten. Dazu müssen Sie Ihren API-Schlüssel einrichten, das SDK-Paket installieren und das Modell initialisieren.

API-Schlüssel einrichten

Sie benötigen einen API-Schlüssel, um die Gemini API zu verwenden. Erstellen Sie einen Schlüssel in Google AI Studio, falls noch nicht geschehen.

API-Schlüssel anfordern

API-Schlüssel sichern

Wir empfehlen dringend, einen API-Schlüssel nicht in Ihr Versionsverwaltungssystem einzuchecken. Verwenden Sie stattdessen einen Secrets-Speicher für Ihren API-Schlüssel.

Bei allen Snippets in dieser Anleitung wird davon ausgegangen, dass Sie auf Ihren API-Schlüssel als Umgebungsvariable zugreifen.

SDK-Paket installieren

Wenn Sie die Gemini API in Ihrer eigenen Anwendung verwenden möchten, müssen Sie das Go SDK-Paket in Ihrem Modulverzeichnis mit einem get-Befehl versehen:

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

Generatives Modell initialisieren

Bevor Sie API-Aufrufe ausführen können, müssen Sie das generative Modell importieren und initialisieren.

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

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

model := client.GenerativeModel("MODEL_NAME")

Beachten Sie bei der Angabe eines Modells Folgendes:

  • Verwenden Sie ein Modell, das für Ihren Anwendungsfall spezifisch ist (z. B. gemini-pro-vision ist für die multimodale Eingabe vorgesehen). In diesem Leitfaden ist in den Anleitungen für jede Implementierung das empfohlene Modell für den jeweiligen Anwendungsfall aufgeführt.

Gängige Anwendungsfälle implementieren

Nachdem Ihr Projekt nun eingerichtet ist, können Sie die Verwendung der Gemini API zum Implementieren verschiedener Anwendungsfälle kennenlernen:

Im Abschnitt „Erweiterte Anwendungsfälle“ finden Sie Informationen zur Gemini API und zu Einbettungen.

Text aus reiner Texteingabe generieren

Wenn die Eingabe nur Text enthält, verwenden Sie das Modell gemini-pro mit der Methode GenerateContent, um eine Textausgabe zu generieren:

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

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

Text aus Text- und Bildeingabe generieren (multimodal)

Gemini bietet ein multimodales Modell (gemini-pro-vision), sodass Sie sowohl Text als auch Bilder eingeben können. Sieh dir die Anforderungen an Bilder für Prompts an.

Wenn die Eingabe der Aufforderung sowohl Text als auch Bilder enthält, verwenden Sie das Modell gemini-pro-vision mit der Methode GenerateContent, um die Textausgabe zu generieren:

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

// For text-and-image input (multimodal), use the gemini-pro-vision model
model := client.GenerativeModel("gemini-pro-vision")

imgData1, err := os.ReadFile(pathToImage1)
if err != nil {
  log.Fatal(err)
}

imgData2, err := os.ReadFile(pathToImage1)
if err != nil {
  log.Fatal(err)
}

prompt := []genai.Part{
  genai.ImageData("jpeg", imgData1),
  genai.ImageData("jpeg", imgData2),
  genai.Text("What's different between these two pictures?"),
}
resp, err := model.GenerateContent(ctx, prompt...)

if err != nil {
  log.Fatal(err)
}

Multi-Turn Conversations (Chat)

Mit Gemini können Sie frei formulierte Unterhaltungen über mehrere Runden hinweg erstellen. Das SDK vereinfacht den Prozess, indem es den Status der Unterhaltung verwaltet. Im Gegensatz zu GenerateContent müssen Sie den Unterhaltungsverlauf also nicht selbst speichern.

Verwenden Sie das Modell gemini-pro und initialisieren Sie den Chat, indem Sie StartChat() aufrufen, um eine Unterhaltung mit mehreren Themen (z. B. Chat) zu erstellen. Verwenden Sie dann SendMessage(), um eine neue Nutzernachricht zu senden. Dadurch werden auch die Nachricht und die Antwort an das Chatprotokoll angehängt.

Es gibt zwei mögliche Optionen für role, die mit dem Inhalt einer Unterhaltung verknüpft sind:

  • user: die Rolle, die die Prompts bereitstellt. Dieser Wert ist der Standardwert für SendMessage-Aufrufe.

  • model: die Rolle, die die Antworten bereitstellt. Diese Rolle kann verwendet werden, wenn StartChat() mit vorhandenem history aufgerufen wird.

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

// For text-only input, use the gemini-pro model
model := client.GenerativeModel("gemini-pro")
// Initialize the chat
cs := model.StartChat()
cs.History = []*genai.Content{
  &genai.Content{
    Parts: []genai.Part{
      genai.Text("Hello, I have 2 dogs in my house."),
    },
    Role: "user",
  },
  &genai.Content{
    Parts: []genai.Part{
      genai.Text("Great to meet you. What would you like to know?"),
    },
    Role: "model",
  },
}

resp, err := cs.SendMessage(ctx, genai.Text("How many paws are in my house?"))
if err != nil {
  log.Fatal(err)
}

Streaming für schnellere Interaktionen

Standardmäßig gibt das Modell eine Antwort zurück, nachdem der gesamte Generierungsprozess abgeschlossen ist. Sie können Interaktionen beschleunigen, indem Sie nicht auf das gesamte Ergebnis warten und stattdessen Streaming verwenden, um Teilergebnisse zu verarbeiten.

Das folgende Beispiel zeigt, wie Sie Streaming mit der Methode GenerateContentStream implementieren, um Text aus einer Eingabeaufforderung für Text und Bilder zu generieren.

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

// For text-and-image input (multimodal), use the gemini-pro-vision model
model := client.GenerativeModel("gemini-pro-vision")

imageBytes, err := os.ReadFile(pathToImage)

img := genai.ImageData("jpeg", imageBytes)
prompt := genai.Text("Tell me a story about this animal")
iter := model.GenerateContentStream(ctx, img, prompt)

for {
  resp, err := iter.Next()
  if err == iterator.Done {
    break
  }
  if err != nil {
    log.Fatal(err)
  }

  // ... print resp
}

Ein ähnlicher Ansatz kann für reine Texteingaben und Chats verwendet werden.

prompt := genai.Text("Tell me a story about a lumberjack and his giant ox")
iter := model.GenerateContentStream(ctx, prompt)

prompt := genai.Text("And how do you feel about that?")
iter := cs.SendMessageStream(ctx, prompt)

Erweiterte Anwendungsfälle implementieren

Die im vorherigen Abschnitt dieser Anleitung beschriebenen häufigen Anwendungsfälle helfen Ihnen, sich mit der Verwendung der Gemini API vertraut zu machen. In diesem Abschnitt werden einige Anwendungsfälle beschrieben, die als komplexer gelten.

Einbettungen verwenden

Das Einbetten ist eine Methode, um Informationen als Liste von Gleitkommazahlen in einem Array darzustellen. Mit Gemini können Sie Text (Wörter, Sätze und Textblöcke) in einer vektorisierten Form darstellen, was das Vergleichen und Gegenüberstellen von Einbettungen erleichtert. Zum Beispiel sollten zwei Texte, die ein ähnliches Thema oder eine ähnliche Stimmung haben, ähnliche Einbettungen haben, die durch mathematische Vergleichstechniken wie Kosinus-Ähnlichkeit identifiziert werden können.

Verwenden Sie das Modell embedding-001 mit der Methode EmbedContent (oder der Methode BatchEmbedContent), um Einbettungen zu generieren. Im folgenden Beispiel wird eine Einbettung für einen einzelnen String generiert:

ctx := context.Background()
// Access your API key as an environment variable (see "Set up your API key" above)
client, err := genai.NewClient(ctx, option.WithAPIKey(os.Getenv("API_KEY")))
if err != nil {
  log.Fatal(err)
}
defer client.Close()
em := client.EmbeddingModel("embedding-001")
res, err := em.EmbedContent(ctx, genai.Text("The quick brown fox jumps over the lazy dog."))

if err != nil {
  panic(err)
}
fmt.Println(res.Embedding.Values)

Anzahl der Tokens

Bei der Verwendung langer Aufforderungen kann es hilfreich sein, Tokens zu zählen, bevor Inhalte an das Modell gesendet werden. Die folgenden Beispiele zeigen, wie CountTokens() für verschiedene Anwendungsfälle verwendet wird:

// For text-only input
text := "Parrots can be green and live a long time."
resp, err := model.CountTokens(ctx, genai.Text(text))
if err != nil {
  log.Fatal(err)
}
fmt.Println(resp.TotalTokens)

// For text-and-image input (multimodal)
text := "Parrots can be green and live a long time."
imageBytes, err := os.ReadFile(pathToImage)
if err != nil {
  log.Fatal(err)
}

resp, err := model.CountTokens(
    ctx,
    genai.Text(text),
    genai.ImageData("png", imageBytes))
  if err != nil {
    log.Fatal(err)
}
fmt.Println(resp.TotalTokens)

Optionen zum Steuern der Inhaltsgenerierung

Sie können die Inhaltsgenerierung steuern, indem Sie Modellparameter konfigurieren und Sicherheitseinstellungen verwenden.

Modellparameter konfigurieren

Jede Aufforderung, die Sie an das Modell senden, enthält Parameterwerte, die steuern, wie das Modell eine Antwort generiert. Das Modell kann für verschiedene Parameterwerte unterschiedliche Ergebnisse generieren. Weitere Informationen zu Modellparametern Die Konfiguration wird für die Lebensdauer der Modellinstanz beibehalten.

// ...

model := client.GenerativeModel("MODEL_NAME")

// Configure model parameters by invoking Set* methods on the model.
model.SetTemperature(0.9)
model.SetTopK(1)

// ...

Sicherheitseinstellungen verwenden

In den Sicherheitseinstellungen kannst du die Wahrscheinlichkeit anpassen, dass du Antworten erhältst, die als schädlich eingestuft werden könnten. Inhalte mit mittlerer und/oder hoher Wahrscheinlichkeit werden standardmäßig durch Sicherheitseinstellungen in allen Dimensionen blockiert. Weitere Informationen zu Sicherheitseinstellungen

So legen Sie eine Sicherheitseinstellung fest:

// ...

model := client.GenerativeModel("MODEL_NAME")

model.SafetySettings = []*genai.SafetySetting{
  {
    Category:  genai.HarmCategoryHarassment,
    Threshold: genai.HarmBlockOnlyHigh,
  },
}

// ...

Sie können auch mehrere Sicherheitseinstellungen festlegen:

// ...

model := client.GenerativeModel("MODEL_NAME")

model.SafetySettings = []*genai.SafetySetting{
  {
    Category:  genai.HarmCategoryHarassment,
    Threshold: genai.HarmBlockOnlyHigh,
  },
  {
    Category:  genai.HarmCategoryHateSpeech,
    Threshold: genai.HarmBlockMediumAndAbove,
  },
}

// ...

Nächste Schritte

  • Beim Prompt-Design werden Aufforderungen erstellt, die die gewünschte Antwort aus Sprachmodellen auslösen. Gut strukturierte Eingabeaufforderungen sind wichtig, um genaue, hochwertige Antworten aus einem Sprachmodell zu gewährleisten. Best Practices für das Schreiben von Prompts

  • Gemini bietet verschiedene Modellvariationen, um den Anforderungen verschiedener Anwendungsfälle gerecht zu werden, z. B. Eingabetypen und -komplexität, Implementierungen für Chat- oder andere Dialogsprachaufgaben sowie Größenbeschränkungen. Weitere Informationen zu den verfügbaren Gemini-Modellen

  • Gemini bietet Optionen zum Anfordern von Erhöhungen der Ratenbegrenzung. Das Ratenlimit für Gemini Pro-Modelle beträgt 60 Anfragen pro Minute (RPM).