🧹 clean: all the missed console.log

This commit is contained in:
LazyYuuki 2025-06-15 19:12:19 +08:00
parent b93f8afda2
commit de5ddaac6d
2 changed files with 20 additions and 22 deletions

View File

@ -47,7 +47,6 @@ export const useAppState = create<AppState>()((set) => ({
} }
: undefined, : undefined,
})) }))
console.log(useAppState.getState().streamingContent)
}, },
updateCurrentToolCall: (toolCall) => { updateCurrentToolCall: (toolCall) => {
set(() => ({ set(() => ({

View File

@ -1,23 +1,23 @@
import { create } from "zustand"; import { create } from 'zustand'
import { ThreadMessage } from "@janhq/core"; import { ThreadMessage } from '@janhq/core'
import { import {
createMessage, createMessage,
deleteMessage as deleteMessageExt, deleteMessage as deleteMessageExt,
} from "@/services/messages"; } from '@/services/messages'
import { useAssistant } from "./useAssistant"; import { useAssistant } from './useAssistant'
type MessageState = { type MessageState = {
messages: Record<string, ThreadMessage[]>; messages: Record<string, ThreadMessage[]>
getMessages: (threadId: string) => ThreadMessage[]; getMessages: (threadId: string) => ThreadMessage[]
setMessages: (threadId: string, messages: ThreadMessage[]) => void; setMessages: (threadId: string, messages: ThreadMessage[]) => void
addMessage: (message: ThreadMessage) => void; addMessage: (message: ThreadMessage) => void
deleteMessage: (threadId: string, messageId: string) => void; deleteMessage: (threadId: string, messageId: string) => void
}; }
export const useMessages = create<MessageState>()((set, get) => ({ export const useMessages = create<MessageState>()((set, get) => ({
messages: {}, messages: {},
getMessages: (threadId) => { getMessages: (threadId) => {
return get().messages[threadId] || []; return get().messages[threadId] || []
}, },
setMessages: (threadId, messages) => { setMessages: (threadId, messages) => {
set((state) => ({ set((state) => ({
@ -25,11 +25,10 @@ export const useMessages = create<MessageState>()((set, get) => ({
...state.messages, ...state.messages,
[threadId]: messages, [threadId]: messages,
}, },
})); }))
}, },
addMessage: (message) => { addMessage: (message) => {
console.log("addMessage: ", message); const currentAssistant = useAssistant.getState().currentAssistant
const currentAssistant = useAssistant.getState().currentAssistant;
const newMessage = { const newMessage = {
...message, ...message,
created_at: message.created_at || Date.now(), created_at: message.created_at || Date.now(),
@ -37,7 +36,7 @@ export const useMessages = create<MessageState>()((set, get) => ({
...message.metadata, ...message.metadata,
assistant: currentAssistant, assistant: currentAssistant,
}, },
}; }
createMessage(newMessage).then((createdMessage) => { createMessage(newMessage).then((createdMessage) => {
set((state) => ({ set((state) => ({
messages: { messages: {
@ -47,19 +46,19 @@ export const useMessages = create<MessageState>()((set, get) => ({
createdMessage, createdMessage,
], ],
}, },
})); }))
}); })
}, },
deleteMessage: (threadId, messageId) => { deleteMessage: (threadId, messageId) => {
deleteMessageExt(threadId, messageId); deleteMessageExt(threadId, messageId)
set((state) => ({ set((state) => ({
messages: { messages: {
...state.messages, ...state.messages,
[threadId]: [threadId]:
state.messages[threadId]?.filter( state.messages[threadId]?.filter(
(message) => message.id !== messageId, (message) => message.id !== messageId
) || [], ) || [],
}, },
})); }))
}, },
})); }))