bot_publish.py

"""
This example is for describing how to create a bot, update a bot and publish a bot to the API.
"""

import logging
import os
import sys
from pathlib import Path

from cozepy import (
    COZE_COM_BASE_URL,
    BotPromptInfo,
    ChatEventType,
    Coze,
    Message,
    MessageContentType,
    TokenAuth,
    setup_logging,
)

# Get an access_token through personal access token or oauth.
coze_api_token = os.getenv("COZE_API_TOKEN")
# The default access is api.coze.com, but if you need to access api.coze.cn,
# please use base_url to configure the api endpoint to access
coze_api_base = os.getenv("COZE_API_BASE") or COZE_COM_BASE_URL
# workspace id
workspace_id = os.getenv("COZE_WORKSPACE_ID") or "your workspace id"
# Whether to print detailed logs
is_debug = os.getenv("DEBUG")

if is_debug:
    setup_logging(logging.DEBUG)

# Init the Coze client through the access_token.
coze = Coze(auth=TokenAuth(token=coze_api_token), base_url=coze_api_base)

# Call the upload file interface to get the avatar id.
avatar_path = "/path/avatar.jpg" if len(sys.argv) < 2 else sys.argv[1]
avatar = coze.files.upload(file=Path(avatar_path))

# Invoke the create interface to create a bot in the draft status.
bot = coze.bots.create(
    # The bot should exist under a space and your space id needs configuration.
    space_id=workspace_id,
    # Bot name
    name="translator bot",
    # Bot avatar
    icon_file_id=avatar.id,
    # Bot system prompt
    prompt_info=BotPromptInfo(prompt="your are a translator, translate the following text from English to Chinese"),
)

# Call the publish interface to publish the bot on the api channel.
coze.bots.publish(bot_id=bot.bot_id)

# Developers can also modify the bot configuration and republish it.
coze.bots.update(
    bot_id=bot.bot_id,
    name="translator bot 2.0",
    prompt_info=BotPromptInfo(prompt="your are a translator, translate the following text from Chinese to English"),
)
coze.bots.publish(bot_id=bot.bot_id, connector_ids=["1024"])

# Call the coze.chat.stream method to create a chat. The create method is a streaming
# chat and will return a Chat Iterator. Developers should iterate the iterator to get
# chat event and handle them.
for event in coze.chat.stream(
    # The published bot's id
    bot_id=bot.bot_id,
    # biz user id, maybe random string
    user_id="user id",
    # user input
    additional_messages=[Message.build_user_question_text("chinese")],
):
    if (
        event.event == ChatEventType.CONVERSATION_MESSAGE_DELTA
        and event.message.content_type == MessageContentType.TEXT
    ):
        print(event.message.content, end="")