Unverified Commit 2f120786 authored by takatost's avatar takatost Committed by GitHub

feat: reorder togetherai (#1951)

parent 6075fee5
...@@ -148,7 +148,9 @@ class AIModel(ABC): ...@@ -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:
......
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
\ No newline at end of file - openai_api_compatible
\ No newline at end of file
...@@ -217,7 +217,9 @@ class ModelProviderFactory: ...@@ -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:
......
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
\ No newline at end of file - gpt-3.5-turbo-instruct
- text-davinci-003
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment