Typescript SDK

npm install @qwen-code/sdk

import { query } from '@qwen-code/sdk';
 
// 单轮查询
const result = query({
  prompt: '当前目录下有哪些文件？',
  options: {
    cwd: '/path/to/project',
  },
});
 
// 遍历消息
for await (const message of result) {
  if (message.type === 'assistant') {
    console.log('助手：', message.message.content);
  } else if (message.type === 'result') {
    console.log('结果：', message.result);
  }
}

const query = qwen.query('Your prompt', {
  timeout: {
    canUseTool: 60000, // 权限回调 60 秒
    mcpRequest: 600000, // MCP 工具调用 10 分钟
    controlRequest: 60000, // 控制请求 60 秒
    streamClose: 15000, // 流关闭等待 15 秒
  },
});

import {
  isSDKUserMessage,
  isSDKAssistantMessage,
  isSDKSystemMessage,
  isSDKResultMessage,
  isSDKPartialAssistantMessage,
} from '@qwen-code/sdk';
 
for await (const message of result) {
  if (isSDKAssistantMessage(message)) {
    // 处理助手消息
  } else if (isSDKResultMessage(message)) {
    // 处理结果消息
  }
}

const q = query({ prompt: 'Hello', options: {} });
 
// 获取会话 ID
const sessionId = q.getSessionId();
 
// 检查是否已关闭
const closed = q.isClosed();
 
// 中断当前操作
await q.interrupt();
 
// 会话中途更改权限模式
await q.setPermissionMode('yolo');
 
// 会话中途更改模型
await q.setModel('qwen-max');
 
// 关闭会话
await q.close();

import { query, type SDKUserMessage } from '@qwen-code/sdk';
 
async function* generateMessages(): AsyncIterable<SDKUserMessage> {
  yield {
    type: 'user',
    session_id: 'my-session',
    message: { role: 'user', content: '创建一个 hello.txt 文件' },
    parent_tool_use_id: null,
  };
 
  // 等待某些条件或用户输入
  yield {
    type: 'user',
    session_id: 'my-session',
    message: { role: 'user', content: '现在读取该文件内容' },
    parent_tool_use_id: null,
  };
}
 
const result = query({
  prompt: generateMessages(),
  options: {
    permissionMode: 'auto-edit',
  },
});
 
for await (const message of result) {
  console.log(message);
}

import { query, type CanUseTool } from '@qwen-code/sdk';
 
const canUseTool: CanUseTool = async (toolName, input, { signal }) => {
  // 允许所有读取操作
  if (toolName.startsWith('read_')) {
    return { behavior: 'allow', updatedInput: input };
  }
 
  // 提示用户进行写入操作（在实际应用中）
  const userApproved = await promptUser(`允许 ${toolName} 吗？`);
 
  if (userApproved) {
    return { behavior: 'allow', updatedInput: input };
  }
 
  return { behavior: 'deny', message: '用户拒绝了该操作' };
};
 
const result = query({
  prompt: '创建一个新文件',
  options: {
    canUseTool,
  },
});

import { query } from '@qwen-code/sdk';
 
const result = query({
  prompt: 'Use the custom tool from my MCP server',
  options: {
    mcpServers: {
      'my-server': {
        command: 'node',
        args: ['path/to/mcp-server.js'],
        env: { PORT: '3000' },
      },
    },
  },
});

{
  content: Array<
    | { type: 'text'; text: string }
    | { type: 'image'; data: string; mimeType: string }
    | { type: 'resource'; uri: string; mimeType?: string; text?: string }
  >;
  isError?: boolean;
}

import { z } from 'zod';
import { query, tool, createSdkMcpServer } from '@qwen-code/sdk';
 
// 使用 Zod schema 定义一个工具
const calculatorTool = tool(
  'calculate_sum',
  '将两个数字相加',
  { a: z.number(), b: z.number() },
  async (args) => ({
    content: [{ type: 'text', text: String(args.a + args.b) }],
  }),
);
 
// 创建 MCP 服务器
const server = createSdkMcpServer({
  name: 'calculator',
  tools: [calculatorTool],
});
 
// 在查询中使用该服务器
const result = query({
  prompt: '42 加 17 等于多少？',
  options: {
    permissionMode: 'yolo',
    mcpServers: {
      calculator: server,
    },
  },
});
 
for await (const message of result) {
  console.log(message);
}

import { query, isAbortError } from '@qwen-code/sdk';
 
const abortController = new AbortController();
 
const result = query({
  prompt: '长时间运行的任务...',
  options: {
    abortController,
  },
});
 
// 5秒后中止
setTimeout(() => abortController.abort(), 5000);
 
try {
  for await (const message of result) {
    console.log(message);
  }
} catch (error) {
  if (isAbortError(error)) {
    console.log('查询已被中止');
  } else {
    throw error;
  }
}

import { AbortError, isAbortError } from '@qwen-code/sdk';
 
try {
  // ... 查询操作
} catch (error) {
  if (isAbortError(error)) {
    // 处理中止情况
  } else {
    // 处理其他错误
  }
}