对话文件管理

import os
import sys
import uuid
import json
from copy import deepcopy

import requests

from http import HTTPStatus

list_session_files_url = "https://dashscope.aliyuncs.com/api/v2/apps/deep-search-agent/session/files/list"
upsert_session_files_url = "https://dashscope.aliyuncs.com/api/v2/apps/deep-search-agent/session/files/upsert"
delete_session_files_url = "https://dashscope.aliyuncs.com/api/v2/apps/deep-search-agent/session/files/delete"

headers = {
    'Authorization': f'Bearer {os.getenv("DASHSCOPE_API_KEY")}',  # 配置 API KEY
    'Content-Type': 'application/json'
}


if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("usage: python operate_session_files.py <session_id>")
        exit(1)

    session_id = sys.argv[1]

    # 1. 获取测试 session 文件
    list_session_files_params = {
        "request_id": str(uuid.uuid4()),
        "input": {
            "session_id": session_id
        },
        "parameters": {}
    }
    response = requests.post(list_session_files_url, json=list_session_files_params, headers=headers)
    session_files_data = {}
    if response.status_code == HTTPStatus.OK:
        print("1. List session files successfully.")
        session_files_data = response.json()["output"]["result"]
        print(json.dumps(response.json(), indent=2, ensure_ascii=False))
    else:
        print(f'code={response.status_code}')
        exit(1)

    # 2. 更新测试 session 文件
    files = session_files_data["files"]
    temp_file = None
    for file in files:
        if '测试/' in file["file_path"]:
            temp_file = file
    if temp_file is None:
        temp_file = deepcopy(files[0])
        temp_file["file_id"] = temp_file["file_id"] + "_test"
        temp_file["file_path"] = "测试/" + temp_file["file_path"]
        files.append(temp_file)

    upsert_session_files_params = {
        "request_id": str(uuid.uuid4()),
        "input": {
            "session_id": session_id,
            "files": files
        },
        "parameters": {}
    }
    response = requests.post(upsert_session_files_url, json=upsert_session_files_params, headers=headers)
    upsert_data = {}
    if response.status_code == HTTPStatus.OK:
        print("2. Upsert session files successfully.")
        print(json.dumps(response.json(), indent=2, ensure_ascii=False))
    else:
        print(f'code={response.status_code}')
        exit(1)

    # 3. 删除测试 session 文件
    delete_session_files_params = {
        "request_id": str(uuid.uuid4()),
        "input": {
            "session_id": session_id,
            "file_ids": [
                temp_file["file_id"]
            ]
        },
        "parameters": {}
    }
    response = requests.post(delete_session_files_url, json=delete_session_files_params, headers=headers)
    delete_data = {}
    if response.status_code == HTTPStatus.OK:
        print("3. Delete session files successfully.")
        print(json.dumps(response.json(), indent=2, ensure_ascii=False))
    else:
        print(f'code={response.status_code}')
        exit(1)

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;

import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.UUID;

public class DeepSearchDemo {

    private static final String LIST_SESSION_FILES_URL = "https://dashscope.aliyuncs.com/api/v2/apps/deep-search-agent/session/files/list";
    private static final String UPSERT_SESSION_FILES_URL = "https://dashscope.aliyuncs.com/api/v2/apps/deep-search-agent/session/files/upsert";
    private static final String DELETE_SESSION_FILES_URL = "https://dashscope.aliyuncs.com/api/v2/apps/deep-search-agent/session/files/delete";

    private static final String DASHSCOPE_API_KEY = System.getenv("DASHSCOPE_API_KEY");


    private static JSONObject buildListParams(String sessionId) {
        JSONObject input = new JSONObject();
        input.put("session_id", sessionId);

        JSONObject params = new JSONObject();
        params.put("input", input);
        params.put("request_id", UUID.randomUUID().toString());
        params.put("parameters", new JSONObject());

        return params;
    }

    private static JSONObject buildUpsertParams(String sessionId, JSONArray files) {
        JSONObject input = new JSONObject();
        input.put("session_id", sessionId);
        input.put("files", files);

        JSONObject params = new JSONObject();
        params.put("input", input);
        params.put("request_id", UUID.randomUUID().toString());
        params.put("parameters", new JSONObject());

        return params;
    }

    private static JSONObject buildDeleteParams(String sessionId, JSONArray fileIds) {
        JSONObject input = new JSONObject();
        input.put("session_id", sessionId);
        input.put("file_ids", fileIds);

        JSONObject params = new JSONObject();
        params.put("input", input);
        params.put("request_id", UUID.randomUUID().toString());
        params.put("parameters", new JSONObject());

        return params;
    }

    private static JSONObject sendPostRequest(String url, JSONObject body) throws Exception {
        URL urlObj = new URL(url);
        HttpURLConnection conn = (HttpURLConnection) urlObj.openConnection();
        conn.setRequestMethod("POST");
        conn.setDoOutput(true);
        conn.setDoInput(true);
        conn.setRequestProperty("Authorization", "Bearer " + DASHSCOPE_API_KEY);
        conn.setRequestProperty("Content-Type", "application/json; charset=utf-8");
        conn.setRequestProperty("Accept", "application/json");
        conn.setRequestProperty("User-Agent", "Java Client");

        try (DataOutputStream os = new DataOutputStream(conn.getOutputStream())) {
            os.write(JSON.toJSONString(body).getBytes("UTF-8"));
            os.flush();
        }
        int responseCode = conn.getResponseCode();

        // 读取响应
        StringBuilder response = new StringBuilder();
        BufferedReader reader;
        if (responseCode >= 200 && responseCode < 300) {
            reader = new BufferedReader(new InputStreamReader(conn.getInputStream()));
        } else {
            reader = new BufferedReader(new InputStreamReader(conn.getErrorStream()));
        }
        String inputLine;
        while ((inputLine = reader.readLine()) != null) {
            response.append(inputLine);
        }
        reader.close();
        if (responseCode != 200) {
            System.err.println("Error response code: " + responseCode);
            System.out.println("Response: " + response.toString());
            System.exit(1);
        }

        return JSON.parseObject(response.toString());
    }


    public static void main(String[] args) throws Exception {
        if (args.length != 1) {
            System.out.println("usage: java DeepSearchDemo <session_id>");
            System.exit(1);
        }

        String sessionId = args[0];

        // 1. List session files
        JSONObject listResponse = sendPostRequest(LIST_SESSION_FILES_URL, buildListParams(sessionId));
        JSONObject result = listResponse.getJSONObject("output").getJSONObject("result");
        System.out.println("1. List session files successfully.");
        System.out.println(JSON.toJSONString(result, true));

        // 2. Upsert session files
        JSONArray files = result.getJSONArray("files");
        JSONObject tempFile = null;
        for (int i = 0; i < files.size(); i++) {
            JSONObject file = files.getJSONObject(i);
            if (file.getString("file_path").contains("测试/")) {
                tempFile = file;
                break;
            }
        }
        if (tempFile == null) {
            tempFile = new JSONObject().fluentPutAll(files.getJSONObject(0));
            tempFile.put("file_id", tempFile.getString("file_id") + "_test");
            tempFile.put("file_path", "测试/" + tempFile.getString("file_path"));
            files.add(tempFile);
        }
        JSONObject upsertResponse = sendPostRequest(UPSERT_SESSION_FILES_URL, buildUpsertParams(sessionId, files));
        result = upsertResponse.getJSONObject("output").getJSONObject("result");
        System.out.println("2. Upsert session files successfully.");
        System.out.println(JSON.toJSONString(result, true));

        // 3. Delete session files
        JSONArray fileIds = new JSONArray();
        fileIds.add(tempFile.getString("file_id"));

        JSONObject deleteResponse = sendPostRequest(DELETE_SESSION_FILES_URL, buildDeleteParams(sessionId, fileIds));
        result = deleteResponse.getJSONObject("output").getJSONObject("result");
        System.out.println("3. Delete session files successfully.");
        System.out.println(JSON.toJSONString(result, true));
    }

}