Function main

sdk/node/examples/basic/test_generate_object.ts:68–255

()

// ============================================================================

async function main() {
  console.log('='.repeat(70));
  console.log('generate_object Integration Test (Node.js SDK + MiniMax)');
  console.log('='.repeat(70));

  const configContent = loadConfig();
  const workspace = fs.mkdtempSync(path.join(os.tmpdir(), 'a3s-genobj-test-'));
  console.log(`Workspace: ${workspace}\n`);

  const agent = await Agent.create(configContent);
  const session = agent.session(workspace, {
    permissionPolicy: { defaultDecision: 'allow' },
  });

  // ── Test 1: Direct tool call — simple extraction ──────────────────────────

  await runTest('generate_object: extract person info (tool mode)', async () => {
    const result = await session.tool('generate_object', {
      schema: {
        type: 'object',
        required: ['name', 'age', 'city'],
        properties: {
          name: { type: 'string' },
          age: { type: 'integer' },
          city: { type: 'string' },
        },
      },
      prompt: 'Extract: "Alice Chen is 28 years old and lives in Shanghai."',
      schema_name: 'person',
      mode: 'tool',
    });

    if (result.exitCode !== 0) throw new Error(`Tool failed: ${result.output}`);
    const parsed = JSON.parse(result.output);
    const obj = parsed.object;
    if (obj.name.toLowerCase() !== 'alice chen') throw new Error(`name: ${obj.name}`);
    if (obj.age !== 28) throw new Error(`age: ${obj.age}`);
    if (!obj.city.toLowerCase().includes('shanghai')) throw new Error(`city: ${obj.city}`);
    console.log(`    Result: ${JSON.stringify(obj)}`);
  });

  // ── Test 2: Enum classification ───────────────────────────────────────────

  await runTest('generate_object: sentiment classification with enum', async () => {
    const result = await session.tool('generate_object', {
      schema: {
        type: 'object',
        required: ['sentiment', 'confidence'],
        properties: {
          sentiment: { type: 'string', enum: ['positive', 'negative', 'neutral'] },
          confidence: { type: 'number', minimum: 0, maximum: 1 },
        },
      },
      prompt: 'Classify sentiment: "This is the worst product I have ever used. Total waste of money."',
      schema_name: 'sentiment',
    });

    if (result.exitCode !== 0) throw new Error(`Tool failed: ${result.output}`);