fix: [#5308] gemini对话总结

This commit is contained in:
DDMeaqua 2024-11-13 14:24:44 +08:00
parent 819d249a09
commit 7eda14f138

View File

@ -51,8 +51,10 @@ export class GeminiProApi implements LLMApi {
console.log("[Proxy Endpoint] ", baseUrl, path);
let chatPath = [baseUrl, path].join("/");
if (!chatPath.includes("gemini-pro")) {
chatPath += chatPath.includes("?") ? "&alt=sse" : "?alt=sse";
}
return chatPath;
}
extractMessage(res: any) {
@ -60,6 +62,7 @@ export class GeminiProApi implements LLMApi {
return (
res?.candidates?.at(0)?.content?.parts.at(0)?.text ||
res?.at(0)?.candidates?.at(0)?.content?.parts.at(0)?.text ||
res?.error?.message ||
""
);
@ -167,6 +170,7 @@ export class GeminiProApi implements LLMApi {
try {
// https://github.com/google-gemini/cookbook/blob/main/quickstarts/rest/Streaming_REST.ipynb
const chatPath = this.path(Google.ChatPath(modelConfig.model));
console.log("[Chat Path] ", chatPath);
const chatPayload = {
method: "POST",