@vectorx/agent-runtime 0.0.0-beta-20251112071515

package/lib/schema/message.schema.js

@@ -0,0 +1,79 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sendMessageInputSchema = exports.messageHistorySchema = exports.messageItemSchema = exports.toolResultMessageSchema = exports.toolCallMessageSchema = exports.baseMessageSchema = exports.messageContentSchema = exports.docUrlContentSchema = exports.videoUrlContentSchema = exports.videoContentSchema = exports.audioContentSchema = exports.imageContentSchema = exports.textContentSchema = void 0;
+const zod_1 = require("zod");
+exports.textContentSchema = zod_1.z.object({
+    type: zod_1.z.literal("text"),
+    text: zod_1.z.string(),
+});
+exports.imageContentSchema = zod_1.z.object({
+    type: zod_1.z.literal("image_url"),
+    image_url: zod_1.z.object({
+        url: zod_1.z.string(),
+    }),
+});
+exports.audioContentSchema = zod_1.z.object({
+    type: zod_1.z.literal("input_audio"),
+    input_audio: zod_1.z.object({
+        data: zod_1.z.string(),
+        format: zod_1.z.string(),
+    }),
+});
+exports.videoContentSchema = zod_1.z.object({
+    type: zod_1.z.literal("video"),
+    video: zod_1.z.array(zod_1.z.string()),
+});
+exports.videoUrlContentSchema = zod_1.z.object({
+    type: zod_1.z.literal("video_url"),
+    video_url: zod_1.z.object({
+        url: zod_1.z.string(),
+    }),
+});
+exports.docUrlContentSchema = zod_1.z.object({
+    type: zod_1.z.literal("doc_url"),
+    doc_url: zod_1.z.array(zod_1.z.string()),
+    file_parsing_strategy: zod_1.z.string().optional(),
+});
+exports.messageContentSchema = zod_1.z.union([
+    exports.textContentSchema,
+    exports.imageContentSchema,
+    exports.audioContentSchema,
+    exports.videoContentSchema,
+    exports.videoUrlContentSchema,
+    exports.docUrlContentSchema,
+]);
+exports.baseMessageSchema = zod_1.z.object({
+    role: zod_1.z.union([zod_1.z.literal("system"), zod_1.z.literal("user"), zod_1.z.literal("assistant")]),
+    content: zod_1.z.union([
+        zod_1.z.string(),
+        zod_1.z.array(exports.messageContentSchema),
+    ]),
+});
+exports.toolCallMessageSchema = zod_1.z.object({
+    role: zod_1.z.literal("tool"),
+    function: zod_1.z.string(),
+    arguments: zod_1.z.record(zod_1.z.any()).optional(),
+});
+exports.toolResultMessageSchema = zod_1.z.object({
+    role: zod_1.z.literal("tool"),
+    function: zod_1.z.string(),
+    name: zod_1.z.string(),
+    content: zod_1.z.union([zod_1.z.string(), zod_1.z.record(zod_1.z.any()), zod_1.z.array(zod_1.z.any())]),
+});
+exports.messageItemSchema = zod_1.z.union([exports.baseMessageSchema, exports.toolCallMessageSchema, exports.toolResultMessageSchema]);
+exports.messageHistorySchema = zod_1.z.array(exports.messageItemSchema);
+exports.sendMessageInputSchema = zod_1.z.object({
+    msg: zod_1.z.string(),
+    history: exports.messageHistorySchema,
+    files: zod_1.z
+        .array(zod_1.z
+        .object({
+        id: zod_1.z.string(),
+        filename: zod_1.z.string(),
+        purpose: zod_1.z.string(),
+        bytes: zod_1.z.number(),
+        created_at: zod_1.z.number(),
+    })
+        .passthrough())
+        .optional(),
+});

package/lib/schema/task.schema.d.ts

@@ -0,0 +1,9 @@
+import { z } from "zod";
+export declare const queryTasksParamsSchema: z.ZodObject<{
+    task_id: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    task_id?: string;
+}, {
+    task_id?: string;
+}>;
+export type QueryTasksParams = z.infer<typeof queryTasksParamsSchema>;

package/lib/schema/task.schema.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.queryTasksParamsSchema = void 0;
+const zod_1 = require("zod");
+exports.queryTasksParamsSchema = zod_1.z.object({
+    task_id: zod_1.z.string().min(1),
+});

package/lib/sse-sender.d.ts

@@ -0,0 +1,32 @@
+import type { EventEmitter } from "events";
+import type { RcbContext, ServerSentEvent } from "@vectorx/functions-framework";
+export type Primitive = number | string | boolean | undefined | null;
+export type BufferLike = number | string | Buffer | ArrayBuffer | ArrayBufferView | DataView;
+export declare namespace sse {
+    interface SseEvent<T = Primitive | object | Buffer> {
+        data: T;
+        comment?: string;
+        event?: string;
+        id?: string;
+        retry?: number;
+    }
+    type Event<T> = SseEvent<T> | string | Buffer;
+    interface IServerSentEvent extends EventEmitter {
+        closed: boolean;
+        setEncoder(encoder: TextEncoder): void;
+        send<T>(event: Event<T> | Event<T>[]): boolean;
+        end(msg: SseEvent): void;
+        on(event: "close", listener: () => void): this;
+        on(event: "error", listener: (errorEvent: SseEvent) => void): this;
+    }
+}
+export declare class SSESender<T> {
+    readonly context: RcbContext;
+    private _status;
+    private _sse;
+    get status(): "initial" | "started" | "ended";
+    constructor(context: RcbContext);
+    get sse(): ServerSentEvent;
+    send(data: sse.Event<T>): void;
+    end(): void;
+}

package/lib/sse-sender.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SSESender = void 0;
+class SSESender {
+    get status() {
+        return this._status;
+    }
+    constructor(context) {
+        this.context = context;
+        this._status = "initial";
+        this._sse = null;
+    }
+    get sse() {
+        var _a, _b;
+        if (!this._sse) {
+            const sse = (_b = (_a = this.context) === null || _a === void 0 ? void 0 : _a.sse) === null || _b === void 0 ? void 0 : _b.call(_a);
+            if (!sse) {
+                throw new Error("Invalid sse");
+            }
+            this._sse = sse;
+            this._status = "started";
+        }
+        return this._sse;
+    }
+    send(data) {
+        this.sse.send(data);
+    }
+    end() {
+        this.sse.end({ data: "[DONE]" });
+        this._status = "ended";
+    }
+}
+exports.SSESender = SSESender;

package/lib/types.d.ts

@@ -0,0 +1,71 @@
+import type { SSESender } from "./sse-sender";
+export interface SendMessageInput {
+    msg: string;
+    history?: {
+        role: "user" | "assistant";
+        content: string;
+    }[];
+    searchEnable?: boolean;
+}
+export interface GetLoneMemoryInput {
+    userId: string;
+    limit?: number;
+    filter?: {
+        startTime?: number;
+        endTime?: number;
+        type?: string;
+    };
+}
+export interface KnowledgeSearchInput {
+    collectionView: string;
+    search: {
+        content: string;
+        documentSetName?: string[];
+        options?: {
+            chunkExpand?: number[];
+            filter?: string;
+            limit?: number;
+        };
+    };
+}
+export interface UserProfile {
+    tags?: string[];
+    preferences?: Record<string, any>;
+}
+export interface DocumentSet {
+    documentSetId: string;
+    documentSetName: string;
+    author: string;
+    fileTitle: string;
+    fileMetaData: string;
+}
+export interface DocumentData {
+    text: string;
+    startPos: number;
+    endPos: number;
+    pre: string[];
+    next: string[];
+    paragraphTitle: string;
+    allParentParagraphTitles: string[];
+    documentSet: DocumentSet;
+}
+export interface KnowledgeSearchResult {
+    score: number;
+    data: DocumentData;
+}
+export interface KnowledgeSearchResponse {
+    data: {
+        documents: KnowledgeSearchResult[];
+        requestId: string;
+    };
+}
+export interface ErrorResponse {
+    code: string;
+    message: string;
+    requestId: string;
+}
+export type AgentCoreSseSender = SSESender<{
+    content: string;
+    [key: string]: unknown;
+}>;
+export type PickRequired<T, K extends keyof T> = T & Required<Pick<T, K>>;

package/lib/types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/lib/utils/agent-helper.d.ts

@@ -0,0 +1,9 @@
+import type { RcbContext } from "@vectorx/functions-framework";
+import queryString from "query-string";
+export declare function parseAgentId(url: string): string;
+export declare function parseAgentTag(url: string): string;
+export declare function parseApiName(url: string): string;
+export declare function parseQueryFromCtx(ctx: RcbContext): queryString.ParsedQuery<string | boolean>;
+export declare function parseTaskIdFromPath(url: string): string | null;
+export declare function genRecordId(): string;
+export declare function genRandomStr(length: number): string;

package/lib/utils/agent-helper.js

@@ -0,0 +1,111 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.parseAgentId = parseAgentId;
+exports.parseAgentTag = parseAgentTag;
+exports.parseApiName = parseApiName;
+exports.parseQueryFromCtx = parseQueryFromCtx;
+exports.parseTaskIdFromPath = parseTaskIdFromPath;
+exports.genRecordId = genRecordId;
+exports.genRandomStr = genRandomStr;
+const crypto = __importStar(require("crypto"));
+const query_string_1 = __importDefault(require("query-string"));
+const AGENT_REG = /\/v1\/aiagent\/agents\/([^\/]+)\/(.+)/;
+function parseAgentId(url) {
+    return parseUrl(url).agentId;
+}
+function parseAgentTag(url) {
+    const agentId = parseAgentId(url);
+    const agentTag = agentId.split("-")[2];
+    return agentTag;
+}
+function parseApiName(url) {
+    try {
+        return parseUrl(url).apiName;
+    }
+    catch (e) {
+        return null;
+    }
+}
+function parseUrl(url) {
+    const { pathname } = new URL(url);
+    const regResult = pathname.match(AGENT_REG);
+    if (!regResult) {
+        throw new Error("Invalid pathname");
+    }
+    const [_, agentId, remainingPath] = regResult;
+    let apiName = remainingPath;
+    if (remainingPath.startsWith("query-tasks/")) {
+        apiName = "query-tasks";
+    }
+    return { agentId, apiName };
+}
+function parseQueryFromCtx(ctx) {
+    var _a;
+    const url = (_a = ctx.httpContext) === null || _a === void 0 ? void 0 : _a.url;
+    if (!url) {
+        throw new Error("Invalid url: " + url);
+    }
+    const [_, queryStr] = url.split("?");
+    return query_string_1.default.parse(queryStr, {
+        parseBooleans: true,
+        parseFragmentIdentifier: true,
+    });
+}
+function parseTaskIdFromPath(url) {
+    try {
+        const { pathname } = new URL(url);
+        const parts = pathname.split("/").filter(Boolean);
+        const agentsIdx = parts.findIndex((p) => p === "agents");
+        if (agentsIdx >= 0 && parts[agentsIdx + 2] === "query-tasks" && parts[agentsIdx + 3]) {
+            return parts[agentsIdx + 3];
+        }
+        return null;
+    }
+    catch (_a) {
+        return null;
+    }
+}
+function genRecordId() {
+    return "record-" + genRandomStr(8);
+}
+function genRandomStr(length) {
+    const randomBytes = crypto.randomBytes(Math.ceil(length / 2));
+    const hexString = randomBytes.toString("hex").slice(0, length);
+    return hexString;
+}

package/lib/utils/index.d.ts

@@ -0,0 +1 @@
+export * from "./agent-helper";

package/lib/utils/index.js

@@ -0,0 +1,17 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+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 });
+__exportStar(require("./agent-helper"), exports);

package/lib/utils/integration-response.d.ts

@@ -0,0 +1,20 @@
+export interface IntegrationResponse {
+    statusCode: number;
+    headers: Record<string, string>;
+    body: {
+        code: string;
+        message: string;
+    };
+}
+export interface ErrorResponse {
+    code: string;
+    message: string;
+    requestId: string;
+}
+export interface BaseResponse<T = any> {
+    code: number;
+    msg: string;
+    data?: T;
+}
+export declare function createIntegrationResponse(statusCode: number, code: string, message: string): IntegrationResponse;
+export declare const OK_MESSAGE = "OK";

package/lib/utils/integration-response.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OK_MESSAGE = void 0;
+exports.createIntegrationResponse = createIntegrationResponse;
+function createIntegrationResponse(statusCode, code, message) {
+    return {
+        statusCode,
+        headers: {
+            "Content-Type": "application/json",
+        },
+        body: {
+            code,
+            message,
+        },
+    };
+}
+exports.OK_MESSAGE = "OK";

package/package.json

@@ -0,0 +1,47 @@
+{
+  "name": "@vectorx/agent-runtime",
+  "version": "0.0.0-beta-20251112071515",
+  "description": "Cloud AI agent runtime",
+  "main": "lib/index.js",
+  "files": [
+    "lib",
+    "types",
+    "README.md"
+  ],
+  "keywords": [
+    "ai",
+    "agent",
+    "cloud",
+    "framework"
+  ],
+  "author": "",
+  "license": "ISC",
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "dependencies": {
+    "@vectorx/ai-sdk": "0.0.0-beta-20251112071515",
+    "@vectorx/functions-framework": "0.0.0-beta-20251112071515",
+    "form-data": "^4.0.0",
+    "langfuse": "^3.38.4",
+    "query-string": "^6.14.1",
+    "zod": "^3.24.2"
+  },
+  "devDependencies": {
+    "@types/form-data": "^2.5.0",
+    "@types/jest": "^29.5.12",
+    "@types/node": "^20.11.24",
+    "@typescript-eslint/eslint-plugin": "^7.1.0",
+    "@typescript-eslint/parser": "^7.1.0",
+    "eslint": "^8.57.0",
+    "jest": "^29.7.0",
+    "ts-node-dev": "^2.0.0",
+    "typescript": "^5.3.3"
+  },
+  "scripts": {
+    "dev": "tsc -w",
+    "build": "rimraf lib types && tsc",
+    "lint": "eslint 'src/**/*.ts'",
+    "fix": "eslint 'src/**/*.ts' --fix"
+  }
+}