chatifai 1.0.23 → 1.0.25

package/dist/index.d.ts

@@ -322,6 +322,8 @@ export declare const ChunkSchema: Schema<any, import("mongoose").Model<any, any,
     created_at?: number | null | undefined;
     updated_at?: number | null | undefined;
     chunk_id?: string | null | undefined;
+    chunk_id_before?: string | null | undefined;
+    chunk_id_after?: string | null | undefined;
     embeddings_meta_data?: {
         model: string;
         dimensions: number;
@@ -350,6 +352,8 @@ export declare const ChunkSchema: Schema<any, import("mongoose").Model<any, any,
     created_at?: number | null | undefined;
     updated_at?: number | null | undefined;
     chunk_id?: string | null | undefined;
+    chunk_id_before?: string | null | undefined;
+    chunk_id_after?: string | null | undefined;
     embeddings_meta_data?: {
         model: string;
         dimensions: number;
@@ -378,6 +382,8 @@ export declare const ChunkSchema: Schema<any, import("mongoose").Model<any, any,
     created_at?: number | null | undefined;
     updated_at?: number | null | undefined;
     chunk_id?: string | null | undefined;
+    chunk_id_before?: string | null | undefined;
+    chunk_id_after?: string | null | undefined;
     embeddings_meta_data?: {
         model: string;
         dimensions: number;
@@ -481,3 +487,108 @@ export declare const FullFileTextSchema: Schema<any, import("mongoose").Model<an
 } & {
     __v: number;
 }>;
+export declare const ThreadSchema: Schema<any, import("mongoose").Model<any, any, any, any, any, any>, {}, {}, {}, {}, {
+    versionKey: false;
+    toJSON: {
+        virtuals: true;
+    };
+    toObject: {
+        virtuals: true;
+    };
+    timestamps: true;
+    id: true;
+}, {
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
+    organization_id: string;
+    user_id: string;
+    agent_id: string;
+    messages_count: number;
+    title?: string | null | undefined;
+    created_at?: number | null | undefined;
+    updated_at?: number | null | undefined;
+    description?: string | null | undefined;
+    query_id?: string | null | undefined;
+}, import("mongoose").Document<unknown, {}, import("mongoose").FlatRecord<{
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
+    organization_id: string;
+    user_id: string;
+    agent_id: string;
+    messages_count: number;
+    title?: string | null | undefined;
+    created_at?: number | null | undefined;
+    updated_at?: number | null | undefined;
+    description?: string | null | undefined;
+    query_id?: string | null | undefined;
+}>, {}> & import("mongoose").FlatRecord<{
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
+    organization_id: string;
+    user_id: string;
+    agent_id: string;
+    messages_count: number;
+    title?: string | null | undefined;
+    created_at?: number | null | undefined;
+    updated_at?: number | null | undefined;
+    description?: string | null | undefined;
+    query_id?: string | null | undefined;
+}> & {
+    _id: import("mongoose").Types.ObjectId;
+} & {
+    __v: number;
+}>;
+export declare const MessageSchema: Schema<any, import("mongoose").Model<any, any, any, any, any, any>, {}, {}, {}, {}, {
+    versionKey: false;
+    toJSON: {
+        virtuals: true;
+    };
+    toObject: {
+        virtuals: true;
+    };
+    timestamps: true;
+    id: true;
+}, {
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
+    organization_id: string;
+    thread_id: string;
+    role: "user" | "agent" | "system";
+    content: string;
+    created_at?: number | null | undefined;
+    updated_at?: number | null | undefined;
+    user_id?: string | null | undefined;
+    agent_id?: string | null | undefined;
+}, import("mongoose").Document<unknown, {}, import("mongoose").FlatRecord<{
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
+    organization_id: string;
+    thread_id: string;
+    role: "user" | "agent" | "system";
+    content: string;
+    created_at?: number | null | undefined;
+    updated_at?: number | null | undefined;
+    user_id?: string | null | undefined;
+    agent_id?: string | null | undefined;
+}>, {}> & import("mongoose").FlatRecord<{
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
+    organization_id: string;
+    thread_id: string;
+    role: "user" | "agent" | "system";
+    content: string;
+    created_at?: number | null | undefined;
+    updated_at?: number | null | undefined;
+    user_id?: string | null | undefined;
+    agent_id?: string | null | undefined;
+}> & {
+    _id: import("mongoose").Types.ObjectId;
+} & {
+    __v: number;
+}>;

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");
@@ -112,6 +112,8 @@ exports.ContextSchema = new mongoose_1.Schema({
 });
 exports.ChunkSchema = new mongoose_1.Schema({
     chunk_id: { type: String },
+    chunk_id_before: { type: String },
+    chunk_id_after: { type: String },
     organization_id: { type: String, required: true },
     context_id: { type: String, required: true },
     text: { type: String, required: true },
@@ -177,3 +179,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

@@ -202,6 +202,8 @@ export declare const zodAgentSchema: z.ZodObject<{
 export declare const zodChunkSchema: z.ZodObject<{
     _id: z.ZodOptional<z.ZodString>;
     chunk_id: z.ZodOptional<z.ZodString>;
+    chunk_id_before: z.ZodOptional<z.ZodString>;
+    chunk_id_after: z.ZodOptional<z.ZodString>;
     organization_id: z.ZodString;
     context_id: z.ZodString;
     text: z.ZodString;
@@ -269,6 +271,8 @@ export declare const zodChunkSchema: z.ZodObject<{
     created_at?: number | undefined;
     updated_at?: number | undefined;
     chunk_id?: string | undefined;
+    chunk_id_before?: string | undefined;
+    chunk_id_after?: string | undefined;
 }, {
     type: string;
     organization_id: string;
@@ -295,6 +299,8 @@ export declare const zodChunkSchema: z.ZodObject<{
     created_at?: number | undefined;
     updated_at?: number | undefined;
     chunk_id?: string | undefined;
+    chunk_id_before?: string | undefined;
+    chunk_id_after?: string | undefined;
 }>;
 export declare const zodFullTextSchema: z.ZodObject<{
     _id: z.ZodOptional<z.ZodString>;
@@ -349,6 +355,7 @@ 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;
@@ -357,6 +364,7 @@ export declare const zodQuerySchema: z.ZodObject<{
     timestamp: number;
     context_ids?: string[] | undefined;
     thread_id?: string | undefined;
+    pre_defined_query_id?: string | undefined;
 }, {
     organization_id: string;
     query: string;
@@ -364,6 +372,72 @@ export declare const zodQuerySchema: z.ZodObject<{
     timestamp: number;
     context_ids?: string[] | undefined;
     thread_id?: 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: {
@@ -390,6 +464,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',
@@ -87,6 +87,8 @@ exports.zodAgentSchema = zod_1.z.object({
 exports.zodChunkSchema = zod_1.z.object({
     _id: zod_1.z.string().optional(),
     chunk_id: zod_1.z.string().optional(),
+    chunk_id_before: zod_1.z.string().optional(),
+    chunk_id_after: zod_1.z.string().optional(),
     organization_id: zod_1.z.string(),
     context_id: zod_1.z.string(),
     text: zod_1.z.string(),
@@ -128,14 +130,39 @@ exports.zodPreDefinedQuerySchema = zod_1.z.object({
     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

@@ -114,6 +114,8 @@ export const ContextSchema = new Schema(
 export const ChunkSchema = new Schema(
   {
     chunk_id: { type: String },
+    chunk_id_before: { type: String },
+    chunk_id_after: { type: String },
     organization_id: { type: String, required: true },
     context_id: { type: String, required: true },
     text: { type: String, required: true },
@@ -188,4 +190,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.23",
+  "version": "1.0.25",
   "description": "SDK for interacting with the Chatifai API",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/types.ts

@@ -92,6 +92,8 @@ export const zodAgentSchema = z.object({
 export const zodChunkSchema = z.object({
   _id: z.string().optional(),
   chunk_id: z.string().optional(),
+  chunk_id_before: z.string().optional(),
+  chunk_id_after: z.string().optional(),
   organization_id: z.string(),
   context_id: z.string(),
   text: z.string(),
@@ -136,15 +138,42 @@ export const zodPreDefinedQuerySchema = z.object({
   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;
@@ -172,6 +201,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',