Skip to content

Commit c49893e

Browse files
committed
Merge branch 'main' of https://github.com/continuedev/continue into dependabot/npm_and_yarn/gui/mdast-util-to-hast-13.2.1
2 parents 4d6ab65 + 34fdeb8 commit c49893e

27 files changed

+2549
-2746
lines changed

binary/package-lock.json

Lines changed: 2 additions & 2 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

core/llm/llms/Gemini.ts

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ import { safeParseToolCallArgs } from "../../tools/parseArgs.js";
1313
import { renderChatMessage, stripImages } from "../../util/messageContent.js";
1414
import { extractBase64FromDataUrl } from "../../util/url.js";
1515
import { BaseLLM } from "../index.js";
16+
import { LlmApiRequestType } from "../openaiTypeConverters.js";
1617
import {
1718
GeminiChatContent,
1819
GeminiChatContentPart,
@@ -41,6 +42,15 @@ class Gemini extends BaseLLM {
4142
maxEmbeddingBatchSize: 100,
4243
};
4344

45+
protected useOpenAIAdapterFor: (LlmApiRequestType | "*")[] = [
46+
"chat",
47+
"embed",
48+
"list",
49+
"rerank",
50+
"streamChat",
51+
"streamFim",
52+
];
53+
4454
// Function to convert completion options to Gemini format
4555
public convertArgs(options: CompletionOptions): GeminiGenerationConfig {
4656
// should be public for use within VertexAI

core/llm/llms/VertexAI.ts

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ import { ChatMessage, CompletionOptions, LLMOptions } from "../../index.js";
55
import { renderChatMessage, stripImages } from "../../util/messageContent.js";
66
import { BaseLLM } from "../index.js";
77

8+
import { LlmApiRequestType } from "../openaiTypeConverters.js";
89
import Anthropic from "./Anthropic.js";
910
import Gemini from "./Gemini.js";
1011

@@ -23,6 +24,15 @@ class VertexAI extends BaseLLM {
2324

2425
private clientPromise: Promise<AuthClient | void>;
2526

27+
protected useOpenAIAdapterFor: (LlmApiRequestType | "*")[] = [
28+
"chat",
29+
"embed",
30+
"list",
31+
"rerank",
32+
"streamChat",
33+
"streamFim",
34+
];
35+
2636
/*
2737
Vertex Supports 3 different URL formats
2838
1. Standard VertexAI: e.g. https://{location}-aiplatform.googleapis.com/v1/projects/{project}/locations/{location}/publishers/google/models/{model}:streamGenerateContent

core/package-lock.json

Lines changed: 111 additions & 35 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)