from cozepy import Coze, TokenAuth, Message, ChatEventType, COZE_CN_BASE_URL


def to_xx(user_id, quest_text):
    """调用Coze API进行对话"""
    coze = Coze(auth=TokenAuth(token="pat_TNHVdiVDn9poKjUXYqVArNwGd9C2cWIIP04ltz4N85WClayks7VuNDF9ewHb5rTi"),
                base_url=COZE_CN_BASE_URL)

    bot_id = "7566918492960620559"
    user_id = user_id

    text = ""
    for event in coze.chat.stream(
            bot_id=bot_id,
            user_id=user_id,
            additional_messages=[
                Message.build_user_question_text(quest_text)
            ]
    ):
        if event.event == ChatEventType.CONVERSATION_MESSAGE_DELTA:
            message = event.message
            text += message.content

    return text


if __name__ == '__main__':
    print(to_xx(user_id="7566918492960620559", quest_text="我想要加入你们公司，需要什么要求"))