@llmbridge/plugin-logging 0.0.4 → 0.1.0

package/dist/index.d.mts

@@ -7,19 +7,9 @@ declare function generateRandomDigits(length: number): string;
 declare class LoggingPlugin implements LLMBridge.Plugin<void, string, void, void> {
     private options;
     constructor(options: LoggingPluginOptions);
-    beforeRun(context: LLMBridge.PluginCompletionContext): Promise<void>;
-    afterRun({ response }: {
-        response: any;
-    }): Promise<void>;
-    beforeExec(params: any): Promise<string>;
-    afterExec({ beforeResponse, response }: {
-        beforeResponse: string;
-        response: any;
-    }): Promise<void>;
-    beforeToolExec(tool: LLMBridge.Tool, counter: number, input: any, context: LLMBridge.PluginCompletionContext): Promise<void>;
-    afterToolExec({ response }: {
-        response: any;
-    }): Promise<void>;
+    wrapRun(next: () => Promise<LLMBridge.Response>, context: LLMBridge.PluginCompletionContext): Promise<LLMBridge.Response>;
+    wrapExec(next: (params: any) => Promise<any>, params: any, context: LLMBridge.PluginCompletionContext): Promise<any>;
+    wrapToolExec(next: (tool: LLMBridge.Tool, counter: number, input: any) => Promise<any>, tool: LLMBridge.Tool, counter: number, input: any, context: LLMBridge.PluginCompletionContext): Promise<any>;
     createLogFiles(requestId: string, type: 'request' | 'response', data: any): void;
     createTxtFile(filePath: string, data: any): void;
     objectToTxt(obj: any, prefix?: string): string;

package/dist/index.d.ts

@@ -7,19 +7,9 @@ declare function generateRandomDigits(length: number): string;
 declare class LoggingPlugin implements LLMBridge.Plugin<void, string, void, void> {
     private options;
     constructor(options: LoggingPluginOptions);
-    beforeRun(context: LLMBridge.PluginCompletionContext): Promise<void>;
-    afterRun({ response }: {
-        response: any;
-    }): Promise<void>;
-    beforeExec(params: any): Promise<string>;
-    afterExec({ beforeResponse, response }: {
-        beforeResponse: string;
-        response: any;
-    }): Promise<void>;
-    beforeToolExec(tool: LLMBridge.Tool, counter: number, input: any, context: LLMBridge.PluginCompletionContext): Promise<void>;
-    afterToolExec({ response }: {
-        response: any;
-    }): Promise<void>;
+    wrapRun(next: () => Promise<LLMBridge.Response>, context: LLMBridge.PluginCompletionContext): Promise<LLMBridge.Response>;
+    wrapExec(next: (params: any) => Promise<any>, params: any, context: LLMBridge.PluginCompletionContext): Promise<any>;
+    wrapToolExec(next: (tool: LLMBridge.Tool, counter: number, input: any) => Promise<any>, tool: LLMBridge.Tool, counter: number, input: any, context: LLMBridge.PluginCompletionContext): Promise<any>;
     createLogFiles(requestId: string, type: 'request' | 'response', data: any): void;
     createTxtFile(filePath: string, data: any): void;
     objectToTxt(obj: any, prefix?: string): string;

package/dist/index.js

@@ -47,26 +47,25 @@ var LoggingPlugin = class {
   constructor(options) {
     this.options = options;
   }
-  async beforeRun(context) {
+  async wrapRun(next, context) {
     console.log("run", context.model);
-  }
-  async afterRun({ response }) {
+    const response = await next();
     const lastResponsePath = import_path.default.join(this.options.folder, "last_response.json");
     import_fs.default.writeFileSync(lastResponsePath, JSON.stringify(response, null, 2));
+    return response;
   }
-  async beforeExec(params) {
+  async wrapExec(next, params, context) {
     const requestId = generateRandomDigits(4);
     this.createLogFiles(requestId, "request", params);
-    return requestId;
-  }
-  async afterExec({ beforeResponse, response }) {
-    this.createLogFiles(beforeResponse, "response", response);
+    const response = await next(params);
+    this.createLogFiles(requestId, "response", response);
+    return response;
   }
-  async beforeToolExec(tool, counter, input, context) {
+  async wrapToolExec(next, tool, counter, input, context) {
     console.log(`run (${counter + 1}/${context.options.tools?.usesLimit}) tool ${tool.name}`, input);
-  }
-  async afterToolExec({ response }) {
-    console.log(`tool response`, response);
+    const result = await next(tool, counter, input);
+    console.log(`tool response`, result);
+    return result;
   }
   createLogFiles(requestId, type, data) {
     const currentDate = (0, import_moment.default)();

package/dist/index.mjs

@@ -10,26 +10,25 @@ var LoggingPlugin = class {
   constructor(options) {
     this.options = options;
   }
-  async beforeRun(context) {
+  async wrapRun(next, context) {
     console.log("run", context.model);
-  }
-  async afterRun({ response }) {
+    const response = await next();
     const lastResponsePath = path.join(this.options.folder, "last_response.json");
     fs.writeFileSync(lastResponsePath, JSON.stringify(response, null, 2));
+    return response;
   }
-  async beforeExec(params) {
+  async wrapExec(next, params, context) {
     const requestId = generateRandomDigits(4);
     this.createLogFiles(requestId, "request", params);
-    return requestId;
-  }
-  async afterExec({ beforeResponse, response }) {
-    this.createLogFiles(beforeResponse, "response", response);
+    const response = await next(params);
+    this.createLogFiles(requestId, "response", response);
+    return response;
   }
-  async beforeToolExec(tool, counter, input, context) {
+  async wrapToolExec(next, tool, counter, input, context) {
     console.log(`run (${counter + 1}/${context.options.tools?.usesLimit}) tool ${tool.name}`, input);
-  }
-  async afterToolExec({ response }) {
-    console.log(`tool response`, response);
+    const result = await next(tool, counter, input);
+    console.log(`tool response`, result);
+    return result;
   }
   createLogFiles(requestId, type, data) {
     const currentDate = moment();

package/package.json

@@ -1,14 +1,15 @@
 {
   "name": "@llmbridge/plugin-logging",
-  "version": "0.0.4",
+  "version": "0.1.0",
   "description": "Logging plugin for LLMBridge",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
   "scripts": {
-    "build": "pnpm clean && tsup src/index.ts --format cjs,esm --dts",
+    "build": "npm run clean && tsup src/index.ts --format cjs,esm --dts",
     "test": "vitest run",
-    "clean": "rm -rf dist"
+    "clean": "rimraf dist",
+    "build:publish": "npm run build && npm publish"
   },
   "keywords": [
     "llm",
@@ -26,7 +27,7 @@
     "moment": "^2.30.1"
   },
   "peerDependencies": {
-    "@llmbridge/core": ">=0.0.3 <0.1.0",
+    "@llmbridge/core": ">=0.1.0 <0.2.0",
     "moment": "^2.29.0",
     "typescript": "^5.0.0"
   },
@@ -34,7 +35,8 @@
     "@types/node": "^18.11.9",
     "typescript": "^5.7.3",
     "tsup": "^8.4.0",
-    "vitest": "^3.0.7"
+    "vitest": "^3.0.7",
+    "rimraf": "^5.0.5"
   },
   "author": "",
   "license": "MIT"