export type CreateApiApiAiV1betaModelsModelStreamGenerateContentArgs = {model: string, body: {model: string, contents: Array<{role?: ('user' | 'model') | undefined, parts: Array<{text: string} | {inlineData: {mimeType: string, data: string}} | {fileData: {mimeType?: string | undefined, fileUri: string}}>}>, config?: ({temperature?: number | undefined, topP?: number | undefined, topK?: number | undefined, candidateCount?: number | undefined, maxOutputTokens?: number | undefined, responseLogprobs?: boolean | undefined, logprobs?: number | undefined, presencePenalty?: number | undefined, frequencyPenalty?: number | undefined, seed?: number | undefined, responseMimeType?: string | undefined, responseJsonSchema: unknown} | {[key: string]: unknown}) | undefined}}; export type CreateApiApiAiV1betaModelsModelStreamGenerateContentOptions = {baseUrl?: string | undefined, apikeyHeaderXGoogApiKey?: string | undefined, fetch?: typeof fetch | undefined}; /** * Gemini 流式对话 * * POST /api/ai/v1beta/models/{model}:streamGenerateContent */ export const createApiApiAiV1betaModelsModelStreamGenerateContent = async (args: CreateApiApiAiV1betaModelsModelStreamGenerateContentArgs, options?: CreateApiApiAiV1betaModelsModelStreamGenerateContentOptions) => { const {baseUrl = "https://aitoearn.cn", fetch: fetchFn = fetch, apikeyHeaderXGoogApiKey} = options ?? {} const query = new URLSearchParams() const headers = new Headers() headers.set('content-type', "application/json") headers.set('accept', "text/event-stream") if (apikeyHeaderXGoogApiKey !== undefined) headers.set("x-goog-api-key", apikeyHeaderXGoogApiKey) const url = `${baseUrl.replace(/\/+$/, '')}/api/ai/v1beta/models/${encodeURIComponent(String(args.model))}:streamGenerateContent${query.size > 0 ? `?${query}` : ''}` const res = await fetchFn(url, {method: "POST", headers, body: JSON.stringify(args.body)}) if (!res.ok) throw new Error(`Request failed with status ${res.status}`, {cause: {status: res.status, body: await res.text()}}) return res.text() };