Skip to content

Commit 0ccf382

Browse files
committed
Fix merge conflicts with new llm providers
1 parent 67ca455 commit 0ccf382

File tree

5 files changed

+59
-7
lines changed

5 files changed

+59
-7
lines changed

.env.example

+3
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,9 @@ GROQ_API_KEY= # Starts with gsk_
1414
OPENROUTER_API_KEY=
1515
GOOGLE_GENERATIVE_AI_API_KEY= # Gemini API key
1616

17+
ALI_BAILIAN_API_KEY= # Ali Bailian API Key
18+
VOLENGINE_API_KEY= # VolEngine API Key
19+
1720
# Speech Synthesis
1821
ELEVENLABS_XI_API_KEY= # API key from elevenlabs
1922

agent/src/index.ts

+10
Original file line numberDiff line numberDiff line change
@@ -248,6 +248,16 @@ export function getTokenForProvider(
248248
return (
249249
character.settings?.secrets?.FAL_API_KEY || settings.FAL_API_KEY
250250
);
251+
case ModelProviderName.ALI_BAILIAN:
252+
return (
253+
character.settings?.secrets?.ALI_BAILIAN_API_KEY ||
254+
settings.ALI_BAILIAN_API_KEY
255+
);
256+
case ModelProviderName.VOLENGINE:
257+
return (
258+
character.settings?.secrets?.VOLENGINE_API_KEY ||
259+
settings.VOLENGINE_API_KEY
260+
);
251261
}
252262
}
253263

packages/core/src/generation.ts

+4
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,8 @@ export async function generateText({
118118
// OPENAI & LLAMACLOUD shared same structure.
119119
case ModelProviderName.OPENAI:
120120
case ModelProviderName.ETERNALAI:
121+
case ModelProviderName.ALI_BAILIAN:
122+
case ModelProviderName.VOLENGINE:
121123
case ModelProviderName.LLAMACLOUD: {
122124
elizaLogger.debug("Initializing OpenAI model.");
123125
const openai = createOpenAI({ apiKey, baseURL: endpoint });
@@ -1151,6 +1153,8 @@ export async function handleProvider(
11511153
switch (provider) {
11521154
case ModelProviderName.OPENAI:
11531155
case ModelProviderName.ETERNALAI:
1156+
case ModelProviderName.ALI_BAILIAN:
1157+
case ModelProviderName.VOLENGINE:
11541158
case ModelProviderName.LLAMACLOUD:
11551159
return await handleOpenAI(options);
11561160
case ModelProviderName.ANTHROPIC:

packages/core/src/models.ts

+38-7
Original file line numberDiff line numberDiff line change
@@ -314,16 +314,47 @@ export const models: Models = {
314314
},
315315
endpoint: settings.GAIANET_SERVER_URL || "http://localhost:8080/v1",
316316
model: {
317-
[ModelClass.SMALL]:
318-
settings.GAIANET_MODEL || "llama3.2",
319-
[ModelClass.MEDIUM]:
320-
settings.GAIANET_MODEL || "llama3.2",
321-
[ModelClass.LARGE]:
322-
settings.GAIANET_MODEL || "llama3.2",
317+
[ModelClass.SMALL]: settings.GAIANET_MODEL || "llama3.2",
318+
[ModelClass.MEDIUM]: settings.GAIANET_MODEL || "llama3.2",
319+
[ModelClass.LARGE]: settings.GAIANET_MODEL || "llama3.2",
323320
[ModelClass.EMBEDDING]:
324321
settings.GAIANET_EMBEDDING_MODEL || "nomic-embed",
325322
},
326-
}
323+
},
324+
[ModelProviderName.ALI_BAILIAN]: {
325+
endpoint: "https://dashscope.aliyuncs.com/compatible-mode/v1",
326+
settings: {
327+
stop: [],
328+
maxInputTokens: 128000,
329+
maxOutputTokens: 8192,
330+
frequency_penalty: 0.4,
331+
presence_penalty: 0.4,
332+
temperature: 0.6,
333+
},
334+
model: {
335+
[ModelClass.SMALL]: "qwen-turbo",
336+
[ModelClass.MEDIUM]: "qwen-plus",
337+
[ModelClass.LARGE]: "qwen-max",
338+
[ModelClass.IMAGE]: "wanx-v1",
339+
},
340+
},
341+
[ModelProviderName.VOLENGINE]: {
342+
endpoint: "https://open.volcengineapi.com/api/v3/",
343+
settings: {
344+
stop: [],
345+
maxInputTokens: 128000,
346+
maxOutputTokens: 8192,
347+
frequency_penalty: 0.4,
348+
presence_penalty: 0.4,
349+
temperature: 0.6,
350+
},
351+
model: {
352+
[ModelClass.SMALL]: "doubao-lite-128k",
353+
[ModelClass.MEDIUM]: "doubao-pro-128k",
354+
[ModelClass.LARGE]: "doubao-pro-128k",
355+
[ModelClass.EMBEDDING]: "doubao-embedding",
356+
},
357+
},
327358
};
328359

329360
export function getModel(provider: ModelProviderName, type: ModelClass) {

packages/core/src/types.ts

+4
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,8 @@ export type Models = {
202202
[ModelProviderName.GALADRIEL]: Model;
203203
[ModelProviderName.FAL]: Model;
204204
[ModelProviderName.GAIANET]: Model;
205+
[ModelProviderName.ALI_BAILIAN]: Model;
206+
[ModelProviderName.VOLENGINE]: Model;
205207
};
206208

207209
/**
@@ -224,6 +226,8 @@ export enum ModelProviderName {
224226
GALADRIEL = "galadriel",
225227
FAL = "falai",
226228
GAIANET = "gaianet",
229+
ALI_BAILIAN = "ali_bailian",
230+
VOLENGINE = "volengine",
227231
}
228232

229233
/**

0 commit comments

Comments
 (0)