Refactor: Relocate SDK files to lib and update imports

Moved all SDK files from the `sdk` directory to the `lib` directory to better align with project structure. Updated all associated import paths across the codebase to reflect this change. Removed unused or commented-out code in `SiteCoordinator.js` for better clarity and maintainability.
This commit is contained in:
geoffsee
2025-05-27 14:48:18 -04:00
committed by Geoff Seemueller
parent fc22278b58
commit 335e8eff11
14 changed files with 10 additions and 28 deletions

View File

@@ -7,24 +7,10 @@ export default class SiteCoordinator extends DurableObject {
this.env = env;
}
// Public method to calculate dynamic max tokens
async dynamicMaxTokens(input, maxOuputTokens) {
return 2000;
// const baseTokenLimit = 1024;
//
//
// const { encode } = await import("gpt-tokenizer/esm/model/gpt-4o");
//
// const inputTokens = Array.isArray(input)
// ? encode(input.map(i => i.content).join(' '))
// : encode(input);
//
// const scalingFactor = inputTokens.length > 300 ? 1.5 : 1;
//
// return Math.min(baseTokenLimit + Math.floor(inputTokens.length * scalingFactor^2), maxOuputTokens);
}
// Public method to retrieve conversation history
async getConversationHistory(conversationId) {
const history = await this.env.KV_STORAGE.get(
`conversations:${conversationId}`,
@@ -33,7 +19,6 @@ export default class SiteCoordinator extends DurableObject {
return JSON.parse(history) || [];
}
// Public method to save a message to the conversation history
async saveConversationHistory(conversationId, message) {
const history = await this.getConversationHistory(conversationId);
history.push(message);
@@ -52,7 +37,6 @@ export default class SiteCoordinator extends DurableObject {
);
}
// New method to get stream data
async getStreamData(streamId) {
const streamEntry = await this.env.KV_STORAGE.get(`streams:${streamId}`);
if (!streamEntry) {
@@ -61,7 +45,6 @@ export default class SiteCoordinator extends DurableObject {
const { data, expirationTimestamp } = JSON.parse(streamEntry);
if (Date.now() > expirationTimestamp) {
// await this.state.storage.delete(streamId); // Clean up expired entry
await this.deleteStreamData(`streams:${streamId}`);
return null;
}
@@ -69,7 +52,6 @@ export default class SiteCoordinator extends DurableObject {
return data;
}
// New method to delete stream data (cleanup)
async deleteStreamData(streamId) {
await this.env.KV_STORAGE.delete(`streams:${streamId}`);
}

View File

@@ -6,7 +6,7 @@ import {
ModelSnapshotType2,
UnionStringArray,
} from "mobx-state-tree";
import ChatSdk from "../sdk/chat-sdk";
import ChatSdk from "../lib/chat-sdk";
export class CerebrasSdk {
static async handleCerebrasStream(

View File

@@ -7,7 +7,7 @@ import {
ModelSnapshotType2,
UnionStringArray,
} from "mobx-state-tree";
import ChatSdk from "../sdk/chat-sdk";
import ChatSdk from "../lib/chat-sdk";
export class ClaudeChatSdk {
private static async streamClaudeResponse(

View File

@@ -6,7 +6,7 @@ import {
ModelSnapshotType2,
UnionStringArray,
} from "mobx-state-tree";
import ChatSdk from "../sdk/chat-sdk";
import ChatSdk from "../lib/chat-sdk";
export class CloudflareAISdk {
static async handleCloudflareAIStream(

View File

@@ -9,7 +9,7 @@ import {
UnionStringArray,
} from "mobx-state-tree";
import Message from "../models/Message";
import ChatSdk from "../sdk/chat-sdk";
import ChatSdk from "../lib/chat-sdk";
export class FireworksAiChatSdk {
private static async streamFireworksResponse(

View File

@@ -1,5 +1,5 @@
import { OpenAI } from "openai";
import ChatSdk from "../sdk/chat-sdk";
import ChatSdk from "../lib/chat-sdk";
import { StreamParams } from "../services/ChatService";
export class GoogleChatSdk {

View File

@@ -6,7 +6,7 @@ import {
ModelSnapshotType2,
UnionStringArray,
} from "mobx-state-tree";
import ChatSdk from "../sdk/chat-sdk";
import ChatSdk from "../lib/chat-sdk";
export class GroqChatSdk {
static async handleGroqStream(

View File

@@ -1,5 +1,5 @@
import { OpenAI } from "openai";
import ChatSdk from "../sdk/chat-sdk";
import ChatSdk from "../lib/chat-sdk";
export class OpenAiChatSdk {
static async handleOpenAiStream(

View File

@@ -1,5 +1,5 @@
import { OpenAI } from "openai";
import ChatSdk from "../sdk/chat-sdk";
import ChatSdk from "../lib/chat-sdk";
export class XaiChatSdk {
static async handleXaiStream(

View File

@@ -1,6 +1,6 @@
import {flow, getSnapshot, types} from 'mobx-state-tree';
import OpenAI from 'openai';
import ChatSdk from '../sdk/chat-sdk';
import ChatSdk from '../lib/chat-sdk';
import Message from "../models/Message";
import O1Message from "../models/O1Message";
import {getModelFamily, ModelFamily} from "../../../src/components/chat/lib/SupportedModels";
@@ -8,7 +8,7 @@ import {OpenAiChatSdk} from "../providers/openai";
import {GroqChatSdk} from "../providers/groq";
import {ClaudeChatSdk} from "../providers/claude";
import {FireworksAiChatSdk} from "../providers/fireworks";
import handleStreamData from "../sdk/handleStreamData";
import handleStreamData from "../lib/handleStreamData";
import {GoogleChatSdk} from "../providers/google";
import {XaiChatSdk} from "../providers/xai";
import {CerebrasSdk} from "../providers/cerebras";