@cogitator-ai/core 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +76 -0
- package/dist/__tests__/agent.test.d.ts +2 -0
- package/dist/__tests__/agent.test.d.ts.map +1 -0
- package/dist/__tests__/agent.test.js +91 -0
- package/dist/__tests__/agent.test.js.map +1 -0
- package/dist/__tests__/base64.test.d.ts +2 -0
- package/dist/__tests__/base64.test.d.ts.map +1 -0
- package/dist/__tests__/base64.test.js +62 -0
- package/dist/__tests__/base64.test.js.map +1 -0
- package/dist/__tests__/calculator.test.d.ts +2 -0
- package/dist/__tests__/calculator.test.d.ts.map +1 -0
- package/dist/__tests__/calculator.test.js +146 -0
- package/dist/__tests__/calculator.test.js.map +1 -0
- package/dist/__tests__/cogitator-memory.test.d.ts +2 -0
- package/dist/__tests__/cogitator-memory.test.d.ts.map +1 -0
- package/dist/__tests__/cogitator-memory.test.js +176 -0
- package/dist/__tests__/cogitator-memory.test.js.map +1 -0
- package/dist/__tests__/datetime.test.d.ts +2 -0
- package/dist/__tests__/datetime.test.d.ts.map +1 -0
- package/dist/__tests__/datetime.test.js +87 -0
- package/dist/__tests__/datetime.test.js.map +1 -0
- package/dist/__tests__/exec.test.d.ts +2 -0
- package/dist/__tests__/exec.test.d.ts.map +1 -0
- package/dist/__tests__/exec.test.js +59 -0
- package/dist/__tests__/exec.test.js.map +1 -0
- package/dist/__tests__/filesystem.test.d.ts +2 -0
- package/dist/__tests__/filesystem.test.d.ts.map +1 -0
- package/dist/__tests__/filesystem.test.js +148 -0
- package/dist/__tests__/filesystem.test.js.map +1 -0
- package/dist/__tests__/google-backend.test.d.ts +5 -0
- package/dist/__tests__/google-backend.test.d.ts.map +1 -0
- package/dist/__tests__/google-backend.test.js +429 -0
- package/dist/__tests__/google-backend.test.js.map +1 -0
- package/dist/__tests__/hash.test.d.ts +2 -0
- package/dist/__tests__/hash.test.d.ts.map +1 -0
- package/dist/__tests__/hash.test.js +50 -0
- package/dist/__tests__/hash.test.js.map +1 -0
- package/dist/__tests__/http.test.d.ts +2 -0
- package/dist/__tests__/http.test.d.ts.map +1 -0
- package/dist/__tests__/http.test.js +64 -0
- package/dist/__tests__/http.test.js.map +1 -0
- package/dist/__tests__/json.test.d.ts +2 -0
- package/dist/__tests__/json.test.d.ts.map +1 -0
- package/dist/__tests__/json.test.js +65 -0
- package/dist/__tests__/json.test.js.map +1 -0
- package/dist/__tests__/logger.test.d.ts +2 -0
- package/dist/__tests__/logger.test.d.ts.map +1 -0
- package/dist/__tests__/logger.test.js +186 -0
- package/dist/__tests__/logger.test.js.map +1 -0
- package/dist/__tests__/random.test.d.ts +2 -0
- package/dist/__tests__/random.test.d.ts.map +1 -0
- package/dist/__tests__/random.test.js +81 -0
- package/dist/__tests__/random.test.js.map +1 -0
- package/dist/__tests__/regex.test.d.ts +2 -0
- package/dist/__tests__/regex.test.d.ts.map +1 -0
- package/dist/__tests__/regex.test.js +75 -0
- package/dist/__tests__/regex.test.js.map +1 -0
- package/dist/__tests__/registry.test.d.ts +2 -0
- package/dist/__tests__/registry.test.d.ts.map +1 -0
- package/dist/__tests__/registry.test.js +102 -0
- package/dist/__tests__/registry.test.js.map +1 -0
- package/dist/__tests__/sleep.test.d.ts +2 -0
- package/dist/__tests__/sleep.test.d.ts.map +1 -0
- package/dist/__tests__/sleep.test.js +29 -0
- package/dist/__tests__/sleep.test.js.map +1 -0
- package/dist/__tests__/tool.test.d.ts +2 -0
- package/dist/__tests__/tool.test.d.ts.map +1 -0
- package/dist/__tests__/tool.test.js +103 -0
- package/dist/__tests__/tool.test.js.map +1 -0
- package/dist/__tests__/uuid.test.d.ts +2 -0
- package/dist/__tests__/uuid.test.d.ts.map +1 -0
- package/dist/__tests__/uuid.test.js +37 -0
- package/dist/__tests__/uuid.test.js.map +1 -0
- package/dist/agent.d.ts +15 -0
- package/dist/agent.d.ts.map +1 -0
- package/dist/agent.js +35 -0
- package/dist/agent.js.map +1 -0
- package/dist/cogitator.d.ts +66 -0
- package/dist/cogitator.d.ts.map +1 -0
- package/dist/cogitator.js +538 -0
- package/dist/cogitator.js.map +1 -0
- package/dist/index.d.ts +19 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +15 -0
- package/dist/index.js.map +1 -0
- package/dist/llm/anthropic.d.ts +19 -0
- package/dist/llm/anthropic.d.ts.map +1 -0
- package/dist/llm/anthropic.js +188 -0
- package/dist/llm/anthropic.js.map +1 -0
- package/dist/llm/base.d.ts +11 -0
- package/dist/llm/base.d.ts.map +1 -0
- package/dist/llm/base.js +9 -0
- package/dist/llm/base.js.map +1 -0
- package/dist/llm/google.d.ts +32 -0
- package/dist/llm/google.d.ts.map +1 -0
- package/dist/llm/google.js +282 -0
- package/dist/llm/google.js.map +1 -0
- package/dist/llm/index.d.ts +22 -0
- package/dist/llm/index.d.ts.map +1 -0
- package/dist/llm/index.js +67 -0
- package/dist/llm/index.js.map +1 -0
- package/dist/llm/ollama.d.ts +20 -0
- package/dist/llm/ollama.d.ts.map +1 -0
- package/dist/llm/ollama.js +134 -0
- package/dist/llm/ollama.js.map +1 -0
- package/dist/llm/openai.d.ts +21 -0
- package/dist/llm/openai.d.ts.map +1 -0
- package/dist/llm/openai.js +154 -0
- package/dist/llm/openai.js.map +1 -0
- package/dist/logger.d.ts +48 -0
- package/dist/logger.d.ts.map +1 -0
- package/dist/logger.js +118 -0
- package/dist/logger.js.map +1 -0
- package/dist/registry.d.ts +15 -0
- package/dist/registry.d.ts.map +1 -0
- package/dist/registry.js +31 -0
- package/dist/registry.js.map +1 -0
- package/dist/tool.d.ts +13 -0
- package/dist/tool.d.ts.map +1 -0
- package/dist/tool.js +44 -0
- package/dist/tool.js.map +1 -0
- package/dist/tools/base64.d.ts +23 -0
- package/dist/tools/base64.d.ts.map +1 -0
- package/dist/tools/base64.js +53 -0
- package/dist/tools/base64.js.map +1 -0
- package/dist/tools/calculator.d.ts +15 -0
- package/dist/tools/calculator.d.ts.map +1 -0
- package/dist/tools/calculator.js +193 -0
- package/dist/tools/calculator.js.map +1 -0
- package/dist/tools/datetime.d.ts +23 -0
- package/dist/tools/datetime.d.ts.map +1 -0
- package/dist/tools/datetime.js +95 -0
- package/dist/tools/datetime.js.map +1 -0
- package/dist/tools/exec.d.ts +46 -0
- package/dist/tools/exec.d.ts.map +1 -0
- package/dist/tools/exec.js +87 -0
- package/dist/tools/exec.js.map +1 -0
- package/dist/tools/filesystem.d.ts +100 -0
- package/dist/tools/filesystem.d.ts.map +1 -0
- package/dist/tools/filesystem.js +186 -0
- package/dist/tools/filesystem.js.map +1 -0
- package/dist/tools/hash.d.ts +19 -0
- package/dist/tools/hash.d.ts.map +1 -0
- package/dist/tools/hash.js +32 -0
- package/dist/tools/hash.js.map +1 -0
- package/dist/tools/http.d.ts +29 -0
- package/dist/tools/http.d.ts.map +1 -0
- package/dist/tools/http.js +82 -0
- package/dist/tools/http.js.map +1 -0
- package/dist/tools/index.d.ts +323 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +50 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/json.d.ts +28 -0
- package/dist/tools/json.d.ts.map +1 -0
- package/dist/tools/json.js +42 -0
- package/dist/tools/json.js.map +1 -0
- package/dist/tools/random.d.ts +29 -0
- package/dist/tools/random.d.ts.map +1 -0
- package/dist/tools/random.js +56 -0
- package/dist/tools/random.js.map +1 -0
- package/dist/tools/regex.d.ts +43 -0
- package/dist/tools/regex.d.ts.map +1 -0
- package/dist/tools/regex.js +68 -0
- package/dist/tools/regex.js.map +1 -0
- package/dist/tools/sleep.d.ts +10 -0
- package/dist/tools/sleep.d.ts.map +1 -0
- package/dist/tools/sleep.js +25 -0
- package/dist/tools/sleep.js.map +1 -0
- package/dist/tools/uuid.d.ts +15 -0
- package/dist/tools/uuid.d.ts.map +1 -0
- package/dist/tools/uuid.js +25 -0
- package/dist/tools/uuid.js.map +1 -0
- package/dist/utils/circuit-breaker.d.ts +127 -0
- package/dist/utils/circuit-breaker.d.ts.map +1 -0
- package/dist/utils/circuit-breaker.js +235 -0
- package/dist/utils/circuit-breaker.js.map +1 -0
- package/dist/utils/fallback.d.ts +66 -0
- package/dist/utils/fallback.d.ts.map +1 -0
- package/dist/utils/fallback.js +99 -0
- package/dist/utils/fallback.js.map +1 -0
- package/dist/utils/index.d.ts +4 -0
- package/dist/utils/index.d.ts.map +1 -0
- package/dist/utils/index.js +4 -0
- package/dist/utils/index.js.map +1 -0
- package/dist/utils/retry.d.ts +51 -0
- package/dist/utils/retry.d.ts.map +1 -0
- package/dist/utils/retry.js +115 -0
- package/dist/utils/retry.js.map +1 -0
- package/package.json +56 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sleep.test.js","sourceRoot":"","sources":["../../src/__tests__/sleep.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAEvC,MAAM,WAAW,GAAG;IAClB,OAAO,EAAE,YAAY;IACrB,KAAK,EAAE,UAAU;IACjB,MAAM,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;CACrC,CAAC;AAEF,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,WAAW,CAAC,CAAC;QAC7D,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;QAEnC,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE;QAC9B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/tool.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,103 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { z } from 'zod';
|
|
3
|
+
import { tool, toolToSchema } from '../tool';
|
|
4
|
+
describe('tool()', () => {
|
|
5
|
+
it('creates a tool with required properties', () => {
|
|
6
|
+
const calculator = tool({
|
|
7
|
+
name: 'calculate',
|
|
8
|
+
description: 'Perform a calculation',
|
|
9
|
+
parameters: z.object({
|
|
10
|
+
expression: z.string(),
|
|
11
|
+
}),
|
|
12
|
+
execute: ({ expression }) => Promise.resolve(eval(expression)),
|
|
13
|
+
});
|
|
14
|
+
expect(calculator.name).toBe('calculate');
|
|
15
|
+
expect(calculator.description).toBe('Perform a calculation');
|
|
16
|
+
expect(calculator.parameters).toBeDefined();
|
|
17
|
+
expect(calculator.execute).toBeInstanceOf(Function);
|
|
18
|
+
});
|
|
19
|
+
it('creates a tool with optional properties', () => {
|
|
20
|
+
const dangerousTool = tool({
|
|
21
|
+
name: 'dangerous',
|
|
22
|
+
description: 'A dangerous tool',
|
|
23
|
+
parameters: z.object({ action: z.string() }),
|
|
24
|
+
execute: () => Promise.resolve('done'),
|
|
25
|
+
sideEffects: ['network', 'filesystem'],
|
|
26
|
+
requiresApproval: true,
|
|
27
|
+
timeout: 5000,
|
|
28
|
+
});
|
|
29
|
+
expect(dangerousTool.sideEffects).toEqual(['network', 'filesystem']);
|
|
30
|
+
expect(dangerousTool.requiresApproval).toBe(true);
|
|
31
|
+
expect(dangerousTool.timeout).toBe(5000);
|
|
32
|
+
});
|
|
33
|
+
it('executes the tool function correctly', async () => {
|
|
34
|
+
const greet = tool({
|
|
35
|
+
name: 'greet',
|
|
36
|
+
description: 'Greet someone',
|
|
37
|
+
parameters: z.object({
|
|
38
|
+
name: z.string(),
|
|
39
|
+
}),
|
|
40
|
+
execute: ({ name }) => Promise.resolve(`Hello, ${name}!`),
|
|
41
|
+
});
|
|
42
|
+
const mockContext = {
|
|
43
|
+
agentId: 'agent_test',
|
|
44
|
+
runId: 'run_test',
|
|
45
|
+
signal: new AbortController().signal,
|
|
46
|
+
};
|
|
47
|
+
const result = await greet.execute({ name: 'World' }, mockContext);
|
|
48
|
+
expect(result).toBe('Hello, World!');
|
|
49
|
+
});
|
|
50
|
+
it('provides toJSON method that returns schema', () => {
|
|
51
|
+
const myTool = tool({
|
|
52
|
+
name: 'test',
|
|
53
|
+
description: 'Test tool',
|
|
54
|
+
parameters: z.object({
|
|
55
|
+
input: z.string(),
|
|
56
|
+
}),
|
|
57
|
+
execute: () => Promise.resolve('result'),
|
|
58
|
+
});
|
|
59
|
+
const schema = myTool.toJSON();
|
|
60
|
+
expect(schema.name).toBe('test');
|
|
61
|
+
expect(schema.description).toBe('Test tool');
|
|
62
|
+
expect(schema.parameters.type).toBe('object');
|
|
63
|
+
});
|
|
64
|
+
});
|
|
65
|
+
describe('toolToSchema()', () => {
|
|
66
|
+
it('converts a tool to JSON Schema format', () => {
|
|
67
|
+
const myTool = tool({
|
|
68
|
+
name: 'search',
|
|
69
|
+
description: 'Search for something',
|
|
70
|
+
parameters: z.object({
|
|
71
|
+
query: z.string().describe('The search query'),
|
|
72
|
+
limit: z.number().optional().describe('Max results'),
|
|
73
|
+
}),
|
|
74
|
+
execute: () => Promise.resolve([]),
|
|
75
|
+
});
|
|
76
|
+
const schema = toolToSchema(myTool);
|
|
77
|
+
expect(schema.name).toBe('search');
|
|
78
|
+
expect(schema.description).toBe('Search for something');
|
|
79
|
+
expect(schema.parameters.type).toBe('object');
|
|
80
|
+
expect(schema.parameters.properties).toHaveProperty('query');
|
|
81
|
+
expect(schema.parameters.properties).toHaveProperty('limit');
|
|
82
|
+
expect(schema.parameters.required).toContain('query');
|
|
83
|
+
expect(schema.parameters.required).not.toContain('limit');
|
|
84
|
+
});
|
|
85
|
+
it('handles complex nested schemas', () => {
|
|
86
|
+
const complexTool = tool({
|
|
87
|
+
name: 'complex',
|
|
88
|
+
description: 'Complex tool',
|
|
89
|
+
parameters: z.object({
|
|
90
|
+
user: z.object({
|
|
91
|
+
name: z.string(),
|
|
92
|
+
age: z.number(),
|
|
93
|
+
}),
|
|
94
|
+
tags: z.array(z.string()),
|
|
95
|
+
}),
|
|
96
|
+
execute: () => Promise.resolve(null),
|
|
97
|
+
});
|
|
98
|
+
const schema = toolToSchema(complexTool);
|
|
99
|
+
expect(schema.parameters.properties).toHaveProperty('user');
|
|
100
|
+
expect(schema.parameters.properties).toHaveProperty('tags');
|
|
101
|
+
});
|
|
102
|
+
});
|
|
103
|
+
//# sourceMappingURL=tool.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool.test.js","sourceRoot":"","sources":["../../src/__tests__/tool.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAE7C,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;IACtB,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,UAAU,GAAG,IAAI,CAAC;YACtB,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,uBAAuB;YACpC,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;gBACnB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;aACvB,CAAC;YACF,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAW,CAAC;SACzE,CAAC,CAAC;QAEH,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1C,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC7D,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;QAC5C,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,aAAa,GAAG,IAAI,CAAC;YACzB,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,kBAAkB;YAC/B,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAC5C,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YACtC,WAAW,EAAE,CAAC,SAAS,EAAE,YAAY,CAAC;YACtC,gBAAgB,EAAE,IAAI;YACtB,OAAO,EAAE,IAAI;SACd,CAAC,CAAC;QAEH,MAAM,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC;QACrE,MAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,KAAK,GAAG,IAAI,CAAC;YACjB,IAAI,EAAE,OAAO;YACb,WAAW,EAAE,eAAe;YAC5B,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;gBACnB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;aACjB,CAAC;YACF,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,IAAI,GAAG,CAAC;SAC1D,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG;YAClB,OAAO,EAAE,YAAY;YACrB,KAAK,EAAE,UAAU;YACjB,MAAM,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACrC,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,WAAW,CAAC,CAAC;QACnE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,MAAM,GAAG,IAAI,CAAC;YAClB,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,WAAW;YACxB,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;gBACnB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;aAClB,CAAC;YACF,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;SACzC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,MAAM,GAAG,IAAI,CAAC;YAClB,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,sBAAsB;YACnC,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;gBACnB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;gBAC9C,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC;aACrD,CAAC;YACF,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;SACnC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QAEpC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,WAAW,GAAG,IAAI,CAAC;YACvB,IAAI,EAAE,SAAS;YACf,WAAW,EAAE,cAAc;YAC3B,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;gBACnB,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;oBACb,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;oBAChB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;iBAChB,CAAC;gBACF,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;aAC1B,CAAC;YACF,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC;SACrC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,YAAY,CAAC,WAAW,CAAC,CAAC;QAEzC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"uuid.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/uuid.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { uuid } from '../tools/uuid';
|
|
3
|
+
const mockContext = {
|
|
4
|
+
agentId: 'agent_test',
|
|
5
|
+
runId: 'run_test',
|
|
6
|
+
signal: new AbortController().signal,
|
|
7
|
+
};
|
|
8
|
+
describe('uuid tool', () => {
|
|
9
|
+
it('generates a single UUID by default', async () => {
|
|
10
|
+
const result = await uuid.execute({}, mockContext);
|
|
11
|
+
expect(result).toHaveProperty('uuid');
|
|
12
|
+
expect(result.uuid).toMatch(/^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i);
|
|
13
|
+
});
|
|
14
|
+
it('generates multiple UUIDs when count is specified', async () => {
|
|
15
|
+
const result = await uuid.execute({ count: 5 }, mockContext);
|
|
16
|
+
expect(result).toHaveProperty('uuids');
|
|
17
|
+
expect(result).toHaveProperty('count', 5);
|
|
18
|
+
const uuids = result.uuids;
|
|
19
|
+
expect(uuids).toHaveLength(5);
|
|
20
|
+
uuids.forEach((id) => {
|
|
21
|
+
expect(id).toMatch(/^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i);
|
|
22
|
+
});
|
|
23
|
+
});
|
|
24
|
+
it('generates unique UUIDs', async () => {
|
|
25
|
+
const result = await uuid.execute({ count: 100 }, mockContext);
|
|
26
|
+
const uuids = result.uuids;
|
|
27
|
+
const uniqueSet = new Set(uuids);
|
|
28
|
+
expect(uniqueSet.size).toBe(100);
|
|
29
|
+
});
|
|
30
|
+
it('has correct metadata', () => {
|
|
31
|
+
expect(uuid.name).toBe('uuid');
|
|
32
|
+
expect(uuid.description).toContain('UUID');
|
|
33
|
+
const schema = uuid.toJSON();
|
|
34
|
+
expect(schema.parameters.properties).toHaveProperty('count');
|
|
35
|
+
});
|
|
36
|
+
});
|
|
37
|
+
//# sourceMappingURL=uuid.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"uuid.test.js","sourceRoot":"","sources":["../../src/__tests__/uuid.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAErC,MAAM,WAAW,GAAG;IAClB,OAAO,EAAE,YAAY;IACrB,KAAK,EAAE,UAAU;IACjB,MAAM,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;CACrC,CAAC;AAEF,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QACtC,MAAM,CAAE,MAA2B,CAAC,IAAI,CAAC,CAAC,OAAO,CAC/C,wEAAwE,CACzE,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC1C,MAAM,KAAK,GAAI,MAA8B,CAAC,KAAK,CAAC;QACpD,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE;YACnB,MAAM,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,wEAAwE,CAAC,CAAC;QAC/F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,WAAW,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAI,MAA8B,CAAC,KAAK,CAAC;QACpD,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC/B,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
package/dist/agent.d.ts
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Agent class implementation
|
|
3
|
+
*/
|
|
4
|
+
import type { Agent as IAgent, AgentConfig, Tool } from '@cogitator-ai/types';
|
|
5
|
+
export declare class Agent implements IAgent {
|
|
6
|
+
readonly id: string;
|
|
7
|
+
readonly name: string;
|
|
8
|
+
readonly config: AgentConfig;
|
|
9
|
+
constructor(config: AgentConfig);
|
|
10
|
+
get model(): string;
|
|
11
|
+
get instructions(): string;
|
|
12
|
+
get tools(): Tool[];
|
|
13
|
+
clone(overrides: Partial<AgentConfig>): Agent;
|
|
14
|
+
}
|
|
15
|
+
//# sourceMappingURL=agent.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,KAAK,IAAI,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AAG9E,qBAAa,KAAM,YAAW,MAAM;IAClC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAC;gBAEjB,MAAM,EAAE,WAAW;IAW/B,IAAI,KAAK,IAAI,MAAM,CAElB;IAED,IAAI,YAAY,IAAI,MAAM,CAEzB;IAED,IAAI,KAAK,IAAI,IAAI,EAAE,CAElB;IAED,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,WAAW,CAAC,GAAG,KAAK;CAM9C"}
|
package/dist/agent.js
ADDED
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Agent class implementation
|
|
3
|
+
*/
|
|
4
|
+
import { nanoid } from 'nanoid';
|
|
5
|
+
export class Agent {
|
|
6
|
+
id;
|
|
7
|
+
name;
|
|
8
|
+
config;
|
|
9
|
+
constructor(config) {
|
|
10
|
+
this.id = config.id ?? `agent_${nanoid(12)}`;
|
|
11
|
+
this.name = config.name;
|
|
12
|
+
this.config = {
|
|
13
|
+
temperature: 0.7,
|
|
14
|
+
maxIterations: 10,
|
|
15
|
+
timeout: 120_000,
|
|
16
|
+
...config,
|
|
17
|
+
};
|
|
18
|
+
}
|
|
19
|
+
get model() {
|
|
20
|
+
return this.config.model;
|
|
21
|
+
}
|
|
22
|
+
get instructions() {
|
|
23
|
+
return this.config.instructions;
|
|
24
|
+
}
|
|
25
|
+
get tools() {
|
|
26
|
+
return this.config.tools ?? [];
|
|
27
|
+
}
|
|
28
|
+
clone(overrides) {
|
|
29
|
+
return new Agent({
|
|
30
|
+
...this.config,
|
|
31
|
+
...overrides,
|
|
32
|
+
});
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
//# sourceMappingURL=agent.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,MAAM,OAAO,KAAK;IACP,EAAE,CAAS;IACX,IAAI,CAAS;IACb,MAAM,CAAc;IAE7B,YAAY,MAAmB;QAC7B,IAAI,CAAC,EAAE,GAAG,MAAM,CAAC,EAAE,IAAI,SAAS,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;QAC7C,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACxB,IAAI,CAAC,MAAM,GAAG;YACZ,WAAW,EAAE,GAAG;YAChB,aAAa,EAAE,EAAE;YACjB,OAAO,EAAE,OAAO;YAChB,GAAG,MAAM;SACV,CAAC;IACJ,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;IAC3B,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;IAClC,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,SAA+B;QACnC,OAAO,IAAI,KAAK,CAAC;YACf,GAAG,IAAI,CAAC,MAAM;YACd,GAAG,SAAS;SACb,CAAC,CAAC;IACL,CAAC;CACF"}
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Cogitator - Main runtime class
|
|
3
|
+
*/
|
|
4
|
+
import type { CogitatorConfig, RunOptions, RunResult } from '@cogitator-ai/types';
|
|
5
|
+
import { type Agent } from './agent';
|
|
6
|
+
import { ToolRegistry } from './registry';
|
|
7
|
+
export declare class Cogitator {
|
|
8
|
+
private config;
|
|
9
|
+
private backends;
|
|
10
|
+
readonly tools: ToolRegistry;
|
|
11
|
+
private memoryAdapter?;
|
|
12
|
+
private contextBuilder?;
|
|
13
|
+
private memoryInitialized;
|
|
14
|
+
private sandboxManager?;
|
|
15
|
+
private sandboxInitialized;
|
|
16
|
+
constructor(config?: CogitatorConfig);
|
|
17
|
+
/**
|
|
18
|
+
* Initialize memory adapter and context builder (lazy, on first run)
|
|
19
|
+
*/
|
|
20
|
+
private initializeMemory;
|
|
21
|
+
/**
|
|
22
|
+
* Initialize sandbox manager (lazy, on first sandboxed tool execution)
|
|
23
|
+
*/
|
|
24
|
+
private initializeSandbox;
|
|
25
|
+
/**
|
|
26
|
+
* Build initial messages for a run, loading history if memory is enabled
|
|
27
|
+
*/
|
|
28
|
+
private buildInitialMessages;
|
|
29
|
+
/**
|
|
30
|
+
* Save a message entry to memory (non-blocking, won't crash on failure)
|
|
31
|
+
*/
|
|
32
|
+
private saveEntry;
|
|
33
|
+
/**
|
|
34
|
+
* Create a span with proper IDs and emit callback
|
|
35
|
+
*/
|
|
36
|
+
private createSpan;
|
|
37
|
+
/**
|
|
38
|
+
* Run an agent with the given input
|
|
39
|
+
*/
|
|
40
|
+
run(agent: Agent, options: RunOptions): Promise<RunResult>;
|
|
41
|
+
/**
|
|
42
|
+
* Stream chat with token callback
|
|
43
|
+
*/
|
|
44
|
+
private streamChat;
|
|
45
|
+
/**
|
|
46
|
+
* Execute a tool
|
|
47
|
+
*/
|
|
48
|
+
private executeTool;
|
|
49
|
+
/**
|
|
50
|
+
* Execute a tool in sandbox (Docker or WASM)
|
|
51
|
+
*/
|
|
52
|
+
private executeInSandbox;
|
|
53
|
+
/**
|
|
54
|
+
* Get or create an LLM backend
|
|
55
|
+
*/
|
|
56
|
+
private getBackend;
|
|
57
|
+
/**
|
|
58
|
+
* Calculate cost based on model and tokens using dynamic model registry
|
|
59
|
+
*/
|
|
60
|
+
private calculateCost;
|
|
61
|
+
/**
|
|
62
|
+
* Close all connections
|
|
63
|
+
*/
|
|
64
|
+
close(): Promise<void>;
|
|
65
|
+
}
|
|
66
|
+
//# sourceMappingURL=cogitator.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cogitator.d.ts","sourceRoot":"","sources":["../src/cogitator.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,KAAK,EACV,eAAe,EACf,UAAU,EACV,SAAS,EAUV,MAAM,qBAAqB,CAAC;AAW7B,OAAO,EAAE,KAAK,KAAK,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAc1C,qBAAa,SAAS;IACpB,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,QAAQ,CAAsC;IACtD,SAAgB,KAAK,EAAE,YAAY,CAAsB;IAEzD,OAAO,CAAC,aAAa,CAAC,CAAgB;IACtC,OAAO,CAAC,cAAc,CAAC,CAAiB;IACxC,OAAO,CAAC,iBAAiB,CAAS;IAElC,OAAO,CAAC,cAAc,CAAC,CAAiB;IACxC,OAAO,CAAC,kBAAkB,CAAS;gBAEvB,MAAM,GAAE,eAAoB;IAIxC;;OAEG;YACW,gBAAgB;IA6D9B;;OAEG;YACW,iBAAiB;IAa/B;;OAEG;YACW,oBAAoB;IA0ClC;;OAEG;YACW,SAAS;IAqBvB;;OAEG;IACH,OAAO,CAAC,UAAU;IA2BlB;;OAEG;IACG,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC;IAkOhE;;OAEG;YACW,UAAU;IA6DxB;;OAEG;YACW,WAAW;IA4CzB;;OAEG;YACW,gBAAgB;IAuF9B;;OAEG;IACH,OAAO,CAAC,UAAU;IAalB;;OAEG;IACH,OAAO,CAAC,aAAa;IAWrB;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAc7B"}
|