From 4569a2492838498ffd680b41bc175da21328a04e Mon Sep 17 00:00:00 2001 From: Saxon Fletcher Date: Mon, 1 Dec 2025 19:11:55 +1000 Subject: [PATCH] Fix clone object assistant (#40932) fix clone object assistant Co-authored-by: Joshen Lim --- apps/studio/state/ai-assistant-state.tsx | 14 ++------------ 1 file changed, 2 insertions(+), 12 deletions(-) diff --git a/apps/studio/state/ai-assistant-state.tsx b/apps/studio/state/ai-assistant-state.tsx index 731f9b05cde..8c7c8534551 100644 --- a/apps/studio/state/ai-assistant-state.tsx +++ b/apps/studio/state/ai-assistant-state.tsx @@ -18,7 +18,7 @@ type SuggestionsType = { prompts?: { label: string; description: string }[] } -export type AssistantMessageType = MessageType & { results?: { [id: string]: any[] } } +export type AssistantMessageType = MessageType export type SqlSnippet = string | { label: string; content: string } @@ -230,7 +230,7 @@ function createChatInstance( ) { return new Chat({ id: options.id, - messages: options.initialMessages, + messages: options.initialMessages.map((message) => sanitizeForCloning(message)), sendAutomaticallyWhen: lastAssistantMessageIsCompleteWithToolCalls, transport: new DefaultChatTransport({ api: `${BASE_PATH}/api/ai/sql/generate-v4`, @@ -461,15 +461,6 @@ export const createAiAssistantState = (): AiAssistantState => { state.suggestions = undefined }, - getCachedSQLResults: ({ messageId, snippetId }: { messageId: string; snippetId?: string }) => { - const chat = state.activeChat - if (!chat || !snippetId) return - - const message = chat.messages.find((msg) => msg.id === messageId) - const results = (message?.results ?? {})[snippetId] - return results - }, - // --- New function to load persisted state --- loadPersistedState: (persistedState: StoredAiAssistantState) => { state.chats = persistedState.chats @@ -535,7 +526,6 @@ export type AiAssistantState = AiAssistantData & { updateMessage: (message: MessageType) => void setSqlSnippets: (snippets: SqlSnippet[]) => void clearSqlSnippets: () => void - getCachedSQLResults: (args: { messageId: string; snippetId?: string }) => any[] | undefined loadPersistedState: (persistedState: StoredAiAssistantState) => void clearStorage: () => Promise }