npm - vzcode - Versions diffs - 2.21.0 → 2.25.0 - Mend

vzcode 2.21.0 → 2.25.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (46) hide show

package/dist/index.html CHANGED Viewed

@@ -20,8 +20,8 @@
       href="https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;500;600;700&display=swap"
       rel="stylesheet"
     />
-    <script type="module" crossorigin src="/assets/index-D6-he0oi.js"></script>
-    <link rel="stylesheet" crossorigin href="/assets/index-fYq6iaqF.css">
+    <script type="module" crossorigin src="/assets/index-CkFweu5-.js"></script>
+    <link rel="stylesheet" crossorigin href="/assets/index-Dyt9tpmu.css">
   </head>
   <body>
     <div id="root"></div>

package/dist/{server/aiChatHandler → llm-streaming-server}/aiEditing.js RENAMED Viewed

@@ -1,6 +1,6 @@
 import { assembleFullPrompt, prepareFilesForPrompt, } from 'editcodewithai';
 import { formatMarkdownFiles } from 'llm-code-format';
-import { createFilesSnapshot, generateFilesUnifiedDiff, } from '../../utils/fileDiff.js';
+import { createFilesSnapshot, generateFilesUnifiedDiff, } from '../utils/fileDiff.js';
 // Dev flag for waiting 1 second before starting the LLM function.
 // Useful for debugging and testing purposes, e.g. checking the typing indicator.
 const delayStart = false;

package/dist/{server/aiChatHandler → llm-streaming-server}/chatOperations.js RENAMED Viewed

@@ -1,6 +1,6 @@
 import { dateToTimestamp } from '@vizhub/viz-utils';
-import { randomId } from '../../randomId.js';
-import { diff } from '../../ot.js';
+import { randomId } from '../randomId.js';
+import { diff } from '../ot.js';
 /**
  * Ensures the chats object exists in the ShareDB document
  */
@@ -402,6 +402,25 @@ export const updateStreamingStatus = (shareDBDoc, chatId, status, isStreaming =
     });
     shareDBDoc.submitOp(op);
 };
+/**
+ * Sets the model for a chat when AI generation starts
+ */
+export const setChatModel = (shareDBDoc, chatId, model) => {
+    DEBUG &&
+        console.log(`ChatOperations: Setting model: "${model}" for chatId: ${chatId}`);
+    const op = diff(shareDBDoc.data, {
+        ...shareDBDoc.data,
+        chats: {
+            ...shareDBDoc.data.chats,
+            [chatId]: {
+                ...shareDBDoc.data.chats[chatId],
+                model: model,
+                updatedAt: dateToTimestamp(new Date()),
+            },
+        },
+    });
+    shareDBDoc.submitOp(op);
+};
 /**
  * Finalizes streaming message and clears streaming state
  */

package/dist/{server/aiChatHandler → llm-streaming-server}/errorHandling.js RENAMED Viewed

@@ -1,5 +1,5 @@
 import { dateToTimestamp } from '@vizhub/viz-utils';
-import { diff } from '../../ot.js';
+import { diff } from '../ot.js';
 /**
  * Handles errors by adding an error message to the chat and clearing AI state
  */

package/dist/llm-streaming-server/index.js ADDED Viewed

@@ -0,0 +1,20 @@
+/**
+ * @vizhub/llm-streaming-server
+ *
+ * Server-side library for streaming LLM responses with ShareDB integration.
+ * This module provides functionality for:
+ * - Creating and managing LLM streaming functions
+ * - Performing AI-assisted code editing
+ * - Managing chat operations with ShareDB
+ * - Validating requests and handling errors
+ */
+// Core LLM streaming functionality
+export { createLLMFunction } from './llmStreaming.js';
+// AI editing operations
+export { performAIChat, performAIEditing, } from './aiEditing.js';
+// ShareDB chat operations
+export { ensureChatsExist, ensureChatExists, addUserMessage, updateAIStatus, updateAIScratchpad, clearAIScratchpadAndStatus, createAIMessage, updateAIMessageContent, setAIStatus, finalizeAIMessage, addDiffToAIMessage, addAIMessage, updateFiles, resolveFileId, createNewFile, createStreamingAIMessage, addStreamingEvent, updateStreamingStatus, finalizeStreamingMessage, } from './chatOperations.js';
+// Request validation
+export { validateRequest } from './validation.js';
+// Error handling
+export { handleError, handleBackgroundError, } from './errorHandling.js';

package/dist/{server/aiChatHandler → llm-streaming-server}/llmStreaming.js RENAMED Viewed

@@ -1,8 +1,8 @@
 import OpenAI from 'openai';
 import { parseMarkdownFiles, StreamingMarkdownParser, } from 'llm-code-format';
 import { mergeFileChanges } from 'editcodewithai';
-import { updateFiles, updateAIScratchpad, createStreamingAIMessage, addStreamingEvent, updateStreamingStatus, finalizeStreamingMessage, } from './chatOperations.js';
-import { formatFiles } from '../prettier.js';
+import { updateFiles, updateAIScratchpad, createStreamingAIMessage, addStreamingEvent, updateStreamingStatus, finalizeStreamingMessage, setChatModel, } from './chatOperations.js';
+import { formatFiles } from '../server/prettier.js';
 // Verbose logs
 const DEBUG = false;
 // Useful for testing/debugging the streaming behavior
@@ -22,8 +22,12 @@ export const createLLMFunction = ({ shareDBDoc, chatId,
 enableReasoningTokens = false, model, aiRequestOptions, }) => {
     return async (fullPrompt) => {
         // Create OpenRouter client for reasoning token support
+        const apiKey = process.env.VZCODE_EDIT_WITH_AI_API_KEY;
+        if (!apiKey) {
+            console.warn('[LLMStreaming] OpenRouter API Key (VZCODE_EDIT_WITH_AI_API_KEY) not found');
+        }
         const openRouterClient = new OpenAI({
-            apiKey: process.env.VZCODE_EDIT_WITH_AI_API_KEY,
+            apiKey: apiKey,
             baseURL: process.env.VZCODE_EDIT_WITH_AI_BASE_URL ||
                 'https://openrouter.ai/api/v1',
             defaultHeaders: {
@@ -36,8 +40,14 @@ enableReasoningTokens = false, model, aiRequestOptions, }) => {
         let currentEditingFileName = null;
         let accumulatedTextChunk = '';
         let currentFileContent = '';
+        // Stream the response with reasoning tokens
+        const modelName = model ||
+            process.env.VZCODE_EDIT_WITH_AI_MODEL_NAME ||
+            'anthropic/claude-haiku-4.5';
         // Create streaming AI message
         createStreamingAIMessage(shareDBDoc, chatId);
+        // Set the model being used for this chat
+        setChatModel(shareDBDoc, chatId, modelName);
         // Set initial content generation status
         updateStreamingStatus(shareDBDoc, chatId, 'Formulating a plan...');
         // Helper to get original file content
@@ -145,25 +155,20 @@ enableReasoningTokens = false, model, aiRequestOptions, }) => {
         const parser = new StreamingMarkdownParser(callbacks);
         const chunks = [];
         let reasoningContent = '';
-        // Stream the response with reasoning tokens
-        const modelName = model ||
-            process.env.VZCODE_EDIT_WITH_AI_MODEL_NAME ||
-            'anthropic/claude-3.5-sonnet';
         // Configure reasoning tokens based on enableReasoningTokens flag
         const requestConfig = {
             model: modelName,
             messages: [{ role: 'user', content: fullPrompt }],
-            usage: { include: true },
             stream: true,
             ...aiRequestOptions,
         };
         // Only include reasoning configuration if reasoning tokens are enabled
-        if (enableReasoningTokens) {
-            requestConfig.reasoning = {
-                effort: 'low',
-                exclude: false,
-            };
-        }
+        // if (enableReasoningTokens) {
+        //   requestConfig.reasoning = {
+        //     effort: 'low',
+        //     exclude: false,
+        //   };
+        // }
         const stream = await openRouterClient.chat.completions.create(requestConfig);
         let reasoningStarted = false;
         let contentStarted = false;

package/dist/server/aiChatHandler/index.js CHANGED Viewed

@@ -1,13 +1,13 @@
-import { validateRequest } from './validation.js';
-import { ensureChatsExist, ensureChatExists, addUserMessage, setAIStatus, } from './chatOperations.js';
-import { createLLMFunction } from './llmStreaming.js';
-import { performAIEditing } from './aiEditing.js';
-import { handleError, handleBackgroundError, } from './errorHandling.js';
+import { validateRequest } from '../../llm-streaming-server/validation.js';
+import { ensureChatsExist, ensureChatExists, addUserMessage, setAIStatus, } from '../../llm-streaming-server/chatOperations.js';
+import { createLLMFunction } from '../../llm-streaming-server/llmStreaming.js';
+import { performAIEditing } from '../../llm-streaming-server/aiEditing.js';
+import { handleError, handleBackgroundError, } from '../../llm-streaming-server/errorHandling.js';
 import { createRunCodeFunction } from '../../runCode.js';
 import { createSubmitOperation } from '../../submitOperation.js';
 import { getGenerationMetadata } from 'editcodewithai';
 const DEBUG = false;
-export const handleAIChatMessage = ({ shareDBDoc, onCreditDeduction, model, aiRequestOptions, }) => async (req, res) => {
+export const handleAIChatMessage = ({ shareDBDoc, onCreditDeduction, model, aiRequestOptions, enableReasoningTokens, }) => async (req, res) => {
     const { content, chatId } = req.body;
     if (DEBUG) {
         console.log('[handleAIChatMessage] content:', content, 'chatId:', chatId, 'shareDBDoc:', shareDBDoc);
@@ -31,6 +31,7 @@ export const handleAIChatMessage = ({ shareDBDoc, onCreditDeduction, model, aiRe
             content,
             model,
             aiRequestOptions,
+            enableReasoningTokens,
             onCreditDeduction,
         }).catch((error) => {
             console.error('Background AI processing error:', error);
@@ -45,12 +46,13 @@ export const handleAIChatMessage = ({ shareDBDoc, onCreditDeduction, model, aiRe
 /**
  * Processes the AI request asynchronously in the background
  */
-const processAIRequestAsync = async ({ shareDBDoc, chatId, content, model, aiRequestOptions, onCreditDeduction, }) => {
+const processAIRequestAsync = async ({ shareDBDoc, chatId, content, model, aiRequestOptions, enableReasoningTokens, onCreditDeduction, }) => {
     try {
         // Create LLM function for streaming
         const llmFunction = createLLMFunction({
             shareDBDoc,
             chatId,
+            enableReasoningTokens: enableReasoningTokens ?? true,
             model,
             aiRequestOptions,
         });

package/package.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "name": "vzcode",
-  "version": "2.21.0",
+  "version": "2.25.0",
   "description": "Multiplayer code editor system",
   "main": "src/index.ts",
   "type": "module",
@@ -11,7 +11,7 @@
   ],
   "scripts": {
     "test": "vitest run",
-    "test-interactive": "cd test/sampleDirectories/playground && node --import \"data:text/javascript,import { register } from 'node:module'; import { pathToFileURL } from 'node:url'; register('ts-node/esm', pathToFileURL('./'));\" ../../../src/server/index.ts",
+    "test-interactive": "cd test/sampleDirectories/playground && tsx ../../../src/server/index.ts",
     "prettier": "prettier {*.*,**/*.*} --write",
     "typecheck": "tsc --noEmit",
     "lint": "eslint . --ext .js,.ts,.tsx",
@@ -110,35 +110,35 @@
   },
   "homepage": "https://github.com/vizhub-core/vzcode#readme",
   "dependencies": {
-    "@codemirror/autocomplete": "^6.19.0",
+    "@codemirror/autocomplete": "^6.20.0",
     "@codemirror/lang-css": "^6.3.1",
     "@codemirror/lang-html": "^6.4.11",
     "@codemirror/lang-javascript": "^6.2.4",
     "@codemirror/lang-json": "^6.0.2",
-    "@codemirror/lang-markdown": "^6.4.0",
-    "@codemirror/lint": "^6.9.0",
+    "@codemirror/lang-markdown": "^6.5.0",
+    "@codemirror/lint": "^6.9.2",
     "@codemirror/state": "^6.5.2",
     "@codemirror/theme-one-dark": "^6.1.3",
-    "@codemirror/view": "^6.38.5",
-    "@langchain/core": "^0.3.78",
-    "@langchain/openai": "^0.6.14",
-    "@lezer/highlight": "^1.2.1",
-    "@livekit/components-react": "^2.9.15",
+    "@codemirror/view": "^6.38.8",
+    "@langchain/core": "^1.0.6",
+    "@langchain/openai": "^1.1.2",
+    "@lezer/highlight": "^1.2.3",
+    "@livekit/components-react": "^2.9.16",
     "@replit/codemirror-indentation-markers": "^6.5.3",
     "@replit/codemirror-interact": "^6.3.1",
     "@replit/codemirror-lang-svelte": "^6.0.0",
     "@replit/codemirror-vscode-keymap": "^6.0.2",
     "@teamwork/websocket-json-stream": "^2.0.0",
-    "@typescript/vfs": "^1.6.1",
-    "@uiw/codemirror-theme-abcdef": "^4.25.2",
-    "@uiw/codemirror-theme-dracula": "^4.25.2",
-    "@uiw/codemirror-theme-eclipse": "^4.25.2",
-    "@uiw/codemirror-theme-github": "^4.25.2",
-    "@uiw/codemirror-theme-material": "^4.25.2",
-    "@uiw/codemirror-theme-nord": "^4.25.2",
-    "@uiw/codemirror-theme-okaidia": "^4.25.2",
-    "@uiw/codemirror-theme-xcode": "^4.25.2",
-    "@uiw/codemirror-themes": "^4.25.2",
+    "@typescript/vfs": "^1.6.2",
+    "@uiw/codemirror-theme-abcdef": "^4.25.3",
+    "@uiw/codemirror-theme-dracula": "^4.25.3",
+    "@uiw/codemirror-theme-eclipse": "^4.25.3",
+    "@uiw/codemirror-theme-github": "^4.25.3",
+    "@uiw/codemirror-theme-material": "^4.25.3",
+    "@uiw/codemirror-theme-nord": "^4.25.3",
+    "@uiw/codemirror-theme-okaidia": "^4.25.3",
+    "@uiw/codemirror-theme-xcode": "^4.25.3",
+    "@uiw/codemirror-themes": "^4.25.3",
     "@valtown/codemirror-ts": "^2.3.1",
     "@vizhub/runtime": "^4.5.0",
     "@vizhub/viz-types": "^0.5.0",
@@ -157,22 +157,22 @@
     "diff2html": "^3.4.52",
     "dotenv": "^17.2.3",
     "editcodewithai": "^2.4.0",
-    "eslint-linter-browserify": "^9.37.0",
+    "eslint-linter-browserify": "^9.39.1",
     "express": "^5.1.0",
     "ignore": "^7.0.5",
     "json0-ot-diff": "^1.1.2",
     "jszip": "^3.10.1",
-    "livekit-server-sdk": "^2.14.0",
+    "livekit-server-sdk": "^2.14.2",
     "llm-code-format": "^3.1.0",
-    "lucide-react": "^0.545.0",
-    "npm": "^11.6.2",
-    "open": "^10.2.0",
+    "lucide-react": "^0.554.0",
+    "npm": "^11.6.3",
+    "open": "^11.0.0",
     "prettier-plugin-svelte": "^3.4.0",
     "react": "^18",
     "react-bootstrap": "^2.10.10",
     "react-dom": "^18",
     "react-markdown": "^10.1.0",
-    "react-router-dom": "^7.9.4",
+    "react-router-dom": "^7.9.6",
     "remark-gfm": "^4.0.1",
     "sharedb": "^5.2.2",
     "sharedb-client-browser": "^5.2.2",
@@ -181,31 +181,32 @@
     "ws": "^8.18.3"
   },
   "devDependencies": {
-    "@eslint/js": "^9.37.0",
-    "@tauri-apps/cli": "^2.8.4",
+    "@eslint/js": "^9.39.1",
+    "@tauri-apps/cli": "^2.9.4",
     "@types/d3-color": "^3.1.3",
     "@types/react": "^18",
     "@types/react-dom": "^18",
-    "@typescript-eslint/eslint-plugin": "^8.46.0",
-    "@typescript-eslint/parser": "^8.46.0",
-    "@vitejs/plugin-react": "^5.0.4",
+    "@typescript-eslint/eslint-plugin": "^8.47.0",
+    "@typescript-eslint/parser": "^8.47.0",
+    "@vitejs/plugin-react": "^5.1.1",
     "concurrently": "^9.2.1",
     "cross-env": "^10.1.0",
-    "eslint": "^9.37.0",
+    "eslint": "^9.39.1",
     "eslint-plugin-jsx-a11y": "^6.10.2",
     "eslint-plugin-react": "^7.37.5",
-    "eslint-plugin-react-hooks": "^7.0.0",
-    "globals": "^16.4.0",
-    "npm-check-updates": "^19.0.0",
+    "eslint-plugin-react-hooks": "^7.0.1",
+    "globals": "^16.5.0",
+    "npm-check-updates": "^19.1.2",
     "prettier": "^3.6.2",
-    "sass": "^1.93.2",
+    "sass": "^1.94.2",
     "ts-node": "^10.9.2",
+    "tsx": "^4.20.6",
     "typescript": "^5.9.3",
-    "vite": "^7.1.9",
-    "vitest": "^3.2.4"
+    "vite": "^7.2.4",
+    "vitest": "^4.0.13"
   },
   "optionalDependencies": {
-    "@rollup/rollup-darwin-arm64": "^4.52.4",
-    "@rollup/rollup-win32-x64-msvc": "^4.52.4"
+    "@rollup/rollup-darwin-arm64": "^4.53.3",
+    "@rollup/rollup-win32-x64-msvc": "^4.53.3"
   }
 }

package/src/client/CodeEditor/getOrCreateEditor.tsx CHANGED Viewed

@@ -70,7 +70,7 @@ import { SparklesSVG } from '../Icons/SparklesSVG';
 import { MINIMAL_EXTENSIONS } from '../featureFlags';
 const DEBUG = false;
-const enableToDoPlugin = false;
+const enableToDoPlugin = true;
 // Define a StateField to store the file name.
 // This should be defined at the module level if it's to be imported by other modules.
@@ -138,6 +138,7 @@ const languageExtensions = {
   html: () => html(htmlConfig),
   css,
   md: markdown,
+  prompt: markdown,
   svelte,
 };
@@ -188,7 +189,7 @@ export const getOrCreateEditor = async ({
   esLintSource,
   rainbowBracketsEnabled = true,
   setIsAIChatOpen,
-  handleSendMessage,
+  setAIChatMessage,
 }: {
   // TODO pass this in from the outside
   paneId?: PaneId;
@@ -225,7 +226,7 @@ export const getOrCreateEditor = async ({
   ) => Promise<readonly Diagnostic[]>;
   rainbowBracketsEnabled?: boolean; // New parameter type
   setIsAIChatOpen: (isAIChatOpen: boolean) => void;
-  handleSendMessage: any; // TODO fix types
+  setAIChatMessage: (message: string) => void;
 }): Promise<ExtendedEditorCacheValue> => {
   // Cache hit
@@ -356,7 +357,10 @@ export const getOrCreateEditor = async ({
     );
     // Enable line wrapping for Markdown files
-    if (fileExtension === 'md') {
+    if (
+      fileExtension === 'md' ||
+      fileExtension === 'prompt'
+    ) {
       extensions.push(EditorView.lineWrapping);
     }
@@ -485,22 +489,25 @@ export const getOrCreateEditor = async ({
           wrap.style.alignItems = 'center';
           wrap.style.justifyContent = 'center';
           wrap.className = 'icon-button icon-button-dark';
+          wrap.style.cursor = 'pointer';
+          // Handle click directly on the wrapper element
+          wrap.addEventListener('click', (e) => {
+            e.preventDefault();
+            e.stopPropagation();
+            setAIChatMessage('Implement the TODO');
+            setIsAIChatOpen(true);
+          });
           const reactContainer =
             document.createElement('div');
           wrap.appendChild(reactContainer);
           const root = createRoot(reactContainer);
           root.render(
-            <div
-              onClick={() => {
-                setIsAIChatOpen(true);
-                // setAIChatMessage('Implement the TODO');
-                handleSendMessage('Implement the TODO');
-              }}
-            >
-              <SparklesSVG width={14} height={14} />
-            </div>,
+            <SparklesSVG width={14} height={14} />,
           );
           return wrap;
         }

package/src/client/CodeEditor/index.tsx CHANGED Viewed

@@ -43,7 +43,7 @@ export const CodeEditor = ({
     codeEditorRef,
     enableAutoFollow,
     setIsAIChatOpen,
-    handleSendMessage,
+    setAIChatMessage,
   } = useContext(VZCodeContext);
   // Set `doc.data.isInteracting` to `true` when the user is interacting
@@ -117,7 +117,7 @@ export const CodeEditor = ({
         aiCopilotEndpoint,
         esLintSource,
         setIsAIChatOpen,
-        handleSendMessage,
+        setAIChatMessage,
       });
       if (isMounted) {
@@ -142,7 +142,7 @@ export const CodeEditor = ({
     aiCopilotEndpoint,
     esLintSource,
     setIsAIChatOpen,
-    handleSendMessage,
+    setAIChatMessage,
   ]);
   // Every time the active file switches from one file to another,

package/src/client/VZSidebar/FileTypeIcon.tsx CHANGED Viewed

@@ -25,6 +25,7 @@ export function getExtensionIcon(fileName: string) {
     case 'json':
       return <JsonSVG />;
     case 'md':
+    case 'prompt':
       return <MarkdownSVG />;
     case 'html':
       return <HtmlSVG />;

package/src/client/VZSidebar/index.tsx CHANGED Viewed

@@ -32,7 +32,7 @@ import { MicSVG } from '../Icons/MicSVG';
 import { sortFileTree } from '../sortFileTree';
 import { SplitPaneResizeContext } from '../SplitPaneResizeContext';
 import { VZCodeContext } from '../VZCodeContext';
-import { AIChat } from './AIChat';
+import { AIChat } from '../../llm-streaming-ui/components/index.js';
 import { Listing } from './Listing';
 import { Search } from './Search';
 import { DeleteConfirmationModal } from './DeleteConfirmationModal';

package/src/llm-streaming-server/README.md ADDED Viewed

@@ -0,0 +1,71 @@
+# LLM Streaming Server
+Server-side library for streaming LLM responses with ShareDB integration.
+## Overview
+This module is being extracted from VZCode's AI chat functionality to become a standalone, reusable library. It provides server-side functionality for:
+- **LLM Streaming**: Creating and managing streaming LLM functions with OpenAI/OpenRouter
+- **AI Code Editing**: Performing AI-assisted code transformations and edits
+- **ShareDB Integration**: Managing chat operations and real-time collaboration
+- **Request Validation**: Validating incoming AI chat requests
+- **Error Handling**: Handling errors gracefully in streaming contexts
+## Current Status
+⚠️ **Work in Progress**: This is Phase 1 of the extraction plan. The files have been copied to this directory and imports have been updated, but the code is not yet decoupled from VZCode.
+## Files
+- `index.ts` - Main entry point exporting all public APIs
+- `llmStreaming.ts` - Core LLM streaming functionality with OpenAI/OpenRouter
+- `aiEditing.ts` - AI-assisted code editing operations
+- `chatOperations.ts` - ShareDB chat operations (messages, events, streaming)
+- `validation.ts` - Request validation utilities
+- `errorHandling.ts` - Error handling for streaming contexts
+## Dependencies
+Currently depends on:
+- `openai` - OpenAI SDK for LLM interactions
+- `sharedb` - Real-time collaborative editing framework
+- `llm-code-format` - Markdown parsing for code blocks
+- `editcodewithai` - Code editing utilities
+- VZCode utilities: `../types.js`, `../ot.js`, `../randomId.js`, etc.
+## Future Plans
+This module will eventually be:
+1. Decoupled from VZCode-specific types and utilities
+2. Published as `@vizhub/llm-streaming-server` on npm
+3. Made framework-agnostic (can work with Express, Fastify, etc.)
+4. Fully documented with usage examples
+See the [main issue](https://github.com/vizhub-core/vzcode/issues/XXX) for the full extraction plan.
+## Usage (Current)
+Currently, this module is imported by `src/server/aiChatHandler/index.ts`:
+```typescript
+import { createLLMFunction } from '../../llm-streaming-server/llmStreaming.js';
+import { performAIEditing } from '../../llm-streaming-server/aiEditing.js';
+import {
+  ensureChatsExist,
+  ensureChatExists,
+  addUserMessage,
+  setAIStatus,
+} from '../../llm-streaming-server/chatOperations.js';
+```
+## Contributing
+As this module is being extracted, please:
+- Keep changes minimal and focused
+- Maintain backward compatibility with VZCode
+- Update tests when modifying functionality
+- Document any new dependencies or breaking changes

package/src/{server/aiChatHandler → llm-streaming-server}/aiEditing.ts RENAMED Viewed

@@ -6,7 +6,7 @@ import { formatMarkdownFiles } from 'llm-code-format';
 import {
   createFilesSnapshot,
   generateFilesUnifiedDiff,
-} from '../../utils/fileDiff.js';
+} from '../utils/fileDiff.js';
 // Dev flag for waiting 1 second before starting the LLM function.
 // Useful for debugging and testing purposes, e.g. checking the typing indicator.

package/src/{server/aiChatHandler → llm-streaming-server}/chatOperations.ts RENAMED Viewed

@@ -1,11 +1,11 @@
 import { dateToTimestamp } from '@vizhub/viz-utils';
-import { randomId } from '../../randomId.js';
+import { randomId } from '../randomId.js';
 import {
   ShareDBDoc,
   StreamingEvent,
   ExtendedVizContent,
-} from '../../types.js';
-import { diff } from '../../ot.js';
+} from '../types.js';
+import { diff } from '../ot.js';
 import {
   VizChatId,
   VizContent,
@@ -543,6 +543,33 @@ export const updateStreamingStatus = (
   shareDBDoc.submitOp(op);
 };
+/**
+ * Sets the model for a chat when AI generation starts
+ */
+export const setChatModel = (
+  shareDBDoc: ShareDBDoc<ExtendedVizContent>,
+  chatId: VizChatId,
+  model: string,
+) => {
+  DEBUG &&
+    console.log(
+      `ChatOperations: Setting model: "${model}" for chatId: ${chatId}`,
+    );
+  const op = diff(shareDBDoc.data, {
+    ...shareDBDoc.data,
+    chats: {
+      ...shareDBDoc.data.chats,
+      [chatId]: {
+        ...shareDBDoc.data.chats[chatId],
+        model: model,
+        updatedAt: dateToTimestamp(new Date()),
+      },
+    },
+  });
+  shareDBDoc.submitOp(op);
+};
 /**
  * Finalizes streaming message and clears streaming state
  */

package/src/{server/aiChatHandler → llm-streaming-server}/errorHandling.ts RENAMED Viewed

@@ -1,5 +1,5 @@
 import { dateToTimestamp } from '@vizhub/viz-utils';
-import { diff } from '../../ot.js';
+import { diff } from '../ot.js';
 /**
  * Handles errors by adding an error message to the chat and clearing AI state