切换OpenAI和Azure OpenAI

async def create_by_custom(query, modelType):
    temperature = 0.2
    frequency_penalty = 0
    presence_penalty = 0
    try:
        if int(modelType) == 2:
            openai.api_type = "open_ai"
            openai.api_version = None
            openai.api_base = "https://api.openai.com/v1"
            openai.api_key = "your key"
            model = 'gpt-4'
            response = openai.ChatCompletion.create(
                model=model,
                messages=query,
                temperature=temperature,
                frequency_penalty=frequency_penalty,
                presence_penalty=presence_penalty,
            )
            return response
        else:
            # model = 'gpt-3.5-turbo'
            # openai.api_key = "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"
            openai.api_type = "azure"
            openai.api_version = "2023-05-15"
            openai.api_base = 'your base'
            openai.api_key = 'your key'
            model = 'gpt-3.5-turbo'
            response = await openai.ChatCompletion.acreate(
                messages=query,
                temperature=temperature,
                frequency_penalty=frequency_penalty,
                presence_penalty=presence_penalty,
                engine="enginedepart-fc"
            )
            return response
    except openai.error.APIConnectionError as e:
        logger.warn(e)
        logger.warn("[CHATGPT] APIConnection failed")
        return "我连接不到你的网络"
    except openai.error.Timeout as e:
        logger.warn(e)
        logger.warn("[CHATGPT] Timeout")
        return "我没有收到你的消息"
    except Exception as e:
        # unknown exception
        logger.exception(e)
        # Session.clear_session(user_id)
        return "请再问我一次吧"

 

posted @ 2023-08-30 17:09  尹言覃少  阅读(83)  评论(0编辑  收藏  举报