update volcengine model

elizaOS · Dec 31, 2024 · 5a7b5c5 · 5a7b5c5
1 parent 7ab53c2
commit 5a7b5c5
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 4 deletions.
diff --git a/.env.example b/.env.example
@@ -162,6 +162,13 @@ LARGE_GAIANET_SERVER_URL=       # Default: https://qwen72b.gaia.domains/v1
 GAIANET_EMBEDDING_MODEL=
 USE_GAIANET_EMBEDDING=          # Set to TRUE for GAIANET/768, leave blank for local
 
+# Volcengine Configuration
+VOLENGINE_MODEL=
+SMALL_VOLENGINE_MODEL=          # Default: doubao-lite-128k
+MEDIUM_VOLENGINE_MODEL=         # Default: doubao-pro-128k
+LARGE_VOLENGINE_MODEL=          # Default: doubao-pro-256k
+EMBEDDING_VOLENGINE_MODEL=      # Default: doubao-embedding
+
 # EVM
 EVM_PRIVATE_KEY=
 EVM_PROVIDER_URL=

diff --git a/packages/core/src/models.ts b/packages/core/src/models.ts
@@ -405,10 +405,21 @@ export const models: Models = {
             temperature: 0.6,
         },
         model: {
-            [ModelClass.SMALL]: "doubao-lite-128k",
-            [ModelClass.MEDIUM]: "doubao-pro-128k",
-            [ModelClass.LARGE]: "doubao-pro-256k",
-            [ModelClass.EMBEDDING]: "doubao-embedding",
+            [ModelClass.SMALL]:
+                settings.SMALL_VOLENGINE_MODEL ||
+                settings.VOLENGINE_MODEL ||
+                "doubao-lite-128k",
+            [ModelClass.MEDIUM]:
+                settings.MEDIUM_VOLENGINE_MODEL ||
+                settings.VOLENGINE_MODEL ||
+                "doubao-pro-128k",
+            [ModelClass.LARGE]:
+                settings.LARGE_VOLENGINE_MODEL ||
+                settings.VOLENGINE_MODEL ||
+                "doubao-pro-256k",
+            [ModelClass.EMBEDDING]:
+                settings.EMBEDDING_VOLENGINE_MODEL ||
+                "doubao-embedding",
         },
     },
     [ModelProviderName.NANOGPT]: {