Wtyczka Google Generative AI

Wtyczka Google Generative AI zapewnia interfejsy modeli Gemini od Google za pomocą Gemini API.

Instalacja

npm i --save @genkit-ai/googleai

Konfiguracja

Aby używać tej wtyczki, określ ją, gdy wywołujesz configureGenkit():

import { googleAI } from '@genkit-ai/googleai';

export default configureGenkit({
  plugins: [googleAI()],
  // ...
});

Wtyczka wymaga klucza interfejsu API Gemini, który możesz uzyskać w Google AI Studio.

Skonfiguruj wtyczkę do korzystania z klucza interfejsu API, wykonując jedną z tych czynności:

• Ustaw zmienną środowiskową GOOGLE_GENAI_API_KEY na swój klucz interfejsu API.

• Podaj klucz interfejsu API podczas inicjowania wtyczki:

  googleAI({ apiKey: yourKey });
  

  Nie umieszczaj jednak klucza interfejsu API bezpośrednio w kodzie. Tej funkcji należy używać tylko w połączeniu z usługą taką jak Cloud Secret Manager lub podobną.

Niektóre modele (np. Gemini 1.5 Pro) są w wersji testowej i są dostępne tylko przez interfejs API v1beta. Możesz określić apiVersion, aby uzyskać dostęp do tych modeli:

configureGenkit({
  plugins: [googleAI({ apiVersion: 'v1beta' })],
});

Możesz też określić wiele wersji, jeśli chcesz używać różnych wersji modeli jednocześnie.

configureGenkit({
  plugins: [googleAI({ apiVersion: ['v1', 'v1beta'] })],
});

Wykorzystanie

Ta wtyczka statycznie eksportuje odwołania do obsługiwanych modeli:

import {
  geminiPro,
  geminiProVision,
  textEmbeddingGecko001,
} from '@genkit-ai/googleai';

Korzystając z tych odwołań, możesz określić, którego modelu używa generate():

const llmResponse = await generate({
  model: geminiPro,
  prompt: 'Tell me a joke.',
});

lub skorzystaj z elementów umieszczanych (np. textEmbeddingGecko001) z użytkownikiem embed lub retrieverami:

const embedding = await embed({
  embedder: textEmbeddingGecko001,
  content: input,
});