Remove file upload functionality and related components

The `FileUploadStore` and all file upload features were removed, simplifying the chat interface. This change eliminates unused code, including file handling logic, attachment management, and UI elements, streamlining the application.
2025-09-08 22:56:46 +00:00 · 2025-05-27 14:15:12 -04:00
parent c04e19611e
commit 47272ba350
13 changed files with 105 additions and 302 deletions
--- a/src/models/Attachment.ts
+++ b/src/models/Attachment.ts
@@ -1,6 +0,0 @@
-import { types } from "mobx-state-tree";
-
-export default types.model("Attachment", {
-  content: types.string,
-  url: types.string,
-});
--- a/src/stores/ClientChatStore.ts
+++ b/src/stores/ClientChatStore.ts
@@ -1,6 +1,5 @@
 import { applySnapshot, flow, Instance, types } from "mobx-state-tree";
 import Message from "../models/Message";
-import IntermediateStep from "../models/IntermediateStep";
 import UserOptionsStore from "./UserOptionsStore";

 const ClientChatStore = types
@@ -10,8 +9,6 @@ const ClientChatStore = types
    isLoading: types.optional(types.boolean, false),
    model: types.optional(types.string, "llama-3.3-70b-versatile"),
    imageModel: types.optional(types.string, "black-forest-labs/flux-1.1-pro"),
-    tools: types.optional(types.array(types.string), []),
-    intermediateSteps: types.array(IntermediateStep),
  })
  .actions((self) => ({
    cleanup() {
@@ -39,7 +36,6 @@ const ClientChatStore = types
        const payload = {
          messages: self.messages.slice(),
          model: self.model,
-          tools: self.tools.slice(),
        };

        yield new Promise((resolve) => setTimeout(resolve, 500));
@@ -96,8 +92,6 @@ const ClientChatStore = types
              self.appendToLastMessage(
                parsedData.data.choices[0]?.delta?.content || "",
              );
-            } else {
-              self.handleIntermediateSteps(parsedData);
            }
          } catch (error) {
            console.error("Error processing stream:", error);
@@ -160,7 +154,6 @@ const ClientChatStore = types
          const payload = {
            messages: self.messages.slice(),
            model: self.model,
-            tools: self.tools.slice(),
          };

          const response = yield fetch("/api/chat", {
@@ -211,8 +204,6 @@ const ClientChatStore = types
                self.appendToLastMessage(
                  parsedData.data.choices[0]?.delta?.content || "",
                );
-              } else {
-                self.handleIntermediateSteps(parsedData);
              }
            } catch (error) {
              console.error("Error processing stream:", error);
@@ -240,31 +231,11 @@ const ClientChatStore = types
    reset() {
      applySnapshot(self, {});
    },
-    addIntermediateStep(stepData) {
-      return;
-    },
-
-    handleIntermediateSteps(eventData: any) {
-      try {
-        self.appendToLastMessage(eventData?.data.trim() + "\n");
-        self.addIntermediateStep({
-          kind: eventData.type,
-          data: eventData.data,
-        });
-      } catch (e) {}
-    },
    removeMessagesAfter(index: number) {
      if (index >= 0 && index < self.messages.length) {
        self.messages.splice(index + 1);
      }
    },
-    setTools(tools: string[]) {
-      self.tools.clear();
-      self.tools.push(...tools);
-    },
-    getTools() {
-      return self.tools.slice();
-    },
    updateLastMessage(content: string) {
      if (self.messages.length > 0) {
        self.messages[self.messages.length - 1].content = content;
--- a/workers/site/api-router.ts
+++ b/workers/site/api-router.ts
@@ -28,15 +28,6 @@ export function createRouter() {
        },
      )

-      .get(
-        "/api/streams/webhook/:streamId",
-        withParams,
-        async ({ streamId }, env, ctx) => {
-          const { chatService } = createServerContext(env, ctx);
-          return chatService.proxyWebhookStream(streamId); // Handles SSE for streamId
-        },
-      )
-
      .post("/api/feedback", async (r, e, c) => {
        const { feedbackService } = createServerContext(e, c);
        return feedbackService.handleFeedback(r);
--- a/workers/site/sdk/assistant-sdk.ts
+++ b/workers/site/sdk/assistant-sdk.ts
@@ -6,13 +6,11 @@ export class AssistantSdk {
    maxTokens?: number;
    userTimezone?: string;
    userLocation?: string;
-    tools?: string[];
  }): string {
    const {
      maxTokens,
      userTimezone = "UTC",
      userLocation = "",
-      tools = [],
    } = params;
    const selectedFewshots = Sdk.selectEquitably?.(few_shots) || few_shots;
    const sdkDate =
@@ -23,17 +21,6 @@ export class AssistantSdk {
    const now = new Date();
    const formattedMinutes = String(now.getMinutes()).padStart(2, "0");
    const currentTime = `${now.getHours()}:${formattedMinutes} ${now.getSeconds()}s`;
-    // const toolsInfo =
-    //   tools
-    //     .map((tool) => {
-    //       switch (tool) {
-    //         // case "user-attachments": return "### Attachments\nUser supplied attachments are normalized to text and will have this header (# Attachment:...) in the message.";
-    //         // case "web-search": return "### Web Search\nResults are optionally available in 'Live Search'.";
-    //         default:
-    //           return `- ${tool}`;
-    //       }
-    //     })
-    //     .join("\n\n") || "- No additional tools selected.";

    return `# Assistant Knowledge
 ## Current Context
--- a/workers/site/sdk/chat-sdk.ts
+++ b/workers/site/sdk/chat-sdk.ts
@@ -8,7 +8,8 @@ export class ChatSdk {
    static async preprocess({
                                messages,
                            }) {
-    // a custom implementation for preprocessing would go here
+        // run processing on messages to generate events/context
+        // removed in this fork due to expenses
        return Message.create({
            role: "assistant",
            content: "",
@@ -25,11 +26,11 @@ export class ChatSdk {
        },
    ) {
        const streamId = crypto.randomUUID();
-    const { messages, model, conversationId, attachments, tools } =
+        const {messages, model, conversationId} =
            await request.json();

        if (!messages?.length) {
-      return new Response("No messages provided", { status: 400 });
+            return new Response("No messages provided", {status: 400});
        }

        const preprocessedContext = await ChatSdk.preprocess({
@@ -39,8 +40,6 @@ export class ChatSdk {
        const objectId = ctx.env.SITE_COORDINATOR.idFromName("stream-index");
        const durableObject = ctx.env.SITE_COORDINATOR.get(objectId);

-    const webhooks =
-      JSON.parse(await durableObject.getStreamData(streamId)) ?? {};

        await durableObject.saveStreamData(
            streamId,
@@ -49,11 +48,8 @@ export class ChatSdk {
                model,
                conversationId,
                timestamp: Date.now(),
-        attachments,
-        tools,
                systemPrompt: ctx.systemPrompt,
-        preprocessedContext,
-        ...webhooks,
+                preprocessedContext
            }),
        );

@@ -83,12 +79,11 @@ export class ChatSdk {
        return durableObject.dynamicMaxTokens(messages, ctx.maxTokens);
    }

-  static buildAssistantPrompt({ maxTokens, tools }) {
+    static buildAssistantPrompt({maxTokens}) {
        return AssistantSdk.getAssistantPrompt({
            maxTokens,
            userTimezone: "UTC",
            userLocation: "USA/unknown",
-      tools,
        });
    }

@@ -97,7 +92,6 @@ export class ChatSdk {
        opts: {
            systemPrompt: any;
            assistantPrompt: string;
-      attachments: any[];
            toolResults: IMessage;
            model: any;
        },
@@ -126,17 +120,6 @@ export class ChatSdk {
            }),
        );

-    const attachmentMessages = (opts.attachments || []).map((attachment) =>
-      Message.create({
-        role: "user",
-        content: `Attachment: ${attachment.content}`,
-      }),
-    );
-
-    if (attachmentMessages.length > 0) {
-      messagesToSend.push(...attachmentMessages);
-    }
-
        messagesToSend.push(
            ...messages
                .filter((message: any) => message.content?.trim())
@@ -145,79 +128,6 @@ export class ChatSdk {

        return messagesToSend;
    }
-
-  static async handleAgentStream(
-    eventSource: EventSource,
-    dataCallback: any,
-  ): Promise<void> {
-    // console.log("sdk::handleWebhookStream::start");
-    let done = false;
-    return new Promise((resolve, reject) => {
-      if (!done) {
-        // console.log("sdk::handleWebhookStream::promise::created");
-        eventSource.onopen = () => {
-          // console.log("sdk::handleWebhookStream::eventSource::open");
-          console.log("Connected to agent");
-        };
-
-        const parseEvent = (data) => {
-          try {
-            return JSON.parse(data);
-          } catch (_) {
-            return data;
-          }
-        };
-        eventSource.onmessage = (event) => {
-          try {
-            if (event.data === "[DONE]") {
-              done = true;
-              console.log("Stream completed");
-
-              eventSource.close();
-              return resolve();
-            }
-
-            dataCallback({ type: "web-search", data: parseEvent(event.data) });
-          } catch (error) {
-            console.log("sdk::handleWebhookStream::eventSource::error");
-            console.error("Error parsing webhook data:", error);
-            dataCallback({ error: "Invalid data format from webhook" });
-          }
-        };
-
-        eventSource.onerror = (error: any) => {
-          console.error("Webhook stream error:", error);
-
-          if (
-            error.error &&
-            error.error.message === "The server disconnected."
-          ) {
-            return resolve();
-          }
-
-          reject(new Error("Failed to stream from webhook"));
-        };
-      }
-    });
-  }
-
-  static sendDoubleNewline(controller, encoder) {
-    const data = {
-      type: "chat",
-      data: {
-        choices: [
-          {
-            index: 0,
-            delta: { content: "\n\n" },
-            logprobs: null,
-            finish_reason: null,
-          },
-        ],
-      },
-    };
-
-    controller.enqueue(encoder.encode(`data: ${JSON.stringify(data)}\n\n`));
-  }
 }

 export default ChatSdk;
--- a/workers/site/sdk/models/cerebras.ts
+++ b/workers/site/sdk/models/cerebras.ts
@@ -21,12 +21,10 @@ export class CerebrasSdk {
        }>,
        _NotCustomized
      >;
-      attachments: any;
      maxTokens: unknown | number | undefined;
      messages: any;
      model: string;
      env: Env;
-      tools: any;
    },
    dataCallback: (data) => void,
  ) {
@@ -35,15 +33,12 @@ export class CerebrasSdk {
      messages,
      env,
      maxTokens,
-      tools,
      systemPrompt,
      model,
-      attachments,
    } = param;

    const assistantPrompt = ChatSdk.buildAssistantPrompt({
      maxTokens: maxTokens,
-      tools: tools,
    });

    const safeMessages = ChatSdk.buildMessageChain(messages, {
@@ -51,7 +46,6 @@ export class CerebrasSdk {
      model,
      assistantPrompt,
      toolResults: preprocessedContext,
-      attachments: attachments,
    });

    const openai = new OpenAI({
--- a/workers/site/sdk/models/claude.ts
+++ b/workers/site/sdk/models/claude.ts
@@ -49,7 +49,6 @@ export class ClaudeChatSdk {
    param: {
      openai: OpenAI;
      systemPrompt: any;
-      disableWebhookGeneration: boolean;
      preprocessedContext: ModelSnapshotType2<
        ModelPropertiesDeclarationToProperties<{
          role: ISimpleType<UnionStringArray<string[]>>;
@@ -57,12 +56,10 @@ export class ClaudeChatSdk {
        }>,
        _NotCustomized
      >;
-      attachments: any;
      maxTokens: unknown | number | undefined;
      messages: any;
      model: string;
      env: Env;
-      tools: any;
    },
    dataCallback: (data) => void,
  ) {
@@ -71,15 +68,12 @@ export class ClaudeChatSdk {
      messages,
      env,
      maxTokens,
-      tools,
      systemPrompt,
      model,
-      attachments,
    } = param;

    const assistantPrompt = ChatSdk.buildAssistantPrompt({
      maxTokens: maxTokens,
-      tools: tools,
    });

    const safeMessages = ChatSdk.buildMessageChain(messages, {
@@ -87,7 +81,6 @@ export class ClaudeChatSdk {
      model,
      assistantPrompt,
      toolResults: preprocessedContext,
-      attachments: attachments,
    });

    const anthropic = new Anthropic({
--- a/workers/site/sdk/models/cloudflareAi.ts
+++ b/workers/site/sdk/models/cloudflareAi.ts
@@ -13,7 +13,6 @@ export class CloudflareAISdk {
    param: {
      openai: OpenAI;
      systemPrompt: any;
-      disableWebhookGeneration: boolean;
      preprocessedContext: ModelSnapshotType2<
        ModelPropertiesDeclarationToProperties<{
          role: ISimpleType<UnionStringArray<string[]>>;
@@ -21,12 +20,10 @@ export class CloudflareAISdk {
        }>,
        _NotCustomized
      >;
-      attachments: any;
      maxTokens: unknown | number | undefined;
      messages: any;
      model: string;
      env: Env;
-      tools: any;
    },
    dataCallback: (data) => void,
  ) {
@@ -35,22 +32,18 @@ export class CloudflareAISdk {
      messages,
      env,
      maxTokens,
-      tools,
      systemPrompt,
      model,
-      attachments,
    } = param;

    const assistantPrompt = ChatSdk.buildAssistantPrompt({
      maxTokens: maxTokens,
-      tools: tools,
    });
    const safeMessages = ChatSdk.buildMessageChain(messages, {
      systemPrompt: systemPrompt,
      model,
      assistantPrompt,
      toolResults: preprocessedContext,
-      attachments: attachments,
    });

    const cfAiURL = `https://api.cloudflare.com/client/v4/accounts/${env.CLOUDFLARE_ACCOUNT_ID}/ai/v1`;
--- a/workers/site/sdk/models/fireworks.ts
+++ b/workers/site/sdk/models/fireworks.ts
@@ -42,7 +42,6 @@ export class FireworksAiChatSdk {
    param: {
      openai: OpenAI;
      systemPrompt: any;
-      disableWebhookGeneration: boolean;
      preprocessedContext: ModelSnapshotType2<
        ModelPropertiesDeclarationToProperties<{
          role: ISimpleType<UnionStringArray<string[]>>;
@@ -50,12 +49,10 @@ export class FireworksAiChatSdk {
        }>,
        _NotCustomized
      >;
-      attachments: any;
      maxTokens: number;
      messages: any;
      model: any;
      env: Env;
-      tools: any;
    },
    dataCallback: (data) => void,
  ) {
@@ -64,15 +61,12 @@ export class FireworksAiChatSdk {
      messages,
      env,
      maxTokens,
-      tools,
      systemPrompt,
      model,
-      attachments,
    } = param;

    const assistantPrompt = ChatSdk.buildAssistantPrompt({
      maxTokens: maxTokens,
-      tools: tools,
    });

    const safeMessages = ChatSdk.buildMessageChain(messages, {
@@ -80,7 +74,6 @@ export class FireworksAiChatSdk {
      model,
      assistantPrompt,
      toolResults: preprocessedContext,
-      attachments: attachments,
    });

    const fireworksOpenAIClient = new OpenAI({
--- a/workers/site/sdk/models/google.ts
+++ b/workers/site/sdk/models/google.ts
@@ -12,15 +12,12 @@ export class GoogleChatSdk {
      messages,
      env,
      maxTokens,
-      tools,
      systemPrompt,
      model,
-      attachments,
    } = param;

    const assistantPrompt = ChatSdk.buildAssistantPrompt({
      maxTokens: maxTokens,
-      tools: tools,
    });

    const safeMessages = ChatSdk.buildMessageChain(messages, {
@@ -28,7 +25,6 @@ export class GoogleChatSdk {
      model,
      assistantPrompt,
      toolResults: preprocessedContext,
-      attachments: attachments,
    });

    const openai = new OpenAI({
--- a/workers/site/sdk/models/groq.ts
+++ b/workers/site/sdk/models/groq.ts
@@ -13,7 +13,6 @@ export class GroqChatSdk {
    param: {
      openai: OpenAI;
      systemPrompt: any;
-      disableWebhookGeneration: boolean;
      preprocessedContext: ModelSnapshotType2<
        ModelPropertiesDeclarationToProperties<{
          role: ISimpleType<UnionStringArray<string[]>>;
@@ -21,12 +20,10 @@ export class GroqChatSdk {
        }>,
        _NotCustomized
      >;
-      attachments: any;
      maxTokens: unknown | number | undefined;
      messages: any;
      model: string;
      env: Env;
-      tools: any;
    },
    dataCallback: (data) => void,
  ) {
@@ -35,22 +32,18 @@ export class GroqChatSdk {
      messages,
      env,
      maxTokens,
-      tools,
      systemPrompt,
      model,
-      attachments,
    } = param;

    const assistantPrompt = ChatSdk.buildAssistantPrompt({
      maxTokens: maxTokens,
-      tools: tools,
    });
    const safeMessages = ChatSdk.buildMessageChain(messages, {
      systemPrompt: systemPrompt,
      model,
      assistantPrompt,
      toolResults: preprocessedContext,
-      attachments: attachments,
    });

    const openai = new OpenAI({
--- a/workers/site/sdk/models/openai.ts
+++ b/workers/site/sdk/models/openai.ts
@@ -7,12 +7,9 @@ export class OpenAiChatSdk {
      openai: OpenAI;
      systemPrompt: any;
      preprocessedContext: any;
-      attachments: any;
      maxTokens: unknown | number | undefined;
      messages: any;
-      disableWebhookGeneration: boolean;
      model: any;
-      tools: any;
    },
    dataCallback: (data: any) => any,
  ) {
@@ -20,9 +17,7 @@ export class OpenAiChatSdk {
      openai,
      systemPrompt,
      maxTokens,
-      tools,
      messages,
-      attachments,
      model,
      preprocessedContext,
    } = ctx;
@@ -33,14 +28,12 @@ export class OpenAiChatSdk {

    const assistantPrompt = ChatSdk.buildAssistantPrompt({
      maxTokens: maxTokens,
-      tools: tools,
    });
    const safeMessages = ChatSdk.buildMessageChain(messages, {
      systemPrompt: systemPrompt,
      model,
      assistantPrompt,
      toolResults: preprocessedContext,
-      attachments: attachments,
    });

    return OpenAiChatSdk.streamOpenAiResponse(
--- a/workers/site/sdk/models/xai.ts
+++ b/workers/site/sdk/models/xai.ts
@@ -7,13 +7,11 @@ export class XaiChatSdk {
      openai: OpenAI;
      systemPrompt: any;
      preprocessedContext: any;
-      attachments: any;
      maxTokens: unknown | number | undefined;
      messages: any;
      disableWebhookGeneration: boolean;
      model: any;
      env: Env;
-      tools: any;
    },
    dataCallback: (data: any) => any,
  ) {
@@ -21,7 +19,6 @@ export class XaiChatSdk {
      openai,
      systemPrompt,
      maxTokens,
-      tools,
      messages,
      attachments,
      env,
@@ -49,7 +46,6 @@ export class XaiChatSdk {

    const assistantPrompt = ChatSdk.buildAssistantPrompt({
      maxTokens: maxTokens,
-      tools: tools,
    });

    const safeMessages = ChatSdk.buildMessageChain(messages, {
@@ -57,7 +53,6 @@ export class XaiChatSdk {
      model,
      assistantPrompt,
      toolResults: preprocessedContext,
-      attachments: attachments,
    });

    const xAiClient = new OpenAI({