diff --git a/genkit-tools/common/src/types/document.ts b/genkit-tools/common/src/types/document.ts
index b389cd4d4e..f84b802def 100644
--- a/genkit-tools/common/src/types/document.ts
+++ b/genkit-tools/common/src/types/document.ts
@@ -39,6 +39,11 @@ export const TextPartSchema = EmptyPartSchema.extend({
   text: z.string(),
 });
 
+/**
+ * Text part.
+ */
+export type TextPart = z.infer<typeof TextPartSchema>;
+
 /**
  * Zod schema for a reasoning part.
  */
@@ -48,9 +53,9 @@ export const ReasoningPartSchema = EmptyPartSchema.extend({
 });
 
 /**
- * Text part.
+ * Reasoning part.
  */
-export type TextPart = z.infer<typeof TextPartSchema>;
+export type ReasoningPart = z.infer<typeof ReasoningPartSchema>;
 
 /**
  * Zod schema of media.