Spaces:
Runtime error
Runtime error
""" | |
This addon is used to make OpenAI model names point to one of these models. The names are semi-random picked. | |
""" | |
model_map = [ | |
{"gpt-4-0613": "mistralai/Mistral-7B-Instruct-v0.3"}, | |
{"davinci-002": ""}, | |
{"gpt-4": "microsoft/WizardLM-2-8x22B"}, | |
{"gpt-4o": "mistralai/Mistral-7B-Instruct-v0.2"}, | |
{"gpt-4-0125-preview": "mistralai/Mixtral-8x7B-Instruct-v0.1"}, | |
{"gpt-4-turbo-preview": "mistralai/Mixtral-8x22B-Instruct-v0.1"}, | |
{"gpt-4-1106-preview": "cognitivecomputations/dolphin-2.6-mixtral-8x7b"}, | |
{"gpt-3.5-turbo-16k": "meta-llama/Meta-Llama-3-8B-Instruct"}, | |
{"gpt-3.5-turbo-instruct-0914": "meta-llama/Meta-Llama-3-70B-Instruct"}, | |
] | |
def translate(model: str) -> str: | |
""" | |
Translate model names to the actual model names | |
""" | |
for i in model_map: | |
if model in i: | |
return i[model] | |
return model | |
def message_translation(messages: list[dict[str, str]]) -> list[dict[str, str]]: | |
""" | |
Translate messages | |
Current Message format | |
messages = [{"role": "user", "content": "Whatever"}] | |
Old format: | |
messages = [{"role": "user", "content": [{"text": "Whatever"}]}] | |
:param messages: list of messages | |
:return: list of messages | |
""" | |
translated_messages = [] | |
for message in messages: | |
# Check if the message content is in the old format | |
if isinstance(message['content'], list): | |
# Extract the text from the old format and assign it to the new format | |
content = message['content'][0]['text'] | |
message['content'] = content | |
# Append the message to the list of translated messages | |
translated_messages.append(message) | |
return translated_messages | |