import tos
from volcenginesdkarkruntime import Ark

class MMMConfig:
    target_size = 300 * 1024
    tos_bucket_name = "ark-auto-2100652488-cn-beijing-default"
    model="doubao-1-5-vision-pro-32k-250115"
    api_key = ('817dff39-5586-4f9b-acba-55004167c0b1')
    client = Ark(api_key=api_key)
    root_path="/data/data/luosy/project/sku_search/temp_img"
    ak = ('AKLTMDIxMmQ5NWQ0MDc1NDAzYjhjZWM4YjQ3MjM3ZDUzMjc')
    sk = ('WTJSbU1UZzVZek5rTldWa05HTTVPVGhqTmpnNE5UWmxaR0prTUdNME56aw==')
    endpoint, region = "tos-cn-beijing.volces.com", "cn-beijing"
    tos_client=tos.TosClientV2(ak, sk, endpoint, region)
    inner_tos_client=tos.TosClientV2(ak, sk, endpoint, region)
    
class LMConfig:
    lm_client = Ark(
        api_key= "817dff39-5586-4f9b-acba-55004167c0b1",
        base_url="https://ark.cn-beijing.volces.com/api/v3"
    )
    model="ep-20241018084532-cgm84"


doubao_ky=("817dff39-5586-4f9b-acba-55004167c0b1","https://ark.cn-beijing.volces.com/api/v3")
ali_ky=("sk-04b63960983445f980d85ff185a17876","https://dashscope.aliyuncs.com/compatible-mode/v1")
doubao_model={
    "text_doubao":"ep-20241018084532-cgm84",
    "text_ds":"deepseek-r1-250120",
    "mm_doubao":"doubao-1-5-vision-pro-32k-250115"
}
ali_model={
    "text_ds":"deepseek-r1",
    "text_dsv3":"deepseek-v3",
    "text_qwen":"qwen-max-latest",
    "mm_tyqw":"qwen-vl-max-latest",
    "mm_qwen":"qwen-vl-plus"
}