fix(js): Export RessourcePartSchema through genkit-tools

Author: shrutip90

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

@@ -28,6 +28,7 @@ const EmptyPartSchema = z.object({
   metadata: z.record(z.unknown()).optional(),
   custom: z.record(z.unknown()).optional(),
   reasoning: z.never().optional(),
+  resource: z.never().optional(),
 });
 
 /**
@@ -147,6 +148,20 @@ export const CustomPartSchema = EmptyPartSchema.extend({
  */
 export type CustomPart = z.infer<typeof CustomPartSchema>;
 
+/**
+ * Zod schema of a resource part.
+ */
+export const ResourcePartSchema = EmptyPartSchema.extend({
+  resource: z.object({
+    uri: z.string(),
+  }),
+});
+
+/**
+ * Resource part.
+ */
+export type ResourcePart = z.infer<typeof ResourcePartSchema>;
+
 // Disclaimer: genkit/js/ai/document.ts defines the following schema, type pair
 // as PartSchema and Part, respectively. genkit-tools cannot retain those names
 // due to it clashing with similar schema in model.ts, and genkit-tools

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

@@ -23,12 +23,14 @@ import {
   TextPartSchema,
   ToolRequestPartSchema,
   ToolResponsePartSchema,
+  ResourcePartSchema,
   type CustomPart,
   type DataPart,
   type MediaPart,
   type TextPart,
   type ToolRequestPart,
   type ToolResponsePart,
+  type ResourcePart,
 } from './document';
 export {
   CustomPartSchema,
@@ -37,12 +39,14 @@ export {
   TextPartSchema,
   ToolRequestPartSchema,
   ToolResponsePartSchema,
+  ResourcePartSchema,
   type CustomPart,
   type DataPart,
   type MediaPart,
   type TextPart,
   type ToolRequestPart,
   type ToolResponsePart,
+  type ResourcePart,
 };
 
 //
@@ -77,6 +81,7 @@ export const PartSchema = z.union([
   DataPartSchema,
   CustomPartSchema,
   ReasoningPartSchema,
+  ResourcePartSchema
 ]);
 
 /**

genkit-tools/genkit-schema.json

@@ -27,6 +27,9 @@
         },
         "reasoning": {
           "not": {}
+        },
+        "resource": {
+          "not": {}
         }
       },
       "required": [
@@ -59,6 +62,9 @@
         },
         "reasoning": {
           "$ref": "#/$defs/CustomPart/properties/reasoning"
+        },
+        "resource": {
+          "$ref": "#/$defs/CustomPart/properties/resource"
         }
       },
       "additionalProperties": false
@@ -118,6 +124,9 @@
         },
         "reasoning": {
           "$ref": "#/$defs/CustomPart/properties/reasoning"
+        },
+        "resource": {
+          "$ref": "#/$defs/CustomPart/properties/resource"
         }
       },
       "required": [
@@ -166,13 +175,61 @@
         },
         "reasoning": {
           "type": "string"
+        },
+        "resource": {
+          "$ref": "#/$defs/CustomPart/properties/resource"
         }
       },
       "required": [
         "reasoning"
       ],
       "additionalProperties": false
     },
+    "ResourcePart": {
+      "type": "object",
+      "properties": {
+        "text": {
+          "$ref": "#/$defs/CustomPart/properties/text"
+        },
+        "media": {
+          "$ref": "#/$defs/CustomPart/properties/media"
+        },
+        "toolRequest": {
+          "$ref": "#/$defs/CustomPart/properties/toolRequest"
+        },
+        "toolResponse": {
+          "$ref": "#/$defs/CustomPart/properties/toolResponse"
+        },
+        "data": {
+          "$ref": "#/$defs/CustomPart/properties/data"
+        },
+        "metadata": {
+          "$ref": "#/$defs/CustomPart/properties/metadata"
+        },
+        "custom": {
+          "$ref": "#/$defs/DataPart/properties/custom"
+        },
+        "reasoning": {
+          "$ref": "#/$defs/CustomPart/properties/reasoning"
+        },
+        "resource": {
+          "type": "object",
+          "properties": {
+            "uri": {
+              "type": "string"
+            }
+          },
+          "required": [
+            "uri"
+          ],
+          "additionalProperties": false
+        }
+      },
+      "required": [
+        "resource"
+      ],
+      "additionalProperties": false
+    },
     "TextPart": {
       "type": "object",
       "properties": {
@@ -199,6 +256,9 @@
         },
         "reasoning": {
           "$ref": "#/$defs/CustomPart/properties/reasoning"
+        },
+        "resource": {
+          "$ref": "#/$defs/CustomPart/properties/resource"
         }
       },
       "required": [
@@ -232,6 +292,9 @@
         },
         "reasoning": {
           "$ref": "#/$defs/CustomPart/properties/reasoning"
+        },
+        "resource": {
+          "$ref": "#/$defs/CustomPart/properties/resource"
         }
       },
       "required": [
@@ -281,6 +344,9 @@
         },
         "reasoning": {
           "$ref": "#/$defs/CustomPart/properties/reasoning"
+        },
+        "resource": {
+          "$ref": "#/$defs/CustomPart/properties/resource"
         }
       },
       "required": [
@@ -1138,6 +1204,9 @@
         },
         {
           "$ref": "#/$defs/ReasoningPart"
+        },
+        {
+          "$ref": "#/$defs/ResourcePart"
         }
       ]
     },