@ainetwork/adk-provider-memory-mongodb 0.1.6 → 0.1.8

package/dist/models/chats.model.d.cts

@@ -1,105 +0,0 @@
-import * as mongoose from 'mongoose';
-import { Schema, Document } from 'mongoose';
-
-declare enum ChatRole {
-    USER = "USER",
-    SYSTEM = "SYSTEM",
-    MODEL = "MODEL"
-}
-declare const SessionObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
-    sessionId: string;
-    userId: string;
-    created_at: number;
-    updated_at: number;
-    title?: string | null | undefined;
-}, Document<unknown, {}, mongoose.FlatRecord<{
-    sessionId: string;
-    userId: string;
-    created_at: number;
-    updated_at: number;
-    title?: string | null | undefined;
-}>, {}> & mongoose.FlatRecord<{
-    sessionId: string;
-    userId: string;
-    created_at: number;
-    updated_at: number;
-    title?: string | null | undefined;
-}> & {
-    _id: mongoose.Types.ObjectId;
-} & {
-    __v: number;
-}>;
-interface SessionDocument extends Document {
-    sessionId: string;
-    userId: string;
-    title?: string;
-    created_at: number;
-    updated_at: number;
-}
-declare const ChatContentObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, {
-    _id: false;
-}, {
-    type: string;
-    parts: any[];
-}, Document<unknown, {}, mongoose.FlatRecord<{
-    type: string;
-    parts: any[];
-}>, {}> & mongoose.FlatRecord<{
-    type: string;
-    parts: any[];
-}> & {
-    _id: mongoose.Types.ObjectId;
-} & {
-    __v: number;
-}>;
-declare const ChatObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
-    sessionId: string;
-    userId: string;
-    role: ChatRole;
-    content: {
-        type: string;
-        parts: any[];
-    };
-    timestamp: number;
-    metadata: any;
-}, Document<unknown, {}, mongoose.FlatRecord<{
-    sessionId: string;
-    userId: string;
-    role: ChatRole;
-    content: {
-        type: string;
-        parts: any[];
-    };
-    timestamp: number;
-    metadata: any;
-}>, {}> & mongoose.FlatRecord<{
-    sessionId: string;
-    userId: string;
-    role: ChatRole;
-    content: {
-        type: string;
-        parts: any[];
-    };
-    timestamp: number;
-    metadata: any;
-}> & {
-    _id: mongoose.Types.ObjectId;
-} & {
-    __v: number;
-}>;
-interface ChatDocument extends Document {
-    sessionId: string;
-    role: ChatRole;
-    content: {
-        type: string;
-        parts: any[];
-    };
-    timestamp: number;
-    metadata?: {
-        [key: string]: unknown;
-    };
-    createdAt: Date;
-    updatedAt: Date;
-}
-
-export { ChatContentObjectSchema, type ChatDocument, ChatObjectSchema, ChatRole, type SessionDocument, SessionObjectSchema };

package/dist/models/chats.model.d.ts

@@ -1,105 +0,0 @@
-import * as mongoose from 'mongoose';
-import { Schema, Document } from 'mongoose';
-
-declare enum ChatRole {
-    USER = "USER",
-    SYSTEM = "SYSTEM",
-    MODEL = "MODEL"
-}
-declare const SessionObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
-    sessionId: string;
-    userId: string;
-    created_at: number;
-    updated_at: number;
-    title?: string | null | undefined;
-}, Document<unknown, {}, mongoose.FlatRecord<{
-    sessionId: string;
-    userId: string;
-    created_at: number;
-    updated_at: number;
-    title?: string | null | undefined;
-}>, {}> & mongoose.FlatRecord<{
-    sessionId: string;
-    userId: string;
-    created_at: number;
-    updated_at: number;
-    title?: string | null | undefined;
-}> & {
-    _id: mongoose.Types.ObjectId;
-} & {
-    __v: number;
-}>;
-interface SessionDocument extends Document {
-    sessionId: string;
-    userId: string;
-    title?: string;
-    created_at: number;
-    updated_at: number;
-}
-declare const ChatContentObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, {
-    _id: false;
-}, {
-    type: string;
-    parts: any[];
-}, Document<unknown, {}, mongoose.FlatRecord<{
-    type: string;
-    parts: any[];
-}>, {}> & mongoose.FlatRecord<{
-    type: string;
-    parts: any[];
-}> & {
-    _id: mongoose.Types.ObjectId;
-} & {
-    __v: number;
-}>;
-declare const ChatObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
-    sessionId: string;
-    userId: string;
-    role: ChatRole;
-    content: {
-        type: string;
-        parts: any[];
-    };
-    timestamp: number;
-    metadata: any;
-}, Document<unknown, {}, mongoose.FlatRecord<{
-    sessionId: string;
-    userId: string;
-    role: ChatRole;
-    content: {
-        type: string;
-        parts: any[];
-    };
-    timestamp: number;
-    metadata: any;
-}>, {}> & mongoose.FlatRecord<{
-    sessionId: string;
-    userId: string;
-    role: ChatRole;
-    content: {
-        type: string;
-        parts: any[];
-    };
-    timestamp: number;
-    metadata: any;
-}> & {
-    _id: mongoose.Types.ObjectId;
-} & {
-    __v: number;
-}>;
-interface ChatDocument extends Document {
-    sessionId: string;
-    role: ChatRole;
-    content: {
-        type: string;
-        parts: any[];
-    };
-    timestamp: number;
-    metadata?: {
-        [key: string]: unknown;
-    };
-    createdAt: Date;
-    updatedAt: Date;
-}
-
-export { ChatContentObjectSchema, type ChatDocument, ChatObjectSchema, ChatRole, type SessionDocument, SessionObjectSchema };

package/dist/models/chats.model.js

@@ -1,13 +0,0 @@
-import {
-  ChatContentObjectSchema,
-  ChatObjectSchema,
-  ChatRole,
-  SessionObjectSchema
-} from "../chunk-N5FAVJ4Z.js";
-export {
-  ChatContentObjectSchema,
-  ChatObjectSchema,
-  ChatRole,
-  SessionObjectSchema
-};
-//# sourceMappingURL=chats.model.js.map

package/implements/session.memory.ts

@@ -1,104 +0,0 @@
-import { randomUUID } from "node:crypto";
-import type { ChatObject, SessionMetadata, SessionObject } from "@ainetwork/adk/types/memory";
-import { ISessionMemory } from "@ainetwork/adk/modules";
-import { MongoDBMemory } from "./base.memory";
-import {
-	ChatDocument,
-	ChatRole,
-  ChatObjectSchema,
-  SessionObjectSchema,
-  SessionDocument
-} from "../models/chats.model";
-import { loggers } from "@ainetwork/adk/utils/logger";
-import { Model } from "mongoose";
-
-export class MongoDBSession extends MongoDBMemory implements ISessionMemory {
-  private chatModel: Model<ChatDocument>;
-  private sessionModel: Model<SessionDocument>;
-
-  constructor(uri: string) {
-    super(uri);
-    const _mongoose = super.getInstance();
-    this.chatModel = _mongoose.model<ChatDocument>("Chat", ChatObjectSchema);
-    this.sessionModel = _mongoose.model<SessionDocument>("Session", SessionObjectSchema);
-  }
-
-  public async getSession(userId: string, sessionId: string): Promise<SessionObject | undefined> {
-		const chats = await this.chatModel.find({ sessionId, userId }).sort({
-			timestamp: 1,
-		});
-    const session = await this.sessionModel.findOne({ sessionId, userId });
-
-		loggers.agent.debug(`Found ${chats.length} chats for session ${sessionId}`);
-
-		const sessionObject: SessionObject = { chats: {} };
-    sessionObject.title = session?.title;
-		chats.forEach((chat: ChatDocument) => {
-			const chatId = chat._id?.toString() || chat.id;
-			sessionObject.chats[chatId] = {
-				role: chat.role as ChatRole,
-				content: chat.content,
-				timestamp: chat.timestamp,
-				metadata: chat.metadata,
-			};
-		});
-
-		return sessionObject;
-  };
-
-	public async createSession(userId: string, sessionId: string, title: string): Promise<SessionMetadata> {
-    const now = Date.now();
-    await this.sessionModel.create({
-      sessionId,
-      userId,
-      title,
-      updated_at: now,
-      created_at: now,
-    });
-
-    return { title, sessionId, updatedAt: now };
-  };
-
-	public async addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void> {
-    const newId = randomUUID();
-    await this.sessionModel.updateOne({ sessionId, userId }, {
-      updated_at: Date.now(),
-    });
-		await this.chatModel.create({
-			sessionId,
-      chatId: newId,
-      userId,
-			role: chat.role,
-			content: chat.content,
-			timestamp: chat.timestamp,
-			metadata: chat.metadata,
-		});
-  };
-
-	public async deleteSession(userId: string, sessionId: string): Promise<void> {
-		const chats = await this.chatModel.find({ userId, sessionId }).sort({
-			timestamp: 1,
-		});
-
-		chats?.forEach((chat: ChatDocument) => {
-      chat.deleteOne();
-		});
-    
-    const session = await this.sessionModel.findOne({ sessionId, userId });
-    session?.deleteOne();
-  };
-
-	public async listSessions(userId: string): Promise<SessionMetadata[]> {
-    const sessions = await this.sessionModel.find({ userId }).sort({
-      updated_at: -1,
-    });
-    const data: SessionMetadata[] = sessions.map((session: SessionDocument) => {
-      return {
-        sessionId: session.sessionId,
-        title: session.title,
-        updatedAt: session.updated_at
-      } as SessionMetadata;
-    })
-    return data;
-  };
-}

package/models/chats.model.ts

@@ -1,99 +0,0 @@
-import { type Document, Schema } from "mongoose";
-
-// ChatRole enum
-export enum ChatRole {
-	USER = "USER",
-	SYSTEM = "SYSTEM",
-	MODEL = "MODEL",
-}
-
-export const SessionObjectSchema = new Schema(
-	{
-		sessionId: {
-			type: String,
-			required: true,
-			index: true,
-		},
-		userId: {
-			type: String,
-			required: true,
-			index: true,
-		},
-		title: {
-			type: String,
-			required: false,
-		},
-		created_at: {
-			type: Number,
-			required: true,
-		},
-		updated_at: {
-			type: Number,
-			required: true,
-		}
-	},
-);
-
-export interface SessionDocument extends Document {
-	sessionId: string;
-	userId: string;
-	title?: string;
-	created_at: number;
-	updated_at: number;
-}
-
-// ChatContentObject schema
-export const ChatContentObjectSchema = new Schema(
-	{
-		type: { type: String, required: true },
-		parts: { type: [Schema.Types.Mixed], required: true },
-	},
-	{ _id: false },
-);
-
-// ChatObject schema - 개별 문서로 저장
-export const ChatObjectSchema = new Schema(
-	{
-		sessionId: {
-			type: String,
-			required: true,
-			index: true,
-		},
-		userId: {
-			type: String,
-			required: true,
-			index: true,
-		},
-		role: {
-			type: String,
-			enum: Object.values(ChatRole),
-			required: true,
-		},
-		content: {
-			type: ChatContentObjectSchema,
-			required: true,
-		},
-		timestamp: {
-			type: Number,
-			required: true,
-		},
-		metadata: {
-			type: Schema.Types.Mixed,
-			default: {},
-		},
-	},
-);
-
-// Chat Document interface
-export interface ChatDocument extends Document {
-	sessionId: string;
-	role: ChatRole;
-	content: {
-		type: string;
-		parts: any[];
-	};
-	timestamp: number;
-	metadata?: { [key: string]: unknown };
-	createdAt: Date;
-	updatedAt: Date;
-}