Skip to content

Commit 560d375

Browse files
authored
feat(ark): add doubao-pro-256k and doubao-embedding-large (#11831)
1 parent 3388d66 commit 560d375

File tree

3 files changed

+18
-1
lines changed

3 files changed

+18
-1
lines changed

api/core/model_runtime/model_providers/volcengine_maas/llm/models.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,10 @@ class ModelConfig(BaseModel):
4040
properties=ModelProperties(context_size=32768, max_tokens=4096, mode=LLMMode.CHAT),
4141
features=[ModelFeature.TOOL_CALL],
4242
),
43+
"Doubao-pro-256k": ModelConfig(
44+
properties=ModelProperties(context_size=262144, max_tokens=4096, mode=LLMMode.CHAT),
45+
features=[],
46+
),
4347
"Doubao-pro-128k": ModelConfig(
4448
properties=ModelProperties(context_size=131072, max_tokens=4096, mode=LLMMode.CHAT),
4549
features=[ModelFeature.TOOL_CALL],

api/core/model_runtime/model_providers/volcengine_maas/text_embedding/models.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ class ModelConfig(BaseModel):
1212

1313
ModelConfigs = {
1414
"Doubao-embedding": ModelConfig(properties=ModelProperties(context_size=4096, max_chunks=32)),
15+
"Doubao-embedding-large": ModelConfig(properties=ModelProperties(context_size=4096, max_chunks=32)),
1516
}
1617

1718

@@ -21,7 +22,7 @@ def get_model_config(credentials: dict) -> ModelConfig:
2122
if not model_configs:
2223
return ModelConfig(
2324
properties=ModelProperties(
24-
context_size=int(credentials.get("context_size", 0)),
25+
context_size=int(credentials.get("context_size", 4096)),
2526
max_chunks=int(credentials.get("max_chunks", 1)),
2627
)
2728
)

api/core/model_runtime/model_providers/volcengine_maas/volcengine_maas.yaml

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -166,6 +166,12 @@ model_credential_schema:
166166
show_on:
167167
- variable: __model_type
168168
value: llm
169+
- label:
170+
en_US: Doubao-pro-256k
171+
value: Doubao-pro-256k
172+
show_on:
173+
- variable: __model_type
174+
value: llm
169175
- label:
170176
en_US: Llama3-8B
171177
value: Llama3-8B
@@ -220,6 +226,12 @@ model_credential_schema:
220226
show_on:
221227
- variable: __model_type
222228
value: text-embedding
229+
- label:
230+
en_US: Doubao-embedding-large
231+
value: Doubao-embedding-large
232+
show_on:
233+
- variable: __model_type
234+
value: text-embedding
223235
- label:
224236
en_US: Custom
225237
zh_Hans: 自定义

0 commit comments

Comments
 (0)