mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-05-13 05:12:55 -07:00
Markdown w/ Frontmatter Agent Parser (#16094)
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
@@ -0,0 +1,342 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
||||||
|
import * as fs from 'node:fs/promises';
|
||||||
|
import * as path from 'node:path';
|
||||||
|
import * as os from 'node:os';
|
||||||
|
import {
|
||||||
|
parseAgentMarkdown,
|
||||||
|
markdownToAgentDefinition,
|
||||||
|
loadAgentsFromDirectory,
|
||||||
|
AgentLoadError,
|
||||||
|
} from './agentLoader.js';
|
||||||
|
import { GEMINI_MODEL_ALIAS_PRO } from '../config/models.js';
|
||||||
|
import type { LocalAgentDefinition } from './types.js';
|
||||||
|
|
||||||
|
describe('loader', () => {
|
||||||
|
let tempDir: string;
|
||||||
|
|
||||||
|
beforeEach(async () => {
|
||||||
|
tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'agent-test-'));
|
||||||
|
});
|
||||||
|
|
||||||
|
afterEach(async () => {
|
||||||
|
if (tempDir) {
|
||||||
|
await fs.rm(tempDir, { recursive: true, force: true });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
async function writeAgentMarkdown(content: string, fileName = 'test.md') {
|
||||||
|
const filePath = path.join(tempDir, fileName);
|
||||||
|
await fs.writeFile(filePath, content);
|
||||||
|
return filePath;
|
||||||
|
}
|
||||||
|
|
||||||
|
describe('parseAgentMarkdown', () => {
|
||||||
|
it('should parse a valid markdown agent file', async () => {
|
||||||
|
const filePath = await writeAgentMarkdown(`---
|
||||||
|
name: test-agent-md
|
||||||
|
description: A markdown agent
|
||||||
|
---
|
||||||
|
You are a markdown agent.`);
|
||||||
|
|
||||||
|
const result = await parseAgentMarkdown(filePath);
|
||||||
|
expect(result).toHaveLength(1);
|
||||||
|
expect(result[0]).toMatchObject({
|
||||||
|
name: 'test-agent-md',
|
||||||
|
description: 'A markdown agent',
|
||||||
|
kind: 'local',
|
||||||
|
system_prompt: 'You are a markdown agent.',
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should parse frontmatter with tools and model config', async () => {
|
||||||
|
const filePath = await writeAgentMarkdown(`---
|
||||||
|
name: complex-agent
|
||||||
|
description: A complex markdown agent
|
||||||
|
tools:
|
||||||
|
- run_shell_command
|
||||||
|
model: gemini-pro
|
||||||
|
temperature: 0.7
|
||||||
|
---
|
||||||
|
System prompt content.`);
|
||||||
|
|
||||||
|
const result = await parseAgentMarkdown(filePath);
|
||||||
|
expect(result).toHaveLength(1);
|
||||||
|
expect(result[0]).toMatchObject({
|
||||||
|
name: 'complex-agent',
|
||||||
|
description: 'A complex markdown agent',
|
||||||
|
tools: ['run_shell_command'],
|
||||||
|
model: 'gemini-pro',
|
||||||
|
temperature: 0.7,
|
||||||
|
system_prompt: 'System prompt content.',
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should throw AgentLoadError if frontmatter is missing', async () => {
|
||||||
|
const filePath = await writeAgentMarkdown(`Just some markdown content.`);
|
||||||
|
await expect(parseAgentMarkdown(filePath)).rejects.toThrow(
|
||||||
|
AgentLoadError,
|
||||||
|
);
|
||||||
|
await expect(parseAgentMarkdown(filePath)).rejects.toThrow(
|
||||||
|
'Invalid markdown format',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should throw AgentLoadError if frontmatter is invalid YAML', async () => {
|
||||||
|
const filePath = await writeAgentMarkdown(`---
|
||||||
|
name: [invalid yaml
|
||||||
|
---
|
||||||
|
Body`);
|
||||||
|
await expect(parseAgentMarkdown(filePath)).rejects.toThrow(
|
||||||
|
AgentLoadError,
|
||||||
|
);
|
||||||
|
await expect(parseAgentMarkdown(filePath)).rejects.toThrow(
|
||||||
|
'YAML frontmatter parsing failed',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should throw AgentLoadError if validation fails (missing required field)', async () => {
|
||||||
|
const filePath = await writeAgentMarkdown(`---
|
||||||
|
name: test-agent
|
||||||
|
# missing description
|
||||||
|
---
|
||||||
|
Body`);
|
||||||
|
await expect(parseAgentMarkdown(filePath)).rejects.toThrow(
|
||||||
|
/Validation failed/,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should throw AgentLoadError if tools list includes forbidden tool', async () => {
|
||||||
|
const filePath = await writeAgentMarkdown(`---
|
||||||
|
name: test-agent
|
||||||
|
description: Test
|
||||||
|
tools:
|
||||||
|
- delegate_to_agent
|
||||||
|
---
|
||||||
|
Body`);
|
||||||
|
await expect(parseAgentMarkdown(filePath)).rejects.toThrow(
|
||||||
|
/tools list cannot include 'delegate_to_agent'/,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should parse a valid remote agent markdown file', async () => {
|
||||||
|
const filePath = await writeAgentMarkdown(`---
|
||||||
|
kind: remote
|
||||||
|
name: remote-agent
|
||||||
|
description: A remote agent
|
||||||
|
agent_card_url: https://example.com/card
|
||||||
|
---
|
||||||
|
`);
|
||||||
|
const result = await parseAgentMarkdown(filePath);
|
||||||
|
expect(result).toHaveLength(1);
|
||||||
|
expect(result[0]).toEqual({
|
||||||
|
kind: 'remote',
|
||||||
|
name: 'remote-agent',
|
||||||
|
description: 'A remote agent',
|
||||||
|
agent_card_url: 'https://example.com/card',
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should infer remote agent kind from agent_card_url', async () => {
|
||||||
|
const filePath = await writeAgentMarkdown(`---
|
||||||
|
name: inferred-remote
|
||||||
|
description: Inferred
|
||||||
|
agent_card_url: https://example.com/inferred
|
||||||
|
---
|
||||||
|
`);
|
||||||
|
const result = await parseAgentMarkdown(filePath);
|
||||||
|
expect(result).toHaveLength(1);
|
||||||
|
expect(result[0]).toEqual({
|
||||||
|
kind: 'remote',
|
||||||
|
name: 'inferred-remote',
|
||||||
|
description: 'Inferred',
|
||||||
|
agent_card_url: 'https://example.com/inferred',
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should parse a remote agent with no body', async () => {
|
||||||
|
const filePath = await writeAgentMarkdown(`---
|
||||||
|
kind: remote
|
||||||
|
name: no-body-remote
|
||||||
|
agent_card_url: https://example.com/card
|
||||||
|
---
|
||||||
|
`);
|
||||||
|
const result = await parseAgentMarkdown(filePath);
|
||||||
|
expect(result).toHaveLength(1);
|
||||||
|
expect(result[0]).toEqual({
|
||||||
|
kind: 'remote',
|
||||||
|
name: 'no-body-remote',
|
||||||
|
agent_card_url: 'https://example.com/card',
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should parse multiple remote agents in a list', async () => {
|
||||||
|
const filePath = await writeAgentMarkdown(`---
|
||||||
|
- kind: remote
|
||||||
|
name: remote-1
|
||||||
|
agent_card_url: https://example.com/1
|
||||||
|
- kind: remote
|
||||||
|
name: remote-2
|
||||||
|
agent_card_url: https://example.com/2
|
||||||
|
---
|
||||||
|
`);
|
||||||
|
const result = await parseAgentMarkdown(filePath);
|
||||||
|
expect(result).toHaveLength(2);
|
||||||
|
expect(result[0]).toEqual({
|
||||||
|
kind: 'remote',
|
||||||
|
name: 'remote-1',
|
||||||
|
agent_card_url: 'https://example.com/1',
|
||||||
|
});
|
||||||
|
expect(result[1]).toEqual({
|
||||||
|
kind: 'remote',
|
||||||
|
name: 'remote-2',
|
||||||
|
agent_card_url: 'https://example.com/2',
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('markdownToAgentDefinition', () => {
|
||||||
|
it('should convert valid Markdown DTO to AgentDefinition with defaults', () => {
|
||||||
|
const markdown = {
|
||||||
|
kind: 'local' as const,
|
||||||
|
name: 'test-agent',
|
||||||
|
description: 'A test agent',
|
||||||
|
system_prompt: 'You are a test agent.',
|
||||||
|
};
|
||||||
|
|
||||||
|
const result = markdownToAgentDefinition(markdown);
|
||||||
|
expect(result).toMatchObject({
|
||||||
|
name: 'test-agent',
|
||||||
|
description: 'A test agent',
|
||||||
|
promptConfig: {
|
||||||
|
systemPrompt: 'You are a test agent.',
|
||||||
|
query: '${query}',
|
||||||
|
},
|
||||||
|
modelConfig: {
|
||||||
|
model: 'inherit',
|
||||||
|
top_p: 0.95,
|
||||||
|
},
|
||||||
|
runConfig: {
|
||||||
|
max_time_minutes: 5,
|
||||||
|
},
|
||||||
|
inputConfig: {
|
||||||
|
inputs: {
|
||||||
|
query: {
|
||||||
|
type: 'string',
|
||||||
|
required: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should pass through model aliases', () => {
|
||||||
|
const markdown = {
|
||||||
|
kind: 'local' as const,
|
||||||
|
name: 'test-agent',
|
||||||
|
description: 'A test agent',
|
||||||
|
model: GEMINI_MODEL_ALIAS_PRO,
|
||||||
|
system_prompt: 'You are a test agent.',
|
||||||
|
};
|
||||||
|
|
||||||
|
const result = markdownToAgentDefinition(
|
||||||
|
markdown,
|
||||||
|
) as LocalAgentDefinition;
|
||||||
|
expect(result.modelConfig.model).toBe(GEMINI_MODEL_ALIAS_PRO);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should pass through unknown model names (e.g. auto)', () => {
|
||||||
|
const markdown = {
|
||||||
|
kind: 'local' as const,
|
||||||
|
name: 'test-agent',
|
||||||
|
description: 'A test agent',
|
||||||
|
model: 'auto',
|
||||||
|
system_prompt: 'You are a test agent.',
|
||||||
|
};
|
||||||
|
|
||||||
|
const result = markdownToAgentDefinition(
|
||||||
|
markdown,
|
||||||
|
) as LocalAgentDefinition;
|
||||||
|
expect(result.modelConfig.model).toBe('auto');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should convert remote agent definition', () => {
|
||||||
|
const markdown = {
|
||||||
|
kind: 'remote' as const,
|
||||||
|
name: 'remote-agent',
|
||||||
|
description: 'A remote agent',
|
||||||
|
agent_card_url: 'https://example.com/card',
|
||||||
|
};
|
||||||
|
|
||||||
|
const result = markdownToAgentDefinition(markdown);
|
||||||
|
expect(result).toEqual({
|
||||||
|
kind: 'remote',
|
||||||
|
name: 'remote-agent',
|
||||||
|
description: 'A remote agent',
|
||||||
|
displayName: undefined,
|
||||||
|
agentCardUrl: 'https://example.com/card',
|
||||||
|
inputConfig: {
|
||||||
|
inputs: {
|
||||||
|
query: {
|
||||||
|
type: 'string',
|
||||||
|
description: 'The task for the agent.',
|
||||||
|
required: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('loadAgentsFromDirectory', () => {
|
||||||
|
it('should load definitions from a directory (Markdown only)', async () => {
|
||||||
|
await writeAgentMarkdown(
|
||||||
|
`---
|
||||||
|
name: agent-1
|
||||||
|
description: Agent 1
|
||||||
|
---
|
||||||
|
Prompt 1`,
|
||||||
|
'valid.md',
|
||||||
|
);
|
||||||
|
|
||||||
|
// Create a non-supported file
|
||||||
|
await fs.writeFile(path.join(tempDir, 'other.txt'), 'content');
|
||||||
|
|
||||||
|
// Create a hidden file
|
||||||
|
await writeAgentMarkdown(
|
||||||
|
`---
|
||||||
|
name: hidden
|
||||||
|
description: Hidden
|
||||||
|
---
|
||||||
|
Hidden`,
|
||||||
|
'_hidden.md',
|
||||||
|
);
|
||||||
|
|
||||||
|
const result = await loadAgentsFromDirectory(tempDir);
|
||||||
|
expect(result.agents).toHaveLength(1);
|
||||||
|
expect(result.agents[0].name).toBe('agent-1');
|
||||||
|
expect(result.errors).toHaveLength(0);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should return empty result if directory does not exist', async () => {
|
||||||
|
const nonExistentDir = path.join(tempDir, 'does-not-exist');
|
||||||
|
const result = await loadAgentsFromDirectory(nonExistentDir);
|
||||||
|
expect(result.agents).toHaveLength(0);
|
||||||
|
expect(result.errors).toHaveLength(0);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should capture errors for malformed individual files', async () => {
|
||||||
|
// Create a malformed Markdown file
|
||||||
|
await writeAgentMarkdown('invalid markdown', 'malformed.md');
|
||||||
|
|
||||||
|
const result = await loadAgentsFromDirectory(tempDir);
|
||||||
|
expect(result.agents).toHaveLength(0);
|
||||||
|
expect(result.errors).toHaveLength(1);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
+101
-92
@@ -1,10 +1,10 @@
|
|||||||
/**
|
/**
|
||||||
* @license
|
* @license
|
||||||
* Copyright 2025 Google LLC
|
* Copyright 2026 Google LLC
|
||||||
* SPDX-License-Identifier: Apache-2.0
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
*/
|
*/
|
||||||
|
|
||||||
import TOML from '@iarna/toml';
|
import yaml from 'js-yaml';
|
||||||
import * as fs from 'node:fs/promises';
|
import * as fs from 'node:fs/promises';
|
||||||
import { type Dirent } from 'node:fs';
|
import { type Dirent } from 'node:fs';
|
||||||
import * as path from 'node:path';
|
import * as path from 'node:path';
|
||||||
@@ -14,40 +14,38 @@ import {
|
|||||||
isValidToolName,
|
isValidToolName,
|
||||||
DELEGATE_TO_AGENT_TOOL_NAME,
|
DELEGATE_TO_AGENT_TOOL_NAME,
|
||||||
} from '../tools/tool-names.js';
|
} from '../tools/tool-names.js';
|
||||||
|
import { FRONTMATTER_REGEX } from '../skills/skillLoader.js';
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* DTO for TOML parsing - represents the raw structure of the TOML file.
|
* DTO for Markdown parsing - represents the structure from frontmatter.
|
||||||
*/
|
*/
|
||||||
interface TomlBaseAgentDefinition {
|
interface FrontmatterBaseAgentDefinition {
|
||||||
name: string;
|
name: string;
|
||||||
display_name?: string;
|
display_name?: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
interface TomlLocalAgentDefinition extends TomlBaseAgentDefinition {
|
interface FrontmatterLocalAgentDefinition
|
||||||
|
extends FrontmatterBaseAgentDefinition {
|
||||||
kind: 'local';
|
kind: 'local';
|
||||||
description: string;
|
description: string;
|
||||||
tools?: string[];
|
tools?: string[];
|
||||||
prompts: {
|
|
||||||
system_prompt: string;
|
system_prompt: string;
|
||||||
query?: string;
|
|
||||||
};
|
|
||||||
model?: {
|
|
||||||
model?: string;
|
model?: string;
|
||||||
temperature?: number;
|
temperature?: number;
|
||||||
};
|
|
||||||
run?: {
|
|
||||||
max_turns?: number;
|
max_turns?: number;
|
||||||
timeout_mins?: number;
|
timeout_mins?: number;
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
interface TomlRemoteAgentDefinition extends TomlBaseAgentDefinition {
|
interface FrontmatterRemoteAgentDefinition
|
||||||
description?: string;
|
extends FrontmatterBaseAgentDefinition {
|
||||||
kind: 'remote';
|
kind: 'remote';
|
||||||
|
description?: string;
|
||||||
agent_card_url: string;
|
agent_card_url: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
type TomlAgentDefinition = TomlLocalAgentDefinition | TomlRemoteAgentDefinition;
|
type FrontmatterAgentDefinition =
|
||||||
|
| FrontmatterLocalAgentDefinition
|
||||||
|
| FrontmatterRemoteAgentDefinition;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Error thrown when an agent definition is invalid or cannot be loaded.
|
* Error thrown when an agent definition is invalid or cannot be loaded.
|
||||||
@@ -87,23 +85,11 @@ const localAgentSchema = z
|
|||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
.optional(),
|
.optional(),
|
||||||
prompts: z.object({
|
|
||||||
system_prompt: z.string().min(1),
|
|
||||||
query: z.string().optional(),
|
|
||||||
}),
|
|
||||||
model: z
|
|
||||||
.object({
|
|
||||||
model: z.string().optional(),
|
model: z.string().optional(),
|
||||||
temperature: z.number().optional(),
|
temperature: z.number().optional(),
|
||||||
})
|
|
||||||
.optional(),
|
|
||||||
run: z
|
|
||||||
.object({
|
|
||||||
max_turns: z.number().int().positive().optional(),
|
max_turns: z.number().int().positive().optional(),
|
||||||
timeout_mins: z.number().int().positive().optional(),
|
timeout_mins: z.number().int().positive().optional(),
|
||||||
})
|
})
|
||||||
.optional(),
|
|
||||||
})
|
|
||||||
.strict();
|
.strict();
|
||||||
|
|
||||||
const remoteAgentSchema = z
|
const remoteAgentSchema = z
|
||||||
@@ -116,22 +102,16 @@ const remoteAgentSchema = z
|
|||||||
})
|
})
|
||||||
.strict();
|
.strict();
|
||||||
|
|
||||||
const remoteAgentsConfigSchema = z
|
|
||||||
.object({
|
|
||||||
remote_agents: z.array(remoteAgentSchema),
|
|
||||||
})
|
|
||||||
.strict();
|
|
||||||
|
|
||||||
// Use a Zod union to automatically discriminate between local and remote
|
// Use a Zod union to automatically discriminate between local and remote
|
||||||
// agent types. This is more robust than manually checking the 'kind' field,
|
// agent types.
|
||||||
// as it correctly handles cases where 'kind' is omitted by relying on
|
|
||||||
// the presence of unique fields like `agent_card_url` or `prompts`.
|
|
||||||
const agentUnionOptions = [
|
const agentUnionOptions = [
|
||||||
{ schema: localAgentSchema, label: 'Local Agent' },
|
{ schema: localAgentSchema, label: 'Local Agent' },
|
||||||
{ schema: remoteAgentSchema, label: 'Remote Agent' },
|
{ schema: remoteAgentSchema, label: 'Remote Agent' },
|
||||||
] as const;
|
] as const;
|
||||||
|
|
||||||
const singleAgentSchema = z.union([
|
const remoteAgentsListSchema = z.array(remoteAgentSchema);
|
||||||
|
|
||||||
|
const markdownFrontmatterSchema = z.union([
|
||||||
agentUnionOptions[0].schema,
|
agentUnionOptions[0].schema,
|
||||||
agentUnionOptions[1].schema,
|
agentUnionOptions[1].schema,
|
||||||
]);
|
]);
|
||||||
@@ -159,15 +139,15 @@ function formatZodError(error: z.ZodError, context: string): string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Parses and validates an agent TOML file. Returns a validated array of RemoteAgentDefinitions or a single LocalAgentDefinition.
|
* Parses and validates an agent Markdown file with frontmatter.
|
||||||
*
|
*
|
||||||
* @param filePath Path to the TOML file.
|
* @param filePath Path to the Markdown file.
|
||||||
* @returns An array of parsed and validated TomlAgentDefinitions.
|
* @returns An array containing the single parsed agent definition.
|
||||||
* @throws AgentLoadError if parsing or validation fails.
|
* @throws AgentLoadError if parsing or validation fails.
|
||||||
*/
|
*/
|
||||||
export async function parseAgentToml(
|
export async function parseAgentMarkdown(
|
||||||
filePath: string,
|
filePath: string,
|
||||||
): Promise<TomlAgentDefinition[]> {
|
): Promise<FrontmatterAgentDefinition[]> {
|
||||||
let content: string;
|
let content: string;
|
||||||
try {
|
try {
|
||||||
content = await fs.readFile(filePath, 'utf-8');
|
content = await fs.readFile(filePath, 'utf-8');
|
||||||
@@ -178,34 +158,44 @@ export async function parseAgentToml(
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
let raw: unknown;
|
// Split frontmatter and body
|
||||||
|
const match = content.match(FRONTMATTER_REGEX);
|
||||||
|
if (!match) {
|
||||||
|
throw new AgentLoadError(
|
||||||
|
filePath,
|
||||||
|
'Invalid markdown format. File must start with YAML frontmatter enclosed in "---".',
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
const frontmatterStr = match[1];
|
||||||
|
const body = match[2] || '';
|
||||||
|
|
||||||
|
let rawFrontmatter: unknown;
|
||||||
try {
|
try {
|
||||||
raw = TOML.parse(content);
|
rawFrontmatter = yaml.load(frontmatterStr);
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
throw new AgentLoadError(
|
throw new AgentLoadError(
|
||||||
filePath,
|
filePath,
|
||||||
`TOML parsing failed: ${(error as Error).message}`,
|
`YAML frontmatter parsing failed: ${(error as Error).message}`,
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check for `remote_agents` array
|
// Handle array of remote agents
|
||||||
if (
|
if (Array.isArray(rawFrontmatter)) {
|
||||||
typeof raw === 'object' &&
|
const result = remoteAgentsListSchema.safeParse(rawFrontmatter);
|
||||||
raw !== null &&
|
|
||||||
'remote_agents' in (raw as Record<string, unknown>)
|
|
||||||
) {
|
|
||||||
const result = remoteAgentsConfigSchema.safeParse(raw);
|
|
||||||
if (!result.success) {
|
if (!result.success) {
|
||||||
throw new AgentLoadError(
|
throw new AgentLoadError(
|
||||||
filePath,
|
filePath,
|
||||||
`Validation failed: ${formatZodError(result.error, 'Remote Agents Config')}`,
|
`Validation failed: ${formatZodError(result.error, 'Remote Agents List')}`,
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
return result.data.remote_agents as TomlAgentDefinition[];
|
return result.data.map((agent) => ({
|
||||||
|
...agent,
|
||||||
|
kind: 'remote',
|
||||||
|
}));
|
||||||
}
|
}
|
||||||
|
|
||||||
// Single Agent Logic
|
const result = markdownFrontmatterSchema.safeParse(rawFrontmatter);
|
||||||
const result = singleAgentSchema.safeParse(raw);
|
|
||||||
|
|
||||||
if (!result.success) {
|
if (!result.success) {
|
||||||
throw new AgentLoadError(
|
throw new AgentLoadError(
|
||||||
@@ -214,27 +204,47 @@ export async function parseAgentToml(
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
const toml = result.data as TomlAgentDefinition;
|
const frontmatter = result.data;
|
||||||
|
|
||||||
// Prevent sub-agents from delegating to other agents (to prevent recursion/complexity)
|
if (frontmatter.kind === 'remote') {
|
||||||
if ('tools' in toml && toml.tools?.includes(DELEGATE_TO_AGENT_TOOL_NAME)) {
|
return [
|
||||||
|
{
|
||||||
|
...frontmatter,
|
||||||
|
kind: 'remote',
|
||||||
|
},
|
||||||
|
];
|
||||||
|
}
|
||||||
|
|
||||||
|
// Local agent validation
|
||||||
|
// Validate tools
|
||||||
|
if (
|
||||||
|
frontmatter.tools &&
|
||||||
|
frontmatter.tools.includes(DELEGATE_TO_AGENT_TOOL_NAME)
|
||||||
|
) {
|
||||||
throw new AgentLoadError(
|
throw new AgentLoadError(
|
||||||
filePath,
|
filePath,
|
||||||
`Validation failed: tools list cannot include '${DELEGATE_TO_AGENT_TOOL_NAME}'. Sub-agents cannot delegate to other agents.`,
|
`Validation failed: tools list cannot include '${DELEGATE_TO_AGENT_TOOL_NAME}'. Sub-agents cannot delegate to other agents.`,
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
return [toml];
|
// Construct the local agent definition
|
||||||
|
const agentDef: FrontmatterLocalAgentDefinition = {
|
||||||
|
...frontmatter,
|
||||||
|
kind: 'local',
|
||||||
|
system_prompt: body.trim(),
|
||||||
|
};
|
||||||
|
|
||||||
|
return [agentDef];
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Converts a TomlAgentDefinition DTO to the internal AgentDefinition structure.
|
* Converts a FrontmatterAgentDefinition DTO to the internal AgentDefinition structure.
|
||||||
*
|
*
|
||||||
* @param toml The parsed TOML definition.
|
* @param markdown The parsed Markdown/Frontmatter definition.
|
||||||
* @returns The internal AgentDefinition.
|
* @returns The internal AgentDefinition.
|
||||||
*/
|
*/
|
||||||
export function tomlToAgentDefinition(
|
export function markdownToAgentDefinition(
|
||||||
toml: TomlAgentDefinition,
|
markdown: FrontmatterAgentDefinition,
|
||||||
): AgentDefinition {
|
): AgentDefinition {
|
||||||
const inputConfig = {
|
const inputConfig = {
|
||||||
inputs: {
|
inputs: {
|
||||||
@@ -246,41 +256,41 @@ export function tomlToAgentDefinition(
|
|||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
if (toml.kind === 'remote') {
|
if (markdown.kind === 'remote') {
|
||||||
return {
|
return {
|
||||||
kind: 'remote',
|
kind: 'remote',
|
||||||
name: toml.name,
|
name: markdown.name,
|
||||||
description: toml.description || '(Loading description...)',
|
description: markdown.description || '(Loading description...)',
|
||||||
displayName: toml.display_name,
|
displayName: markdown.display_name,
|
||||||
agentCardUrl: toml.agent_card_url,
|
agentCardUrl: markdown.agent_card_url,
|
||||||
inputConfig,
|
inputConfig,
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
// If a model is specified, use it. Otherwise, inherit
|
// If a model is specified, use it. Otherwise, inherit
|
||||||
const modelName = toml.model?.model || 'inherit';
|
const modelName = markdown.model || 'inherit';
|
||||||
|
|
||||||
return {
|
return {
|
||||||
kind: 'local',
|
kind: 'local',
|
||||||
name: toml.name,
|
name: markdown.name,
|
||||||
description: toml.description,
|
description: markdown.description,
|
||||||
displayName: toml.display_name,
|
displayName: markdown.display_name,
|
||||||
promptConfig: {
|
promptConfig: {
|
||||||
systemPrompt: toml.prompts.system_prompt,
|
systemPrompt: markdown.system_prompt,
|
||||||
query: toml.prompts.query,
|
query: '${query}',
|
||||||
},
|
},
|
||||||
modelConfig: {
|
modelConfig: {
|
||||||
model: modelName,
|
model: modelName,
|
||||||
temp: toml.model?.temperature ?? 1,
|
temp: markdown.temperature ?? 1,
|
||||||
top_p: 0.95,
|
top_p: 0.95,
|
||||||
},
|
},
|
||||||
runConfig: {
|
runConfig: {
|
||||||
max_turns: toml.run?.max_turns,
|
max_turns: markdown.max_turns,
|
||||||
max_time_minutes: toml.run?.timeout_mins || 5,
|
max_time_minutes: markdown.timeout_mins || 5,
|
||||||
},
|
},
|
||||||
toolConfig: toml.tools
|
toolConfig: markdown.tools
|
||||||
? {
|
? {
|
||||||
tools: toml.tools,
|
tools: markdown.tools,
|
||||||
}
|
}
|
||||||
: undefined,
|
: undefined,
|
||||||
inputConfig,
|
inputConfig,
|
||||||
@@ -289,7 +299,8 @@ export function tomlToAgentDefinition(
|
|||||||
|
|
||||||
/**
|
/**
|
||||||
* Loads all agents from a specific directory.
|
* Loads all agents from a specific directory.
|
||||||
* Ignores non-TOML files and files starting with _.
|
* Ignores files starting with _ and non-supported extensions.
|
||||||
|
* Supported extensions: .md
|
||||||
*
|
*
|
||||||
* @param dir Directory path to scan.
|
* @param dir Directory path to scan.
|
||||||
* @returns Object containing successfully loaded agents and any errors.
|
* @returns Object containing successfully loaded agents and any errors.
|
||||||
@@ -319,21 +330,19 @@ export async function loadAgentsFromDirectory(
|
|||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
const files = dirEntries
|
const files = dirEntries.filter(
|
||||||
.filter(
|
|
||||||
(entry) =>
|
(entry) =>
|
||||||
entry.isFile() &&
|
entry.isFile() &&
|
||||||
entry.name.endsWith('.toml') &&
|
!entry.name.startsWith('_') &&
|
||||||
!entry.name.startsWith('_'),
|
entry.name.endsWith('.md'),
|
||||||
)
|
);
|
||||||
.map((entry) => entry.name);
|
|
||||||
|
|
||||||
for (const file of files) {
|
for (const entry of files) {
|
||||||
const filePath = path.join(dir, file);
|
const filePath = path.join(dir, entry.name);
|
||||||
try {
|
try {
|
||||||
const tomls = await parseAgentToml(filePath);
|
const agentDefs = await parseAgentMarkdown(filePath);
|
||||||
for (const toml of tomls) {
|
for (const def of agentDefs) {
|
||||||
const agent = tomlToAgentDefinition(toml);
|
const agent = markdownToAgentDefinition(def);
|
||||||
result.agents.push(agent);
|
result.agents.push(agent);
|
||||||
}
|
}
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
@@ -19,9 +19,9 @@ import {
|
|||||||
PREVIEW_GEMINI_MODEL,
|
PREVIEW_GEMINI_MODEL,
|
||||||
PREVIEW_GEMINI_MODEL_AUTO,
|
PREVIEW_GEMINI_MODEL_AUTO,
|
||||||
} from '../config/models.js';
|
} from '../config/models.js';
|
||||||
import * as tomlLoader from './toml-loader.js';
|
import * as tomlLoader from './agentLoader.js';
|
||||||
|
|
||||||
vi.mock('./toml-loader.js', () => ({
|
vi.mock('./agentLoader.js', () => ({
|
||||||
loadAgentsFromDirectory: vi
|
loadAgentsFromDirectory: vi
|
||||||
.fn()
|
.fn()
|
||||||
.mockResolvedValue({ agents: [], errors: [] }),
|
.mockResolvedValue({ agents: [], errors: [] }),
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ import { Storage } from '../config/storage.js';
|
|||||||
import { coreEvents, CoreEvent } from '../utils/events.js';
|
import { coreEvents, CoreEvent } from '../utils/events.js';
|
||||||
import type { Config } from '../config/config.js';
|
import type { Config } from '../config/config.js';
|
||||||
import type { AgentDefinition } from './types.js';
|
import type { AgentDefinition } from './types.js';
|
||||||
import { loadAgentsFromDirectory } from './toml-loader.js';
|
import { loadAgentsFromDirectory } from './agentLoader.js';
|
||||||
import { CodebaseInvestigatorAgent } from './codebase-investigator.js';
|
import { CodebaseInvestigatorAgent } from './codebase-investigator.js';
|
||||||
import { CliHelpAgent } from './cli-help-agent.js';
|
import { CliHelpAgent } from './cli-help-agent.js';
|
||||||
import { A2AClientManager } from './a2a-client-manager.js';
|
import { A2AClientManager } from './a2a-client-manager.js';
|
||||||
|
|||||||
@@ -1,373 +0,0 @@
|
|||||||
/**
|
|
||||||
* @license
|
|
||||||
* Copyright 2025 Google LLC
|
|
||||||
* SPDX-License-Identifier: Apache-2.0
|
|
||||||
*/
|
|
||||||
|
|
||||||
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
||||||
import * as fs from 'node:fs/promises';
|
|
||||||
import * as path from 'node:path';
|
|
||||||
import * as os from 'node:os';
|
|
||||||
import {
|
|
||||||
parseAgentToml,
|
|
||||||
tomlToAgentDefinition,
|
|
||||||
loadAgentsFromDirectory,
|
|
||||||
AgentLoadError,
|
|
||||||
} from './toml-loader.js';
|
|
||||||
import { GEMINI_MODEL_ALIAS_PRO } from '../config/models.js';
|
|
||||||
import type { LocalAgentDefinition } from './types.js';
|
|
||||||
|
|
||||||
describe('toml-loader', () => {
|
|
||||||
let tempDir: string;
|
|
||||||
|
|
||||||
beforeEach(async () => {
|
|
||||||
tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'agent-test-'));
|
|
||||||
});
|
|
||||||
|
|
||||||
afterEach(async () => {
|
|
||||||
if (tempDir) {
|
|
||||||
await fs.rm(tempDir, { recursive: true, force: true });
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
async function writeAgentToml(content: string, fileName = 'test.toml') {
|
|
||||||
const filePath = path.join(tempDir, fileName);
|
|
||||||
await fs.writeFile(filePath, content);
|
|
||||||
return filePath;
|
|
||||||
}
|
|
||||||
|
|
||||||
describe('parseAgentToml', () => {
|
|
||||||
it('should parse a valid MVA TOML file', async () => {
|
|
||||||
const filePath = await writeAgentToml(`
|
|
||||||
name = "test-agent"
|
|
||||||
description = "A test agent"
|
|
||||||
[prompts]
|
|
||||||
system_prompt = "You are a test agent."
|
|
||||||
`);
|
|
||||||
|
|
||||||
const result = await parseAgentToml(filePath);
|
|
||||||
expect(result).toHaveLength(1);
|
|
||||||
expect(result[0]).toMatchObject({
|
|
||||||
name: 'test-agent',
|
|
||||||
description: 'A test agent',
|
|
||||||
prompts: {
|
|
||||||
system_prompt: 'You are a test agent.',
|
|
||||||
},
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should parse a valid remote agent TOML file', async () => {
|
|
||||||
const filePath = await writeAgentToml(`
|
|
||||||
kind = "remote"
|
|
||||||
name = "remote-agent"
|
|
||||||
description = "A remote agent"
|
|
||||||
agent_card_url = "https://example.com/card"
|
|
||||||
`);
|
|
||||||
|
|
||||||
const result = await parseAgentToml(filePath);
|
|
||||||
expect(result).toHaveLength(1);
|
|
||||||
expect(result[0]).toEqual({
|
|
||||||
kind: 'remote',
|
|
||||||
name: 'remote-agent',
|
|
||||||
description: 'A remote agent',
|
|
||||||
agent_card_url: 'https://example.com/card',
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should infer remote agent kind from agent_card_url', async () => {
|
|
||||||
const filePath = await writeAgentToml(`
|
|
||||||
name = "inferred-remote"
|
|
||||||
description = "Inferred"
|
|
||||||
agent_card_url = "https://example.com/inferred"
|
|
||||||
`);
|
|
||||||
|
|
||||||
const result = await parseAgentToml(filePath);
|
|
||||||
expect(result).toHaveLength(1);
|
|
||||||
expect(result[0]).toEqual({
|
|
||||||
kind: 'remote',
|
|
||||||
name: 'inferred-remote',
|
|
||||||
description: 'Inferred',
|
|
||||||
agent_card_url: 'https://example.com/inferred',
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should parse a remote agent without description', async () => {
|
|
||||||
const filePath = await writeAgentToml(`
|
|
||||||
kind = "remote"
|
|
||||||
name = "no-description-remote"
|
|
||||||
agent_card_url = "https://example.com/card"
|
|
||||||
`);
|
|
||||||
|
|
||||||
const result = await parseAgentToml(filePath);
|
|
||||||
expect(result).toHaveLength(1);
|
|
||||||
expect(result[0]).toEqual({
|
|
||||||
kind: 'remote',
|
|
||||||
name: 'no-description-remote',
|
|
||||||
agent_card_url: 'https://example.com/card',
|
|
||||||
});
|
|
||||||
expect(result[0].description).toBeUndefined();
|
|
||||||
|
|
||||||
// defined after conversion to AgentDefinition
|
|
||||||
const agentDef = tomlToAgentDefinition(result[0]);
|
|
||||||
expect(agentDef.description).toBe('(Loading description...)');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should parse multiple agents in one file', async () => {
|
|
||||||
const filePath = await writeAgentToml(`
|
|
||||||
[[remote_agents]]
|
|
||||||
kind = "remote"
|
|
||||||
name = "agent-1"
|
|
||||||
description = "Remote 1"
|
|
||||||
agent_card_url = "https://example.com/1"
|
|
||||||
|
|
||||||
[[remote_agents]]
|
|
||||||
kind = "remote"
|
|
||||||
name = "agent-2"
|
|
||||||
description = "Remote 2"
|
|
||||||
agent_card_url = "https://example.com/2"
|
|
||||||
`);
|
|
||||||
|
|
||||||
const result = await parseAgentToml(filePath);
|
|
||||||
expect(result).toHaveLength(2);
|
|
||||||
expect(result[0].name).toBe('agent-1');
|
|
||||||
expect(result[0].kind).toBe('remote');
|
|
||||||
expect(result[1].name).toBe('agent-2');
|
|
||||||
expect(result[1].kind).toBe('remote');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should allow omitting kind in remote_agents block', async () => {
|
|
||||||
const filePath = await writeAgentToml(`
|
|
||||||
[[remote_agents]]
|
|
||||||
name = "implicit-remote-1"
|
|
||||||
agent_card_url = "https://example.com/1"
|
|
||||||
|
|
||||||
[[remote_agents]]
|
|
||||||
name = "implicit-remote-2"
|
|
||||||
agent_card_url = "https://example.com/2"
|
|
||||||
`);
|
|
||||||
|
|
||||||
const result = await parseAgentToml(filePath);
|
|
||||||
expect(result).toHaveLength(2);
|
|
||||||
expect(result[0]).toMatchObject({
|
|
||||||
kind: 'remote',
|
|
||||||
name: 'implicit-remote-1',
|
|
||||||
agent_card_url: 'https://example.com/1',
|
|
||||||
});
|
|
||||||
expect(result[1]).toMatchObject({
|
|
||||||
kind: 'remote',
|
|
||||||
name: 'implicit-remote-2',
|
|
||||||
agent_card_url: 'https://example.com/2',
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should throw AgentLoadError if file reading fails', async () => {
|
|
||||||
const filePath = path.join(tempDir, 'non-existent.toml');
|
|
||||||
await expect(parseAgentToml(filePath)).rejects.toThrow(AgentLoadError);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should throw AgentLoadError if TOML parsing fails', async () => {
|
|
||||||
const filePath = await writeAgentToml('invalid toml [');
|
|
||||||
await expect(parseAgentToml(filePath)).rejects.toThrow(AgentLoadError);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should throw AgentLoadError if validation fails (missing required field)', async () => {
|
|
||||||
const filePath = await writeAgentToml(`
|
|
||||||
name = "test-agent"
|
|
||||||
# missing description
|
|
||||||
[prompts]
|
|
||||||
system_prompt = "You are a test agent."
|
|
||||||
`);
|
|
||||||
await expect(parseAgentToml(filePath)).rejects.toThrow(
|
|
||||||
/Validation failed/,
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should throw AgentLoadError if name is not a slug', async () => {
|
|
||||||
const filePath = await writeAgentToml(`
|
|
||||||
name = "Test Agent!"
|
|
||||||
description = "A test agent"
|
|
||||||
[prompts]
|
|
||||||
system_prompt = "You are a test agent."
|
|
||||||
`);
|
|
||||||
await expect(parseAgentToml(filePath)).rejects.toThrow(
|
|
||||||
/Name must be a valid slug/,
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should throw AgentLoadError if delegate_to_agent is included in tools', async () => {
|
|
||||||
const filePath = await writeAgentToml(`
|
|
||||||
name = "test-agent"
|
|
||||||
description = "A test agent"
|
|
||||||
tools = ["run_shell_command", "delegate_to_agent"]
|
|
||||||
[prompts]
|
|
||||||
system_prompt = "You are a test agent."
|
|
||||||
`);
|
|
||||||
|
|
||||||
await expect(parseAgentToml(filePath)).rejects.toThrow(
|
|
||||||
/tools list cannot include 'delegate_to_agent'/,
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should throw AgentLoadError if tools contains invalid names', async () => {
|
|
||||||
const filePath = await writeAgentToml(`
|
|
||||||
name = "test-agent"
|
|
||||||
description = "A test agent"
|
|
||||||
tools = ["not-a-tool"]
|
|
||||||
[prompts]
|
|
||||||
system_prompt = "You are a test agent."
|
|
||||||
`);
|
|
||||||
await expect(parseAgentToml(filePath)).rejects.toThrow(
|
|
||||||
/Validation failed:[\s\S]*tools.0: Invalid tool name/,
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should throw AgentLoadError if file contains both single and multiple agents', async () => {
|
|
||||||
const filePath = await writeAgentToml(`
|
|
||||||
name = "top-level-agent"
|
|
||||||
description = "I should not be here"
|
|
||||||
[prompts]
|
|
||||||
system_prompt = "..."
|
|
||||||
|
|
||||||
[[remote_agents]]
|
|
||||||
kind = "remote"
|
|
||||||
name = "array-agent"
|
|
||||||
description = "I am in an array"
|
|
||||||
agent_card_url = "https://example.com/card"
|
|
||||||
`);
|
|
||||||
|
|
||||||
await expect(parseAgentToml(filePath)).rejects.toThrow(
|
|
||||||
/Validation failed/,
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should show both options in error message when validation fails ambiguously', async () => {
|
|
||||||
const filePath = await writeAgentToml(`
|
|
||||||
name = "ambiguous-agent"
|
|
||||||
description = "I have neither prompts nor card"
|
|
||||||
`);
|
|
||||||
await expect(parseAgentToml(filePath)).rejects.toThrow(
|
|
||||||
/Validation failed: Agent Definition:\n\(Local Agent\) prompts: Required\n\(Remote Agent\) agent_card_url: Required/,
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('tomlToAgentDefinition', () => {
|
|
||||||
it('should convert valid TOML to AgentDefinition with defaults', () => {
|
|
||||||
const toml = {
|
|
||||||
kind: 'local' as const,
|
|
||||||
name: 'test-agent',
|
|
||||||
description: 'A test agent',
|
|
||||||
prompts: {
|
|
||||||
system_prompt: 'You are a test agent.',
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
const result = tomlToAgentDefinition(toml);
|
|
||||||
expect(result).toMatchObject({
|
|
||||||
name: 'test-agent',
|
|
||||||
description: 'A test agent',
|
|
||||||
promptConfig: {
|
|
||||||
systemPrompt: 'You are a test agent.',
|
|
||||||
},
|
|
||||||
modelConfig: {
|
|
||||||
model: 'inherit',
|
|
||||||
top_p: 0.95,
|
|
||||||
},
|
|
||||||
runConfig: {
|
|
||||||
max_time_minutes: 5,
|
|
||||||
},
|
|
||||||
inputConfig: {
|
|
||||||
inputs: {
|
|
||||||
query: {
|
|
||||||
type: 'string',
|
|
||||||
required: false,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should pass through model aliases', () => {
|
|
||||||
const toml = {
|
|
||||||
kind: 'local' as const,
|
|
||||||
name: 'test-agent',
|
|
||||||
description: 'A test agent',
|
|
||||||
model: {
|
|
||||||
model: GEMINI_MODEL_ALIAS_PRO,
|
|
||||||
},
|
|
||||||
prompts: {
|
|
||||||
system_prompt: 'You are a test agent.',
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
const result = tomlToAgentDefinition(toml) as LocalAgentDefinition;
|
|
||||||
expect(result.modelConfig.model).toBe(GEMINI_MODEL_ALIAS_PRO);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should pass through unknown model names (e.g. auto)', () => {
|
|
||||||
const toml = {
|
|
||||||
kind: 'local' as const,
|
|
||||||
name: 'test-agent',
|
|
||||||
description: 'A test agent',
|
|
||||||
model: {
|
|
||||||
model: 'auto',
|
|
||||||
},
|
|
||||||
prompts: {
|
|
||||||
system_prompt: 'You are a test agent.',
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
const result = tomlToAgentDefinition(toml) as LocalAgentDefinition;
|
|
||||||
expect(result.modelConfig.model).toBe('auto');
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('loadAgentsFromDirectory', () => {
|
|
||||||
it('should load definitions from a directory', async () => {
|
|
||||||
await writeAgentToml(
|
|
||||||
`
|
|
||||||
name = "agent-1"
|
|
||||||
description = "Agent 1"
|
|
||||||
[prompts]
|
|
||||||
system_prompt = "Prompt 1"
|
|
||||||
`,
|
|
||||||
'valid.toml',
|
|
||||||
);
|
|
||||||
|
|
||||||
// Create a non-TOML file
|
|
||||||
await fs.writeFile(path.join(tempDir, 'other.txt'), 'content');
|
|
||||||
|
|
||||||
// Create a hidden file
|
|
||||||
await writeAgentToml(
|
|
||||||
`
|
|
||||||
name = "hidden"
|
|
||||||
description = "Hidden"
|
|
||||||
[prompts]
|
|
||||||
system_prompt = "Hidden"
|
|
||||||
`,
|
|
||||||
'_hidden.toml',
|
|
||||||
);
|
|
||||||
|
|
||||||
const result = await loadAgentsFromDirectory(tempDir);
|
|
||||||
expect(result.agents).toHaveLength(1);
|
|
||||||
expect(result.agents[0].name).toBe('agent-1');
|
|
||||||
expect(result.errors).toHaveLength(0);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should return empty result if directory does not exist', async () => {
|
|
||||||
const nonExistentDir = path.join(tempDir, 'does-not-exist');
|
|
||||||
const result = await loadAgentsFromDirectory(nonExistentDir);
|
|
||||||
expect(result.agents).toHaveLength(0);
|
|
||||||
expect(result.errors).toHaveLength(0);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should capture errors for malformed individual files', async () => {
|
|
||||||
// Create a malformed TOML file
|
|
||||||
await writeAgentToml('invalid toml [', 'malformed.toml');
|
|
||||||
|
|
||||||
const result = await loadAgentsFromDirectory(tempDir);
|
|
||||||
expect(result.agents).toHaveLength(0);
|
|
||||||
expect(result.errors).toHaveLength(1);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -29,7 +29,7 @@ export interface SkillDefinition {
|
|||||||
isBuiltin?: boolean;
|
isBuiltin?: boolean;
|
||||||
}
|
}
|
||||||
|
|
||||||
const FRONTMATTER_REGEX = /^---\r?\n([\s\S]*?)\r?\n---\r?\n([\s\S]*)/;
|
export const FRONTMATTER_REGEX = /^---\r?\n([\s\S]*?)\r?\n---\r?\n([\s\S]*)/;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Discovers and loads all skills in the provided directory.
|
* Discovers and loads all skills in the provided directory.
|
||||||
|
|||||||
Reference in New Issue
Block a user