chatifai 1.0.24 → 1.0.26

package/dist/index.js

@@ -14,7 +14,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.FullFileTextSchema = exports.PreDefinedQuerySchema = exports.ChunkSchema = exports.ContextSchema = exports.AgentSchema = exports.UserSchema = exports.OrganizationSchema = void 0;
+exports.MessageSchema = exports.ThreadSchema = exports.FullFileTextSchema = exports.PreDefinedQuerySchema = exports.ChunkSchema = exports.ContextSchema = exports.AgentSchema = exports.UserSchema = exports.OrganizationSchema = void 0;
 // Export types for consumers
 __exportStar(require("./types"), exports);
 const mongoose_1 = require("mongoose");
@@ -147,16 +147,19 @@ exports.ChunkSchema = new mongoose_1.Schema({
 });
 exports.PreDefinedQuerySchema = new mongoose_1.Schema({
     organization_id: { type: String },
-    title: { type: String, minlength: 1, maxlength: 100 },
+    query_name: { type: String, minlength: 1, maxlength: 100 },
+    query_hint: { type: String, minlength: 1, maxlength: 100 },
     query: { type: String, required: true },
-    context_ids: { type: [String] },
+    instructions: { type: String },
+    contexts: { type: [Object] },
     status: {
         type: String,
         enum: ['active', 'inactive', 'archived'],
         default: 'active',
         required: true,
     },
-    users_ids: { type: [String] },
+    users: { type: [Object] },
+    click_count: { type: Number },
     created_at: { type: Number },
     updated_at: { type: Number },
 }, {
@@ -179,3 +182,38 @@ exports.FullFileTextSchema = new mongoose_1.Schema({
     id: true,
     // encryptionType: 'csfle',
 });
+exports.ThreadSchema = new mongoose_1.Schema({
+    organization_id: { type: String, required: true },
+    user_id: { type: String, required: true },
+    agent_id: { type: String, required: true },
+    title: { type: String, minlength: 1, maxlength: 200 },
+    description: { type: String },
+    query_id: { type: String },
+    messages_count: { type: Number, default: 0 },
+    created_at: { type: Number },
+    updated_at: { type: Number },
+}, {
+    versionKey: false,
+    toJSON: { virtuals: true },
+    toObject: { virtuals: true },
+    timestamps: true,
+    id: true,
+    // encryptionType: 'csfle',
+});
+exports.MessageSchema = new mongoose_1.Schema({
+    thread_id: { type: String, required: true },
+    organization_id: { type: String, required: true },
+    user_id: { type: String },
+    agent_id: { type: String },
+    content: { type: String, required: true },
+    role: { type: String, enum: ['user', 'agent', 'system'], required: true },
+    created_at: { type: Number },
+    updated_at: { type: Number },
+}, {
+    versionKey: false,
+    toJSON: { virtuals: true },
+    toObject: { virtuals: true },
+    timestamps: true,
+    id: true,
+    // encryptionType: 'csfle',
+});

package/dist/types.d.ts

@@ -321,33 +321,230 @@ export declare const zodFullTextSchema: z.ZodObject<{
 export declare const zodPreDefinedQuerySchema: z.ZodObject<{
     _id: z.ZodOptional<z.ZodString>;
     organization_id: z.ZodOptional<z.ZodString>;
-    title: z.ZodOptional<z.ZodString>;
+    query_name: z.ZodOptional<z.ZodString>;
+    query_hint: z.ZodOptional<z.ZodString>;
     query: z.ZodString;
-    context_ids: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    instructions: z.ZodOptional<z.ZodString>;
+    contexts: z.ZodOptional<z.ZodArray<z.ZodObject<{
+        _id: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        organization_id: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        title: z.ZodOptional<z.ZodString>;
+        ai_description: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        note: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        file_id: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        file_url: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        file_gcs_url: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        file_name: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        type: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        size: z.ZodOptional<z.ZodOptional<z.ZodNumber>>;
+        tokens: z.ZodOptional<z.ZodOptional<z.ZodNumber>>;
+        is_active: z.ZodOptional<z.ZodOptional<z.ZodDefault<z.ZodBoolean>>>;
+        created_at: z.ZodOptional<z.ZodOptional<z.ZodNumber>>;
+        updated_at: z.ZodOptional<z.ZodOptional<z.ZodNumber>>;
+        uploaded_by: z.ZodOptional<z.ZodOptional<z.ZodObject<{
+            first_name: z.ZodOptional<z.ZodString>;
+            last_name: z.ZodOptional<z.ZodString>;
+            email: z.ZodOptional<z.ZodString>;
+        }, "strip", z.ZodTypeAny, {
+            email?: string | undefined;
+            first_name?: string | undefined;
+            last_name?: string | undefined;
+        }, {
+            email?: string | undefined;
+            first_name?: string | undefined;
+            last_name?: string | undefined;
+        }>>>;
+        bucket_name: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        processing_status: z.ZodOptional<z.ZodOptional<z.ZodEnum<["pending", "processing", "completed", "failed"]>>>;
+    }, "strip", z.ZodTypeAny, {
+        _id?: string | undefined;
+        title?: string | undefined;
+        type?: string | undefined;
+        created_at?: number | undefined;
+        updated_at?: number | undefined;
+        organization_id?: string | undefined;
+        ai_description?: string | undefined;
+        note?: string | undefined;
+        file_id?: string | undefined;
+        file_url?: string | undefined;
+        file_gcs_url?: string | undefined;
+        file_name?: string | undefined;
+        size?: number | undefined;
+        tokens?: number | undefined;
+        is_active?: boolean | undefined;
+        uploaded_by?: {
+            email?: string | undefined;
+            first_name?: string | undefined;
+            last_name?: string | undefined;
+        } | undefined;
+        bucket_name?: string | undefined;
+        processing_status?: "pending" | "processing" | "completed" | "failed" | undefined;
+    }, {
+        _id?: string | undefined;
+        title?: string | undefined;
+        type?: string | undefined;
+        created_at?: number | undefined;
+        updated_at?: number | undefined;
+        organization_id?: string | undefined;
+        ai_description?: string | undefined;
+        note?: string | undefined;
+        file_id?: string | undefined;
+        file_url?: string | undefined;
+        file_gcs_url?: string | undefined;
+        file_name?: string | undefined;
+        size?: number | undefined;
+        tokens?: number | undefined;
+        is_active?: boolean | undefined;
+        uploaded_by?: {
+            email?: string | undefined;
+            first_name?: string | undefined;
+            last_name?: string | undefined;
+        } | undefined;
+        bucket_name?: string | undefined;
+        processing_status?: "pending" | "processing" | "completed" | "failed" | undefined;
+    }>, "many">>;
     status: z.ZodDefault<z.ZodEnum<["active", "inactive", "archived"]>>;
-    users_ids: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    users: z.ZodOptional<z.ZodArray<z.ZodObject<{
+        _id: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        organization_memberships: z.ZodOptional<z.ZodOptional<z.ZodArray<z.ZodString, "many">>>;
+        clerk_user_id: z.ZodOptional<z.ZodString>;
+        email: z.ZodOptional<z.ZodString>;
+        first_name: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        last_name: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        active_until: z.ZodOptional<z.ZodOptional<z.ZodNumber>>;
+        notes: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        avatar: z.ZodOptional<z.ZodOptional<z.ZodString>>;
+        user_role: z.ZodOptional<z.ZodOptional<z.ZodDefault<z.ZodEnum<["admin", "member", "client"]>>>>;
+        created_at: z.ZodOptional<z.ZodOptional<z.ZodNumber>>;
+        updated_at: z.ZodOptional<z.ZodOptional<z.ZodNumber>>;
+    }, "strip", z.ZodTypeAny, {
+        _id?: string | undefined;
+        created_at?: number | undefined;
+        updated_at?: number | undefined;
+        organization_memberships?: string[] | undefined;
+        clerk_user_id?: string | undefined;
+        email?: string | undefined;
+        first_name?: string | undefined;
+        last_name?: string | undefined;
+        active_until?: number | undefined;
+        notes?: string | undefined;
+        avatar?: string | undefined;
+        user_role?: "admin" | "member" | "client" | undefined;
+    }, {
+        _id?: string | undefined;
+        created_at?: number | undefined;
+        updated_at?: number | undefined;
+        organization_memberships?: string[] | undefined;
+        clerk_user_id?: string | undefined;
+        email?: string | undefined;
+        first_name?: string | undefined;
+        last_name?: string | undefined;
+        active_until?: number | undefined;
+        notes?: string | undefined;
+        avatar?: string | undefined;
+        user_role?: "admin" | "member" | "client" | undefined;
+    }>, "many">>;
+    click_count: z.ZodOptional<z.ZodNumber>;
     created_at: z.ZodOptional<z.ZodNumber>;
     updated_at: z.ZodOptional<z.ZodNumber>;
 }, "strip", z.ZodTypeAny, {
     status: "active" | "inactive" | "archived";
     query: string;
     _id?: string | undefined;
-    title?: string | undefined;
     created_at?: number | undefined;
     updated_at?: number | undefined;
     organization_id?: string | undefined;
-    context_ids?: string[] | undefined;
-    users_ids?: string[] | undefined;
+    instructions?: string | undefined;
+    query_name?: string | undefined;
+    query_hint?: string | undefined;
+    contexts?: {
+        _id?: string | undefined;
+        title?: string | undefined;
+        type?: string | undefined;
+        created_at?: number | undefined;
+        updated_at?: number | undefined;
+        organization_id?: string | undefined;
+        ai_description?: string | undefined;
+        note?: string | undefined;
+        file_id?: string | undefined;
+        file_url?: string | undefined;
+        file_gcs_url?: string | undefined;
+        file_name?: string | undefined;
+        size?: number | undefined;
+        tokens?: number | undefined;
+        is_active?: boolean | undefined;
+        uploaded_by?: {
+            email?: string | undefined;
+            first_name?: string | undefined;
+            last_name?: string | undefined;
+        } | undefined;
+        bucket_name?: string | undefined;
+        processing_status?: "pending" | "processing" | "completed" | "failed" | undefined;
+    }[] | undefined;
+    users?: {
+        _id?: string | undefined;
+        created_at?: number | undefined;
+        updated_at?: number | undefined;
+        organization_memberships?: string[] | undefined;
+        clerk_user_id?: string | undefined;
+        email?: string | undefined;
+        first_name?: string | undefined;
+        last_name?: string | undefined;
+        active_until?: number | undefined;
+        notes?: string | undefined;
+        avatar?: string | undefined;
+        user_role?: "admin" | "member" | "client" | undefined;
+    }[] | undefined;
+    click_count?: number | undefined;
 }, {
     query: string;
     _id?: string | undefined;
-    title?: string | undefined;
     status?: "active" | "inactive" | "archived" | undefined;
     created_at?: number | undefined;
     updated_at?: number | undefined;
     organization_id?: string | undefined;
-    context_ids?: string[] | undefined;
-    users_ids?: string[] | undefined;
+    instructions?: string | undefined;
+    query_name?: string | undefined;
+    query_hint?: string | undefined;
+    contexts?: {
+        _id?: string | undefined;
+        title?: string | undefined;
+        type?: string | undefined;
+        created_at?: number | undefined;
+        updated_at?: number | undefined;
+        organization_id?: string | undefined;
+        ai_description?: string | undefined;
+        note?: string | undefined;
+        file_id?: string | undefined;
+        file_url?: string | undefined;
+        file_gcs_url?: string | undefined;
+        file_name?: string | undefined;
+        size?: number | undefined;
+        tokens?: number | undefined;
+        is_active?: boolean | undefined;
+        uploaded_by?: {
+            email?: string | undefined;
+            first_name?: string | undefined;
+            last_name?: string | undefined;
+        } | undefined;
+        bucket_name?: string | undefined;
+        processing_status?: "pending" | "processing" | "completed" | "failed" | undefined;
+    }[] | undefined;
+    users?: {
+        _id?: string | undefined;
+        created_at?: number | undefined;
+        updated_at?: number | undefined;
+        organization_memberships?: string[] | undefined;
+        clerk_user_id?: string | undefined;
+        email?: string | undefined;
+        first_name?: string | undefined;
+        last_name?: string | undefined;
+        active_until?: number | undefined;
+        notes?: string | undefined;
+        avatar?: string | undefined;
+        user_role?: "admin" | "member" | "client" | undefined;
+    }[] | undefined;
+    click_count?: number | undefined;
 }>;
 export declare const zodQuerySchema: z.ZodObject<{
     thread_id: z.ZodOptional<z.ZodString>;
@@ -355,21 +552,89 @@ export declare const zodQuerySchema: z.ZodObject<{
     organization_id: z.ZodString;
     user_id: z.ZodString;
     query: z.ZodString;
+    pre_defined_query_id: z.ZodOptional<z.ZodString>;
     timestamp: z.ZodNumber;
 }, "strip", z.ZodTypeAny, {
     organization_id: string;
     query: string;
     user_id: string;
     timestamp: number;
-    context_ids?: string[] | undefined;
     thread_id?: string | undefined;
+    context_ids?: string[] | undefined;
+    pre_defined_query_id?: string | undefined;
 }, {
     organization_id: string;
     query: string;
     user_id: string;
     timestamp: number;
-    context_ids?: string[] | undefined;
     thread_id?: string | undefined;
+    context_ids?: string[] | undefined;
+    pre_defined_query_id?: string | undefined;
+}>;
+export declare const zodThreadSchema: z.ZodObject<{
+    _id: z.ZodOptional<z.ZodString>;
+    organization_id: z.ZodString;
+    user_id: z.ZodString;
+    agent_id: z.ZodOptional<z.ZodString>;
+    title: z.ZodOptional<z.ZodString>;
+    description: z.ZodOptional<z.ZodString>;
+    created_at: z.ZodOptional<z.ZodNumber>;
+    updated_at: z.ZodOptional<z.ZodNumber>;
+    query_id: z.ZodOptional<z.ZodString>;
+    messages_count: z.ZodOptional<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    organization_id: string;
+    user_id: string;
+    _id?: string | undefined;
+    title?: string | undefined;
+    created_at?: number | undefined;
+    updated_at?: number | undefined;
+    description?: string | undefined;
+    agent_id?: string | undefined;
+    query_id?: string | undefined;
+    messages_count?: number | undefined;
+}, {
+    organization_id: string;
+    user_id: string;
+    _id?: string | undefined;
+    title?: string | undefined;
+    created_at?: number | undefined;
+    updated_at?: number | undefined;
+    description?: string | undefined;
+    agent_id?: string | undefined;
+    query_id?: string | undefined;
+    messages_count?: number | undefined;
+}>;
+export declare const zodMessageSchema: z.ZodObject<{
+    _id: z.ZodOptional<z.ZodString>;
+    thread_id: z.ZodString;
+    organization_id: z.ZodString;
+    user_id: z.ZodOptional<z.ZodString>;
+    agent_id: z.ZodOptional<z.ZodString>;
+    role: z.ZodEnum<["user", "agent", "system"]>;
+    content: z.ZodString;
+    created_at: z.ZodOptional<z.ZodNumber>;
+    updated_at: z.ZodOptional<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    organization_id: string;
+    thread_id: string;
+    role: "user" | "agent" | "system";
+    content: string;
+    _id?: string | undefined;
+    created_at?: number | undefined;
+    updated_at?: number | undefined;
+    user_id?: string | undefined;
+    agent_id?: string | undefined;
+}, {
+    organization_id: string;
+    thread_id: string;
+    role: "user" | "agent" | "system";
+    content: string;
+    _id?: string | undefined;
+    created_at?: number | undefined;
+    updated_at?: number | undefined;
+    user_id?: string | undefined;
+    agent_id?: string | undefined;
 }>;
 export type EventAttributeType = {
     http_request: {
@@ -396,6 +661,8 @@ export type ModelType = (typeof modelsList)[number];
 export type UserRoleType = (typeof userRoles)[number];
 export type FullFileTextType = z.infer<typeof zodFullTextSchema>;
 export type QueryType = z.infer<typeof zodQuerySchema>;
+export type ThreadType = z.infer<typeof zodThreadSchema>;
+export type MessageType = z.infer<typeof zodMessageSchema>;
 export declare enum FileType {
     PDF = "application/pdf",
     WORD = "application/vnd.openxmlformats-officedocument.wordprocessingml.document",

package/dist/types.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.QueuesList = exports.FileType = exports.zodQuerySchema = exports.zodPreDefinedQuerySchema = exports.zodFullTextSchema = exports.zodChunkSchema = exports.zodAgentSchema = exports.zodContextSchema = exports.zodUserSchema = exports.zodOrganizationSchema = exports.userRoles = exports.modelsList = void 0;
+exports.QueuesList = exports.FileType = exports.zodMessageSchema = exports.zodThreadSchema = exports.zodQuerySchema = exports.zodPreDefinedQuerySchema = exports.zodFullTextSchema = exports.zodChunkSchema = exports.zodAgentSchema = exports.zodContextSchema = exports.zodUserSchema = exports.zodOrganizationSchema = exports.userRoles = exports.modelsList = void 0;
 const zod_1 = require("zod");
 exports.modelsList = [
     'gemini-2.5-flash-lite',
@@ -122,22 +122,50 @@ exports.zodFullTextSchema = zod_1.z.object({
 exports.zodPreDefinedQuerySchema = zod_1.z.object({
     _id: zod_1.z.string().optional(),
     organization_id: zod_1.z.string().optional(),
-    title: zod_1.z.string().min(1).max(100).optional(),
+    query_name: zod_1.z.string().min(1).max(100).optional(),
+    query_hint: zod_1.z.string().min(1).max(100).optional(),
     query: zod_1.z.string().min(1),
-    context_ids: zod_1.z.array(zod_1.z.string()).optional(),
+    instructions: zod_1.z.string().optional(),
+    contexts: zod_1.z.array(exports.zodContextSchema.partial()).optional(),
     status: zod_1.z.enum(['active', 'inactive', 'archived']).default('active'),
-    users_ids: zod_1.z.array(zod_1.z.string()).optional(),
+    users: zod_1.z.array(exports.zodUserSchema.partial()).optional(),
+    click_count: zod_1.z.number().optional(),
     created_at: zod_1.z.number().optional(),
     updated_at: zod_1.z.number().optional(),
 });
+// Zod schema for handling user queries
 exports.zodQuerySchema = zod_1.z.object({
     thread_id: zod_1.z.string().optional(),
     context_ids: zod_1.z.array(zod_1.z.string()).optional(),
     organization_id: zod_1.z.string(),
     user_id: zod_1.z.string(),
     query: zod_1.z.string().min(1),
+    pre_defined_query_id: zod_1.z.string().optional(),
     timestamp: zod_1.z.number(),
 });
+exports.zodThreadSchema = zod_1.z.object({
+    _id: zod_1.z.string().optional(),
+    organization_id: zod_1.z.string(),
+    user_id: zod_1.z.string(),
+    agent_id: zod_1.z.string().optional(),
+    title: zod_1.z.string().min(1).max(200).optional(),
+    description: zod_1.z.string().optional(),
+    created_at: zod_1.z.number().optional(),
+    updated_at: zod_1.z.number().optional(),
+    query_id: zod_1.z.string().optional(),
+    messages_count: zod_1.z.number().optional(),
+});
+exports.zodMessageSchema = zod_1.z.object({
+    _id: zod_1.z.string().optional(),
+    thread_id: zod_1.z.string(),
+    organization_id: zod_1.z.string(),
+    user_id: zod_1.z.string().optional(),
+    agent_id: zod_1.z.string().optional(),
+    role: zod_1.z.enum(['user', 'agent', 'system']),
+    content: zod_1.z.string(),
+    created_at: zod_1.z.number().optional(),
+    updated_at: zod_1.z.number().optional(),
+});
 var FileType;
 (function (FileType) {
     FileType["PDF"] = "application/pdf";

package/index.ts

@@ -153,16 +153,19 @@ export const ChunkSchema = new Schema(
 export const PreDefinedQuerySchema = new Schema(
   {
     organization_id: { type: String },
-    title: { type: String, minlength: 1, maxlength: 100 },
+    query_name: { type: String, minlength: 1, maxlength: 100 },
+    query_hint: { type: String, minlength: 1, maxlength: 100 },
     query: { type: String, required: true },
-    context_ids: { type: [String] },
+    instructions: { type: String },
+    contexts: { type: [Object] },
     status: {
       type: String,
       enum: ['active', 'inactive', 'archived'],
       default: 'active',
       required: true,
     },
-    users_ids: { type: [String] },
+    users: { type: [Object] },
+    click_count: { type: Number },
     created_at: { type: Number },
     updated_at: { type: Number },
   },
@@ -190,4 +193,47 @@ export const FullFileTextSchema = new Schema(
     id: true,
     // encryptionType: 'csfle',
   }
+);
+
+export const ThreadSchema = new Schema(
+  {
+    organization_id: { type: String, required: true },
+    user_id: { type: String, required: true },
+    agent_id: { type: String, required: true },
+    title: { type: String, minlength: 1, maxlength: 200 },
+    description: { type: String },
+    query_id: { type: String },
+    messages_count: { type: Number, default: 0 },
+    created_at: { type: Number },
+    updated_at: { type: Number },
+  },
+  {
+    versionKey: false,
+    toJSON: { virtuals: true },
+    toObject: { virtuals: true },
+    timestamps: true,
+    id: true,
+    // encryptionType: 'csfle',
+  }
+);
+
+export const MessageSchema = new Schema(
+  {
+    thread_id: { type: String, required: true },
+    organization_id: { type: String, required: true },
+    user_id: { type: String },
+    agent_id: { type: String },
+    content: { type: String, required: true },
+    role: { type: String, enum: ['user', 'agent', 'system'], required: true },
+    created_at: { type: Number },
+    updated_at: { type: Number },
+  },
+  {
+    versionKey: false,
+    toJSON: { virtuals: true },
+    toObject: { virtuals: true },
+    timestamps: true,
+    id: true,
+    // encryptionType: 'csfle',
+  }
 );

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chatifai",
-  "version": "1.0.24",
+  "version": "1.0.26",
   "description": "SDK for interacting with the Chatifai API",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/types.ts

@@ -129,24 +129,54 @@ export const zodFullTextSchema = z.object({
 export const zodPreDefinedQuerySchema = z.object({
   _id: z.string().optional(),
   organization_id: z.string().optional(),
-  title: z.string().min(1).max(100).optional(),
+  query_name: z.string().min(1).max(100).optional(),
+  query_hint: z.string().min(1).max(100).optional(), // A short hint to help users understand the purpose of the query
   query: z.string().min(1),
-  context_ids: z.array(z.string()).optional(),
+  instructions: z.string().optional(),
+  contexts: z.array(zodContextSchema.partial()).optional(),
   status: z.enum(['active', 'inactive', 'archived']).default('active'),
-  users_ids: z.array(z.string()).optional(),
+  users: z.array(zodUserSchema.partial()).optional(),
+  click_count: z.number().optional(),
   created_at: z.number().optional(),
   updated_at: z.number().optional(),
 });
 
+// Zod schema for handling user queries
 export const zodQuerySchema = z.object({
   thread_id: z.string().optional(), // If there is not thread id provided, a new thread will be created
   context_ids: z.array(z.string()).optional(), // Run on all organization files unless specific file ids are provided
   organization_id: z.string(), // Will be used to fetch the agent, contexts, chunks, etc.
   user_id: z.string(),
   query: z.string().min(1),
+  pre_defined_query_id: z.string().optional(),
   timestamp: z.number(),
 })
 
+export const zodThreadSchema = z.object({
+  _id: z.string().optional(),
+  organization_id: z.string(),
+  user_id: z.string(),
+  agent_id: z.string().optional(),
+  title: z.string().min(1).max(200).optional(),
+  description: z.string().optional(),
+  created_at: z.number().optional(),
+  updated_at: z.number().optional(),
+  query_id: z.string().optional(),
+  messages_count: z.number().optional(),
+});
+
+export const zodMessageSchema = z.object({
+  _id: z.string().optional(),
+  thread_id: z.string(),
+  organization_id: z.string(),
+  user_id: z.string().optional(), // If user_id is not provided, it means the message is from the agent
+  agent_id: z.string().optional(),
+  role: z.enum(['user', 'agent', 'system']),
+  content: z.string(),
+  created_at: z.number().optional(),
+  updated_at: z.number().optional(),
+});
+
 export type EventAttributeType = {
   http_request: {
     client_ip?: string;
@@ -174,6 +204,8 @@ export type ModelType = (typeof modelsList)[number];
 export type UserRoleType = (typeof userRoles)[number];
 export type FullFileTextType = z.infer<typeof zodFullTextSchema>;
 export type QueryType = z.infer<typeof zodQuerySchema>;
+export type ThreadType = z.infer<typeof zodThreadSchema>;
+export type MessageType = z.infer<typeof zodMessageSchema>;
 
 export enum FileType {
   PDF = 'application/pdf',