import { describe, it, expect } from 'vitest';
import { extractToolCalls } from '../../../../src/core/providers/tool-utils';
describe('extractToolCalls', () => {
it('should extract a simple tool call', () => {
const content = `
Here is a tool call:
\`\`\`json
{
"tool": "test_tool",
"arguments": {
"arg1": "value1"
}
}
\`\`\`
`;
const result = extractToolCalls(content);
expect(result).toHaveLength(1);
expect(result[0]).toEqual({
tool: 'test_tool',
arguments: {
arg1: 'value1',
},
});
});
it('should extract a tool call with nested objects', () => {
const content = `
\`\`\`json
{
"tool": "complex_tool",
"arguments": {
"nested": {
"level2": {
"level3": "value"
}
},
"array": [1, 2, 3]
}
}
\`\`\`
`;
const result = extractToolCalls(content);
expect(result).toHaveLength(1);
expect(result[0]).toEqual({
tool: 'complex_tool',
arguments: {
nested: {
level2: {
level3: 'value',
},
},
array: [1, 2, 3],
},
});
});
it('should handle multiple tool calls', () => {
const content = `
First call:
\`\`\`json
{ "tool": "tool1", "arguments": {} }
\`\`\`
Second call:
\`\`\`json
{ "tool": "tool2", "arguments": {} }
\`\`\`
`;
const result = extractToolCalls(content);
expect(result).toHaveLength(2);
expect(result[0].tool).toBe('tool1');
expect(result[1].tool).toBe('tool2');
});
it('should ignore invalid JSON', () => {
const content = `
\`\`\`json
{ "tool": "broken", "arguments": {
\`\`\`
`;
const result = extractToolCalls(content);
expect(result).toHaveLength(0);
});
it('should ignore blocks without "tool" key', () => {
const content = `
\`\`\`json
{ "not_a_tool": "value" }
\`\`\`
`;
const result = extractToolCalls(content);
expect(result).toHaveLength(0);
});
it('should extract tool calls from a JSON array', () => {
const content = `
\`\`\`json
[
{ "tool": "tool1", "arguments": {} },
{ "tool": "tool2", "arguments": {} }
]
\`\`\`
`;
const result = extractToolCalls(content);
expect(result).toHaveLength(2);
expect(result[0].tool).toBe('tool1');
expect(result[1].tool).toBe('tool2');
});
it('should return empty array for empty string', () => {
const result = extractToolCalls('');
expect(result).toHaveLength(0);
});
it('should return empty array when no code blocks present', () => {
const content = 'Just some regular text without code blocks';
const result = extractToolCalls(content);
expect(result).toHaveLength(0);
});
it('should extract tool calls from non-JSON code blocks via fallback', () => {
const content = `
\`\`\`javascript
{ "tool": "test", "arguments": {} }
\`\`\`
`;
const result = extractToolCalls(content);
expect(result).toHaveLength(1);
expect(result[0].tool).toBe('test');
});
it('should handle mixed valid and invalid tool calls', () => {
const content = `
Valid call:
\`\`\`json
{ "tool": "valid_tool", "arguments": {} }
\`\`\`
Invalid JSON:
\`\`\`json
{ "tool": "broken",
\`\`\`
Non-JSON block:
\`\`\`text
{ "tool": "text_tool", "arguments": {} }
\`\`\`
`;
const result = extractToolCalls(content);
expect(result).toHaveLength(1);
expect(result[0].tool).toBe('valid_tool');
});
it('should extract tool call from raw JSON without code blocks (fallback)', () => {
const content = '{ "tool": "fallback_tool", "arguments": { "arg": "val" } }';
const result = extractToolCalls(content);
expect(result).toHaveLength(1);
expect(result[0]).toEqual({
tool: 'fallback_tool',
arguments: { arg: 'val' }
});
});
it('should extract tool call from raw JSON embedded in text (fallback)', () => {
const content = 'I will use this tool: { "tool": "embedded_tool", "arguments": {} } to do something.';
const result = extractToolCalls(content);
expect(result).toHaveLength(1);
expect(result[0].tool).toBe('embedded_tool');
});
});
