librechat-data-provider 0.1.3 → 0.1.4

package/dist/index.es.js

@@ -1,4 +1,5 @@
 import axios from 'axios';
+import { z } from 'zod';
 import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
 
 /******************************************************************************
@@ -342,6 +343,94 @@ var EModelEndpoint;
     EModelEndpoint["gptPlugins"] = "gptPlugins";
     EModelEndpoint["anthropic"] = "anthropic";
 })(EModelEndpoint || (EModelEndpoint = {}));
+var eModelEndpointSchema = z.nativeEnum(EModelEndpoint);
+var tMessageSchema = z.object({
+    messageId: z.string(),
+    conversationId: z.string(),
+    clientId: z.string(),
+    parentMessageId: z.string(),
+    sender: z.string(),
+    text: z.string(),
+    isCreatedByUser: z.boolean(),
+    error: z.boolean(),
+    createdAt: z.string(),
+    updatedAt: z.string(),
+});
+var tPluginAuthConfigSchema = z.object({
+    authField: z.string(),
+    label: z.string(),
+    description: z.string(),
+});
+var tPluginSchema = z.object({
+    name: z.string(),
+    pluginKey: z.string(),
+    description: z.string(),
+    icon: z.string(),
+    authConfig: z.array(tPluginAuthConfigSchema),
+    authenticated: z.boolean().optional(),
+    isButton: z.boolean().optional(),
+});
+var tExampleSchema = z.object({
+    input: z.object({
+        content: z.string(),
+    }),
+    output: z.object({
+        content: z.string(),
+    }),
+});
+var tAgentOptionsSchema = z.object({
+    agent: z.string(),
+    skipCompletion: z.boolean(),
+    model: z.string(),
+    temperature: z.number(),
+});
+var tConversationSchema = z.object({
+    conversationId: z.string().nullable(),
+    title: z.string(),
+    user: z.string().optional(),
+    endpoint: eModelEndpointSchema.nullable(),
+    suggestions: z.array(z.string()).optional(),
+    messages: z.array(z.string()).optional(),
+    tools: z.array(tPluginSchema).optional(),
+    createdAt: z.string(),
+    updatedAt: z.string(),
+    systemMessage: z.string().nullable().optional(),
+    modelLabel: z.string().nullable().optional(),
+    examples: z.array(tExampleSchema).optional(),
+    chatGptLabel: z.string().nullable().optional(),
+    userLabel: z.string().optional(),
+    model: z.string().optional(),
+    promptPrefix: z.string().nullable().optional(),
+    temperature: z.number().optional(),
+    topP: z.number().optional(),
+    topK: z.number().optional(),
+    context: z.string().nullable().optional(),
+    top_p: z.number().optional(),
+    frequency_penalty: z.number().optional(),
+    presence_penalty: z.number().optional(),
+    jailbreak: z.boolean().optional(),
+    jailbreakConversationId: z.string().nullable().optional(),
+    conversationSignature: z.string().nullable().optional(),
+    parentMessageId: z.string().optional(),
+    clientId: z.string().nullable().optional(),
+    invocationId: z.number().nullable().optional(),
+    toneStyle: z.string().nullable().optional(),
+    maxOutputTokens: z.number().optional(),
+    agentOptions: tAgentOptionsSchema.nullable().optional(),
+});
+var tPresetSchema = tConversationSchema
+    .omit({
+    conversationId: true,
+    createdAt: true,
+    updatedAt: true,
+    title: true,
+})
+    .merge(z.object({
+    conversationId: z.string().optional(),
+    presetId: z.string().nullable().optional(),
+    title: z.string().nullable().optional(),
+}));
+
 var Side;
 (function (Side) {
     Side["Top"] = "top";
@@ -384,7 +473,7 @@ var useGetConversationByIdQuery = function (id, config) {
 var useGetConversationByIdMutation = function (id) {
     var queryClient = useQueryClient();
     return useMutation(function () { return getConversationById(id); }, {
-        // onSuccess: (res: t.TConversation) => {
+        // onSuccess: (res: s.TConversation) => {
         onSuccess: function () {
             queryClient.invalidateQueries([QueryKeys.conversation, id]);
         },
@@ -760,7 +849,7 @@ var SSE = function (url, options) {
 
 function createPayload(submission) {
     var conversation = submission.conversation, message = submission.message, endpointOption = submission.endpointOption;
-    var conversationId = conversation.conversationId;
+    var conversationId = tConversationSchema.parse(conversation).conversationId;
     var endpoint = endpointOption.endpoint;
     var endpointUrlMap = {
         azureOpenAI: '/api/ask/azureOpenAI',
@@ -777,4 +866,4 @@ function createPayload(submission) {
     return { server: server, payload: payload };
 }
 
-export { EModelEndpoint, QueryKeys, SSE, Side, abortRequestWithMessage, clearAllConversations, createPayload, createPreset, deleteConversation, deletePreset, getAIEndpoints, getAvailablePlugins, getConversationById, getConversations, getLoginGoogle, getMessagesByConvoId, getPresets, getSearchEnabled, getStartupConfig, getUser, login, logout, refreshToken, register, requestPasswordReset, resetPassword, searchConversations, setAcceptLanguageHeader, setTokenHeader, updateConversation, updatePreset, updateTokenCount, updateUserPlugins, useAbortRequestWithMessage, useAvailablePluginsQuery, useClearConversationsMutation, useCreatePresetMutation, useDeleteConversationMutation, useDeletePresetMutation, useGetConversationByIdMutation, useGetConversationByIdQuery, useGetConversationsQuery, useGetEndpointsQuery, useGetMessagesByConvoId, useGetPresetsQuery, useGetSearchEnabledQuery, useGetStartupConfig, useGetUserQuery, useLoginUserMutation, useLogoutUserMutation, useRefreshTokenMutation, useRegisterUserMutation, useRequestPasswordResetMutation, useResetPasswordMutation, useSearchQuery, useUpdateConversationMutation, useUpdatePresetMutation, useUpdateTokenCountMutation, useUpdateUserPluginsMutation };
+export { EModelEndpoint, QueryKeys, SSE, Side, abortRequestWithMessage, clearAllConversations, createPayload, createPreset, deleteConversation, deletePreset, eModelEndpointSchema, getAIEndpoints, getAvailablePlugins, getConversationById, getConversations, getLoginGoogle, getMessagesByConvoId, getPresets, getSearchEnabled, getStartupConfig, getUser, login, logout, refreshToken, register, requestPasswordReset, resetPassword, searchConversations, setAcceptLanguageHeader, setTokenHeader, tAgentOptionsSchema, tConversationSchema, tExampleSchema, tMessageSchema, tPluginAuthConfigSchema, tPluginSchema, tPresetSchema, updateConversation, updatePreset, updateTokenCount, updateUserPlugins, useAbortRequestWithMessage, useAvailablePluginsQuery, useClearConversationsMutation, useCreatePresetMutation, useDeleteConversationMutation, useDeletePresetMutation, useGetConversationByIdMutation, useGetConversationByIdQuery, useGetConversationsQuery, useGetEndpointsQuery, useGetMessagesByConvoId, useGetPresetsQuery, useGetSearchEnabledQuery, useGetStartupConfig, useGetUserQuery, useLoginUserMutation, useLogoutUserMutation, useRefreshTokenMutation, useRegisterUserMutation, useRequestPasswordResetMutation, useResetPasswordMutation, useSearchQuery, useUpdateConversationMutation, useUpdatePresetMutation, useUpdateTokenCountMutation, useUpdateUserPluginsMutation };

package/dist/index.js

@@ -1,6 +1,7 @@
 'use strict';
 
 var axios = require('axios');
+var zod = require('zod');
 var reactQuery = require('@tanstack/react-query');
 
 /******************************************************************************
@@ -344,6 +345,94 @@ exports.EModelEndpoint = void 0;
     EModelEndpoint["gptPlugins"] = "gptPlugins";
     EModelEndpoint["anthropic"] = "anthropic";
 })(exports.EModelEndpoint || (exports.EModelEndpoint = {}));
+var eModelEndpointSchema = zod.z.nativeEnum(exports.EModelEndpoint);
+var tMessageSchema = zod.z.object({
+    messageId: zod.z.string(),
+    conversationId: zod.z.string(),
+    clientId: zod.z.string(),
+    parentMessageId: zod.z.string(),
+    sender: zod.z.string(),
+    text: zod.z.string(),
+    isCreatedByUser: zod.z.boolean(),
+    error: zod.z.boolean(),
+    createdAt: zod.z.string(),
+    updatedAt: zod.z.string(),
+});
+var tPluginAuthConfigSchema = zod.z.object({
+    authField: zod.z.string(),
+    label: zod.z.string(),
+    description: zod.z.string(),
+});
+var tPluginSchema = zod.z.object({
+    name: zod.z.string(),
+    pluginKey: zod.z.string(),
+    description: zod.z.string(),
+    icon: zod.z.string(),
+    authConfig: zod.z.array(tPluginAuthConfigSchema),
+    authenticated: zod.z.boolean().optional(),
+    isButton: zod.z.boolean().optional(),
+});
+var tExampleSchema = zod.z.object({
+    input: zod.z.object({
+        content: zod.z.string(),
+    }),
+    output: zod.z.object({
+        content: zod.z.string(),
+    }),
+});
+var tAgentOptionsSchema = zod.z.object({
+    agent: zod.z.string(),
+    skipCompletion: zod.z.boolean(),
+    model: zod.z.string(),
+    temperature: zod.z.number(),
+});
+var tConversationSchema = zod.z.object({
+    conversationId: zod.z.string().nullable(),
+    title: zod.z.string(),
+    user: zod.z.string().optional(),
+    endpoint: eModelEndpointSchema.nullable(),
+    suggestions: zod.z.array(zod.z.string()).optional(),
+    messages: zod.z.array(zod.z.string()).optional(),
+    tools: zod.z.array(tPluginSchema).optional(),
+    createdAt: zod.z.string(),
+    updatedAt: zod.z.string(),
+    systemMessage: zod.z.string().nullable().optional(),
+    modelLabel: zod.z.string().nullable().optional(),
+    examples: zod.z.array(tExampleSchema).optional(),
+    chatGptLabel: zod.z.string().nullable().optional(),
+    userLabel: zod.z.string().optional(),
+    model: zod.z.string().optional(),
+    promptPrefix: zod.z.string().nullable().optional(),
+    temperature: zod.z.number().optional(),
+    topP: zod.z.number().optional(),
+    topK: zod.z.number().optional(),
+    context: zod.z.string().nullable().optional(),
+    top_p: zod.z.number().optional(),
+    frequency_penalty: zod.z.number().optional(),
+    presence_penalty: zod.z.number().optional(),
+    jailbreak: zod.z.boolean().optional(),
+    jailbreakConversationId: zod.z.string().nullable().optional(),
+    conversationSignature: zod.z.string().nullable().optional(),
+    parentMessageId: zod.z.string().optional(),
+    clientId: zod.z.string().nullable().optional(),
+    invocationId: zod.z.number().nullable().optional(),
+    toneStyle: zod.z.string().nullable().optional(),
+    maxOutputTokens: zod.z.number().optional(),
+    agentOptions: tAgentOptionsSchema.nullable().optional(),
+});
+var tPresetSchema = tConversationSchema
+    .omit({
+    conversationId: true,
+    createdAt: true,
+    updatedAt: true,
+    title: true,
+})
+    .merge(zod.z.object({
+    conversationId: zod.z.string().optional(),
+    presetId: zod.z.string().nullable().optional(),
+    title: zod.z.string().nullable().optional(),
+}));
+
 exports.Side = void 0;
 (function (Side) {
     Side["Top"] = "top";
@@ -386,7 +475,7 @@ var useGetConversationByIdQuery = function (id, config) {
 var useGetConversationByIdMutation = function (id) {
     var queryClient = reactQuery.useQueryClient();
     return reactQuery.useMutation(function () { return getConversationById(id); }, {
-        // onSuccess: (res: t.TConversation) => {
+        // onSuccess: (res: s.TConversation) => {
         onSuccess: function () {
             queryClient.invalidateQueries([exports.QueryKeys.conversation, id]);
         },
@@ -762,7 +851,7 @@ var SSE = function (url, options) {
 
 function createPayload(submission) {
     var conversation = submission.conversation, message = submission.message, endpointOption = submission.endpointOption;
-    var conversationId = conversation.conversationId;
+    var conversationId = tConversationSchema.parse(conversation).conversationId;
     var endpoint = endpointOption.endpoint;
     var endpointUrlMap = {
         azureOpenAI: '/api/ask/azureOpenAI',
@@ -786,6 +875,7 @@ exports.createPayload = createPayload;
 exports.createPreset = createPreset;
 exports.deleteConversation = deleteConversation;
 exports.deletePreset = deletePreset;
+exports.eModelEndpointSchema = eModelEndpointSchema;
 exports.getAIEndpoints = getAIEndpoints;
 exports.getAvailablePlugins = getAvailablePlugins;
 exports.getConversationById = getConversationById;
@@ -805,6 +895,13 @@ exports.resetPassword = resetPassword;
 exports.searchConversations = searchConversations;
 exports.setAcceptLanguageHeader = setAcceptLanguageHeader;
 exports.setTokenHeader = setTokenHeader;
+exports.tAgentOptionsSchema = tAgentOptionsSchema;
+exports.tConversationSchema = tConversationSchema;
+exports.tExampleSchema = tExampleSchema;
+exports.tMessageSchema = tMessageSchema;
+exports.tPluginAuthConfigSchema = tPluginAuthConfigSchema;
+exports.tPluginSchema = tPluginSchema;
+exports.tPresetSchema = tPresetSchema;
 exports.updateConversation = updateConversation;
 exports.updatePreset = updatePreset;
 exports.updateTokenCount = updateTokenCount;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "librechat-data-provider",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "data services for librechat apps",
   "main": "dist/index.js",
   "module": "dist/index.es.js",
@@ -25,7 +25,8 @@
   "homepage": "https://github.com/danny-avila/LibreChat#readme",
   "dependencies": {
     "@tanstack/react-query": "^4.28.0",
-    "axios": "^1.3.4"
+    "axios": "^1.3.4",
+    "zod": "^3.21.4"
   },
   "devDependencies": {
     "@babel/preset-env": "^7.21.5",

package/src/createPayload.ts

@@ -1,8 +1,9 @@
-import type { TConversation, TSubmission, EModelEndpoint } from './types';
+import { tConversationSchema } from './schemas';
+import type { TSubmission, EModelEndpoint } from './types';
 
 export default function createPayload(submission: TSubmission) {
   const { conversation, message, endpointOption } = submission;
-  const { conversationId } = conversation as TConversation;
+  const { conversationId } = tConversationSchema.parse(conversation);
   const { endpoint } = endpointOption as { endpoint: EModelEndpoint };
 
   const endpointUrlMap = {

package/src/data-service.ts

@@ -1,4 +1,5 @@
 import * as t from './types';
+import * as s from './schemas';
 import request from './request';
 import * as endpoints from './api-endpoints';
 
@@ -23,11 +24,11 @@ export function clearAllConversations(): Promise<unknown> {
   return request.post(endpoints.deleteConversation(), { arg: {} });
 }
 
-export function getMessagesByConvoId(id: string): Promise<t.TMessage[]> {
+export function getMessagesByConvoId(id: string): Promise<s.TMessage[]> {
   return request.get(endpoints.messages(id));
 }
 
-export function getConversationById(id: string): Promise<t.TConversation> {
+export function getConversationById(id: string): Promise<s.TConversation> {
   return request.get(endpoints.conversationById(id));
 }
 
@@ -37,19 +38,19 @@ export function updateConversation(
   return request.post(endpoints.updateConversation(), { arg: payload });
 }
 
-export function getPresets(): Promise<t.TPreset[]> {
+export function getPresets(): Promise<s.TPreset[]> {
   return request.get(endpoints.presets());
 }
 
-export function createPreset(payload: t.TPreset): Promise<t.TPreset[]> {
+export function createPreset(payload: s.TPreset): Promise<s.TPreset[]> {
   return request.post(endpoints.presets(), payload);
 }
 
-export function updatePreset(payload: t.TPreset): Promise<t.TPreset[]> {
+export function updatePreset(payload: s.TPreset): Promise<s.TPreset[]> {
   return request.post(endpoints.presets(), payload);
 }
 
-export function deletePreset(arg: t.TPreset | object): Promise<t.TPreset[]> {
+export function deletePreset(arg: s.TPreset | object): Promise<s.TPreset[]> {
   return request.post(endpoints.deletePreset(), arg);
 }
 
@@ -106,7 +107,7 @@ export const resetPassword = (payload: t.TResetPassword) => {
   return request.post(endpoints.resetPassword(), payload);
 };
 
-export const getAvailablePlugins = (): Promise<t.TPlugin[]> => {
+export const getAvailablePlugins = (): Promise<s.TPlugin[]> => {
   return request.get(endpoints.plugins());
 };
 

package/src/react-query-service.ts

@@ -7,6 +7,7 @@ import {
   QueryObserverResult,
 } from '@tanstack/react-query';
 import * as t from './types';
+import * as s from './schemas';
 import * as dataService from './data-service';
 
 export enum QueryKeys {
@@ -47,9 +48,9 @@ export const useGetUserQuery = (
 
 export const useGetMessagesByConvoId = (
   id: string,
-  config?: UseQueryOptions<t.TMessage[]>,
-): QueryObserverResult<t.TMessage[]> => {
-  return useQuery<t.TMessage[]>(
+  config?: UseQueryOptions<s.TMessage[]>,
+): QueryObserverResult<s.TMessage[]> => {
+  return useQuery<s.TMessage[]>(
     [QueryKeys.messages, id],
     () => dataService.getMessagesByConvoId(id),
     {
@@ -63,9 +64,9 @@ export const useGetMessagesByConvoId = (
 
 export const useGetConversationByIdQuery = (
   id: string,
-  config?: UseQueryOptions<t.TConversation>,
-): QueryObserverResult<t.TConversation> => {
-  return useQuery<t.TConversation>(
+  config?: UseQueryOptions<s.TConversation>,
+): QueryObserverResult<s.TConversation> => {
+  return useQuery<s.TConversation>(
     [QueryKeys.conversation, id],
     () => dataService.getConversationById(id),
     {
@@ -79,10 +80,10 @@ export const useGetConversationByIdQuery = (
 
 //This isn't ideal because its just a query and we're using mutation, but it was the only way
 //to make it work with how the Chat component is structured
-export const useGetConversationByIdMutation = (id: string): UseMutationResult<t.TConversation> => {
+export const useGetConversationByIdMutation = (id: string): UseMutationResult<s.TConversation> => {
   const queryClient = useQueryClient();
   return useMutation(() => dataService.getConversationById(id), {
-    // onSuccess: (res: t.TConversation) => {
+    // onSuccess: (res: s.TConversation) => {
     onSuccess: () => {
       queryClient.invalidateQueries([QueryKeys.conversation, id]);
     },
@@ -174,13 +175,13 @@ export const useGetEndpointsQuery = (): QueryObserverResult<t.TEndpointsConfig>
 };
 
 export const useCreatePresetMutation = (): UseMutationResult<
-  t.TPreset[],
+  s.TPreset[],
   unknown,
-  t.TPreset,
+  s.TPreset,
   unknown
 > => {
   const queryClient = useQueryClient();
-  return useMutation((payload: t.TPreset) => dataService.createPreset(payload), {
+  return useMutation((payload: s.TPreset) => dataService.createPreset(payload), {
     onSuccess: () => {
       queryClient.invalidateQueries([QueryKeys.presets]);
     },
@@ -188,13 +189,13 @@ export const useCreatePresetMutation = (): UseMutationResult<
 };
 
 export const useUpdatePresetMutation = (): UseMutationResult<
-  t.TPreset[],
+  s.TPreset[],
   unknown,
-  t.TPreset,
+  s.TPreset,
   unknown
 > => {
   const queryClient = useQueryClient();
-  return useMutation((payload: t.TPreset) => dataService.updatePreset(payload), {
+  return useMutation((payload: s.TPreset) => dataService.updatePreset(payload), {
     onSuccess: () => {
       queryClient.invalidateQueries([QueryKeys.presets]);
     },
@@ -202,9 +203,9 @@ export const useUpdatePresetMutation = (): UseMutationResult<
 };
 
 export const useGetPresetsQuery = (
-  config?: UseQueryOptions<t.TPreset[]>,
-): QueryObserverResult<t.TPreset[], unknown> => {
-  return useQuery<t.TPreset[]>([QueryKeys.presets], () => dataService.getPresets(), {
+  config?: UseQueryOptions<s.TPreset[]>,
+): QueryObserverResult<s.TPreset[], unknown> => {
+  return useQuery<s.TPreset[]>([QueryKeys.presets], () => dataService.getPresets(), {
     refetchOnWindowFocus: false,
     refetchOnReconnect: false,
     refetchOnMount: false,
@@ -213,13 +214,13 @@ export const useGetPresetsQuery = (
 };
 
 export const useDeletePresetMutation = (): UseMutationResult<
-  t.TPreset[],
+  s.TPreset[],
   unknown,
-  t.TPreset | object,
+  s.TPreset | object,
   unknown
 > => {
   const queryClient = useQueryClient();
-  return useMutation((payload: t.TPreset | object) => dataService.deletePreset(payload), {
+  return useMutation((payload: s.TPreset | object) => dataService.deletePreset(payload), {
     onSuccess: () => {
       queryClient.invalidateQueries([QueryKeys.presets]);
     },
@@ -323,8 +324,8 @@ export const useResetPasswordMutation = (): UseMutationResult<
   return useMutation((payload: t.TResetPassword) => dataService.resetPassword(payload));
 };
 
-export const useAvailablePluginsQuery = (): QueryObserverResult<t.TPlugin[]> => {
-  return useQuery<t.TPlugin[]>(
+export const useAvailablePluginsQuery = (): QueryObserverResult<s.TPlugin[]> => {
+  return useQuery<s.TPlugin[]>(
     [QueryKeys.availablePlugins],
     () => dataService.getAvailablePlugins(),
     {

package/src/schemas.ts

@@ -0,0 +1,121 @@
+import { z } from 'zod';
+
+export enum EModelEndpoint {
+  azureOpenAI = 'azureOpenAI',
+  openAI = 'openAI',
+  bingAI = 'bingAI',
+  chatGPT = 'chatGPT',
+  chatGPTBrowser = 'chatGPTBrowser',
+  google = 'google',
+  gptPlugins = 'gptPlugins',
+  anthropic = 'anthropic',
+}
+
+export const eModelEndpointSchema = z.nativeEnum(EModelEndpoint);
+
+export const tMessageSchema = z.object({
+  messageId: z.string(),
+  conversationId: z.string(),
+  clientId: z.string(),
+  parentMessageId: z.string(),
+  sender: z.string(),
+  text: z.string(),
+  isCreatedByUser: z.boolean(),
+  error: z.boolean(),
+  createdAt: z.string(),
+  updatedAt: z.string(),
+});
+
+export type TMessage = z.infer<typeof tMessageSchema>;
+
+export const tPluginAuthConfigSchema = z.object({
+  authField: z.string(),
+  label: z.string(),
+  description: z.string(),
+});
+
+export type TPluginAuthConfig = z.infer<typeof tPluginAuthConfigSchema>;
+
+export const tPluginSchema = z.object({
+  name: z.string(),
+  pluginKey: z.string(),
+  description: z.string(),
+  icon: z.string(),
+  authConfig: z.array(tPluginAuthConfigSchema),
+  authenticated: z.boolean().optional(),
+  isButton: z.boolean().optional(),
+});
+
+export type TPlugin = z.infer<typeof tPluginSchema>;
+
+export const tExampleSchema = z.object({
+  input: z.object({
+    content: z.string(),
+  }),
+  output: z.object({
+    content: z.string(),
+  }),
+});
+
+export type TExample = z.infer<typeof tExampleSchema>;
+
+export const tAgentOptionsSchema = z.object({
+  agent: z.string(),
+  skipCompletion: z.boolean(),
+  model: z.string(),
+  temperature: z.number(),
+});
+
+export const tConversationSchema = z.object({
+  conversationId: z.string().nullable(),
+  title: z.string(),
+  user: z.string().optional(),
+  endpoint: eModelEndpointSchema.nullable(),
+  suggestions: z.array(z.string()).optional(),
+  messages: z.array(z.string()).optional(),
+  tools: z.array(tPluginSchema).optional(),
+  createdAt: z.string(),
+  updatedAt: z.string(),
+  systemMessage: z.string().nullable().optional(),
+  modelLabel: z.string().nullable().optional(),
+  examples: z.array(tExampleSchema).optional(),
+  chatGptLabel: z.string().nullable().optional(),
+  userLabel: z.string().optional(),
+  model: z.string().optional(),
+  promptPrefix: z.string().nullable().optional(),
+  temperature: z.number().optional(),
+  topP: z.number().optional(),
+  topK: z.number().optional(),
+  context: z.string().nullable().optional(),
+  top_p: z.number().optional(),
+  frequency_penalty: z.number().optional(),
+  presence_penalty: z.number().optional(),
+  jailbreak: z.boolean().optional(),
+  jailbreakConversationId: z.string().nullable().optional(),
+  conversationSignature: z.string().nullable().optional(),
+  parentMessageId: z.string().optional(),
+  clientId: z.string().nullable().optional(),
+  invocationId: z.number().nullable().optional(),
+  toneStyle: z.string().nullable().optional(),
+  maxOutputTokens: z.number().optional(),
+  agentOptions: tAgentOptionsSchema.nullable().optional(),
+});
+
+export type TConversation = z.infer<typeof tConversationSchema>;
+
+export const tPresetSchema = tConversationSchema
+  .omit({
+    conversationId: true,
+    createdAt: true,
+    updatedAt: true,
+    title: true,
+  })
+  .merge(
+    z.object({
+      conversationId: z.string().optional(),
+      presetId: z.string().nullable().optional(),
+      title: z.string().nullable().optional(),
+    }),
+  );
+
+export type TPreset = z.infer<typeof tPresetSchema>;