Export textEmbeddingGecko001 for googleai (#217)

firebase · May 22, 2024 · b89cb53 · b89cb53
1 parent 4feb8af
commit b89cb53
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/docs/plugins/google-genai.md b/docs/plugins/google-genai.md
@@ -61,7 +61,11 @@ configureGenkit({
 This plugin statically exports references to its supported models:
 
 ```js
-import { geminiPro, geminiProVision } from '@genkit-ai/googleai';
+import {
+  geminiPro,
+  geminiProVision,
+  textEmbeddingGecko001,
+} from '@genkit-ai/googleai';
 ```
 
 You can use these references to specify which model `generate()` uses:
@@ -72,3 +76,12 @@ const llmResponse = await generate({
   prompt: 'Tell me a joke.',
 });
 ```
+
+or use embedders (ex. `textEmbeddingGecko001`) with `embed` or retrievers:
+
+```js
+const embedding = await embed({
+  embedder: textEmbeddingGecko001,
+  content: input,
+});
+```
diff --git a/js/plugins/googleai/src/index.ts b/js/plugins/googleai/src/index.ts
@@ -17,6 +17,7 @@
 import { genkitPlugin, Plugin } from '@genkit-ai/core';
 import {
   SUPPORTED_MODELS as EMBEDDER_MODELS,
+  textEmbeddingGecko001,
   textEmbeddingGeckoEmbedder,
 } from './embedder.js';
 import {
@@ -28,7 +29,13 @@ import {
   SUPPORTED_V15_MODELS,
   SUPPORTED_V1_MODELS,
 } from './gemini.js';
-export { gemini15Flash, gemini15Pro, geminiPro, geminiProVision };
+export {
+  gemini15Flash,
+  gemini15Pro,
+  geminiPro,
+  geminiProVision,
+  textEmbeddingGecko001,
+};
 
 export interface PluginOptions {
   apiKey?: string;