#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@Time    :   2024/08/21 18:22:06
@Author  :   ChenHao
@Description  :   
@Contact :   jerrychen1990@gmail.com
'''


from liteai.core import ModelCard, ModelType

ZHIPU_MODELS = [
    ModelCard(name="glm-4-0520", description="glm-4-0520", provider="zhipu"),
    ModelCard(name="glm-4-air", description="glm-4-air", provider="zhipu"),
    ModelCard(name="glm-4-flash", description="glm-4-flash", provider="zhipu"),
    ModelCard(name="glm-4v", description="glm-4v", provider="zhipu", support_vision=True)
]

OPENAI_MODELS = [
    ModelCard(name="gpt-4o", description="gpt-4o", provider="openai"),
    ModelCard(name="gpt-4", description="gpt-4", provider="openai"),
    ModelCard(name="gpt-4o-mini", description="gpt-4o-mini", provider="openai")
]

MINIMAX_MODELS = [
    ModelCard(name="abab6.5s-chat", description="abab6.5s-chat", provider="minimax"),
    ModelCard(name="speech-01-turbo", description="speech-01-turbo", provider="minimax", model_type=ModelType.TTS)
]

QWEN_MODELS = [
    ModelCard(name="qwen-turbo", description="qwen-turbo", provider="qwen"),
    ModelCard(name="qwen-vl-plus", description="qwen-vl-plus", provider="qwen", support_vision=True)
]


ALL_MODELS = ZHIPU_MODELS + OPENAI_MODELS + MINIMAX_MODELS + QWEN_MODELS
