Add setup callback to @hook (#64)

Author: mattyhogan

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pompeii-labs/magma",
-  "version": "1.8.1",
+  "version": "1.8.2",
   "description": "The Typescript framework to build AI agents quickly and easily",
   "keywords": [
     "Agents",

src/decorators.ts

@@ -135,6 +135,7 @@ export function middleware<T extends MagmaMiddlewareTriggerType>(
  * @param hookName name of the hook
  * @param options configuration options for the hook
  * @param options.session session configuration for the hook
+ * @param options.setup arguments to pass to the agent's setup function before the hook handler is called
  * Examples:
  * @hook('notification') -> POST /hooks/notification
  * @hook('notification', { session: 'default' })
@@ -143,7 +144,7 @@ export function middleware<T extends MagmaMiddlewareTriggerType>(
  */
 export function hook(
     hookName: string,
-    options: { session?: MagmaHook['session']; description?: string } = {}
+    options: { session?: MagmaHook['session']; description?: string; setup?: MagmaHook['setup'] } = {}
 ) {
     return function <R extends void>(
         target: object,
@@ -153,6 +154,7 @@ export function hook(
                 _hookName?: string;
                 _session?: MagmaHook['session'];
                 _description?: string;
+                _setup?: MagmaHook['setup'];
             }
         >
     ) {
@@ -163,6 +165,7 @@ export function hook(
         descriptor.value._hookName = hookName;
         descriptor.value._session = options.session;
         descriptor.value._description = options.description;
+        descriptor.value._setup = options.setup;
     };
 }
 

src/helpers/utilities.ts

@@ -94,6 +94,7 @@ export function loadHooks(target: any): MagmaHook[] {
                 name: method['_hookName'],
                 handler: method.bind(target),
                 session: (method as any)['_session'],
+                setup: (method as any)['_setup'],
                 description: (method as any)['_description'],
             } as MagmaHook);
         }

src/types/utilities/hooks.ts

@@ -5,5 +5,6 @@ export type MagmaHook = {
     name: string;
     handler: (request: Request, response: Response, agent: MagmaAgent) => Promise<void>;
     session?: 'default' | (string & {}) | ((req: Request) => string | Promise<string>);
+    setup?: ((req: Request) => Record<string, any> | Promise<Record<string, any>>) | Record<string, any>;
     description?: string;
 };