Class Pipe

packages/core/src/pipes/pipes.ts:88–419

}

export class Pipe {
	private request: Request;
	private pipe: any;
	private tools: Record<string, (...args: any[]) => Promise<any>>;
	private maxCalls: number;
	private hasTools: boolean;
	private prod: boolean;
	private baseUrl: string;
	private entityApiKey?: string;

	constructor(options: PipeOptions) {
		this.prod = options.prod ?? isProd();
		this.baseUrl = getApiUrl(this.prod);

		this.request = new Request({
			apiKey: options.apiKey,
			baseUrl: this.baseUrl,
		});
		this.pipe = options;
		this.entityApiKey = options.apiKey;

		delete this.pipe.prod;
		delete this.pipe.apiKey;

		this.tools = this.getToolsFromPipe(this.pipe);
		this.maxCalls = options.maxCalls || 100; // TODO: Find a sane default.
		this.hasTools = Object.keys(this.tools).length > 0;
	}

	private getToolsFromPipe(
		pipe: Pipe,
	): Record<string, (...args: any[]) => Promise<any>> {
		const tools: Record<string, (...args: any[]) => Promise<any>> = {};
		if (pipe.tools && Array.isArray(pipe.tools)) {
			pipe.tools.forEach((tool: Tool) => {
				tools[tool.function.name] = tool.run;
			});
		}
		return tools;
	}

	private async runTools(toolCalls: ToolCallResult[]): Promise<Message[]> {
		const toolPromises = toolCalls.map(async (toolCall: ToolCallResult) => {
			const toolName = toolCall.function.name;
			const toolParameters = JSON.parse(toolCall.function.arguments);
			const toolFunction = this.tools[toolName];

			if (!toolFunction) {
				throw new Error(
					`Tool ${toolName} not found. If this is intentional, please set runTools to false to disable tool execution by default.`,
				);
			}

			const toolResponse = await toolFunction(toolParameters);

			return {
				tool_call_id: toolCall.id,
				role: 'tool' as MessageRole,