Spaces:
Running
Running
File size: 2,800 Bytes
f871fed |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
import apiClient from './client'
import {
SourceChatSession,
SourceChatSessionWithMessages,
CreateSourceChatSessionRequest,
UpdateSourceChatSessionRequest,
SendMessageRequest
} from '@/lib/types/api'
export const sourceChatApi = {
// Session management
createSession: async (sourceId: string, data: Omit<CreateSourceChatSessionRequest, 'source_id'>) => {
// Extract clean ID without "source:" prefix for the request body
const cleanId = sourceId.startsWith('source:') ? sourceId.slice(7) : sourceId
const response = await apiClient.post<SourceChatSession>(
`/sources/${sourceId}/chat/sessions`,
{ ...data, source_id: cleanId } // Include source_id in the request body
)
return response.data
},
listSessions: async (sourceId: string) => {
const response = await apiClient.get<SourceChatSession[]>(
`/sources/${sourceId}/chat/sessions`
)
return response.data
},
getSession: async (sourceId: string, sessionId: string) => {
const response = await apiClient.get<SourceChatSessionWithMessages>(
`/sources/${sourceId}/chat/sessions/${sessionId}`
)
return response.data
},
updateSession: async (sourceId: string, sessionId: string, data: UpdateSourceChatSessionRequest) => {
const response = await apiClient.put<SourceChatSession>(
`/sources/${sourceId}/chat/sessions/${sessionId}`,
data
)
return response.data
},
deleteSession: async (sourceId: string, sessionId: string) => {
await apiClient.delete(`/sources/${sourceId}/chat/sessions/${sessionId}`)
},
// Messaging with streaming
sendMessage: (sourceId: string, sessionId: string, data: SendMessageRequest) => {
// Get auth token using the same logic as apiClient interceptor
let token = null
if (typeof window !== 'undefined') {
const authStorage = localStorage.getItem('auth-storage')
if (authStorage) {
try {
const { state } = JSON.parse(authStorage)
if (state?.token) {
token = state.token
}
} catch (error) {
console.error('Error parsing auth storage:', error)
}
}
}
// Use relative URL to leverage Next.js rewrites
// This works both in dev (Next.js proxy) and production (Docker network)
const url = `/api/sources/${sourceId}/chat/sessions/${sessionId}/messages`
// Use fetch with ReadableStream for SSE
return fetch(url, {
method: 'POST',
headers: {
'Content-Type': 'application/json',
...(token && { 'Authorization': `Bearer ${token}` })
},
body: JSON.stringify(data)
}).then(response => {
if (!response.ok) {
throw new Error(`HTTP error! status: ${response.status}`)
}
return response.body
})
}
}
|