Information extraction

import os
import requests

'''
API name: Information Extraction
API path: https://dashscope.aliyuncs.com/api/v2/apps/zhiwen-chat/extraction
Environment requirements: Python >= 3.7
'''

url = "https://dashscope.aliyuncs.com/api/v2/apps/zhiwen-chat/extraction"
request_headers = {
    "Authorization": os.environ.get("DASHSCOPE_API_KEY"),  # If you have not configured an environment variable, replace this with your API key.
    "Content-Type": "application/json"
}
# Information extraction
# Bidding document extraction: BID_EXTRACTION. Resume extraction: RESUME_EXTRACTION. Entity extraction: ENTITY_EXTRACTION.
request_body = {
    "capabilityType": "ENTITY_EXTRACTION",
    "userPrompt": "You are an AI assistant that specializes in named entity recognition (NER). Your task is to accurately extract all names of people, places, and organizations from the given text and organize them in the specified dictionary 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())