Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2025-12-29 23:12:38 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

refactor(hooks): improve error handling in useSessions hook

Browse Source 
- Remove redundant agentId checks as they're handled by the API client
- Add consistent error formatting using formatErrorMessageWithPrefix
- Update error messages for all session operations
This commit is contained in:
icarus 2025-09-20 00:52:39 +08:00
parent f5acddbfeb
commit 77c64cf868
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/renderer/src/hooks/agents/useSessions.ts
View File

@ -1,4 +1,5 @@
import { CreateSessionForm, UpdateSessionForm } from '@renderer/types'
import { formatErrorMessageWithPrefix } from '@renderer/utils/error'
import { useCallback } from 'react'
import { useTranslation } from 'react-i18next'
import useSWR from 'swr'
@ -11,9 +12,6 @@ export const useSessions = (agentId: string) => {
const key = client.getSessionPaths(agentId).base
const fetcher = async () => {
if (!agentId) {
return []
}
const data = await client.listSessions(agentId)
return data.data
}
@ -21,12 +19,11 @@ export const useSessions = (agentId: string) => {
const createSession = useCallback(
async (form: CreateSessionForm) => {
if (!agentId) return
try {
const result = await client.createSession(agentId, form)
mutate((prev) => [...(prev ?? []), result])
} catch (error) {
window.toast.error(t('agent.session.create.error.failed'))
window.toast.error(formatErrorMessageWithPrefix(error, t('agent.session.create.error.failed')))
}
},
[agentId, client, mutate, t]
@ -35,12 +32,14 @@ export const useSessions = (agentId: string) => {
// TODO: including messages field
const getSession = useCallback(
async (id: string) => {
if (!agentId) return
const result = await client.getSession(agentId, id)
mutate((prev) => prev?.map((session) => (session.id === result.id ? result : session)))
return result
try {
const result = await client.getSession(agentId, id)
mutate((prev) => prev?.map((session) => (session.id === result.id ? result : session)))
} catch (error) {
window.toast.error(formatErrorMessageWithPrefix(error, t('agent.session.get.error.failed')))
}
},
[agentId, client, mutate]
[agentId, client, mutate, t]
)
const deleteSession = useCallback(
@ -50,7 +49,7 @@ export const useSessions = (agentId: string) => {
await client.deleteSession(agentId, id)
mutate((prev) => prev?.filter((session) => session.id !== id))
} catch (error) {
window.toast.error(t('agent.session.delete.error.failed'))
window.toast.error(formatErrorMessageWithPrefix(error, t('agent.session.delete.error.failed')))
}
},
[agentId, client, mutate, t]
@ -63,7 +62,7 @@ export const useSessions = (agentId: string) => {
const result = await client.updateSession(agentId, form)
mutate((prev) => prev?.map((session) => (session.id === form.id ? result : session)))
} catch (error) {
window.toast.error(t('agent.session.update.error.failed'))
window.toast.error(formatErrorMessageWithPrefix(error, t('agent.session.update.error.failed')))
}
},
[agentId, client, mutate, t]