Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
FilipZmijewski committed Dec 2, 2024
1 parent 00a2572 commit bd5cd23
Showing 1 changed file with 11 additions and 3 deletions.
14 changes: 11 additions & 3 deletions libs/langchain-community/src/utils/ibm.ts
Original file line number Diff line number Diff line change
Expand Up @@ -14,13 +14,11 @@ import { z } from "zod";
import { ChatGeneration } from "@langchain/core/outputs";
import { AIMessageChunk } from "@langchain/core/messages";
import { ToolCall } from "@langchain/core/messages/tool";
import { WatsonxAuth, WatsonxInit } from "../types/ibm.js";
import {
CallbackManager,
CallbackManagerForLLMRun,
} from "@langchain/core/callbacks/manager";
import {
CallbackHandler,
InvokeRequestCallback,
RecieveResponseCallback,
RequestCallbacks,
Expand All @@ -29,6 +27,7 @@ import {
BaseCallbackHandler,
CallbackHandlerMethods,
} from "@langchain/core/callbacks/base";
import { WatsonxAuth, WatsonxInit } from "../types/ibm.js";

export const authenticateAndSetInstance = ({
watsonxAIApikey,
Expand Down Expand Up @@ -215,6 +214,7 @@ export class WatsonxCallbackManager extends CallbackManager {
) {
super(parentRunId, { ...callbackManager });
}

static fromHandlers(
handlers: CallbackHandlerMethods & RequestCallbacks
): WatsonxCallbackManager {
Expand All @@ -230,35 +230,43 @@ export class WatsonxCallbackManager extends CallbackManager {
> = {};
watsonxCallbackNames.forEach((item) => {
watsonxHandlers[item] = handlers[item];
delete handlers[item];
Reflect.deleteProperty(watsonxHandlers, item);
});
const manager = super.fromHandlers(handlers);
class WatsonxHandler extends BaseCallbackHandler {
name = "watsonxHandler";

requestCallback: InvokeRequestCallback;

responseCallback: RecieveResponseCallback;

constructor() {
super();
Object.assign(this, watsonxHandlers);
}

get watsonxHandlers() {
return {
requestCallback: this.requestCallback,
responseCallback: this.responseCallback,
};
}
}

manager.addHandler(new WatsonxHandler());
return manager;
}
}

export class WatsonxBaseCallbackHandler extends BaseCallbackHandler {
name: string;

constructor() {
super();
}

requestCallback: InvokeRequestCallback;

responseCallback: RecieveResponseCallback;

get watsonxHandlers() {
Expand Down

0 comments on commit bd5cd23

Please sign in to comment.