Tag categorization

import os
import requests

'''
API name: Tag Categorization
API path: https://dashscope.aliyuncs.com/api/v2/apps/zhiwen-chat/tagging
Environment requirements: Python >= 3.7
'''

url = "https://dashscope.aliyuncs.com/api/v2/apps/zhiwen-chat/tagging"
request_headers = {
    "Authorization": os.environ.get("DASHSCOPE_API_KEY"),  # If you have not configured the environment variable, replace this with your API key.
    "Content-Type": "application/json"
}
# Tag categorization: TAG_CLASSIFY
request_body = {
    "capabilityType": "TAG_CLASSIFY",
    "userPrompt": "Analyze the business scope of a company, extract key information, and provide the output in a structured format. Stop after the output is complete and do not add new content.",
    "stream": True,
    "fileIdList": [
        "file_zhiwen_XXX"  # Replace this with your document ID.
    ]
}

with requests.post(url, json=request_body, headers=request_headers) as response:
    if request_body["stream"]:
        for line in response.iter_lines():
            if line:
                line = line.decode('utf-8')
                if line.startswith('data:'):
                    print("Received:", line[5:].strip())
    else:
        print("Received:", response.json())