63 lines
2.9 KiB
TypeScript
63 lines
2.9 KiB
TypeScript
|
import { type ClientOptions, OpenAIClient } from "@langchain/openai";
|
||
|
|
import { StructuredTool } from "@langchain/core/tools";
|
||
|
|
import { Runnable, RunnableConfig } from "@langchain/core/runnables";
|
||
|
|
import type { OpenAIAssistantFinish, OpenAIAssistantAction, OpenAIToolType } from "./schema.js";
|
||
|
|
type ThreadMessage = any;
|
||
|
|
type RequiredActionFunctionToolCall = any;
|
||
|
|
type ExtractRunOutput<AsAgent extends boolean | undefined> = AsAgent extends true ? OpenAIAssistantFinish | OpenAIAssistantAction[] : ThreadMessage[] | RequiredActionFunctionToolCall[];
|
||
|
|
export type OpenAIAssistantRunnableInput<AsAgent extends boolean | undefined = undefined> = {
|
||
|
|
client?: OpenAIClient;
|
||
|
|
clientOptions?: ClientOptions;
|
||
|
|
assistantId: string;
|
||
|
|
pollIntervalMs?: number;
|
||
|
|
asAgent?: AsAgent;
|
||
|
|
};
|
||
|
|
export declare class OpenAIAssistantRunnable<AsAgent extends boolean | undefined, RunInput extends Record<string, any> = Record<string, any>> extends Runnable<RunInput, ExtractRunOutput<AsAgent>> {
|
||
|
|
lc_namespace: string[];
|
||
|
|
private client;
|
||
|
|
assistantId: string;
|
||
|
|
pollIntervalMs: number;
|
||
|
|
asAgent?: AsAgent;
|
||
|
|
constructor(fields: OpenAIAssistantRunnableInput<AsAgent>);
|
||
|
|
static createAssistant<AsAgent extends boolean>({ model, name, instructions, tools, client, clientOptions, asAgent, pollIntervalMs, fileIds, }: Omit<OpenAIAssistantRunnableInput<AsAgent>, "assistantId"> & {
|
||
|
|
model: string;
|
||
|
|
name?: string;
|
||
|
|
instructions?: string;
|
||
|
|
tools?: OpenAIToolType | Array<StructuredTool>;
|
||
|
|
fileIds?: string[];
|
||
|
|
}): Promise<OpenAIAssistantRunnable<AsAgent, Record<string, any>>>;
|
||
|
|
invoke(input: RunInput, _options?: RunnableConfig): Promise<ExtractRunOutput<AsAgent>>;
|
||
|
|
/**
|
||
|
|
* Delete an assistant.
|
||
|
|
*
|
||
|
|
* @link {https://platform.openai.com/docs/api-reference/assistants/deleteAssistant}
|
||
|
|
* @returns {Promise<AssistantDeleted>}
|
||
|
|
*/
|
||
|
|
deleteAssistant(): Promise<OpenAIClient.Beta.Assistants.AssistantDeleted>;
|
||
|
|
/**
|
||
|
|
* Retrieves an assistant.
|
||
|
|
*
|
||
|
|
* @link {https://platform.openai.com/docs/api-reference/assistants/getAssistant}
|
||
|
|
* @returns {Promise<OpenAIClient.Beta.Assistants.Assistant>}
|
||
|
|
*/
|
||
|
|
getAssistant(): Promise<OpenAIClient.Beta.Assistants.Assistant>;
|
||
|
|
/**
|
||
|
|
* Modifies an assistant.
|
||
|
|
*
|
||
|
|
* @link {https://platform.openai.com/docs/api-reference/assistants/modifyAssistant}
|
||
|
|
* @returns {Promise<OpenAIClient.Beta.Assistants.Assistant>}
|
||
|
|
*/
|
||
|
|
modifyAssistant<AsAgent extends boolean>({ model, name, instructions, fileIds, }: Omit<OpenAIAssistantRunnableInput<AsAgent>, "assistantId" | "tools"> & {
|
||
|
|
model?: string;
|
||
|
|
name?: string;
|
||
|
|
instructions?: string;
|
||
|
|
fileIds?: string[];
|
||
|
|
}): Promise<OpenAIClient.Beta.Assistants.Assistant>;
|
||
|
|
private _parseStepsInput;
|
||
|
|
private _createRun;
|
||
|
|
private _createThreadAndRun;
|
||
|
|
private _waitForRun;
|
||
|
|
private _getResponse;
|
||
|
|
}
|
||
|
|
export {};
|