Alibaba Cloud Model Studio:コード機能 (Qwen-Coder)

リクエスト例

import os
from openai import OpenAI

client = OpenAI(
    # API キーはリージョンによって異なります。API キーを取得するには、https://www.alibabacloud.com/help/model-studio/get-api-key をご参照ください
    # 環境変数を設定していない場合は、次の行を Alibaba Cloud Model Studio API キーに置き換えてください：api_key="sk-xxx",
    api_key=os.getenv("DASHSCOPE_API_KEY"),
    base_url="https://dashscope-intl.aliyuncs.com/compatible-mode/v1",
)
completion = client.chat.completions.create(
    model="qwen3-coder-next", 
    messages=[
        {'role': 'system', 'content': 'You are a helpful assistant.'},
        {'role': 'user', 'content': '整数 n をパラメーターとして受け取り、n 未満のすべての素数を含むリストを返す Python 関数 find_prime_numbers を記述してください。コード以外のコンテンツや Markdown コードブロックは出力しないでください。'}],
)
print(completion.choices[0].message.content)

応答

def find_prime_numbers(n):
    if n <= 2:
        return []
    
    primes = []
    for num in range(2, n):
        is_prime = True
        for i in range(2, int(num ** 0.5) + 1):
            if num % i == 0:
                is_prime = False
                break
        if is_prime:
            primes.append(num)
    
    return primes

リクエスト例

import OpenAI from "openai";

const client = new OpenAI(
    {
        // API キーはリージョンによって異なります。API キーの取得：https://www.alibabacloud.com/help/model-studio/get-api-key
        // 設定されていない場合は、次の行をご利用の API キーに置き換えてください：apiKey: "sk-xxx",
        apiKey: process.env.DASHSCOPE_API_KEY,
        // 北京リージョンのモデルを使用する場合は、base_url を https://dashscope.aliyuncs.com/compatible-mode/v1 に置き換えてください
        baseURL: "https://dashscope-intl.aliyuncs.com/compatible-mode/v1"
    }
);

async function main() {
    const completion = await client.chat.completions.create({
        model: "qwen3-coder-next",
        messages: [
            { role: "system", content: "You are a helpful assistant." },
            { role: "user", content: "整数 n をパラメーターとして受け取り、n 未満のすべての素数を含むリストを返す Python 関数 find_prime_numbers を記述してください。コード以外のコンテンツや Markdown コードブロックは出力しないでください。" }
        ],
    });
    console.log(completion.choices[0].message.content);
}

main();

レスポンス

def find_prime_numbers(n):
    if n <= 2:
        return []
    
    primes = []
    for num in range(2, n):
        is_prime = True
        for i in range(2, int(num ** 0.5) + 1):
            if num % i == 0:
                is_prime = False
                break
        if is_prime:
            primes.append(num)
    
    return primes

リクエスト例

curl -X POST https://dashscope-intl.aliyuncs.com/compatible-mode/v1/chat/completions \
-H "Authorization: Bearer $DASHSCOPE_API_KEY" \
-H "Content-Type: application/json" \
-d '{
    "model": "qwen3-coder-next",
    "messages": [
        {
            "role": "system",
            "content": "You are a helpful assistant."
        },
        {
            "role": "user", 
            "content": "整数 n をパラメーターとして受け取り、n 未満のすべての素数を含むリストを返す Python 関数 find_prime_numbers を記述してください。コード以外のコンテンツや Markdown コードブロックは出力しないでください。"
        }
    ]
}'

応答

{
    "model": "qwen3-coder-next",
    "id": "chatcmpl-3123d5cb-01b8-9a90-98cc-5bffbb369xxx",
    "choices": [
        {
            "message": {
                "content": "def find_prime_numbers(n):\n    if n <= 2:\n        return []\n    \n    primes = []\n    for num in range(2, n):\n        is_prime = True\n        for i in range(2, int(num ** 0.5) + 1):\n            if num % i == 0:\n                is_prime = False\n                break\n        if is_prime:\n            primes.append(num)\n    \n    return primes",
                "role": "assistant"
            },
            "index": 0,
            "finish_reason": "stop"
        }
    ],
    "created": 1770108104,
    "object": "chat.completion",
    "usage": {
        "total_tokens": 155,
        "completion_tokens": 89,
        "prompt_tokens": 66
    }
}

リクエスト例

import dashscope
import os

dashscope.base_http_api_url = 'https://dashscope-intl.aliyuncs.com/api/v1'

messages = [
    {
        "role": "system",
        "content": "You are a helpful assistant."
    },
    {
        "role": "user",
        "content": "整数 n をパラメーターとして受け取り、n 未満のすべての素数を含むリストを返す Python 関数 find_prime_numbers を記述してください。コード以外のコンテンツや Markdown コードブロックは出力しないでください。"
    }
]

response = dashscope.Generation.call(
    # API キーはリージョンによって異なります。API キーを取得するには、https://www.alibabacloud.com/help/model-studio/get-api-key をご参照ください
    # 環境変数を設定していない場合は、次の行を Alibaba Cloud Model Studio API キーに置き換えてください：api_key = "sk-xxx",
    api_key=os.getenv("DASHSCOPE_API_KEY"),
    model="qwen3-coder-next",
    messages=messages,
    result_format="message"
)

if response.status_code == 200:
    print(response.output.choices[0].message.content)
else:
    print(f"HTTP status code: {response.status_code}")
    print(f"Error code: {response.code}")
    print(f"Error message: {response.message}")

応答

def find_prime_numbers(n):
    if n <= 2:
        return []
    
    primes = []
    for num in range(2, n):
        is_prime = True
        for i in range(2, int(num ** 0.5) + 1):
            if num % i == 0:
                is_prime = False
                break
        if is_prime:
            primes.append(num)
    
    return primes

リクエスト例

import java.util.Arrays;
import com.alibaba.dashscope.aigc.generation.Generation;
import com.alibaba.dashscope.aigc.generation.GenerationResult;
import com.alibaba.dashscope.aigc.generation.GenerationParam;
import com.alibaba.dashscope.common.Message;
import com.alibaba.dashscope.common.Role;
import com.alibaba.dashscope.exception.ApiException;
import com.alibaba.dashscope.exception.InputRequiredException;
import com.alibaba.dashscope.exception.NoApiKeyException;
import com.alibaba.dashscope.protocol.Protocol;


public class Main {
    public static GenerationResult callWithMessage()
            throws NoApiKeyException, ApiException, InputRequiredException {
        String apiKey = System.getenv("DASHSCOPE_API_KEY");
        Generation gen = new Generation(Protocol.HTTP.getValue(), "https://dashscope-intl.aliyuncs.com/api/v1");
        Message sysMsg = Message.builder()
                .role(Role.SYSTEM.getValue())
                .content("You are a helpful assistant.").build();
        Message userMsg = Message.builder()
                .role(Role.USER.getValue())
                .content("整数 n をパラメーターとして受け取り、n 未満のすべての素数を含むリストを返す Python 関数 find_prime_numbers を記述してください。コード以外のコンテンツや Markdown コードブロックは出力しないでください。").build();
        GenerationParam param = GenerationParam.builder()
                .apiKey(apiKey)
                .model("qwen3-coder-next")
                .messages(Arrays.asList(sysMsg, userMsg))
                .resultFormat(GenerationParam.ResultFormat.MESSAGE)
                .build();
        return gen.call(param);
    }
    public static void main(String[] args){
        try {
            GenerationResult result = callWithMessage();
            System.out.println(result.getOutput().getChoices().get(0).getMessage().getContent());
        } catch (ApiException | NoApiKeyException | InputRequiredException e) {
            System.err.println("Request exception: " + e.getMessage());
            e.printStackTrace();
        }
    }
}

応答

def find_prime_numbers(n):
    if n <= 2:
        return []
    
    primes = []
    for num in range(2, n):
        is_prime = True
        for i in range(2, int(num ** 0.5) + 1):
            if num % i == 0:
                is_prime = False
                break
        if is_prime:
            primes.append(num)
    
    return primes

リクエスト例

curl -X POST "https://dashscope-intl.aliyuncs.com/api/v1/services/aigc/text-generation/generation" \
-H "Authorization: Bearer $DASHSCOPE_API_KEY" \
-H "Content-Type: application/json" \
-d '{
    "model": "qwen3-coder-next",
    "input":{
        "messages":[      
            {
                "role": "system",
                "content": "You are a helpful assistant."
            },
            {
                "role": "user",
                "content": "整数 n をパラメーターとして受け取り、n 未満のすべての素数を含むリストを返す Python 関数 find_prime_numbers を記述してください。コード以外のコンテンツや Markdown コードブロックは出力しないでください。"
            }
        ]
    },
    "parameters": {
        "result_format": "message"
    }
}'

応答

{
    "output": {
        "choices": [
            {
                "message": {
                    "content": "def find_prime_numbers(n):\n    if n <= 2:\n        return []\n    \n    primes = []\n    for num in range(2, n):\n        is_prime = True\n        for i in range(2, int(num ** 0.5) + 1):\n            if num % i == 0:\n                is_prime = False\n                break\n        if is_prime:\n            primes.append(num)\n    \n    return primes",
                    "role": "assistant"
                },
                "finish_reason": "stop"
            }
        ]
    },
    "usage": {
        "total_tokens": 155,
        "input_tokens": 66,
        "output_tokens": 89
    },
    "request_id": "dd78b1cf-8029-46bb-9bea-b794ded7bxxx"
}

import os
import json
from openai import OpenAI

client = OpenAI(
    # API キーはリージョンによって異なります。API キーの取得：https://www.alibabacloud.com/help/model-studio/get-api-key
    # 設定されていない場合は、次の行をご利用の API キーに置き換えてください：api_key="sk-xxx",
    api_key=os.getenv("DASHSCOPE_API_KEY"),
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
)

tools = [
    {
        "type": "function",
        "function": {
            "name": "write_file",
            "description": "指定されたファイルにコンテンツを書き込みます。ファイルが存在しない場合は作成します。",
            "parameters": {
                "type": "object",
                "properties": {
                    "path": {
                        "type": "string",
                        "description": "オブジェクトファイルの相対パスまたは絶対パス"
                    },
                    "content": {
                        "type": "string",
                        "description": "ファイルに書き込む文字列コンテンツ"
                    }
                },
                "required": ["path", "content"]
            }
        }
    }
]

# ツール関数の実装
def write_file(path: str, content: str) -> str:
    """ファイルにコンテンツを書き込みます。"""
    try:
        # ディレクトリが存在することを確認します
        os.makedirs(os.path.dirname(path),
                    exist_ok=True) if os.path.dirname(path) else None
        with open(path, 'w', encoding='utf-8') as f:
            f.write(content)
        return f"成功：ファイル '{path}' が書き込まれました"
    except Exception as e:
        return f"エラー：ファイルの書き込み中に例外が発生しました - {str(e)}"


messages = [{"role": "user", "content": "クイックソートの Python スクリプトを書き、quick_sort.py という名前を付けてください"}]

completion = client.chat.completions.create(
    model="qwen3-coder-next",
    messages=messages,
    tools=tools
)

assistant_output = completion.choices[0].message
if assistant_output.content is None:
    assistant_output.content = ""
messages.append(assistant_output)

# ツール呼び出しが不要な場合は、コンテンツを直接出力します
if assistant_output.tool_calls is None:
    print(f"ツール呼び出しは不要です。直接の応答：{assistant_output.content}")
else:
    # ツール呼び出しループに入ります
    while assistant_output.tool_calls is not None:
        for tool_call in assistant_output.tool_calls:
            tool_call_id = tool_call.id
            func_name = tool_call.function.name
            arguments = json.loads(tool_call.function.arguments)
            print(f"ツール [{func_name}] をパラメーター：{arguments} で呼び出しています")
            # ツールを実行します
            tool_result = write_file(**arguments)
            # ツール応答を構築します
            tool_message = {
                "role": "tool",
                "tool_call_id": tool_call_id,
                "content": tool_result,
            }
            print(f"ツール応答：{tool_message['content']}")
            messages.append(tool_message)
        # モデルを再度呼び出して、要約された自然言語の応答を取得します
        response = client.chat.completions.create(
            model="qwen3-coder-next",
            messages=messages,
            tools=tools
        )
        assistant_output = response.choices[0].message
        if assistant_output.content is None:
            assistant_output.content = ""
        messages.append(assistant_output)
    print(f"最終的なモデルの応答：{assistant_output.content}")

応答

ツール [write_file] を引数：{'content': 'def quick_sort(arr):\\n    if len(arr) <= 1:\\n        return arr\\n    pivot = arr[len(arr) // 2]\\n    left = [x for x in arr if x < pivot]\\n    middle = [x for x in arr if x == pivot]\\n    right = [x for x in arr if x > pivot]\\n    return quick_sort(left) + middle + quick_sort(right)\\n\\nif __name__ == \\"__main__\\":\\n    example_list = [3, 6, 8, 10, 1, 2, 1]\\n    print(\\"Original list:\\", example_list)\\n    sorted_list = quick_sort(example_list)\\n    print(\\"Sorted list:\\", sorted_list)', 'path': 'quick_sort.py'} で呼び出しています
ツールからの返信：成功：ファイル 'quick_sort.py' が書き込まれました
モデルの最終応答：OKです。クイックソートの Python 実装を含むファイル `quick_sort.py` を作成しました。このファイルを実行してサンプル出力を確認できます。さらに変更や説明が必要な場合はお知らせください！

import OpenAI from "openai";
import fs from "fs/promises";
import path from "path";

const client = new OpenAI({
    // API キーはリージョンによって異なります。API キーの取得：https://www.alibabacloud.com/help/model-studio/get-api-key
    // 環境変数を設定していない場合は、次の行を Model Studio API キーに置き換えてください：apiKey: "sk-xxx",
    apiKey: process.env.DASHSCOPE_API_KEY,
    baseURL: "https://dashscope-intl.aliyuncs.com/compatible-mode/v1"
});

const tools = [
    {
        "type": "function",
        "function": {
            "name": "write_file",
            "description": "指定されたファイルにコンテンツを書き込み、存在しない場合は作成します。",
            "parameters": {
                "type": "object",
                "properties": {
                    "path": {
                        "type": "string",
                        "description": "オブジェクトファイルの相対パスまたは絶対パス"
                    },
                    "content": {
                        "type": "string",
                        "description": "ファイルに書き込む文字列コンテンツ"
                    }
                },
                "required": ["path", "content"]
            }
        }
    }
];

// ツール関数の実装
async function write_file(filePath, content) {
    try {
        // セキュリティ：ファイル書き込みはデフォルトで無効になっています。有効にするには、コメントを解除し、パスのセキュリティを確認してください。
        // const dir = path.dirname(filePath);
        // if (dir) {
        //     await fs.mkdir(dir, { recursive: true });
        // }
        // await fs.writeFile(filePath, content, "utf-8");
        return `成功：ファイル '${filePath}' が書き込まれました`;
    } catch (error) {
        return `エラー：ファイルの書き込み中に例外が発生しました - ${error.message}`;
    }
}

const messages = [{"role": "user", "content": "クイックソートの Python コードを書き、ファイル名を quick_sort.py としてください"}];

async function main() {
    const completion = await client.chat.completions.create({
        model: "qwen3-coder-next",
        messages: messages,
        tools: tools
    });

    let assistant_output = completion.choices[0].message;
    // コンテンツが null でないことを確認
    if (!assistant_output.content) assistant_output.content = "";
    messages.push(assistant_output);

    // ツール呼び出しが不要な場合は、コンテンツを直接表示
    if (!assistant_output.tool_calls) {
        console.log(`ツール呼び出しは不要です。直接の応答：${assistant_output.content}`);
    } else {
        // ツール呼び出しループに入る
        while (assistant_output.tool_calls) {
            for (const tool_call of assistant_output.tool_calls) {
                const tool_call_id = tool_call.id;
                const func_name = tool_call.function.name;
                const args = JSON.parse(tool_call.function.arguments);
                console.log(`ツール [${func_name}] を引数で呼び出しています：`, args);
                // ツールを実行
                const tool_result = await write_file(args.path, args.content);
                // ツールリターンメッセージを構築
                const tool_message = {
                    "role": "tool",
                    "tool_call_id": tool_call_id,
                    "content": tool_result
                };
                console.log(`ツールからの返信：${tool_message.content}`);
                messages.push(tool_message);
            }
            // モデルを再度呼び出して、要約された自然言語の応答を取得
            const response = await client.chat.completions.create({
                model: "qwen3-coder-next",
                messages: messages,
                tools: tools
            });
            assistant_output = response.choices[0].message;
            if (!assistant_output.content) assistant_output.content = "";
            messages.push(assistant_output);
        }
        console.log(`モデルの最終応答：${assistant_output.content}`);
    }
}

main();

応答

ツール [write_file] を引数で呼び出しています： {
  content: 'def quick_sort(arr):\\n    if len(arr) <= 1:\\n        return arr\\n    pivot = arr[len(arr) // 2]\\n    left = [x for x in arr if x < pivot]\\n    middle = [x for x in arr if x == pivot]\\n    right = [x for x in arr if x > pivot]\\n    return quick_sort(left) + middle + quick_sort(right)\\n\\nif __name__ == \\"__main__\\":\\n    example_list = [3, 6, 8, 10, 1, 2, 1]\\n    print(\\"Original list:\\", example_list)\\n    sorted_list = quick_sort(example_list)\\n    print(\\"Sorted list:\\", sorted_list)',
  path: 'quick_sort.py'
}
ツールからの返信：成功：ファイル 'quick_sort.py' が書き込まれました
モデルの最終応答：`quick_sort.py` ファイルは、クイックソートの Python 実装で正常に作成されました。このファイルを実行して、サンプルリストのソート結果を確認できます。さらに変更や説明が必要な場合はお知らせください！

この例は、ツール呼び出しプロセスの最初のステップを示しています：リクエストを行い、ツールを呼び出すモデルの意図を取得します。

curl -X POST https://dashscope-intl.aliyuncs.com/compatible-mode/v1/chat/completions \
-H "Authorization: Bearer $DASHSCOPE_API_KEY" \
-H "Content-Type: application/json" \
-d '{
    "model": "qwen3-coder-next",
    "messages": [
        {
            "role": "user", 
            "content": "クイックソートの Python コードを書き、ファイル名を quick_sort.py としてください"
        }
    ],
    "tools": [
        {
            "type": "function",
            "function": {
                "name": "write_file",
                "description": "指定されたファイルにコンテンツを書き込み、存在しない場合は作成します。",
                "parameters": {
                    "type": "object",
                    "properties": {
                        "path": {
                            "type": "string",
                            "description": "オブジェクトファイルの相対パスまたは絶対パス"
                        },
                        "content": {
                            "type": "string",
                            "description": "ファイルに書き込む文字列コンテンツ"
                        }
                    },
                    "required": ["path", "content"]
                }
            }
        }
    ]
}'

応答

{
    "choices": [
        {
            "message": {
                "content": "",
                "role": "assistant",
                "tool_calls": [
                    {
                        "index": 0,
                        "id": "call_0ca7505bb6e44471a40511e5",
                        "type": "function",
                        "function": {
                            "name": "write_file",
                            "arguments": "{\"content\": \"def quick_sort(arr):\\\\n    if len(arr) <= 1:\\\\n        return arr\\\\n    pivot = arr[len(arr) // 2]\\\\n    left = [x for x in arr if x < pivot]\\\\n    middle = [x for x in arr if x == pivot]\\\\n    right = [x for x in arr if x > pivot]\\\\n    return quick_sort(left) + middle + quick_sort(right)\\\\n\\\\nif __name__ == \\\\\\\"__main__\\\\\\\":\\\\n    example_list = [3, 6, 8, 10, 1, 2, 1]\\\\n    print(\\\\\\\"Original list:\\\\\\\", example_list)\\\\n    sorted_list = quick_sort(example_list)\\\\n    print(\\\\\\\"Sorted list:\\\\\\\", sorted_list)\", \"path\": \"quick_sort.py\"}"
                        }
                    }
                ]
            },
            "finish_reason": "tool_calls",
            "index": 0,
            "logprobs": null
        }
    ],
    "object": "chat.completion",
    "usage": {
        "prompt_tokens": 494,
        "completion_tokens": 193,
        "total_tokens": 687,
        "prompt_tokens_details": {
            "cached_tokens": 0
        }
    },
    "created": 1761620025,
    "system_fingerprint": null,
    "model": "qwen3-coder-next",
    "id": "chatcmpl-20e96159-beea-451f-b3a4-d13b218112b5"
}

import os
import json
import dashscope

dashscope.base_http_api_url = 'https://dashscope-intl.aliyuncs.com/api/v1'

tools = [
    {
        "type": "function",
        "function": {
            "name": "write_file",
            "description": "指定されたファイルにコンテンツを書き込み、存在しない場合は作成します。",
            "parameters": {
                "type": "object",
                "properties": {
                    "path": {
                        "type": "string",
                        "description": "オブジェクトファイルの相対パスまたは絶対パス"
                    },
                    "content": {
                        "type": "string",
                        "description": "ファイルに書き込む文字列コンテンツ"
                    }
                },
                "required": ["path", "content"]
            }
        }
    }
]

# ツール関数の実装
def write_file(path: str, content: str) -> str:
    """ファイルコンテンツを書き込みます"""
    try:
        # セキュリティ上の理由から、ファイル書き込み機能はデフォルトで無効になっています。使用するには、コードのコメントを解除し、パスが安全であることを確認してください。
        # os.makedirs(os.path.dirname(path),exist_ok=True) if os.path.dirname(path) else None
        # with open(path, 'w', encoding='utf-8') as f:
        #     f.write(content)
        return f"成功：ファイル '{path}' が書き込まれました"
    except Exception as e:
        return f"エラー：ファイルの書き込み中に例外が発生しました - {str(e)}"


messages = [{"role": "user", "content": "クイックソートの Python コードを書き、ファイル名を quick_sort.py としてください"}]

response = dashscope.Generation.call(
    # 設定されていない場合は、次の行をご利用の API キーに置き換えてください：api_key="sk-xxx",
    api_key=os.getenv('DASHSCOPE_API_KEY'),
    model='qwen3-coder-next',
    messages=messages,
    tools=tools,
    result_format='message'
)

if response.status_code == 200:
    assistant_output = response.output.choices[0].message
    messages.append(assistant_output)

    # ツール呼び出しが不要な場合は、コンテンツを直接表示します
    if "tool_calls" not in assistant_output or not assistant_output["tool_calls"]:
        print(f"ツール呼び出しは不要です。直接の応答：{assistant_output['content']}")
    else:
        # ツール呼び出しループに入ります
        while "tool_calls" in assistant_output and assistant_output["tool_calls"]:
            for tool_call in assistant_output["tool_calls"]:
                func_name = tool_call["function"]["name"]
                arguments = json.loads(tool_call["function"]["arguments"])
                tool_call_id = tool_call.get("id")
                print(f"ツール [{func_name}] を引数で呼び出しています：{arguments}")
                # ツールを実行します
                tool_result = write_file(**arguments)
                # ツールリターンメッセージを構築します
                tool_message = {
                    "role": "tool",
                    "content": tool_result,
                    "tool_call_id": tool_call_id
                }
                print(f"ツールからの返信：{tool_message['content']}")
                messages.append(tool_message)
            # モデルを再度呼び出して、要約された自然言語の応答を取得します
            response = dashscope.Generation.call(
                api_key=os.getenv('DASHSCOPE_API_KEY'),
                model='qwen3-coder-next',
                messages=messages,
                tools=tools,
                result_format='message'
            )
            if response.status_code == 200:
                print(f"モデルの最終応答：{response.output.choices[0].message.content}")
                assistant_output = response.output.choices[0].message
                messages.append(assistant_output)
            else:
                print(f"要約応答生成中のエラー：{response}")
                break
else:
    print(f"実行エラー：{response}")

応答

ツール [write_file] を引数で呼び出しています：{'content': 'def quick_sort(arr):\\n    if len(arr) <= 1:\\n        return arr\\n    pivot = arr[len(arr) // 2]\\n    left = [x for x in arr if x < pivot]\\n    middle = [x for x in arr if x == pivot]\\n    right = [x for x in arr if x > pivot]\\n    return quick_sort(left) + middle + quick_sort(right)\\n\\nif __name__ == \\"__main__\\":\\n    example_list = [3, 6, 8, 10, 1, 2, 1]\\n    print(\\"Original list:\\", example_list)\\n    sorted_list = quick_sort(example_list)\\n    print(\\"Sorted list:\\", sorted_list)', 'path': 'quick_sort.py'}
ツールからの返信：成功：ファイル 'quick_sort.py' が書き込まれました
モデルの最終応答：`quick_sort.py` ファイルは、クイックソートの Python 実装で正常に作成されました。このファイルを実行して、サンプルリストのソート結果を確認できます。さらに変更や説明が必要な場合はお知らせください！

import com.alibaba.dashscope.aigc.generation.Generation;
import com.alibaba.dashscope.aigc.generation.GenerationParam;
import com.alibaba.dashscope.aigc.generation.GenerationResult;
import com.alibaba.dashscope.common.Message;
import com.alibaba.dashscope.common.Role;
import com.alibaba.dashscope.protocol.Protocol;
import com.alibaba.dashscope.exception.NoApiKeyException;
import com.alibaba.dashscope.exception.InputRequiredException;
import com.alibaba.dashscope.tools.FunctionDefinition;
import com.alibaba.dashscope.tools.ToolCallBase;
import com.alibaba.dashscope.tools.ToolCallFunction;
import com.alibaba.dashscope.tools.ToolFunction;
import com.alibaba.dashscope.utils.JsonUtils;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;

import java.io.File;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Main {

    /**
     * ファイルにコンテンツを書き込みます。
     * @param arguments モデルからの JSON 文字列で、必要なツールパラメーターが含まれています。
     * @return ツール実行後の結果文字列。
     */
    public static String writeFile(String arguments) {
        try {
            ObjectMapper objectMapper = new ObjectMapper();
            JsonNode argsNode = objectMapper.readTree(arguments);
            String path = argsNode.get("path").asText();
            String content = argsNode.get("content").asText();

            // セキュリティ：ファイル書き込みはデフォルトで無効になっています。有効にするには、コメントを解除し、パスのセキュリティを確認してください。
            // File file = new File(path);
            // File parentDir = file.getParentFile();
            // if (parentDir != null && !parentDir.exists()) {
            //     parentDir.mkdirs();
            // }
            // Files.write(Paths.get(path), content.getBytes(StandardCharsets.UTF_8));
            return "成功：ファイル '" + path + "' が書き込まれました。";
        } catch (Exception e) {
            return "エラー：ファイルの書き込み中に例外が発生しました - " + e.getMessage();
        }
    }

    public static void main(String[] args) {
        try {
            // ツールパラメーターのスキーマを定義
            String writePropertyParams =
                    "{\"type\":\"object\",\"properties\":{\"path\":{\"type\":\"string\",\"description\":\"ターゲットファイルの相対パスまたは絶対パス\"},\"content\":{\"type\":\"string\",\"description\":\"ファイルに書き込む文字列コンテンツ\"}},\"required\":[\"path\",\"content\"]}";

            FunctionDefinition writeFileFunction = FunctionDefinition.builder()
                    .name("write_file")
                    .description("指定されたファイルにコンテンツを書き込み、存在しない場合はファイルを作成します。")
                    .parameters(JsonUtils.parseString(writePropertyParams).getAsJsonObject())
                    .build();

            Generation gen = new Generation(Protocol.HTTP.getValue(), "https://dashscope-intl.aliyuncs.com/api/v1");

            String userInput = "クイックソートの Python スクリプトを書き、quick_sort.py という名前を付けてください";
            List<Message> messages = new ArrayList<>();
            messages.add(Message.builder().role(Role.USER.getValue()).content(userInput).build());

            // モデルへの最初の呼び出し
            GenerationParam param = GenerationParam.builder()
                    .model("qwen3-coder-next")
                    .apiKey(System.getenv("DASHSCOPE_API_KEY"))
                    .messages(messages)
                    .tools(Arrays.asList(ToolFunction.builder().function(writeFileFunction).build()))
                    .resultFormat(GenerationParam.ResultFormat.MESSAGE)
                    .build();

            GenerationResult result = gen.call(param);
            Message assistantOutput = result.getOutput().getChoices().get(0).getMessage();
            messages.add(assistantOutput);

            // ツール呼び出しが不要な場合は、コンテンツを直接出力します。
            if (assistantOutput.getToolCalls() == null || assistantOutput.getToolCalls().isEmpty()) {
                System.out.println("ツール呼び出しは不要です。直接の返信： " + assistantOutput.getContent());
            } else {
                // ツール呼び出しループに入る
                while (assistantOutput.getToolCalls() != null && !assistantOutput.getToolCalls().isEmpty()) {
                    for (ToolCallBase toolCall : assistantOutput.getToolCalls()) {
                        ToolCallFunction functionCall = (ToolCallFunction) toolCall;
                        String funcName = functionCall.getFunction().getName();
                        String arguments = functionCall.getFunction().getArguments();
                        System.out.println("ツール [" + funcName + "] を呼び出しています、引数： " + arguments);

                        // ツールを実行
                        String toolResult = writeFile(arguments);

                        // ツールリターンメッセージを構築
                        Message toolMessage = Message.builder()
                                .role("tool")
                                .toolCallId(toolCall.getId())
                                .content(toolResult)
                                .build();
                        System.out.println("ツールからの返信： " + toolMessage.getContent());
                        messages.add(toolMessage);
                    }

                    // モデルを再度呼び出して、要約された自然言語の応答を取得
                    param.setMessages(messages);
                    result = gen.call(param);
                    assistantOutput = result.getOutput().getChoices().get(0).getMessage();
                    messages.add(assistantOutput);
                }
                System.out.println("最終的なモデルの返信： " + assistantOutput.getContent());
            }

        } catch (NoApiKeyException | InputRequiredException e) {
            System.err.println("エラー： " + e.getMessage());
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

応答

ツール [write_file] を引数で呼び出しています：{"content": "def quick_sort(arr):\\n    if len(arr) <= 1:\\n        return arr\\n    pivot = arr[len(arr) // 2]\\n    left = [x for x in arr if x < pivot]\\n    middle = [x for x in arr if x == pivot]\\n    right = [x for x in arr if x > pivot]\\n    return quick_sort(left) + middle + quick_sort(right)\\n\\nif __name__ == \\\"__main__\\\":\\n    example_array = [3, 6, 8, 10, 1, 2, 1]\\n    print(\\\"Original array:\\\", example_array)\\n    sorted_array = quick_sort(example_array)\\n    print(\\\"Sorted array:\\\", sorted_array)", "path": "quick_sort.py"}
ツールからの返信：成功：ファイル 'quick_sort.py' が書き込まれました
モデルの最終応答：Python コードファイル `quick_sort.py` を正常に作成しました。これには `quick_sort` 関数とその使用例が含まれています。ターミナルやエディターで実行して、クイックソート機能をテストできます。

例：ツール呼び出しの最初のステップ—リクエストを行い、モデルの意図を取得します。

curl --location "https://dashscope-intl.aliyuncs.com/api/v1/services/aigc/text-generation/generation" \
--header "Authorization: Bearer $DASHSCOPE_API_KEY" \
--header "Content-Type: application/json" \
--data '{
    "model": "qwen3-coder-next",
    "input": {
        "messages": [{
            "role": "user",
            "content": "クイックソートの Python コードを書き、ファイル名を quick_sort.py としてください"
        }]
    },
    "parameters": {
        "result_format": "message",
        "tools": [
        {
            "type": "function",
            "function": {
                "name": "write_file",
                "description": "指定されたファイルにコンテンツを書き込み、存在しない場合は作成します。",
                "parameters": {
                    "type": "object",
                    "properties": {
                        "path": {
                            "type": "string",
                            "description": "オブジェクトファイルの相対パスまたは絶対パス"
                        },
                        "content": {
                            "type": "string",
                            "description": "ファイルに書き込む文字列コンテンツ"
                        }
                    },
                    "required": ["path", "content"]
                }
            }
        }
    ]
    }
}'

応答

{
    "output": {
        "choices": [
            {
                "finish_reason": "tool_calls",
                "message": {
                    "role": "assistant",
                    "tool_calls": [
                        {
                            "function": {
                                "name": "write_file",
                                "arguments": "{\"content\": \"def quick_sort(arr):\\\\n    if len(arr) <= 1:\\\\n        return arr\\\\n    pivot = arr[len(arr) // 2]\\\\n    left = [x for x in arr if x < pivot]\\\\n    middle = [x for x in arr if x == pivot]\\\\n    right = [x for x in arr if x > pivot]\\\\n    return quick_sort(left) + middle + quick_sort(right)\\\\n\\\\nif __name__ == \\\\\\\"__main__\\\\\\\":\\\\n    example_list = [3, 6, 8, 10, 1, 2, 1]\\\\n    print(\\\\\\\"Original list:\\\\\\\", example_list)\\\\n    sorted_list = quick_sort(example_list)\\\\n    print(\\\\\\\"Sorted list:\\\\\\\", sorted_list), \"path\": \"quick_sort.py\"}"
                            },
                            "index": 0,
                            "id": "call_645b149bbd274e8bb3789aae",
                            "type": "function"
                        }
                    ],
                    "content": ""
                }
            }
        ]
    },
    "usage": {
        "total_tokens": 684,
        "output_tokens": 193,
        "input_tokens": 491,
        "prompt_tokens_details": {
            "cached_tokens": 0
        }
    },
    "request_id": "d2386acd-fce3-9d0f-8015-c5f3a8bf9f5c"
}