Spaces:
Sleeping
Sleeping
File size: 2,781 Bytes
a7c9ddd a60c0af a7c9ddd a60c0af | 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 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 | // Use empty string for same-origin requests in production (HF Spaces)
// Only fallback to localhost:8000 if VITE_API_BASE_URL is truly undefined
const baseURL =
import.meta.env.VITE_API_BASE_URL !== undefined
? import.meta.env.VITE_API_BASE_URL
: "http://localhost:8000";
export interface ResearchRequest {
topic: string;
search_api?: string;
}
export interface ResearchStreamEvent {
type: string;
[key: string]: unknown;
}
export interface StreamOptions {
signal?: AbortSignal;
}
export async function runResearchStream(
payload: ResearchRequest,
onEvent: (event: ResearchStreamEvent) => void,
options: StreamOptions = {}
): Promise<void> {
const response = await fetch(`${baseURL}/research/stream`, {
method: "POST",
headers: {
"Content-Type": "application/json",
Accept: "text/event-stream"
},
body: JSON.stringify(payload),
signal: options.signal
});
if (!response.ok) {
const errorText = await response.text().catch(() => "");
throw new Error(
errorText || `Research request failed, status code: ${response.status}`
);
}
const body = response.body;
if (!body) {
throw new Error("Browser does not support streaming responses, unable to get research progress");
}
const reader = body.getReader();
const decoder = new TextDecoder("utf-8");
let buffer = "";
while (true) {
const { value, done } = await reader.read();
buffer += decoder.decode(value || new Uint8Array(), { stream: !done });
let boundary = buffer.indexOf("\n\n");
while (boundary !== -1) {
const rawEvent = buffer.slice(0, boundary).trim();
buffer = buffer.slice(boundary + 2);
if (rawEvent.startsWith("data:")) {
const dataPayload = rawEvent.slice(5).trim();
if (dataPayload) {
try {
const event = JSON.parse(dataPayload) as ResearchStreamEvent;
onEvent(event);
if (event.type === "error" || event.type === "done") {
return;
}
} catch (error) {
console.error("Failed to parse streaming event:", error, dataPayload);
}
}
}
boundary = buffer.indexOf("\n\n");
}
if (done) {
// Handle possible trailing events
if (buffer.trim()) {
const rawEvent = buffer.trim();
if (rawEvent.startsWith("data:")) {
const dataPayload = rawEvent.slice(5).trim();
if (dataPayload) {
try {
const event = JSON.parse(dataPayload) as ResearchStreamEvent;
onEvent(event);
} catch (error) {
console.error("Failed to parse streaming event:", error, dataPayload);
}
}
}
}
break;
}
}
}
|