clean up

Author: Kitenite

apps/web/client/src/app/project/[id]/_components/right-panel/chat-tab/chat-messages/message-content/index.tsx

@@ -1,8 +1,8 @@
 import type { ToolUIPart, UIMessage } from 'ai';
 import { observer } from 'mobx-react-lite';
+import { useMemo } from 'react';
 import { MarkdownRenderer } from '../markdown-renderer';
 import { ToolCallDisplay } from './tool-call-display';
-import { useMemo } from 'react';
 
 export const MessageContent = observer(
     ({
@@ -16,17 +16,18 @@ export const MessageContent = observer(
         applied: boolean;
         isStream: boolean;
     }) => {
-        if (!parts) {
-            return null;
-        }
         // Find the index of the last tool-*** part
-        const lastToolInvocationIdx = useMemo(() => 
+        const lastToolInvocationIdx = useMemo(() =>
             parts?.map((part, index) => ({ type: part.type, index }))
                 .filter(item => item.type.startsWith('tool-'))
                 .pop()?.index ?? -1,
             [parts]
         );
 
+        if (!parts.length) {
+            return null;
+        }
+
         const renderedParts = parts.map((part, idx) => {
             if (part.type === 'text') {
                 return (

apps/web/client/src/app/project/[id]/_components/right-panel/chat-tab/chat-messages/message-content/tool-call-display.tsx

@@ -14,12 +14,12 @@ import {
 import type { WebSearchResult } from '@onlook/models';
 import { Icons } from '@onlook/ui/icons/index';
 import { cn } from '@onlook/ui/utils';
+import type { ToolUIPart } from 'ai';
 import { type z } from 'zod';
 import { BashCodeDisplay } from '../../code-display/bash-code-display';
 import { CollapsibleCodeBlock } from '../../code-display/collapsible-code-block';
 import { SearchSourcesDisplay } from '../../code-display/search-sources-display';
 import { ToolCallSimple } from './tool-call-simple';
-import type { ToolUIPart } from 'ai';
 
 export const ToolCallDisplay = ({
     messageId,
@@ -160,7 +160,7 @@ export const ToolCallDisplay = ({
         if (toolName === SEARCH_REPLACE_MULTI_EDIT_FILE_TOOL_NAME) {
             const args = toolInvocation.input as z.infer<typeof SEARCH_REPLACE_MULTI_EDIT_FILE_TOOL_PARAMETERS>;
             const filePath = args?.file_path;
-            const codeContent = args?.edits.map((edit) => edit.new_string).join('\n...\n');
+            const codeContent = args?.edits?.map((edit) => edit.new_string).join('\n...\n');
             if (!filePath || !codeContent) {
                 return (
                     <ToolCallSimple

apps/web/client/src/app/project/[id]/_components/right-panel/chat-tab/error.tsx

@@ -18,8 +18,8 @@ export const ErrorSection = observer(() => {
 
     const sendFixError = async () => {
         try {
-            const message = await editorEngine.chat.addFixErrorMessage();
-            sendMessageToChat(ChatType.FIX);
+            await editorEngine.chat.addFixErrorMessage();
+            await sendMessageToChat(ChatType.FIX);
         } catch (error) {
             console.error('Error sending fix error message', error);
             toast.error('Failed to send fix error message. Please try again.');

apps/web/client/src/components/store/editor/chat/conversation.ts

@@ -55,7 +55,6 @@ export class ConversationManager {
             }
             const newConversation = await api.chat.conversation.upsert.mutate({
                 projectId: this.editorEngine.projectId,
-                suggestions: [],
             });
             this.current = {
                 conversation: newConversation,
@@ -167,12 +166,10 @@ export class ConversationManager {
             messageId: message.id,
             checkpoints: newCheckpoints,
         });
-        console.log('attachCommitToUserMessage', userMessage);
         await this.addOrReplaceMessage(userMessage);
     }
 
     async addOrReplaceMessage(message: ChatMessage) {
-        console.log('addOrReplaceMessage', message);
         if (!this.current) {
             console.error('No conversation found');
             return;

apps/web/client/src/components/store/hosting/provider.tsx

@@ -156,7 +156,6 @@ export const HostingProvider = ({ children }: HostingProviderProps) => {
                 deployment: {
                     status: DeploymentStatus.FAILED,
                     error: error instanceof Error ? error.message : 'Unknown error',
-                    envVars: {},
                 },
             });
             return {

apps/web/client/src/components/tools/tools.ts

@@ -1,4 +1,6 @@
+import { getToolSetFromType } from '@/app/api/chat/helperts';
 import type { EditorEngine } from '@/components/store/editor/engine';
+import type { ToolCall } from '@ai-sdk/provider-utils';
 import {
     BASH_EDIT_TOOL_NAME,
     BASH_EDIT_TOOL_PARAMETERS,
@@ -55,8 +57,6 @@ import {
     handleWriteFileTool
 } from './handlers';
 import { EMPTY_TOOL_PARAMETERS } from './helpers';
-import type { ToolCall } from '@ai-sdk/provider-utils';
-import { getToolSetFromType } from '@/app/api/chat/helperts';
 
 interface ClientToolMap extends Record<string, {
     name: string;
@@ -81,13 +81,13 @@ const TOOL_HANDLERS: ClientToolMap = {
     [READ_STYLE_GUIDE_TOOL_NAME]: {
         name: READ_STYLE_GUIDE_TOOL_NAME,
         inputSchema: EMPTY_TOOL_PARAMETERS,
-        handler: async (editorEngine: EditorEngine) =>
+        handler: async (args: unknown, editorEngine: EditorEngine) =>
             handleReadStyleGuideTool(editorEngine),
     },
     [ONLOOK_INSTRUCTIONS_TOOL_NAME]: {
         name: ONLOOK_INSTRUCTIONS_TOOL_NAME,
         inputSchema: EMPTY_TOOL_PARAMETERS,
-        handler: async () => ONLOOK_INSTRUCTIONS,
+        handler: async (_args: unknown, _editorEngine: EditorEngine) => ONLOOK_INSTRUCTIONS,
     },
     [SEARCH_REPLACE_EDIT_FILE_TOOL_NAME]: {
         name: SEARCH_REPLACE_EDIT_FILE_TOOL_NAME,
@@ -176,17 +176,17 @@ export async function handleToolCall(toolCall: ToolCall<string, unknown>, editor
         const currentChatMode = editorEngine.state.chatMode;
         const availableTools = await getToolSetFromType(currentChatMode);
 
-        if (!availableTools[toolName]) {            
+        if (!availableTools[toolName]) {
             toast.error(`Tool "${toolName}" not available in ask mode`, {
                 description: `Switch to build mode to use this tool.`,
                 duration: 2000,
             });
-            
+
             throw new Error(`Tool "${toolName}" is not available in ${currentChatMode} mode`);
         }
 
         const clientTool = TOOL_HANDLERS[toolName];
-        
+
         if (!clientTool) {
             throw new Error(`Unknown tool call: ${toolName}`);
         }