adk-typescript/dist/artifacts/InMemoryArtifactService.js

TypeScript port of Google's Agent Development Kit (ADK)

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.InMemoryArtifactService = void 0;
/**
 * An in-memory implementation of the artifact service.
 */
class InMemoryArtifactService {
    constructor() {
        this.artifacts = new Map();
    }
    /**
     * Checks if the filename has a user namespace.
     *
     * @param filename The filename to check
     * @returns True if the filename has a user namespace (starts with "user:"), false otherwise
     */
    fileHasUserNamespace(filename) {
        return filename.startsWith('user:');
    }
    /**
     * Constructs the artifact path.
     *
     * @param appName The name of the application
     * @param userId The ID of the user
     * @param sessionId The ID of the session
     * @param filename The name of the artifact file
     * @returns The constructed artifact path
     */
    artifactPath(appName, userId, sessionId, filename) {
        if (this.fileHasUserNamespace(filename)) {
            return `${appName}/${userId}/user/${filename}`;
        }
        return `${appName}/${userId}/${sessionId}/${filename}`;
    }
    /**
     * Saves an artifact to the artifact service storage.
     *
     * @param params The artifact parameters
     * @returns The revision ID
     */
    saveArtifact(params) {
        const { appName, userId, sessionId, filename, artifact } = params;
        if (!artifact) {
            throw new Error('Cannot save empty artifact');
        }
        const path = this.artifactPath(appName, userId, sessionId, filename);
        if (!this.artifacts.has(path)) {
            this.artifacts.set(path, []);
        }
        const versions = this.artifacts.get(path);
        const version = versions.length;
        versions.push(artifact);
        return version;
    }
    /**
     * Gets an artifact from the artifact service storage.
     *
     * @param params The artifact parameters
     * @returns The artifact or undefined if not found
     */
    loadArtifact(params) {
        const { appName, userId, sessionId, filename, version } = params;
        const path = this.artifactPath(appName, userId, sessionId, filename);
        const versions = this.artifacts.get(path);
        if (!versions || versions.length === 0) {
            return undefined;
        }
        const versionIndex = version !== undefined ? version : versions.length - 1;
        return versions[versionIndex];
    }
    /**
     * Lists all the artifact filenames within a session.
     *
     * @param params The artifact parameters
     * @returns A list of all artifact filenames within a session
     */
    listArtifactKeys(params) {
        const { appName, userId, sessionId } = params;
        const sessionPrefix = `${appName}/${userId}/${sessionId}/`;
        const userNamespacePrefix = `${appName}/${userId}/user/`;
        const filenames = [];
        for (const path of this.artifacts.keys()) {
            if (path.startsWith(sessionPrefix)) {
                const filename = path.substring(sessionPrefix.length);
                filenames.push(filename);
            }
            else if (path.startsWith(userNamespacePrefix)) {
                const filename = path.substring(userNamespacePrefix.length);
                filenames.push(filename);
            }
        }
        return filenames.sort();
    }
    /**
     * Deletes an artifact.
     *
     * @param params The artifact parameters
     */
    async deleteArtifact(params) {
        const { appName, userId, sessionId, filename } = params;
        const path = this.artifactPath(appName, userId, sessionId, filename);
        this.artifacts.delete(path);
    }
    /**
     * Lists all versions of an artifact.
     *
     * @param params The artifact parameters
     * @returns A list of all available versions of the artifact
     */
    listVersions(params) {
        const { appName, userId, sessionId, filename } = params;
        const path = this.artifactPath(appName, userId, sessionId, filename);
        const versions = this.artifacts.get(path);
        if (!versions || versions.length === 0) {
            return [];
        }
        return Array.from({ length: versions.length }, (_, i) => i);
    }
}
exports.InMemoryArtifactService = InMemoryArtifactService;