Adds factory 'createAgent' function

Protofy-xyz · Nov 22, 2024 · 87c2ff8 · 87c2ff8
1 parent 09ddd73
commit 87c2ff8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/packages/protofy/src/Agent.ts b/packages/protofy/src/Agent.ts
@@ -137,6 +137,10 @@ export class Agent {
     }
 }
 
+export function createAgent(data: AgentData, parent?: Agent) {
+    return new Agent(data, parent);
+}
+
 export class AgentInterface {
     input: AgentIOInterface;
     output: AgentIOInterface;

diff --git a/packages/protofy/tests/agent.test.ts b/packages/protofy/tests/agent.test.ts
@@ -1,4 +1,4 @@
-import { Agent } from '../src/Agent';
+import { Agent, createAgent } from '../src/Agent';
 import { z } from 'zod';
 import { zodToJsonSchema } from 'zod-to-json-schema';
 
@@ -25,7 +25,7 @@ describe('Agents basic behavior', () => {
       tags: ['child']
     });
 
-    agent = new Agent({
+    agent = createAgent({
       id: 'getDisplayInfo',
       name: 'getDisplayInfo',
       description: 'Get display info of a user',