From e45c8fc2442ee8103a9a72ddbfb3cb3976a19577 Mon Sep 17 00:00:00 2001
From: Nate Kelley <nate@buster.so>
Date: Wed, 17 Sep 2025 15:15:29 -0600
Subject: [PATCH] Reasoning extend

---
 packages/database/src/schemas/message-schemas.ts       | 8 +++++---
 packages/server-shared/src/chats/chat-message.types.ts | 8 +++-----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/packages/database/src/schemas/message-schemas.ts b/packages/database/src/schemas/message-schemas.ts
index b6fe2f243..e253fd73d 100644
--- a/packages/database/src/schemas/message-schemas.ts
+++ b/packages/database/src/schemas/message-schemas.ts
@@ -81,7 +81,7 @@ const ReasoningMessage_FilesSchema = z.object({
   files: z.record(z.string(), ReasoningFileSchema),
 });
 
-const ReasoingMessage_ThoughtFileTypeSchema = z.enum([
+const ReasoningMessage_ThoughtFileTypeSchema = z.enum([
   'metric_file',
   'dashboard_file',
   'collection',
@@ -94,7 +94,7 @@ const ReasoingMessage_ThoughtFileTypeSchema = z.enum([
 
 const ReasoningMessage_PillSchema = z.object({
   text: z.string(),
-  type: ReasoingMessage_ThoughtFileTypeSchema,
+  type: ReasoningMessage_ThoughtFileTypeSchema,
   id: z.string(),
 });
 
@@ -143,7 +143,9 @@ export type ChatMessageResponseMessage_Text = z.infer<typeof ResponseMessage_Tex
 export type ChatMessageResponseMessage_File = z.infer<typeof ResponseMessage_FileSchema>;
 export type ReasoningFileType = z.infer<typeof ReasoningFileTypeSchema>;
 export type ResponseMessageFileType = z.infer<typeof ResponseMessageFileTypeSchema>;
-export type ReasoingMessage_ThoughtFileType = z.infer<typeof ReasoingMessage_ThoughtFileTypeSchema>;
+export type ReasoingMessage_ThoughtFileType = z.infer<
+  typeof ReasoningMessage_ThoughtFileTypeSchema
+>;
 
 // Message metadata schema
 export const MessageMetadataSchema = z.object({
diff --git a/packages/server-shared/src/chats/chat-message.types.ts b/packages/server-shared/src/chats/chat-message.types.ts
index 7d5d62136..f70b6018f 100644
--- a/packages/server-shared/src/chats/chat-message.types.ts
+++ b/packages/server-shared/src/chats/chat-message.types.ts
@@ -84,10 +84,8 @@ const ReasoningMessage_FilesSchema = z.object({
   files: z.record(z.string(), ReasoningFileSchema),
 });
 
-const ReasoingMessage_ThoughtFileTypeSchema = z.enum([
-  'metric_file',
-  'dashboard_file',
-  'collection',
+const ReasoningMessage_ThoughtFileTypeSchema = z.enum([
+  ...AssetTypeSchema.exclude(['chat']).options,
   'dataset',
   'term',
   'topic',
@@ -97,7 +95,7 @@ const ReasoingMessage_ThoughtFileTypeSchema = z.enum([
 
 const ReasoningMessage_PillSchema = z.object({
   text: z.string(),
-  type: ReasoingMessage_ThoughtFileTypeSchema,
+  type: ReasoningMessage_ThoughtFileTypeSchema,
   id: z.string(),
 });