@llmbridge/plugin-logging 0.0.4 → 0.1.1

package/dist/index.d.mts

@@ -1,28 +1,28 @@
 import { LLMBridge } from '@llmbridge/core';
 
+interface Logger {
+    log?(message?: any, ...optionalParams: any[]): void;
+    info?(message?: any, ...optionalParams: any[]): void;
+    warn?(message?: any, ...optionalParams: any[]): void;
+    error?(message?: any, ...optionalParams: any[]): void;
+    debug?(message?: any, ...optionalParams: any[]): void;
+}
 interface LoggingPluginOptions {
     folder: string;
+    logger?: Logger;
 }
 declare function generateRandomDigits(length: number): string;
 declare class LoggingPlugin implements LLMBridge.Plugin<void, string, void, void> {
     private options;
+    private logger;
     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>;
+    private log;
+    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;
 }
 
-export { LoggingPlugin, type LoggingPluginOptions, generateRandomDigits };
+export { type Logger, LoggingPlugin, type LoggingPluginOptions, generateRandomDigits };

package/dist/index.d.ts

@@ -1,28 +1,28 @@
 import { LLMBridge } from '@llmbridge/core';
 
+interface Logger {
+    log?(message?: any, ...optionalParams: any[]): void;
+    info?(message?: any, ...optionalParams: any[]): void;
+    warn?(message?: any, ...optionalParams: any[]): void;
+    error?(message?: any, ...optionalParams: any[]): void;
+    debug?(message?: any, ...optionalParams: any[]): void;
+}
 interface LoggingPluginOptions {
     folder: string;
+    logger?: Logger;
 }
 declare function generateRandomDigits(length: number): string;
 declare class LoggingPlugin implements LLMBridge.Plugin<void, string, void, void> {
     private options;
+    private logger;
     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>;
+    private log;
+    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;
 }
 
-export { LoggingPlugin, type LoggingPluginOptions, generateRandomDigits };
+export { type Logger, LoggingPlugin, type LoggingPluginOptions, generateRandomDigits };

package/dist/index.js

@@ -46,27 +46,36 @@ function generateRandomDigits(length) {
 var LoggingPlugin = class {
   constructor(options) {
     this.options = options;
+    this.logger = options.logger;
   }
-  async beforeRun(context) {
-    console.log("run", context.model);
+  log(message) {
+    if (!this.logger) {
+      return;
+    }
+    const logFn = this.logger.info || this.logger.log;
+    if (logFn) {
+      logFn.call(this.logger, message);
+    }
   }
-  async afterRun({ response }) {
+  async wrapRun(next, context) {
+    this.log(`run ${context.model}`);
+    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) {
-    console.log(`run (${counter + 1}/${context.options.tools?.usesLimit}) tool ${tool.name}`, input);
-  }
-  async afterToolExec({ response }) {
-    console.log(`tool response`, response);
+  async wrapToolExec(next, tool, counter, input, context) {
+    this.log(`run (${counter + 1}/${context.options.tools?.usesLimit}) tool ${tool.name} ${JSON.stringify(input)}`);
+    const result = await next(tool, counter, input);
+    this.log(`tool response ${JSON.stringify(result)}`);
+    return result;
   }
   createLogFiles(requestId, type, data) {
     const currentDate = (0, import_moment.default)();

package/dist/index.mjs

@@ -9,27 +9,36 @@ function generateRandomDigits(length) {
 var LoggingPlugin = class {
   constructor(options) {
     this.options = options;
+    this.logger = options.logger;
   }
-  async beforeRun(context) {
-    console.log("run", context.model);
+  log(message) {
+    if (!this.logger) {
+      return;
+    }
+    const logFn = this.logger.info || this.logger.log;
+    if (logFn) {
+      logFn.call(this.logger, message);
+    }
   }
-  async afterRun({ response }) {
+  async wrapRun(next, context) {
+    this.log(`run ${context.model}`);
+    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) {
-    console.log(`run (${counter + 1}/${context.options.tools?.usesLimit}) tool ${tool.name}`, input);
-  }
-  async afterToolExec({ response }) {
-    console.log(`tool response`, response);
+  async wrapToolExec(next, tool, counter, input, context) {
+    this.log(`run (${counter + 1}/${context.options.tools?.usesLimit}) tool ${tool.name} ${JSON.stringify(input)}`);
+    const result = await next(tool, counter, input);
+    this.log(`tool response ${JSON.stringify(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.1",
   "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"