# coding=utf-8
import os
import sys
import uuid
import json
import requests
from http import HTTPStatus
file_expose_url = "https://dashscope.aliyuncs.com/api/v2/apps/deep-search-agent/file/expose"
headers = {
'Authorization': f'Bearer {os.getenv("DASHSCOPE_API_KEY", "")}', # Configure the API key.
'Content-Type': 'application/json'
}
def get_file_export_info(file_path: str) -> dict:
api_key = os.getenv("DASHSCOPE_API_KEY")
if not api_key:
print("please set DASHSCOPE_API_KEY environment variable")
exit(1)
# Export the file.
file_expose_params = {
"input": {
"request_id": str(uuid.uuid4()),
"writing_result_path": file_path,
"writing_result_type": "html_url"
},
"parameters": {}
}
response = requests.post(file_expose_url, json=file_expose_params, headers=headers)
if response.status_code == HTTPStatus.OK:
print("file exported successfully.")
export_info = response.json()["output"]["result"]
return export_info
else:
print(f'code={response.status_code}')
exit(1)
if __name__ == "__main__":
if len(sys.argv) != 2:
print("usage: python deep_search_demo.py <file_path>")
exit(1)
file_path = sys.argv[1]
export_info = get_file_export_info(file_path)
print(json.dumps(export_info, indent=2, ensure_ascii=False))