feat: reorder togetherai (#1951)

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

View File

@ -148,7 +148,9 @@ class AIModel(ABC):
position_map = {}
if os.path.exists(position_file_path):
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
for model_schema_yaml_path in model_schema_yaml_paths:

View File

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

View File

@ -217,7 +217,9 @@ class ModelProviderFactory:
position_map = {}
if os.path.exists(position_file_path):
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
for model_provider_dir_path in model_provider_dir_paths:

View File

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