feat(core): Improve Langsmith traces for AI executions (#9081)

Signed-off-by: Oleg Ivaniv <me@olegivaniv.com>
This commit is contained in:
oleg
2024-04-08 22:51:49 +02:00
committed by GitHub
parent 3bcfef95f6
commit 936682eeaa
18 changed files with 99 additions and 26 deletions

View File

@@ -65,6 +65,7 @@
"recast": "0.21.5",
"title-case": "3.0.3",
"transliteration": "2.3.5",
"xml2js": "0.6.2"
"xml2js": "0.6.2",
"@langchain/core": "0.1.41"
}
}

View File

@@ -19,6 +19,7 @@ import type { WorkflowHooks } from './WorkflowHooks';
import type { NodeOperationError } from './errors/node-operation.error';
import type { NodeApiError } from './errors/node-api.error';
import type { AxiosProxyConfig } from 'axios';
import type { CallbackManager as CallbackManagerLC } from '@langchain/core/callbacks/manager';
export interface IAdditionalCredentialOptions {
oauth2?: IOAuth2Options;
@@ -842,6 +843,7 @@ export type IExecuteFunctions = ExecuteFunctions.GetNodeParameterFn &
executeWorkflow(
workflowInfo: IExecuteWorkflowInfo,
inputData?: INodeExecutionData[],
parentCallbackManager?: CallbackManager,
): Promise<any>;
getInputConnectionData(
inputName: ConnectionTypes,
@@ -881,6 +883,8 @@ export type IExecuteFunctions = ExecuteFunctions.GetNodeParameterFn &
getBinaryDataBuffer(itemIndex: number, propertyName: string): Promise<Buffer>;
copyInputItems(items: INodeExecutionData[], properties: string[]): IDataObject[];
};
getParentCallbackManager(): CallbackManager | undefined;
};
export interface IExecuteSingleFunctions extends BaseExecutionFunctions {
@@ -2028,6 +2032,7 @@ export interface IWorkflowExecuteAdditionalData {
loadedWorkflowData?: IWorkflowBase;
loadedRunData?: any;
parentWorkflowSettings?: IWorkflowSettings;
parentCallbackManager?: CallbackManager;
},
) => Promise<any>;
executionId?: string;
@@ -2059,6 +2064,7 @@ export interface IWorkflowExecuteAdditionalData {
nodeType?: string;
},
) => Promise<void>;
parentCallbackManager?: CallbackManager;
}
export type WorkflowExecuteMode =
@@ -2583,3 +2589,5 @@ export type BannerName =
export type Functionality = 'regular' | 'configuration-node' | 'pairedItem';
export type Result<T, E> = { ok: true; result: T } | { ok: false; error: E };
export type CallbackManager = CallbackManagerLC;