@vectorx/agent-runtime 0.1.3 → 0.4.0

package/{types → lib}/agent-runner.d.ts

@@ -9,4 +9,6 @@ export declare class AgentDriver {
     private getHistoryMessagesFunc;
     private getAgentInfoFunc;
     private getConversationsFunc;
+    private queryTasksFunc;
+    private uploadFileFunc;
 }

package/lib/agent-runner.js

@@ -30,6 +30,8 @@ class AgentDriver {
             }
             const apiFuncs = {
                 "POST:send-message": this.sendMessageFunc.bind(this),
+                "POST:upload-file": this.uploadFileFunc.bind(this),
+                "GET:query-tasks": this.queryTasksFunc.bind(this),
                 "GET:messages": this.getHistoryMessagesFunc.bind(this),
                 "GET:info": this.getAgentInfoFunc.bind(this),
                 "GET:conversations": this.getConversationsFunc.bind(this),
@@ -90,5 +92,33 @@ class AgentDriver {
             }
         });
     }
+    queryTasksFunc(event, context) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            try {
+                const pathTaskId = (0, agent_helper_1.parseTaskIdFromPath)(context.httpContext.url);
+                const query = (0, agent_helper_1.parseQueryFromCtx)(context);
+                const parsed = schema_1.queryTasksParamsSchema.passthrough().parse({
+                    task_id: query.task_id || pathTaskId,
+                });
+                return yield ((_b = (_a = this.agent).queryTasks) === null || _b === void 0 ? void 0 : _b.call(_a, parsed.task_id));
+            }
+            catch (e) {
+                return (0, integration_response_1.createIntegrationResponse)(500, codes_1.CODES.AGENT_API_ERROR, `Agent API 'queryTasks' failed, message: ${e}`);
+            }
+        });
+    }
+    uploadFileFunc(event, context) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            try {
+                const parsedEvent = schema_1.uploadFileInputSchema.passthrough().parse(event);
+                return yield ((_b = (_a = this.agent).uploadFile) === null || _b === void 0 ? void 0 : _b.call(_a, parsedEvent));
+            }
+            catch (e) {
+                return (0, integration_response_1.createIntegrationResponse)(500, codes_1.CODES.AGENT_API_ERROR, `Agent API 'uploadFile' failed, message: ${e}`);
+            }
+        });
+    }
 }
 exports.AgentDriver = AgentDriver;

package/{types → lib}/agent-types.d.ts

@@ -79,3 +79,40 @@ export interface GetAgentInfoResponse {
         agentInfo: AgentInfo;
     };
 }
+export interface QueryTasksData {
+    request_id: string;
+    output: {
+        task_id: string;
+        task_status: "PENDING" | "RUNNING" | "SUCCEEDED" | "FAILED" | "CANCELED" | "UNKNOWN";
+        results?: Array<{
+            url?: string;
+            [key: string]: any;
+        }>;
+        task_metrics?: {
+            TOTAL: number;
+            SUCCEEDED: number;
+            FAILED: number;
+        };
+    };
+    usage?: {
+        image_count?: number;
+    };
+}
+export type RawQueryTasksResponse = QueryTasksData;
+export type QueryTasksResponse = BaseResponse<QueryTasksData>;
+export interface UploadFileInput {
+    file: File | Buffer | string;
+    filename: string;
+    purpose?: string;
+}
+export interface UploadFileData {
+    id: string;
+    object: string;
+    bytes: number;
+    created_at: number;
+    filename: string;
+    purpose: string;
+    status: string;
+    status_details?: any;
+}
+export type UploadFileResponse = BaseResponse<UploadFileData>;

package/{types → lib}/agent.d.ts

@@ -1,6 +1,6 @@
 import { type AI, type models } from "@vectorx/ai-sdk";
 import { type RcbContext } from "@vectorx/functions-framework";
-import type { CreateRecordPairResponse, GetAgentInfoResponse, GetConversationsResponse, GetHistoryMessagesResponse, KnowledgeBaseRetrieveResponse } from "./agent-types";
+import type { CreateRecordPairResponse, GetAgentInfoResponse, GetConversationsResponse, GetHistoryMessagesResponse, KnowledgeBaseRetrieveResponse, QueryTasksResponse, UploadFileResponse } from "./agent-types";
 import { type CreateRecordPairParams, type GetHistoryMessagesParams, type KnowledgeSearchInput } from "./schema";
 import { SSESender } from "./sse-sender";
 export declare enum AgentType {
@@ -11,7 +11,8 @@ export declare enum AgentType {
     AI_SDK_CREATE_RECORD_PAIR = "AI_SDK_CREATE_RECORD_PAIR",
     AI_SDK_GET_HISTORY_MESSAGES = "AI_SDK_GET_HISTORY_MESSAGES",
     AI_SDK_GET_CONVERSATIONS = "AI_SDK_GET_CONVERSATIONS",
-    Ai_SDK_POST_KNOWLEDGE_BASE = "Ai_SDK_POST_KNOWLEDGE_BASE"
+    Ai_SDK_POST_KNOWLEDGE_BASE = "Ai_SDK_POST_KNOWLEDGE_BASE",
+    AI_SDK_QUERY_TASKS = "AI_SDK_QUERY_TASKS"
 }
 export declare enum AgentEventType {
     FRAMEWORK_EVENT = "FRAMEWORK_EVENT",
@@ -28,10 +29,12 @@ export declare class AgentRuntime {
     get version(): string;
     constructor(context: RcbContext);
     private createLoggedRequest;
-    protected createModel(modelName: models.ModelName): models.ReactModel;
+    protected createModel(modelName: models.ModelName): models.ReActModel;
     protected createRecordPair(params: CreateRecordPairParams): Promise<CreateRecordPairResponse>;
     protected getHistoryMessages(params: GetHistoryMessagesParams): Promise<GetHistoryMessagesResponse>;
     protected getConversations(): Promise<GetConversationsResponse>;
     getAgentInfo(): Promise<GetAgentInfoResponse>;
     knowledgeBaseRetrieve(params: KnowledgeSearchInput): Promise<KnowledgeBaseRetrieveResponse>;
+    queryTasks(task_id?: string): Promise<QueryTasksResponse>;
+    uploadFile(input: any): Promise<UploadFileResponse>;
 }

package/lib/agent.js

@@ -25,6 +25,7 @@ var AgentType;
     AgentType["AI_SDK_GET_HISTORY_MESSAGES"] = "AI_SDK_GET_HISTORY_MESSAGES";
     AgentType["AI_SDK_GET_CONVERSATIONS"] = "AI_SDK_GET_CONVERSATIONS";
     AgentType["Ai_SDK_POST_KNOWLEDGE_BASE"] = "Ai_SDK_POST_KNOWLEDGE_BASE";
+    AgentType["AI_SDK_QUERY_TASKS"] = "AI_SDK_QUERY_TASKS";
 })(AgentType || (exports.AgentType = AgentType = {}));
 var AgentEventType;
 (function (AgentEventType) {
@@ -573,5 +574,145 @@ class AgentRuntime {
             }
         });
     }
+    queryTasks(task_id) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const validated = schema_1.queryTasksParamsSchema.passthrough().parse({ task_id });
+            this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, safeJsonStringify({
+                type: AgentEventType.FRAMEWORK_EVENT,
+                scene: AgentType.AI_SDK_QUERY_TASKS,
+                method: "GET",
+                url: `https://dashscope.aliyuncs.com/api/v1/tasks/${validated.task_id}`,
+                headers: {},
+                data: { task_id: validated.task_id },
+                eventId: this.context.eventID,
+                timestamp: new Date().toISOString(),
+            }));
+            try {
+                const token = yield this.ai.tokenManager.getValidToken();
+                const response = yield this.request.get({
+                    url: `https://dashscope.aliyuncs.com/api/v1/tasks/${validated.task_id}`,
+                    headers: {
+                        Authorization: `Bearer ${token}`,
+                    },
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_QUERY_TASKS,
+                    method: "GET",
+                    url: `https://dashscope.aliyuncs.com/api/v1/tasks/${validated.task_id}`,
+                    status: response.status,
+                    data: response.data,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                }));
+                if (response.statusCode >= 200 && response.statusCode < 300) {
+                    return {
+                        code: 0,
+                        msg: "success",
+                        data: response.data,
+                    };
+                }
+                else {
+                    return {
+                        code: response.status,
+                        msg: response.statusText || "请求失败",
+                        data: response.data,
+                    };
+                }
+            }
+            catch (error) {
+                this.logger.logAccesslog(functions_framework_1.LogLevel.ERROR, safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_QUERY_TASKS,
+                    method: "GET",
+                    url: `https://dashscope.aliyuncs.com/api/v1/tasks/${validated.task_id}`,
+                    error: error instanceof Error ? error.message : "查询任务失败",
+                    stack: error instanceof Error ? error.stack : undefined,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                }));
+                return {
+                    code: -1,
+                    msg: error instanceof Error ? error.message : "查询任务失败",
+                };
+            }
+        });
+    }
+    uploadFile(input) {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                const validatedInput = schema_1.uploadFileInputSchema.passthrough().parse(input);
+                this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: "AI_SDK_UPLOAD_FILE",
+                    method: "POST",
+                    url: "https://dashscope.aliyuncs.com/compatible-mode/v1/files",
+                    headers: {},
+                    data: { filename: validatedInput.filename, purpose: validatedInput.purpose },
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                }));
+                const token = yield this.ai.tokenManager.getValidToken();
+                const fileBuffer = Buffer.from(validatedInput.fileContent, "base64");
+                const formData = {
+                    file: {
+                        value: fileBuffer,
+                        options: {
+                            filename: validatedInput.filename,
+                            contentType: validatedInput.mimeType || "application/octet-stream",
+                        },
+                    },
+                    purpose: validatedInput.purpose || "file-extract",
+                };
+                const response = yield this.request.upload({
+                    url: "https://dashscope.aliyuncs.com/compatible-mode/v1/files",
+                    headers: {
+                        Authorization: `Bearer ${token}`,
+                    },
+                    data: formData,
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: "AI_SDK_UPLOAD_FILE",
+                    method: "POST",
+                    url: "https://dashscope.aliyuncs.com/compatible-mode/v1/files",
+                    status: response.status,
+                    data: response.data,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                }));
+                if (response.statusCode >= 200 && response.statusCode < 300) {
+                    return {
+                        code: 0,
+                        msg: "success",
+                        data: response.data,
+                    };
+                }
+                else {
+                    return {
+                        code: response.status,
+                        msg: response.statusText || "文件上传失败",
+                        data: response.data,
+                    };
+                }
+            }
+            catch (error) {
+                this.logger.logAccesslog(functions_framework_1.LogLevel.ERROR, safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: "AI_SDK_UPLOAD_FILE",
+                    method: "POST",
+                    url: "https://dashscope.aliyuncs.com/compatible-mode/v1/files",
+                    error: error instanceof Error ? error.message : "文件上传失败",
+                    stack: error instanceof Error ? error.stack : undefined,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                }));
+                return {
+                    code: -1,
+                    msg: error instanceof Error ? error.message : "文件上传失败",
+                };
+            }
+        });
+    }
 }
 exports.AgentRuntime = AgentRuntime;

package/lib/schema/file.schema.d.ts

@@ -0,0 +1,18 @@
+import { z } from "zod";
+export declare const uploadFileInputSchema: z.ZodObject<{
+    filename: z.ZodString;
+    purpose: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    fileContent: z.ZodString;
+    mimeType: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    filename?: string;
+    purpose?: string;
+    fileContent?: string;
+    mimeType?: string;
+}, {
+    filename?: string;
+    purpose?: string;
+    fileContent?: string;
+    mimeType?: string;
+}>;
+export type UploadFileInputSchema = z.infer<typeof uploadFileInputSchema>;

package/lib/schema/file.schema.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.uploadFileInputSchema = void 0;
+const zod_1 = require("zod");
+exports.uploadFileInputSchema = zod_1.z.object({
+    filename: zod_1.z.string().min(1, "文件名不能为空"),
+    purpose: zod_1.z.string().optional().default("file-extract"),
+    fileContent: zod_1.z.string().min(1, "文件内容不能为空"),
+    mimeType: zod_1.z.string().optional(),
+});

package/{types → lib}/schema/index.d.ts

@@ -2,3 +2,5 @@ export * from "./message.schema";
 export * from "./conversation.schema";
 export * from "./memory.schema";
 export * from "./knowledge.schema";
+export * from "./task.schema";
+export * from "./file.schema";

package/lib/schema/index.js

@@ -18,3 +18,5 @@ __exportStar(require("./message.schema"), exports);
 __exportStar(require("./conversation.schema"), exports);
 __exportStar(require("./memory.schema"), exports);
 __exportStar(require("./knowledge.schema"), exports);
+__exportStar(require("./task.schema"), exports);
+__exportStar(require("./file.schema"), exports);

package/{types → lib}/schema/message.schema.d.ts

@@ -58,11 +58,11 @@ export declare const videoContentSchema: z.ZodObject<{
     type: z.ZodLiteral<"video">;
     video: z.ZodArray<z.ZodString, "many">;
 }, "strip", z.ZodTypeAny, {
-    type?: "video";
     video?: string[];
-}, {
     type?: "video";
+}, {
     video?: string[];
+    type?: "video";
 }>;
 export declare const videoUrlContentSchema: z.ZodObject<{
     type: z.ZodLiteral<"video_url">;
@@ -140,11 +140,11 @@ export declare const messageContentSchema: z.ZodUnion<[z.ZodObject<{
     type: z.ZodLiteral<"video">;
     video: z.ZodArray<z.ZodString, "many">;
 }, "strip", z.ZodTypeAny, {
-    type?: "video";
     video?: string[];
-}, {
     type?: "video";
+}, {
     video?: string[];
+    type?: "video";
 }>, z.ZodObject<{
     type: z.ZodLiteral<"video_url">;
     video_url: z.ZodObject<{
@@ -223,11 +223,11 @@ export declare const baseMessageSchema: z.ZodObject<{
         type: z.ZodLiteral<"video">;
         video: z.ZodArray<z.ZodString, "many">;
     }, "strip", z.ZodTypeAny, {
-        type?: "video";
         video?: string[];
-    }, {
         type?: "video";
+    }, {
         video?: string[];
+        type?: "video";
     }>, z.ZodObject<{
         type: z.ZodLiteral<"video_url">;
         video_url: z.ZodObject<{
@@ -249,7 +249,7 @@ export declare const baseMessageSchema: z.ZodObject<{
         };
     }>]>, "many">]>;
 }, "strip", z.ZodTypeAny, {
-    role?: "system" | "user" | "assistant";
+    role?: "user" | "assistant" | "system";
     content?: string | ({
         type?: "text";
         text?: string;
@@ -265,8 +265,8 @@ export declare const baseMessageSchema: z.ZodObject<{
             format?: string;
         };
     } | {
-        type?: "video";
         video?: string[];
+        type?: "video";
     } | {
         type?: "video_url";
         video_url?: {
@@ -274,7 +274,7 @@ export declare const baseMessageSchema: z.ZodObject<{
         };
     })[];
 }, {
-    role?: "system" | "user" | "assistant";
+    role?: "user" | "assistant" | "system";
     content?: string | ({
         type?: "text";
         text?: string;
@@ -290,8 +290,8 @@ export declare const baseMessageSchema: z.ZodObject<{
             format?: string;
         };
     } | {
-        type?: "video";
         video?: string[];
+        type?: "video";
     } | {
         type?: "video_url";
         video_url?: {
@@ -386,11 +386,11 @@ export declare const messageItemSchema: z.ZodUnion<[z.ZodObject<{
         type: z.ZodLiteral<"video">;
         video: z.ZodArray<z.ZodString, "many">;
     }, "strip", z.ZodTypeAny, {
-        type?: "video";
         video?: string[];
-    }, {
         type?: "video";
+    }, {
         video?: string[];
+        type?: "video";
     }>, z.ZodObject<{
         type: z.ZodLiteral<"video_url">;
         video_url: z.ZodObject<{
@@ -412,7 +412,7 @@ export declare const messageItemSchema: z.ZodUnion<[z.ZodObject<{
         };
     }>]>, "many">]>;
 }, "strip", z.ZodTypeAny, {
-    role?: "system" | "user" | "assistant";
+    role?: "user" | "assistant" | "system";
     content?: string | ({
         type?: "text";
         text?: string;
@@ -428,8 +428,8 @@ export declare const messageItemSchema: z.ZodUnion<[z.ZodObject<{
             format?: string;
         };
     } | {
-        type?: "video";
         video?: string[];
+        type?: "video";
     } | {
         type?: "video_url";
         video_url?: {
@@ -437,7 +437,7 @@ export declare const messageItemSchema: z.ZodUnion<[z.ZodObject<{
         };
     })[];
 }, {
-    role?: "system" | "user" | "assistant";
+    role?: "user" | "assistant" | "system";
     content?: string | ({
         type?: "text";
         text?: string;
@@ -453,8 +453,8 @@ export declare const messageItemSchema: z.ZodUnion<[z.ZodObject<{
             format?: string;
         };
     } | {
-        type?: "video";
         video?: string[];
+        type?: "video";
     } | {
         type?: "video_url";
         video_url?: {
@@ -547,11 +547,11 @@ export declare const messageHistorySchema: z.ZodArray<z.ZodUnion<[z.ZodObject<{
         type: z.ZodLiteral<"video">;
         video: z.ZodArray<z.ZodString, "many">;
     }, "strip", z.ZodTypeAny, {
-        type?: "video";
         video?: string[];
-    }, {
         type?: "video";
+    }, {
         video?: string[];
+        type?: "video";
     }>, z.ZodObject<{
         type: z.ZodLiteral<"video_url">;
         video_url: z.ZodObject<{
@@ -573,7 +573,7 @@ export declare const messageHistorySchema: z.ZodArray<z.ZodUnion<[z.ZodObject<{
         };
     }>]>, "many">]>;
 }, "strip", z.ZodTypeAny, {
-    role?: "system" | "user" | "assistant";
+    role?: "user" | "assistant" | "system";
     content?: string | ({
         type?: "text";
         text?: string;
@@ -589,8 +589,8 @@ export declare const messageHistorySchema: z.ZodArray<z.ZodUnion<[z.ZodObject<{
             format?: string;
         };
     } | {
-        type?: "video";
         video?: string[];
+        type?: "video";
     } | {
         type?: "video_url";
         video_url?: {
@@ -598,7 +598,7 @@ export declare const messageHistorySchema: z.ZodArray<z.ZodUnion<[z.ZodObject<{
         };
     })[];
 }, {
-    role?: "system" | "user" | "assistant";
+    role?: "user" | "assistant" | "system";
     content?: string | ({
         type?: "text";
         text?: string;
@@ -614,8 +614,8 @@ export declare const messageHistorySchema: z.ZodArray<z.ZodUnion<[z.ZodObject<{
             format?: string;
         };
     } | {
-        type?: "video";
         video?: string[];
+        type?: "video";
     } | {
         type?: "video_url";
         video_url?: {
@@ -710,11 +710,11 @@ export declare const sendMessageInputSchema: z.ZodObject<{
             type: z.ZodLiteral<"video">;
             video: z.ZodArray<z.ZodString, "many">;
         }, "strip", z.ZodTypeAny, {
-            type?: "video";
             video?: string[];
-        }, {
             type?: "video";
+        }, {
             video?: string[];
+            type?: "video";
         }>, z.ZodObject<{
             type: z.ZodLiteral<"video_url">;
             video_url: z.ZodObject<{
@@ -736,7 +736,7 @@ export declare const sendMessageInputSchema: z.ZodObject<{
             };
         }>]>, "many">]>;
     }, "strip", z.ZodTypeAny, {
-        role?: "system" | "user" | "assistant";
+        role?: "user" | "assistant" | "system";
         content?: string | ({
             type?: "text";
             text?: string;
@@ -752,8 +752,8 @@ export declare const sendMessageInputSchema: z.ZodObject<{
                 format?: string;
             };
         } | {
-            type?: "video";
             video?: string[];
+            type?: "video";
         } | {
             type?: "video_url";
             video_url?: {
@@ -761,7 +761,7 @@ export declare const sendMessageInputSchema: z.ZodObject<{
             };
         })[];
     }, {
-        role?: "system" | "user" | "assistant";
+        role?: "user" | "assistant" | "system";
         content?: string | ({
             type?: "text";
             text?: string;
@@ -777,8 +777,8 @@ export declare const sendMessageInputSchema: z.ZodObject<{
                 format?: string;
             };
         } | {
-            type?: "video";
             video?: string[];
+            type?: "video";
         } | {
             type?: "video_url";
             video_url?: {
@@ -816,7 +816,7 @@ export declare const sendMessageInputSchema: z.ZodObject<{
 }, "strip", z.ZodTypeAny, {
     msg?: string;
     history?: ({
-        role?: "system" | "user" | "assistant";
+        role?: "user" | "assistant" | "system";
         content?: string | ({
             type?: "text";
             text?: string;
@@ -832,8 +832,8 @@ export declare const sendMessageInputSchema: z.ZodObject<{
                 format?: string;
             };
         } | {
-            type?: "video";
             video?: string[];
+            type?: "video";
         } | {
             type?: "video_url";
             video_url?: {
@@ -853,7 +853,7 @@ export declare const sendMessageInputSchema: z.ZodObject<{
 }, {
     msg?: string;
     history?: ({
-        role?: "system" | "user" | "assistant";
+        role?: "user" | "assistant" | "system";
         content?: string | ({
             type?: "text";
             text?: string;
@@ -869,8 +869,8 @@ export declare const sendMessageInputSchema: z.ZodObject<{
                 format?: string;
             };
         } | {
-            type?: "video";
             video?: string[];
+            type?: "video";
         } | {
             type?: "video_url";
             video_url?: {

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/{types → lib}/utils/agent-helper.d.ts

@@ -4,5 +4,6 @@ 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

@@ -40,11 +40,12 @@ 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\/([^\\]+)\/([^\\]+)/;
+const AGENT_REG = /\/v1\/aiagent\/agents\/([^\/]+)\/(.+)/;
 function parseAgentId(url) {
     return parseUrl(url).agentId;
 }
@@ -67,7 +68,11 @@ function parseUrl(url) {
     if (!regResult) {
         throw new Error("Invalid pathname");
     }
-    const [_, agentId, apiName] = regResult;
+    const [_, agentId, remainingPath] = regResult;
+    let apiName = remainingPath;
+    if (remainingPath.startsWith("query-tasks/")) {
+        apiName = "query-tasks";
+    }
     return { agentId, apiName };
 }
 function parseQueryFromCtx(ctx) {
@@ -82,6 +87,20 @@ function parseQueryFromCtx(ctx) {
         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);
 }

package/package.json

@@ -1,9 +1,8 @@
 {
   "name": "@vectorx/agent-runtime",
-  "version": "0.1.3",
+  "version": "0.4.0",
   "description": "Cloud AI agent runtime",
   "main": "lib/index.js",
-  "types": "types/index.d.ts",
   "files": [
     "lib",
     "types",
@@ -21,8 +20,9 @@
     "node": ">=18.0.0"
   },
   "dependencies": {
-    "@vectorx/ai-sdk": "0.1.3",
-    "@vectorx/functions-framework": "0.1.3",
+    "@vectorx/ai-sdk": "0.4.0",
+    "@vectorx/functions-framework": "0.4.0",
+    "langfuse": "^3.38.4",
     "query-string": "^6.14.1",
     "zod": "^3.24.2"
   },