Function callGeminiChat

outdated/gemini/gemini.ts:905–976

(
  prompt: string,
  useSearch: boolean = false,
  imageData?: string
)

}

async function callGeminiChat(
  prompt: string,
  useSearch: boolean = false,
  imageData?: string
): Promise<string> {
  const client = await getGeminiClient();
  const modelKey = useSearch ? CONFIG_KEYS.GEMINI_SEARCH_MODEL : CONFIG_KEYS.GEMINI_CHAT_MODEL;
  const modelName = getConfig(modelKey);

  const activePromptKey = useSearch ? CONFIG_KEYS.GEMINI_SEARCH_ACTIVE_PROMPT : CONFIG_KEYS.GEMINI_CHAT_ACTIVE_PROMPT;
  const systemPromptName = getConfig(activePromptKey);
  const prompts = JSON.parse(getConfig(CONFIG_KEYS.GEMINI_PROMPTS, "{}"));
  const systemPrompt = systemPromptName ? prompts[systemPromptName] || "你是一个乐于助人的人工智能助手。" : "你是一个乐于助人的人工智能助手。";

  const baseParts: any[] = [{ text: prompt }];
  if (imageData) {
    baseParts.push({ inlineData: { mimeType: "image/png", data: imageData } });
  }
  let contents: any[] = [{ role: "user", parts: baseParts }];

  if (getConfig(CONFIG_KEYS.GEMINI_CONTEXT_ENABLED) === "on" && !useSearch) {
    const history = JSON.parse(getConfig(CONFIG_KEYS.GEMINI_CHAT_HISTORY, "[]"));
    const cleanHistory = history.filter((item: any) => {
      return item.role && item.parts && item.parts.every((part: any) => 
        part.text && typeof part.text === 'string' && !part.inlineData
      );
    });
    
    if (history.length > 0 && history.some((item: any) => !item.role)) {

      ConfigManager.set(CONFIG_KEYS.GEMINI_CHAT_HISTORY, "[]");
    } else {
      contents = [...cleanHistory, ...contents];
    }
  }

  const maxTokens = parseInt(getConfig(CONFIG_KEYS.GEMINI_MAX_TOKENS, "0"));
  const tools = useSearch ? [{ googleSearch: {} }] : undefined;

  let response;
  try {
    response = await client.generateContent({
      model: modelName,
      contents,
      systemInstruction: systemPrompt,
      safetySettings: createSafetySettings(),
      maxOutputTokens: maxTokens > 0 ? maxTokens : undefined,
      tools
    });

  } catch (error: any) {

    throw error;
  }

  if (getConfig(CONFIG_KEYS.GEMINI_CONTEXT_ENABLED) === "on" && !useSearch) {
    const history = JSON.parse(getConfig(CONFIG_KEYS.GEMINI_CHAT_HISTORY, "[]"));
    history.push(