vzcode 1.38.0 → 1.40.0

package/src/server/aiChatHandler/aiEditing.js

@@ -1,42 +0,0 @@
-import { performAiEdit } from 'editcodewithai';
-import {
-  clearAIScratchpadAndStatus,
-  updateFiles,
-  setIsInteracting,
-} from './chatOperations.js';
-
-/**
- * Performs AI editing operations on the files
- */
-export const performAIEditing = async (
-  shareDBDoc,
-  chatId,
-  content,
-  files,
-  llmFunction,
-) => {
-  const editResult = await performAiEdit({
-    prompt: content,
-    files,
-    llmFunction,
-    apiKey: process.env.VIZHUB_EDIT_WITH_AI_API_KEY,
-  });
-
-  // Clear the scratchpad and update status
-  clearAIScratchpadAndStatus(
-    shareDBDoc,
-    chatId,
-    'Done editing with AI.',
-  );
-
-  // Apply AI edits to files
-  updateFiles(shareDBDoc, editResult.changedFiles);
-
-  // Wait for propagation
-  await new Promise((resolve) => setTimeout(resolve, 100));
-
-  // Unset isInteracting
-  setIsInteracting(shareDBDoc, false);
-
-  return editResult;
-};

package/src/server/aiChatHandler/llmStreaming.js

@@ -1,105 +0,0 @@
-import { StreamingMarkdownParser } from 'llm-code-format';
-import { ChatOpenAI } from '@langchain/openai';
-import {
-  updateAIStatus,
-  updateAIScratchpad,
-} from './chatOperations.js';
-
-const debug = false;
-
-/**
- * Creates and configures the LLM function for streaming
- */
-export const createLLMFunction = (shareDBDoc, chatId) => {
-  return async (fullPrompt) => {
-    const chatModel = new ChatOpenAI({
-      modelName:
-        process.env.VIZHUB_EDIT_WITH_AI_MODEL_NAME ||
-        'anthropic/claude-sonnet-4',
-      configuration: {
-        apiKey: process.env.VIZHUB_EDIT_WITH_AI_API_KEY,
-        baseURL: process.env.VIZHUB_EDIT_WITH_AI_BASE_URL,
-        defaultHeaders: {
-          'HTTP-Referer': 'https://vizhub.com',
-          'X-Title': 'VizHub',
-        },
-      },
-      streaming: true,
-    });
-
-    let fullContent = '';
-    let generationId = '';
-
-    // Track expected state to avoid race conditions
-    let expectedAiScratchpad =
-      shareDBDoc.data.chats[chatId]?.aiScratchpad || '';
-
-    // Throttle updates to avoid "op too long" errors
-    let lastUpdateTime = 0;
-    const UPDATE_THROTTLE_MS = 100;
-
-    // Define callbacks for streaming parser
-    const callbacks = {
-      onFileNameChange: (fileName, format) => {
-        debug &&
-          console.log(
-            `File changed to: ${fileName} (${format})`,
-          );
-        updateAIStatus(
-          shareDBDoc,
-          chatId,
-          'Editing ' + fileName,
-        );
-      },
-      onCodeLine: (line) => {
-        debug && console.log(`Code line: ${line}`);
-      },
-      onNonCodeLine: (line) => {
-        debug && console.log(`Comment/text: ${line}`);
-      },
-    };
-
-    const parser = new StreamingMarkdownParser(callbacks);
-
-    // Stream the response
-    const stream = await chatModel.stream(fullPrompt);
-    for await (const chunk of stream) {
-      if (chunk.content) {
-        const chunkContent = String(chunk.content);
-        fullContent += chunkContent;
-
-        // Throttle updates
-        const now = Date.now();
-        const shouldUpdate =
-          now - lastUpdateTime >= UPDATE_THROTTLE_MS;
-
-        if (shouldUpdate) {
-          updateAIScratchpad(
-            shareDBDoc,
-            chatId,
-            fullContent,
-          );
-          expectedAiScratchpad = fullContent;
-          lastUpdateTime = now;
-        }
-
-        parser.processChunk(chunkContent);
-      }
-
-      if (!generationId && chunk.lc_kwargs?.id) {
-        generationId = chunk.lc_kwargs.id;
-      }
-    }
-    parser.flushRemaining();
-
-    // Submit final update if there's pending content
-    if (expectedAiScratchpad !== fullContent) {
-      updateAIScratchpad(shareDBDoc, chatId, fullContent);
-    }
-
-    return {
-      content: fullContent,
-      generationId: generationId,
-    };
-  };
-};