Adds function runner to agent network

packages/protofy/src/protocols/function.ts

@@ -1,7 +1,7 @@
 import { Agent } from "../Agent";
 
 //function runner
-export default (agent: Agent) => {
+export default (agent: Agent, params) => {
     //check if the interface is a function
     if(agent.getProtocol().type !== 'function') {
         throw new Error('Error: Invalid protocol type, expected function')
@@ -16,5 +16,8 @@ export default (agent: Agent) => {
         throw new Error('Error: Invalid function, expected function')
     }
 
-
+    //call the function
+    return agent.getProtocol().config.fn(
+        ...(Array.isArray(params) ? params : [params])
+    );
 }

packages/protofy/tests/function.test.ts

@@ -0,0 +1,58 @@
+import { Agent } from '../src/Agent';
+import { z } from 'zod';
+import { zodToJsonSchema } from 'zod-to-json-schema';
+import functionRunner from '../src/protocols/function';
+
+let paramsSchema;
+let returnSchema: z.ZodString;
+let agent: Agent;
+let childAgent: Agent;
+
+describe('Function Agents', () => {
+  beforeEach(() => {
+    paramsSchema = z.tuple([z.object({
+      id: z.string().uuid(),
+      name: z.string().min(1),
+      age: z.number().min(18),
+      email: z.string().email(),
+    })]);
+
+    returnSchema = z.string()
+
+    agent = new Agent({
+      id: 'getDisplayInfo',
+      name: 'getDisplayInfo',
+      description: 'Get display info of a user',
+      tags: ['user', 'display'],
+      interface: {
+        protocol: {
+          type: 'function',
+          config: {
+            fn: (user) => {
+              return user.name + ', ' + user.age
+            }
+          }
+        },
+        input: {
+          shape: zodToJsonSchema(paramsSchema, "params"),
+          protocol: {
+            config: {test: 'test'} 
+          }
+        },
+        output: {
+          shape: zodToJsonSchema(returnSchema, "displayInfo")
+        }
+      }
+    })
+  });
+
+  it('Should be able to call the function and get the result back', () => {
+    expect(functionRunner(agent, {
+        id: 1,
+        name: 'John Doe',
+        age: 30,
+        email: 'a@a.com'
+    })).toBe('John Doe, 30');
+  });
+
+});