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.
This commit is contained in:
geoffsee
2025-05-27 14:15:12 -04:00
committed by Geoff Seemueller
parent c04e19611e
commit 47272ba350
13 changed files with 105 additions and 302 deletions

View File

@@ -5,219 +5,129 @@ import {IMessage} from "../../../src/stores/ClientChatStore";
import {getModelFamily} from "../../../src/components/chat/SupportedModels";
export class ChatSdk {
static async preprocess({
messages,
}) {
// a custom implementation for preprocessing would go here
return Message.create({
role: "assistant",
content: "",
});
}
static async handleChatRequest(
request: Request,
ctx: {
openai: OpenAI;
systemPrompt: any;
maxTokens: any;
env: Env;
},
) {
const streamId = crypto.randomUUID();
const { messages, model, conversationId, attachments, tools } =
await request.json();
if (!messages?.length) {
return new Response("No messages provided", { status: 400 });
static async preprocess({
messages,
}) {
// run processing on messages to generate events/context
// removed in this fork due to expenses
return Message.create({
role: "assistant",
content: "",
});
}
const preprocessedContext = await ChatSdk.preprocess({
messages,
});
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,
JSON.stringify({
messages,
model,
conversationId,
timestamp: Date.now(),
attachments,
tools,
systemPrompt: ctx.systemPrompt,
preprocessedContext,
...webhooks,
}),
);
return new Response(
JSON.stringify({
streamUrl: `/api/streams/${streamId}`,
}),
{
headers: {
"Content-Type": "application/json",
static async handleChatRequest(
request: Request,
ctx: {
openai: OpenAI;
systemPrompt: any;
maxTokens: any;
env: Env;
},
},
);
}
) {
const streamId = crypto.randomUUID();
const {messages, model, conversationId} =
await request.json();
static async calculateMaxTokens(
messages: any[],
ctx: Record<string, any> & {
env: Env;
maxTokens: number;
},
) {
const objectId = ctx.env.SITE_COORDINATOR.idFromName(
"dynamic-token-counter",
);
const durableObject = ctx.env.SITE_COORDINATOR.get(objectId);
return durableObject.dynamicMaxTokens(messages, ctx.maxTokens);
}
if (!messages?.length) {
return new Response("No messages provided", {status: 400});
}
static buildAssistantPrompt({ maxTokens, tools }) {
return AssistantSdk.getAssistantPrompt({
maxTokens,
userTimezone: "UTC",
userLocation: "USA/unknown",
tools,
});
}
const preprocessedContext = await ChatSdk.preprocess({
messages,
});
static buildMessageChain(
messages: any[],
opts: {
systemPrompt: any;
assistantPrompt: string;
attachments: any[];
toolResults: IMessage;
model: any;
},
) {
const modelFamily = getModelFamily(opts.model);
const objectId = ctx.env.SITE_COORDINATOR.idFromName("stream-index");
const durableObject = ctx.env.SITE_COORDINATOR.get(objectId);
const messagesToSend = [];
messagesToSend.push(
Message.create({
role:
opts.model.includes("o1") ||
opts.model.includes("gemma") ||
modelFamily === "claude" ||
modelFamily === "google"
? "assistant"
: "system",
content: opts.systemPrompt.trim(),
}),
);
await durableObject.saveStreamData(
streamId,
JSON.stringify({
messages,
model,
conversationId,
timestamp: Date.now(),
systemPrompt: ctx.systemPrompt,
preprocessedContext
}),
);
messagesToSend.push(
Message.create({
role: "assistant",
content: opts.assistantPrompt.trim(),
}),
);
const attachmentMessages = (opts.attachments || []).map((attachment) =>
Message.create({
role: "user",
content: `Attachment: ${attachment.content}`,
}),
);
if (attachmentMessages.length > 0) {
messagesToSend.push(...attachmentMessages);
return new Response(
JSON.stringify({
streamUrl: `/api/streams/${streamId}`,
}),
{
headers: {
"Content-Type": "application/json",
},
},
);
}
messagesToSend.push(
...messages
.filter((message: any) => message.content?.trim())
.map((message: any) => Message.create(message)),
);
static async calculateMaxTokens(
messages: any[],
ctx: Record<string, any> & {
env: Env;
maxTokens: number;
},
) {
const objectId = ctx.env.SITE_COORDINATOR.idFromName(
"dynamic-token-counter",
);
const durableObject = ctx.env.SITE_COORDINATOR.get(objectId);
return durableObject.dynamicMaxTokens(messages, ctx.maxTokens);
}
return messagesToSend;
}
static buildAssistantPrompt({maxTokens}) {
return AssistantSdk.getAssistantPrompt({
maxTokens,
userTimezone: "UTC",
userLocation: "USA/unknown",
});
}
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");
};
static buildMessageChain(
messages: any[],
opts: {
systemPrompt: any;
assistantPrompt: string;
toolResults: IMessage;
model: any;
},
) {
const modelFamily = getModelFamily(opts.model);
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");
const messagesToSend = [];
eventSource.close();
return resolve();
}
messagesToSend.push(
Message.create({
role:
opts.model.includes("o1") ||
opts.model.includes("gemma") ||
modelFamily === "claude" ||
modelFamily === "google"
? "assistant"
: "system",
content: opts.systemPrompt.trim(),
}),
);
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" });
}
};
messagesToSend.push(
Message.create({
role: "assistant",
content: opts.assistantPrompt.trim(),
}),
);
eventSource.onerror = (error: any) => {
console.error("Webhook stream error:", error);
messagesToSend.push(
...messages
.filter((message: any) => message.content?.trim())
.map((message: any) => Message.create(message)),
);
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`));
}
return messagesToSend;
}
}
export default ChatSdk;