AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 20:59:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: reorder togetherai (#1951)

Browse Source
This commit is contained in:
takatost 2024-01-05 17:04:37 +08:00 committed by GitHub
parent 6075fee556
commit 2f120786fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 37 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/model_runtime/model_providers/__base/ai_model.py
View File

@ -148,7 +148,9 @@ class AIModel(ABC):
position_map = {} position_map = {}
if os.path.exists(position_file_path): if os.path.exists(position_file_path):
with open(position_file_path, 'r', encoding='utf-8') as f: with open(position_file_path, 'r', encoding='utf-8') as f:
position_map = yaml.safe_load(f) positions = yaml.safe_load(f)
# convert list to dict with key as model provider name, value as index
position_map = {position: index for index, position in enumerate(positions)}
# traverse all model_schema_yaml_paths # traverse all model_schema_yaml_paths
for model_schema_yaml_path in model_schema_yaml_paths: for model_schema_yaml_path in model_schema_yaml_paths:

39
api/core/model_runtime/model_providers/_position.yaml
View File

@ -1,19 +1,20 @@
openai: 0 - openai
anthropic: 1 - anthropic
azure_openai: 2 - azure_openai
google: 3 - google
replicate: 4 - replicate
huggingface_hub: 5 - huggingface_hub
cohere: 6 - cohere
zhipuai: 7 - togetherai
baichuan: 8 - zhipuai
spark: 9 - baichuan
minimax: 10 - spark
tongyi: 11 - minimax
wenxin: 12 - tongyi
jina: 13 - wenxin
chatglm: 14 - jina
xinference: 15 - chatglm
openllm: 16 - xinference
localai: 17 - openllm
openai_api_compatible: 18 - localai
- openai_api_compatible

4
api/core/model_runtime/model_providers/model_provider_factory.py
View File

@ -217,7 +217,9 @@ class ModelProviderFactory:
position_map = {} position_map = {}
if os.path.exists(position_file_path): if os.path.exists(position_file_path):
with open(position_file_path, 'r', encoding='utf-8') as f: with open(position_file_path, 'r', encoding='utf-8') as f:
position_map = yaml.safe_load(f) positions = yaml.safe_load(f)
# convert list to dict with key as model provider name, value as index
position_map = {position: index for index, position in enumerate(positions)}
# traverse all model_provider_dir_paths # traverse all model_provider_dir_paths
for model_provider_dir_path in model_provider_dir_paths: for model_provider_dir_path in model_provider_dir_paths:

20
api/core/model_runtime/model_providers/openai/llm/_position.yaml
View File

@ -1,9 +1,11 @@
gpt-4: 0 - gpt-4
gpt-4-32k: 1 - gpt-4-32k
gpt-4-1106-preview: 2 - gpt-4-1106-preview
gpt-4-vision-preview: 3 - gpt-4-vision-preview
gpt-3.5-turbo: 4 - gpt-3.5-turbo
gpt-3.5-turbo-16k: 5 - gpt-3.5-turbo-16k
gpt-3.5-turbo-1106: 6 - gpt-3.5-turbo-16k-0613
gpt-3.5-turbo-instruct: 7 - gpt-3.5-turbo-1106
text-davinci-003: 8 - gpt-3.5-turbo-0613
- gpt-3.5-turbo-instruct
- text-davinci-003