diff --git a/genkit-tools/common/src/types/retrievers.ts b/genkit-tools/common/src/types/retrievers.ts
index 3f8d40bbd..ce9772940 100644
--- a/genkit-tools/common/src/types/retrievers.ts
+++ b/genkit-tools/common/src/types/retrievers.ts
@@ -15,7 +15,7 @@
  */
 
 //
-// IMPORTANT: Keep this file in sync with genkit/ai/src/retrievers.ts!
+// IMPORTANT: Keep this file in sync with js/ai/src/retriever.ts!
 //
 import { z } from 'zod';
 import { PartSchema } from './model';
@@ -26,6 +26,12 @@ export const DocumentDataSchema = z.object({
 });
 export type DocumentData = z.infer<typeof DocumentDataSchema>;
 
+const RetrieverRequestSchema = z.object({
+  query: DocumentDataSchema,
+  options: z.any().optional(),
+});
+export type RetrieverRequest = z.infer<typeof RetrieverRequestSchema>;
+
 const RetrieverResponseSchema = z.object({
   documents: z.array(DocumentDataSchema),
 });