Add RetrieverRequest to tools-common (#836)

firebase · Aug 29, 2024 · b042428 · b042428
1 parent cc18c45
commit b042428
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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),
 });