Update main.ts
Browse files
main.ts
CHANGED
|
@@ -637,8 +637,7 @@ class OpenAICompatibleServer {
|
|
| 637 |
} else if (this.googleAI.isImageGenerationModel(requestedModel)) {
|
| 638 |
responseText = await this.googleAI.generateOrEditImage(content, requestedModel);
|
| 639 |
} else if (requestedModel.endsWith("-search")) {
|
| 640 |
-
const
|
| 641 |
-
const searchMessages = [{ ...lastMessage, content: query }];
|
| 642 |
responseText = await this.googleAI.generateContentWithGrounding(searchMessages, requestedModel.slice(0, -"-search".length));
|
| 643 |
} else {
|
| 644 |
responseText = await this.googleAI.generateContent(body.messages, requestedModel, false);
|
|
|
|
| 637 |
} else if (this.googleAI.isImageGenerationModel(requestedModel)) {
|
| 638 |
responseText = await this.googleAI.generateOrEditImage(content, requestedModel);
|
| 639 |
} else if (requestedModel.endsWith("-search")) {
|
| 640 |
+
const searchMessages = [{ ...lastMessage, content: content }];
|
|
|
|
| 641 |
responseText = await this.googleAI.generateContentWithGrounding(searchMessages, requestedModel.slice(0, -"-search".length));
|
| 642 |
} else {
|
| 643 |
responseText = await this.googleAI.generateContent(body.messages, requestedModel, false);
|