SDK em TypeScript

npm install @qwen-code/sdk

import { query } from '@qwen-code/sdk';
 
// Consulta de uma única etapa
const result = query({
  prompt: 'Quais arquivos estão no diretório atual?',
  options: {
    cwd: '/caminho/para/o/projeto',
  },
});
 
// Iterar sobre as mensagens
for await (const message of result) {
  if (message.type === 'assistant') {
    console.log('Assistente:', message.message.content);
  } else if (message.type === 'result') {
    console.log('Resultado:', message.result);
  }
}

const query = qwen.query('Seu prompt', {
  timeout: {
    canUseTool: 60000, // 60 segundos para o retorno de chamada de permissão
    mcpRequest: 600000, // 10 minutos para chamadas de ferramentas MCP
    controlRequest: 60000, // 60 segundos para solicitações de controle
    streamClose: 15000, // 15 segundos para a espera de fechamento do fluxo
  },
});

import {
  isSDKUserMessage,
  isSDKAssistantMessage,
  isSDKSystemMessage,
  isSDKResultMessage,
  isSDKPartialAssistantMessage,
} from '@qwen-code/sdk';
 
for await (const message of result) {
  if (isSDKAssistantMessage(message)) {
    // Tratar mensagem do assistente
  } else if (isSDKResultMessage(message)) {
    // Tratar mensagem de resultado
  }
}

const q = query({ prompt: 'Olá', options: {} });
 
// Obter o ID da sessão
const sessionId = q.getSessionId();
 
// Verificar se está fechada
const closed = q.isClosed();
 
// Interromper a operação atual
await q.interrupt();
 
// Alterar o modo de permissão durante a sessão
await q.setPermissionMode('yolo');
 
// Alterar o modelo durante a sessão
await q.setModel('qwen-max');
 
// Fechar a sessão
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: 'Crie um arquivo hello.txt' },
    parent_tool_use_id: null,
  };
 
  // Aguarde alguma condição ou entrada do usuário
  yield {
    type: 'user',
    session_id: 'my-session',
    message: { role: 'user', content: 'Agora leia o arquivo novamente' },
    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 }) => {
  // Permite todas as operações de leitura
  if (toolName.startsWith('read_')) {
    return { behavior: 'allow', updatedInput: input };
  }
 
  // Solicita confirmação do usuário para operações de gravação (em um aplicativo real)
  const userApproved = await promptUser(`Permitir ${toolName}?`);
 
  if (userApproved) {
    return { behavior: 'allow', updatedInput: input };
  }
 
  return { behavior: 'deny', message: 'Usuário negou a operação' };
};
 
const result = query({
  prompt: 'Criar um novo arquivo',
  options: {
    canUseTool,
  },
});

import { query } from '@qwen-code/sdk';
 
const result = query({
  prompt: 'Use a ferramenta personalizada do meu servidor MCP',
  options: {
    mcpServers: {
      'meu-servidor': {
        command: 'node',
        args: ['caminho/para/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';
 
// Define uma ferramenta com esquema Zod
const calculatorTool = tool(
  'calculate_sum',
  'Adiciona dois números',
  { a: z.number(), b: z.number() },
  async (args) => ({
    content: [{ type: 'text', text: String(args.a + args.b) }],
  }),
);
 
// Cria o servidor MCP
const server = createSdkMcpServer({
  name: 'calculator',
  tools: [calculatorTool],
});
 
// Usa o servidor em uma consulta
const result = query({
  prompt: 'Quanto é 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: 'Tarefa de longa duração...',
  options: {
    abortController,
  },
});
 
// Abortar após 5 segundos
setTimeout(() => abortController.abort(), 5000);
 
try {
  for await (const message of result) {
    console.log(message);
  }
} catch (error) {
  if (isAbortError(error)) {
    console.log('A consulta foi abortada');
  } else {
    throw error;
  }
}

import { AbortError, isAbortError } from '@qwen-code/sdk';
 
try {
  // ... operações de consulta
} catch (error) {
  if (isAbortError(error)) {
    // Tratar o abortamento
  } else {
    // Tratar outros erros
  }
}