Summary generation

import os
import requests

'''
API name: Summary generation
API path: https://dashscope.aliyuncs.com/api/v2/apps/zhiwen-chat/summary
Environment requirements: Python >= 3.7
'''

url = "https://dashscope.aliyuncs.com/api/v2/apps/zhiwen-chat/summary"
request_headers = {
    "Authorization": os.environ.get("DASHSCOPE_API_KEY"),  # If you have not configured the environment variable, replace your API key here.
    "Content-Type": "application/json"
}
# Summary generation: SUMMARY_GEN
request_body = {
    "capabilityType": "SUMMARY_GEN",
    "userPrompt": "What are these documents about? Summarize them in Markdown format.",
    "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())