feat(genkit-tools): add /listValues endpoint to expose defaultModel

genkit-tools/common/src/manager/manager.ts

@@ -198,6 +198,32 @@ export class RuntimeManager {
     return response.data as Record<string, Action>;
   }
 
+  /**
+   * Retrieves all valuess.
+   */
+  async listValues(
+    input: apis.ListValuesRequest
+  ): Promise<Record<string, unknown>> {
+    const runtime = input.runtimeId
+      ? this.getRuntimeById(input.runtimeId)
+      : this.getMostRecentRuntime();
+    if (!runtime) {
+      throw new Error(
+        input?.runtimeId
+          ? `No runtime found with ID ${input.runtimeId}.`
+          : 'No runtimes found. Make sure your app is running using `genkit start -- ...`. See getting started documentation.'
+      );
+    }
+    const response = await axios
+      .get(`${runtime.reflectionServerUrl}/api/values`, {
+        params: {
+          type: input.type,
+        },
+      })
+      .catch((err) => this.httpErrorHandler(err, 'Error listing values.'));
+    return response.data as Record<string, unknown>;
+  }
+
   /**
    * Runs an action.
    */

genkit-tools/common/src/server/router.ts

@@ -134,6 +134,13 @@ export const TOOLS_SERVER_ROUTER = (manager: RuntimeManager) =>
         return manager.listActions(input);
       }),
 
+    /** Retrieves all values. */
+    listValues: loggedProcedure
+      .input(apis.ListValuesRequestSchema)
+      .query(async ({ input }): Promise<Record<string, unknown>> => {
+        return manager.listValues(input);
+      }),
+
     /** Generate a .prompt file from messages and model config. */
     createPrompt: loggedProcedure
       .input(apis.CreatePromptRequestSchema)

genkit-tools/common/src/types/apis.ts

@@ -80,6 +80,22 @@ export const ListActionsRequestSchema = z
 
 export type ListActionsRequest = z.infer<typeof ListActionsRequestSchema>;
 
+export const ListValuesRequestSchema = z.object({
+  runtimeId: z
+    .string()
+    .optional()
+    .describe(
+      'ID of the Genkit runtime to run the action on. Typically $pid-$port.'
+    ),
+  type: z
+    .enum(['defaultModel'])
+    .describe(
+      "The type of values to fetch. Currently only supports 'defaultModel'"
+    ),
+});
+
+export type ListValuesRequest = z.infer<typeof ListValuesRequestSchema>;
+
 export const RunActionRequestSchema = z.object({
   runtimeId: z
     .string()

js/core/src/reflection.ts

@@ -151,6 +151,22 @@ export class ReflectionServer {
       await this.stop();
     });
 
+    server.get('/api/values', async (req, response, next) => {
+      logger.debug('Fetching values.');
+      try {
+        const type = req.query.type;
+        if (!type) {
+          response.status(400).send('Query parameter "type" is required.');
+          return;
+        }
+        const values = await this.registry.listValues(type as string);
+        response.send(values);
+      } catch (err) {
+        const { message, stack } = err as Error;
+        next({ message, stack });
+      }
+    });
+
     server.get('/api/actions', async (_, response, next) => {
       logger.debug('Fetching actions.');
       try {

js/genkit/src/genkit.ts

@@ -46,6 +46,7 @@ import {
   type GenerationCommonConfigSchema,
   type IndexerParams,
   type ModelArgument,
+  type ModelReference,
   type Part,
   type PromptConfig,
   type PromptGenerateOptions,
@@ -953,7 +954,7 @@ export class Genkit implements HasRegistry {
       this.registry.registerValue(
         'defaultModel',
         'defaultModel',
-        this.options.model
+        modelName(this.options.model)
       );
     }
     if (this.options.promptDir !== null) {
@@ -1086,3 +1087,19 @@ let disableReflectionApi = false;
 export function __disableReflectionApi() {
   disableReflectionApi = true;
 }
+
+/** Helper method to map ModelArgument to string */
+function modelName(
+  modelArg: ModelArgument<any> | undefined
+): string | undefined {
+  if (modelArg === undefined) {
+    return undefined;
+  }
+  if (typeof modelArg === 'string') {
+    return modelArg;
+  }
+  if ((modelArg as ModelReference<any>).name) {
+    return (modelArg as ModelReference<any>).name;
+  }
+  return (modelArg as ModelAction).__action.name;
+}