"""

This example is about how to list workspaces.

"""

import logging

import os

from typing import Optional

from cozepy import (

COZE_CN_BASE_URL,

Coze,

DeviceOAuthApp, # noqa

TokenAuth,

)

from cozepy.log import setup_logging

def get_coze_api_base() -> str:

# The default access is api.coze.cn, but if you need to access api.coze.com,

# please use base_url to configure the api endpoint to access

coze_api_base = os.getenv("COZE_API_BASE")

if coze_api_base:

return coze_api_base

return COZE_CN_BASE_URL # default

def get_coze_api_token(workspace_id: Optional[str] = None) -> str:

# Get an access_token through personal access token or oauth.

coze_api_token = os.getenv("COZE_API_TOKEN")

if coze_api_token:

return coze_api_token

coze_api_base = get_coze_api_base()

device_oauth_app = DeviceOAuthApp(client_id="57294420732781205987760324720643.app.coze", base_url=coze_api_base)

device_code = device_oauth_app.get_device_code(workspace_id)

print(f"Please Open: {device_code.verification_url} to get the access token")

return device_oauth_app.get_access_token(device_code=device_code.device_code, poll=True).access_token

# Init the Coze client through the access_token.

coze = Coze(auth=TokenAuth(token=get_coze_api_token()), base_url=get_coze_api_base())

# coze user id

user_id = os.getenv("COZE_USER_ID")

# coze account id

coze_account_id = os.getenv("COZE_ACCOUNT_ID")

def setup_examples_logger():

coze_log = os.getenv("COZE_LOG")

if coze_log:

setup_logging(logging.getLevelNamesMapping().get(coze_log.upper(), logging.INFO))

setup_examples_logger()

# Call the api to get workspace list.

workspaces = coze.workspaces.list(

user_id=user_id,

coze_account_id=coze_account_id,

)

for workspace in workspaces:

# workspaces is an iterator. Traversing workspaces will automatically turn pages and

# get all workspace results.

print(workspace.model_dump_json(indent=2))

print("logid", workspaces.response.logid)