xt8 commited on
Commit
ab1758d
·
verified ·
1 Parent(s): 9e4db6f

Update main.ts

Browse files
Files changed (1) hide show
  1. main.ts +1 -2
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 query = content.substring(8).trim();
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);