Spaces:
Running
Running
| import { base } from "$app/paths"; | |
| import { browser } from "$app/environment"; | |
| import superjson from "superjson"; | |
| import ObjectId from "bson-objectid"; | |
| superjson.registerCustom<ObjectId, string>( | |
| { | |
| isApplicable: (value): value is ObjectId => { | |
| if (typeof value !== "string" && ObjectId.isValid(value)) { | |
| const str = value.toString(); | |
| return /^[0-9a-fA-F]{24}$/.test(str); | |
| } | |
| return false; | |
| }, | |
| serialize: (value) => value.toString(), | |
| deserialize: (value) => new ObjectId(value), | |
| }, | |
| "ObjectId" | |
| ); | |
| type FetchFn = typeof globalThis.fetch; | |
| interface ApiResponse<T = unknown> { | |
| data: T | null; | |
| error: unknown; | |
| status: number; | |
| } | |
| async function apiCall<T = unknown>( | |
| fetcher: FetchFn, | |
| url: string, | |
| method: string, | |
| body?: unknown, | |
| query?: Record<string, string | number | undefined> | |
| ): Promise<ApiResponse<T>> { | |
| const u = new URL(url); | |
| if (query) { | |
| for (const [k, v] of Object.entries(query)) { | |
| if (v !== undefined && v !== null) { | |
| u.searchParams.set(k, String(v)); | |
| } | |
| } | |
| } | |
| const init: RequestInit = { method }; | |
| if (body !== undefined && body !== null) { | |
| init.headers = { "Content-Type": "application/json" }; | |
| init.body = JSON.stringify(body); | |
| } | |
| const res = await fetcher(u.toString(), init); | |
| if (!res.ok) { | |
| let errorBody: unknown; | |
| try { | |
| errorBody = await res.json(); | |
| } catch { | |
| errorBody = await res.text().catch(() => res.statusText); | |
| } | |
| return { data: null, error: errorBody, status: res.status }; | |
| } | |
| // Handle empty responses (e.g. POST /user/settings returns empty body) | |
| const text = await res.text(); | |
| if (!text) { | |
| return { data: null, error: null, status: res.status }; | |
| } | |
| return { data: text as unknown as T, error: null, status: res.status }; | |
| } | |
| function endpoint(fetcher: FetchFn, baseUrl: string) { | |
| return { | |
| get(opts?: { query?: Record<string, string | number | undefined> }) { | |
| return apiCall(fetcher, baseUrl, "GET", undefined, opts?.query); | |
| }, | |
| post(body?: unknown) { | |
| return apiCall(fetcher, baseUrl, "POST", body); | |
| }, | |
| patch(body?: unknown) { | |
| return apiCall(fetcher, baseUrl, "PATCH", body); | |
| }, | |
| delete() { | |
| return apiCall(fetcher, baseUrl, "DELETE"); | |
| }, | |
| }; | |
| } | |
| export function useAPIClient({ | |
| fetch: customFetch, | |
| origin, | |
| }: { | |
| fetch?: FetchFn; | |
| origin?: string; | |
| } = {}) { | |
| const fetcher = customFetch ?? globalThis.fetch; | |
| const baseUrl = browser | |
| ? `${window.location.origin}${base}/api/v2` | |
| : `${origin ?? `http://localhost:5173`}${base}/api/v2`; | |
| return { | |
| conversations: Object.assign( | |
| // client.conversations({ id: "..." }) — returns endpoint for /conversations/:id | |
| (params: { id: string }) => ({ | |
| ...endpoint(fetcher, `${baseUrl}/conversations/${params.id}`), | |
| message: (msgParams: { messageId: string }) => | |
| endpoint(fetcher, `${baseUrl}/conversations/${params.id}/message/${msgParams.messageId}`), | |
| registry: Object.assign( | |
| endpoint(fetcher, `${baseUrl}/conversations/${params.id}/registry`), | |
| { | |
| stop: (serviceId: string) => | |
| endpoint( | |
| fetcher, | |
| `${baseUrl}/conversations/${params.id}/registry/${encodeURIComponent(serviceId)}/stop` | |
| ), | |
| } | |
| ), | |
| files: (name: string) => | |
| endpoint( | |
| fetcher, | |
| `${baseUrl}/conversations/${params.id}/files/${name.split("/").map(encodeURIComponent).join("/")}` | |
| ), | |
| runs: (runId: string) => | |
| endpoint( | |
| fetcher, | |
| `${baseUrl}/conversations/${params.id}/runs/${encodeURIComponent(runId)}` | |
| ), | |
| }), | |
| // client.conversations.get(), .delete() | |
| { | |
| ...endpoint(fetcher, `${baseUrl}/conversations`), | |
| "import-share": endpoint(fetcher, `${baseUrl}/conversations/import-share`), | |
| } | |
| ), | |
| user: { | |
| ...endpoint(fetcher, `${baseUrl}/user`), | |
| settings: endpoint(fetcher, `${baseUrl}/user/settings`), | |
| reports: endpoint(fetcher, `${baseUrl}/user/reports`), | |
| "billing-orgs": endpoint(fetcher, `${baseUrl}/user/billing-orgs`), | |
| }, | |
| models: Object.assign( | |
| // client.models({ id: "org/name" }) — per-model detail (providers, parameters). | |
| // Model ids may contain a slash; the path resolves to /models/[namespace]([/model]). | |
| (params: { id: string }) => endpoint(fetcher, `${baseUrl}/models/${params.id}`), | |
| { | |
| ...endpoint(fetcher, `${baseUrl}/models`), | |
| old: endpoint(fetcher, `${baseUrl}/models/old`), | |
| } | |
| ), | |
| spaces: { | |
| deploy: endpoint(fetcher, `${baseUrl}/spaces/deploy`), | |
| }, | |
| "public-config": endpoint(fetcher, `${baseUrl}/public-config`), | |
| "feature-flags": endpoint(fetcher, `${baseUrl}/feature-flags`), | |
| debug: { | |
| config: endpoint(fetcher, `${baseUrl}/debug/config`), | |
| }, | |
| export: endpoint(fetcher, `${baseUrl}/export`), | |
| }; | |
| } | |
| // eslint-disable-next-line @typescript-eslint/no-explicit-any | |
| export function handleResponse(response: ApiResponse<any>): any { | |
| if (response.error) { | |
| throw new Error(JSON.stringify(response.error)); | |
| } | |
| if (response.data === null) { | |
| return null; | |
| } | |
| return superjson.parse( | |
| typeof response.data === "string" ? response.data : JSON.stringify(response.data) | |
| ); | |
| } | |