fix(js/plugins/compat-oai): Add missing exports for defining compat models (#3357)

Author: ssbushi

js/plugins/compat-oai/src/index.ts

@@ -19,6 +19,31 @@ import { genkitPlugin } from 'genkit/plugin';
 import { ActionType } from 'genkit/registry';
 import { OpenAI, type ClientOptions } from 'openai';
 
+export {
+  SpeechConfigSchema,
+  TranscriptionConfigSchema,
+  compatOaiSpeechModelRef,
+  compatOaiTranscriptionModelRef,
+  defineCompatOpenAISpeechModel,
+  defineCompatOpenAITranscriptionModel,
+  type SpeechRequestBuilder,
+  type TranscriptionRequestBuilder,
+} from './audio.js';
+export { defineCompatOpenAIEmbedder } from './embedder.js';
+export {
+  ImageGenerationCommonConfigSchema,
+  compatOaiImageModelRef,
+  defineCompatOpenAIImageModel,
+  type ImageRequestBuilder,
+} from './image.js';
+export {
+  ChatCompletionCommonConfigSchema,
+  compatOaiModelRef,
+  defineCompatOpenAIModel,
+  openAIModelRunner,
+  type ModelRequestBuilder,
+} from './model.js';
+
 export interface PluginOptions extends Partial<ClientOptions> {
   name: string;
   initializer?: (ai: Genkit, client: OpenAI) => Promise<void>;
@@ -98,13 +123,13 @@ export const openAICompatible = (options: PluginOptions) => {
         await options.resolver(ai, client, actionType, actionName);
       }
     },
-    async () => {
-      if (options.listActions) {
-        if (listActionsCache) return listActionsCache;
-        listActionsCache = await options.listActions(client);
-        return listActionsCache;
-      }
-    }
+    options.listActions
+      ? async () => {
+          if (listActionsCache) return listActionsCache;
+          listActionsCache = await options.listActions!(client);
+          return listActionsCache;
+        }
+      : undefined
   );
 };
 

js/testapps/compat-oai/src/index.ts

@@ -15,6 +15,11 @@
  * limitations under the License.
  */
 
+import {
+  compatOaiModelRef,
+  defineCompatOpenAIModel,
+  openAICompatible,
+} from '@genkit-ai/compat-oai';
 import { deepSeek } from '@genkit-ai/compat-oai/deepseek';
 import { openAI } from '@genkit-ai/compat-oai/openai';
 import { xAI } from '@genkit-ai/compat-oai/xai';
@@ -24,8 +29,31 @@ import { genkit, z } from 'genkit';
 
 dotenv.config();
 
+const DECLARED_MODELS = ['gemini-2.0-flash'];
+
 const ai = genkit({
-  plugins: [openAI(), deepSeek(), xAI()],
+  plugins: [
+    openAI(),
+    deepSeek(),
+    xAI(),
+    openAICompatible({
+      name: 'goog-oai',
+      baseURL: 'https://generativelanguage.googleapis.com/v1beta/openai/',
+      apiKey: process.env['GEMINI_API_KEY'],
+      initializer: async (ai, client) => {
+        for (const model of DECLARED_MODELS) {
+          defineCompatOpenAIModel({
+            ai,
+            name: `goog-oai/${model}`,
+            client,
+            modelRef: compatOaiModelRef({
+              name: 'goog-oai/gemini-2.0-flash',
+            }),
+          });
+        }
+      },
+    }),
+  ],
 });
 
 export const jokeFlow = ai.defineFlow(
@@ -37,9 +65,8 @@ export const jokeFlow = ai.defineFlow(
   async (subject) => {
     const llmResponse = await ai.generate({
       prompt: `tell me a joke about ${subject}`,
-      model: xAI.model('grok-3-mini', {
-        temperature: 0.5,
-        reasoningEffort: 'low',
+      model: compatOaiModelRef({
+        name: 'goog-oai/gemini-2.0-flash',
       }),
     });
     return llmResponse.text;
@@ -86,8 +113,6 @@ export const webSearchFlow = ai.defineFlow(
   }
 );
 
-//  genkit flow:run embedFlow \"hello world\"
-
 export const embedFlow = ai.defineFlow(
   {
     name: 'embedFlow',