librechat-data-provider 0.7.3 → 0.7.5

package/src/types/agents.ts

@@ -0,0 +1,220 @@
+/* eslint-disable @typescript-eslint/no-namespace */
+import { StepTypes, ContentTypes, ToolCallTypes } from './runs';
+import type { FunctionToolCall } from './assistants';
+
+export namespace Agents {
+  export type MessageType = 'human' | 'ai' | 'generic' | 'system' | 'function' | 'tool' | 'remove';
+
+  export type ImageDetail = 'auto' | 'low' | 'high';
+
+  export type MessageContentText = {
+    type: ContentTypes.TEXT;
+    text: string;
+    tool_call_ids?: string[];
+  };
+
+  export type MessageContentImageUrl = {
+    type: ContentTypes.IMAGE_URL;
+    image_url: string | { url: string; detail?: ImageDetail };
+  };
+
+  export type MessageContentComplex =
+    | MessageContentText
+    | MessageContentImageUrl
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    | (Record<string, any> & { type?: ContentTypes | string })
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    | (Record<string, any> & { type?: never });
+
+  export type MessageContent = string | MessageContentComplex[];
+
+  /**
+   * A call to a tool.
+   */
+  export type ToolCall = {
+    /** Type ("tool_call") according to Assistants Tool Call Structure */
+    type: ToolCallTypes.TOOL_CALL;
+    /** The name of the tool to be called */
+    name: string;
+
+    /** The arguments to the tool call */
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    args?: string | Record<string, any>;
+
+    /** If provided, an identifier associated with the tool call */
+    id?: string;
+    /** If provided, the output of the tool call */
+    output?: string;
+  };
+
+  export type ToolEndEvent = {
+    /** The Step Id of the Tool Call */
+    id: string;
+    /** The Completed Tool Call */
+    tool_call?: ToolCall;
+    /** The content index of the tool call */
+    index: number;
+  };
+
+  export type ToolCallContent = {
+    type: ContentTypes.TOOL_CALL;
+    tool_call?: ToolCall;
+  };
+
+  /**
+   * A chunk of a tool call (e.g., as part of a stream).
+   * When merging ToolCallChunks (e.g., via AIMessageChunk.__add__),
+   * all string attributes are concatenated. Chunks are only merged if their
+   * values of `index` are equal and not None.
+   *
+   * @example
+   * ```ts
+   * const leftChunks = [
+   *   {
+   *     name: "foo",
+   *     args: '{"a":',
+   *     index: 0
+   *   }
+   * ];
+   *
+   * const leftAIMessageChunk = new AIMessageChunk({
+   *   content: "",
+   *   tool_call_chunks: leftChunks
+   * });
+   *
+   * const rightChunks = [
+   *   {
+   *     name: undefined,
+   *     args: '1}',
+   *     index: 0
+   *   }
+   * ];
+   *
+   * const rightAIMessageChunk = new AIMessageChunk({
+   *   content: "",
+   *   tool_call_chunks: rightChunks
+   * });
+   *
+   * const result = leftAIMessageChunk.concat(rightAIMessageChunk);
+   * // result.tool_call_chunks is equal to:
+   * // [
+   * //   {
+   * //     name: "foo",
+   * //     args: '{"a":1}'
+   * //     index: 0
+   * //   }
+   * // ]
+   * ```
+   *
+   * @property {string} [name] - If provided, a substring of the name of the tool to be called
+   * @property {string} [args] - If provided, a JSON substring of the arguments to the tool call
+   * @property {string} [id] - If provided, a substring of an identifier for the tool call
+   * @property {number} [index] - If provided, the index of the tool call in a sequence
+   */
+  export type ToolCallChunk = {
+    name?: string;
+
+    args?: string;
+
+    id?: string;
+
+    index?: number;
+
+    type?: 'tool_call_chunk';
+  };
+
+  /** Event names are of the format: on_[runnable_type]_(start|stream|end).
+
+  Runnable types are one of:
+
+  llm - used by non chat models
+  chat_model - used by chat models
+  prompt -- e.g., ChatPromptTemplate
+  tool -- LangChain tools
+  chain - most Runnables are of this type
+  Further, the events are categorized as one of:
+
+  start - when the runnable starts
+  stream - when the runnable is streaming
+  end - when the runnable ends
+  start, stream and end are associated with slightly different data payload.
+
+  Please see the documentation for EventData for more details. */
+  export type EventName = string;
+  export type RunStep = {
+    type: StepTypes;
+    id: string; // #new
+    runId?: string; // #new
+    index: number; // #new
+    stepIndex?: number; // #new
+    stepDetails: StepDetails;
+    usage: null | {
+      // Define usage structure if it's ever non-null
+      // prompt_tokens: number; // #new
+      // completion_tokens: number; // #new
+      // total_tokens: number; // #new
+    };
+  };
+  /**
+   * Represents a run step delta i.e. any changed fields on a run step during
+   * streaming.
+   */
+  export interface RunStepDeltaEvent {
+    /**
+     * The identifier of the run step, which can be referenced in API endpoints.
+     */
+    id: string;
+    /**
+     * The delta containing the fields that have changed on the run step.
+     */
+    delta: ToolCallDelta;
+  }
+  export type StepDetails = MessageCreationDetails | ToolCallsDetails;
+  export type MessageCreationDetails = {
+    type: StepTypes.MESSAGE_CREATION;
+    message_creation: {
+      message_id: string;
+    };
+  };
+  export type ToolCallsDetails = {
+    type: StepTypes.TOOL_CALLS;
+    tool_calls: AgentToolCall[];
+  };
+  export type ToolCallDelta = {
+    type: StepTypes.TOOL_CALLS | string;
+    tool_calls?: ToolCallChunk[];
+  };
+  export type AgentToolCall = FunctionToolCall | ToolCall;
+  export interface ExtendedMessageContent {
+    type?: string;
+    text?: string;
+    input?: string;
+    index?: number;
+    id?: string;
+    name?: string;
+  }
+  /**
+   * Represents a message delta i.e. any changed fields on a message during
+   * streaming.
+   */
+  export interface MessageDeltaEvent {
+    /**
+     * The identifier of the message, which can be referenced in API endpoints.
+     */
+    id: string;
+    /**
+     * The delta containing the fields that have changed on the Message.
+     */
+    delta: MessageDelta;
+  }
+  /**
+   * The delta containing the fields that have changed on the Message.
+   */
+  export interface MessageDelta {
+    /**
+     * The content of the message in array of text and/or images.
+     */
+    content?: MessageContentComplex[];
+  }
+  export type ContentType = ContentTypes.TEXT | ContentTypes.IMAGE_URL | string;
+}

package/src/types/assistants.ts

@@ -1,5 +1,7 @@
 import type { OpenAPIV3 } from 'openapi-types';
-import type { AssistantsEndpoint } from 'src/schemas';
+import type { AssistantsEndpoint, AgentProvider } from 'src/schemas';
+import type { ContentTypes } from './runs';
+import type { Agents } from './agents';
 import type { TFile } from './files';
 
 export type Schema = OpenAPIV3.SchemaObject & { description?: string };
@@ -13,6 +15,7 @@ export type Metadata = {
 };
 
 export enum Tools {
+  execute_code = 'execute_code',
   code_interpreter = 'code_interpreter',
   file_search = 'file_search',
   retrieval = 'retrieval',
@@ -21,6 +24,7 @@ export enum Tools {
 
 export enum EToolResources {
   code_interpreter = 'code_interpreter',
+  execute_code = 'execute_code',
   file_search = 'file_search',
 }
 
@@ -66,17 +70,20 @@ export interface FileSearchResource {
   vector_store_ids?: Array<string>;
 }
 
+/* Assistant types */
+
 export type Assistant = {
   id: string;
   created_at: number;
   description: string | null;
-  file_ids: string[];
+  file_ids?: string[];
   instructions: string | null;
+  conversation_starters?: string[];
   metadata: Metadata | null;
   model: string;
   name: string | null;
   object: string;
-  tools: FunctionTool[];
+  tools?: FunctionTool[];
   tool_resources?: ToolResources;
 };
 
@@ -87,6 +94,7 @@ export type AssistantCreateParams = {
   description?: string | null;
   file_ids?: string[];
   instructions?: string | null;
+  conversation_starters?: string[];
   metadata?: Metadata | null;
   name?: string | null;
   tools?: Array<FunctionTool | string>;
@@ -99,6 +107,7 @@ export type AssistantUpdateParams = {
   description?: string | null;
   file_ids?: string[];
   instructions?: string | null;
+  conversation_starters?: string[] | null;
   metadata?: Metadata | null;
   name?: string | null;
   tools?: Array<FunctionTool | string>;
@@ -133,6 +142,122 @@ export type File = {
   purpose: 'fine-tune' | 'fine-tune-results' | 'assistants' | 'assistants_output';
 };
 
+/* Agent types */
+
+export type AgentParameterValue = number | null;
+
+export type AgentModelParameters = {
+  temperature: AgentParameterValue;
+  max_context_tokens: AgentParameterValue;
+  max_output_tokens: AgentParameterValue;
+  top_p: AgentParameterValue;
+  frequency_penalty: AgentParameterValue;
+  presence_penalty: AgentParameterValue;
+};
+
+export interface AgentToolResources {
+  execute_code?: ExecuteCodeResource;
+  file_search?: AgentFileSearchResource;
+}
+export interface ExecuteCodeResource {
+  /**
+   * A list of file IDs made available to the `execute_code` tool.
+   * There can be a maximum of 20 files associated with the tool.
+   */
+  file_ids?: Array<string>;
+}
+
+export interface AgentFileSearchResource {
+  /**
+   * The ID of the vector store attached to this agent. There
+   * can be a maximum of 1 vector store attached to the agent.
+   */
+  vector_store_ids?: Array<string>;
+  /**
+   * A list of file IDs made available to the `file_search` tool.
+   * To be used before vector stores are implemented.
+   */
+  file_ids?: Array<string>;
+}
+
+export type Agent = {
+  id: string;
+  name: string | null;
+  author?: string | null;
+  authorName?: string | null;
+  description: string | null;
+  created_at: number;
+  avatar: AgentAvatar | null;
+  instructions: string | null;
+  tools?: string[];
+  projectIds?: string[];
+  tool_kwargs?: Record<string, unknown>;
+  metadata?: Record<string, unknown>;
+  provider: AgentProvider;
+  model: string | null;
+  model_parameters: AgentModelParameters;
+  conversation_starters?: string[];
+  isCollaborative?: boolean;
+  tool_resources?: AgentToolResources;
+};
+
+export type TAgentsMap = Record<string, Agent | undefined>;
+
+export type AgentCreateParams = {
+  name?: string | null;
+  description?: string | null;
+  avatar?: AgentAvatar | null;
+  file_ids?: string[];
+  instructions?: string | null;
+  tools?: Array<FunctionTool | string>;
+  provider: AgentProvider;
+  model: string | null;
+  model_parameters: AgentModelParameters;
+};
+
+export type AgentUpdateParams = {
+  name?: string | null;
+  description?: string | null;
+  avatar?: AgentAvatar | null;
+  file_ids?: string[];
+  instructions?: string | null;
+  tools?: Array<FunctionTool | string>;
+  tool_resources?: ToolResources;
+  provider?: AgentProvider;
+  model?: string | null;
+  model_parameters?: AgentModelParameters;
+  projectIds?: string[];
+  removeProjectIds?: string[];
+  isCollaborative?: boolean;
+};
+
+export type AgentListParams = {
+  limit?: number;
+  before?: string | null;
+  after?: string | null;
+  order?: 'asc' | 'desc';
+  provider?: AgentProvider;
+};
+
+export type AgentListResponse = {
+  object: string;
+  data: Agent[];
+  first_id: string;
+  last_id: string;
+  has_more: boolean;
+};
+
+export type AgentFile = {
+  file_id: string;
+  id?: string;
+  temp_file_id?: string;
+  bytes: number;
+  created_at: number;
+  filename: string;
+  object: string;
+  purpose: 'fine-tune' | 'fine-tune-results' | 'agents' | 'agents_output';
+};
+
 /**
  * Details of the Code Interpreter tool call the run step was involved in.
  * Includes the tool call ID, the code interpreter definition, and the type of tool call.
@@ -245,25 +370,6 @@ export enum AnnotationTypes {
   FILE_PATH = 'file_path',
 }
 
-export enum ContentTypes {
-  TEXT = 'text',
-  TOOL_CALL = 'tool_call',
-  IMAGE_FILE = 'image_file',
-  ERROR = 'error',
-}
-
-export enum StepTypes {
-  TOOL_CALLS = 'tool_calls',
-  MESSAGE_CREATION = 'message_creation',
-}
-
-export enum ToolCallTypes {
-  FUNCTION = 'function',
-  RETRIEVAL = 'retrieval',
-  FILE_SEARCH = 'file_search',
-  CODE_INTERPRETER = 'code_interpreter',
-}
-
 export enum StepStatus {
   IN_PROGRESS = 'in_progress',
   CANCELLED = 'cancelled',
@@ -302,6 +408,7 @@ export type ContentPart = (
   | RetrievalToolCall
   | FileSearchToolCall
   | FunctionToolCall
+  | Agents.AgentToolCall
   | ImageFile
   | Text
 ) &
@@ -309,13 +416,20 @@ export type ContentPart = (
 
 export type TMessageContentParts =
   | { type: ContentTypes.ERROR; text: Text & PartMetadata }
-  | { type: ContentTypes.TEXT; text: Text & PartMetadata }
+  | { type: ContentTypes.TEXT; text: string | (Text & PartMetadata); tool_call_ids?: string[] }
   | {
       type: ContentTypes.TOOL_CALL;
-      tool_call: (CodeToolCall | RetrievalToolCall | FileSearchToolCall | FunctionToolCall) &
+      tool_call: (
+        | CodeToolCall
+        | RetrievalToolCall
+        | FileSearchToolCall
+        | FunctionToolCall
+        | Agents.AgentToolCall
+      ) &
         PartMetadata;
     }
-  | { type: ContentTypes.IMAGE_FILE; image_file: ImageFile & PartMetadata };
+  | { type: ContentTypes.IMAGE_FILE; image_file: ImageFile & PartMetadata }
+  | Agents.MessageContentImageUrl;
 
 export type StreamContentData = TMessageContentParts & {
   /** The index of the current content part */
@@ -375,14 +489,15 @@ export type ActionMetadata = {
   oauth_client_secret?: string;
 };
 
+/* Assistant types */
+
 export type Action = {
   action_id: string;
-  assistant_id: string;
   type?: string;
   settings?: Record<string, unknown>;
   metadata: ActionMetadata;
   version: number | string;
-};
+} & ({ assistant_id: string; agent_id?: never } | { assistant_id?: never; agent_id: string });
 
 export type AssistantAvatar = {
   filepath: string;
@@ -392,6 +507,7 @@ export type AssistantAvatar = {
 export type AssistantDocument = {
   user: string;
   assistant_id: string;
+  conversation_starters?: string[];
   avatar?: AssistantAvatar;
   access_level?: number;
   file_ids?: string[];
@@ -400,6 +516,13 @@ export type AssistantDocument = {
   updatedAt?: Date;
 };
 
+/* Agent types */
+
+export type AgentAvatar = {
+  filepath: string;
+  source: string;
+};
+
 export enum FilePurpose {
   Vision = 'vision',
   FineTune = 'fine-tune',

package/src/types/files.ts

@@ -7,6 +7,7 @@ export enum FileSources {
   openai = 'openai',
   s3 = 's3',
   vectordb = 'vectordb',
+  execute_code = 'execute_code',
 }
 
 export const checkOpenAIStorage = (source: string) =>
@@ -15,13 +16,16 @@ export const checkOpenAIStorage = (source: string) =>
 export enum FileContext {
   avatar = 'avatar',
   unknown = 'unknown',
+  agents = 'agents',
   assistants = 'assistants',
+  execute_code = 'execute_code',
   image_generation = 'image_generation',
   assistants_output = 'assistants_output',
   message_attachment = 'message_attachment',
   filename = 'filename',
   updatedAt = 'updatedAt',
   source = 'source',
+  filterSource = 'filterSource',
   context = 'context',
   bytes = 'bytes',
 }
@@ -127,6 +131,7 @@ export type BatchFile = {
 
 export type DeleteFilesBody = {
   files: BatchFile[];
+  agent_id?: string;
   assistant_id?: string;
   tool_resource?: EToolResources;
 };

package/src/types/mutations.ts

@@ -8,6 +8,9 @@ import {
   FunctionTool,
   AssistantDocument,
   Action,
+  Agent,
+  AgentCreateParams,
+  AgentUpdateParams,
 } from './assistants';
 
 export type MutationOptions<
@@ -41,6 +44,8 @@ export type DeletePresetOptions = MutationOptions<PresetDeleteResponse, types.TP
 
 export type LogoutOptions = MutationOptions<unknown, undefined>;
 
+/* Assistant mutations */
+
 export type AssistantAvatarVariables = {
   assistant_id: string;
   model: string;
@@ -94,6 +99,51 @@ export type DeleteActionVariables = {
 
 export type DeleteActionOptions = MutationOptions<void, DeleteActionVariables>;
 
+/* Agent mutations */
+
+export type AgentAvatarVariables = {
+  agent_id: string;
+  formData: FormData;
+  postCreation?: boolean;
+};
+
+export type UpdateAgentActionVariables = {
+  agent_id: string;
+  action_id?: string;
+  metadata: ActionMetadata;
+  functions: FunctionTool[];
+};
+
+export type UploadAgentAvatarOptions = MutationOptions<Agent, AgentAvatarVariables>;
+
+export type CreateAgentMutationOptions = MutationOptions<Agent, AgentCreateParams>;
+
+export type UpdateAgentVariables = {
+  agent_id: string;
+  data: AgentUpdateParams;
+};
+
+export type UpdateAgentMutationOptions = MutationOptions<Agent, UpdateAgentVariables>;
+
+export type DeleteAgentBody = {
+  agent_id: string;
+};
+
+export type DeleteAgentMutationOptions = MutationOptions<void, Pick<DeleteAgentBody, 'agent_id'>>;
+
+export type UpdateAgentActionResponse = [Agent, Action];
+export type UpdateAgentActionOptions = MutationOptions<
+  UpdateAgentActionResponse,
+  UpdateAgentActionVariables
+>;
+
+export type DeleteAgentActionVariables = {
+  agent_id: string;
+  action_id: string;
+};
+
+export type DeleteAgentActionOptions = MutationOptions<void, DeleteAgentActionVariables>;
+
 export type DeleteConversationOptions = MutationOptions<
   types.TDeleteConversationResponse,
   types.TDeleteConversationRequest
@@ -105,10 +155,22 @@ export type CreateSharedLinkOptions = MutationOptions<
   types.TSharedLink,
   Partial<types.TSharedLink>
 >;
+
+export type updateTagsInConvoOptions = MutationOptions<
+  types.TTagConversationResponse,
+  types.TTagConversationRequest
+>;
+
 export type UpdateSharedLinkOptions = MutationOptions<
   types.TSharedLink,
   Partial<types.TSharedLink>
 >;
+
+export type ArchiveConvoOptions = MutationOptions<
+  types.TArchiveConversationResponse,
+  types.TArchiveConversationRequest
+>;
+
 export type DeleteSharedLinkOptions = MutationOptions<types.TSharedLink, { shareId: string }>;
 
 export type TUpdatePromptContext =
@@ -173,3 +235,16 @@ export type UpdatePromptPermOptions = MutationOptions<
   unknown,
   types.TError
 >;
+
+export type UpdateConversationTagOptions = MutationOptions<
+  types.TConversationTag,
+  types.TConversationTagRequest
+>;
+export type DeleteConversationTagOptions = MutationOptions<types.TConversationTag, string>;
+
+export type AcceptTermsMutationOptions = MutationOptions<
+  types.TAcceptTermsResponse,
+  void,
+  unknown,
+  void
+>;

package/src/types/queries.ts

@@ -1,6 +1,7 @@
 import type { InfiniteData } from '@tanstack/react-query';
-import type { TMessage, TConversation, TSharedLink } from '../schemas';
 import type * as t from '../types';
+import type { TMessage, TConversation, TSharedLink, TConversationTag } from '../schemas';
+
 export type Conversation = {
   id: string;
   createdAt: number;
@@ -15,9 +16,10 @@ export type ConversationListParams = {
   before?: string | null;
   after?: string | null;
   order?: 'asc' | 'desc';
-  pageNumber: string; // Add this line
+  pageNumber: string;
   conversationId?: string;
   isArchived?: boolean;
+  tags?: string[];
 };
 
 // Type for the response from the conversation list API
@@ -68,3 +70,5 @@ export type AllPromptGroupsFilterRequest = {
 };
 
 export type AllPromptGroupsResponse = t.TPromptGroup[];
+
+export type ConversationTagsResponse = TConversationTag[];

package/src/types/runs.ts

@@ -0,0 +1,22 @@
+export enum ContentTypes {
+  TEXT = 'text',
+  TEXT_DELTA = 'text_delta',
+  TOOL_CALL = 'tool_call',
+  IMAGE_FILE = 'image_file',
+  IMAGE_URL = 'image_url',
+  ERROR = 'error',
+}
+
+export enum StepTypes {
+  TOOL_CALLS = 'tool_calls',
+  MESSAGE_CREATION = 'message_creation',
+}
+
+export enum ToolCallTypes {
+  FUNCTION = 'function',
+  RETRIEVAL = 'retrieval',
+  FILE_SEARCH = 'file_search',
+  CODE_INTERPRETER = 'code_interpreter',
+  /* Agents Tool Call */
+  TOOL_CALL = 'tool_call',
+}