Function callGemini

uai/uai.ts:159–186

(
    provider: Provider,
    prompt: string,
    content: string,
    timeout: number
)

): Promise<string> {
    const url = `${trimBase(provider.base_url)}/v1beta/models/${provider.model}:generateContent`;
    const headers: Record<string, string> = { "Content-Type": "application/json" };
    const params: Record<string, string> = {};

    if (provider.auth_method === "query_param" || provider.auth_method === "bearer_token") {
        params["key"] = provider.api_key;
    }
    if (provider.auth_method === "api_key_header") {
        headers["x-goog-api-key"] = provider.api_key;
    }

    const response = await axios.post(
        url,
        {
            contents: [{ role: "user", parts: [{ text: `${prompt}\n\n${content}` }] }]
        },
        { headers, params, timeout }
    );

    const parts = response.data?.candidates?.[0]?.content?.parts || [];
    return parts.map((p: any) => p.text || "").join("");
}

async function callAI(provider: Provider, prompt: string, content: string, timeout: number): Promise<string> {
    if (provider.type === "gemini") {
        return await callGemini(provider, prompt, content, timeout);
    }
    return await callOpenAI(provider, prompt, content, timeout);
}

// ========== 消息收集 ==========
type MessageData = { time: string; sender: string; text: string };