@vectorx/agent-runtime 0.1.2 → 0.3.0

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

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

package/lib/agent-runner.js

@@ -30,6 +30,7 @@ class AgentDriver {
             }
             const apiFuncs = {
                 "POST:send-message": this.sendMessageFunc.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 +91,21 @@ 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}`);
+            }
+        });
+    }
 }
 exports.AgentDriver = AgentDriver;

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

@@ -79,3 +79,24 @@ 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>;

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 } 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,11 @@ 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>;
 }

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,69 @@ 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 : "查询任务失败",
+                };
+            }
+        });
+    }
 }
 exports.AgentRuntime = AgentRuntime;

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

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

package/lib/schema/index.js

@@ -18,3 +18,4 @@ __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);

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.2",
+  "version": "0.3.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.2",
-    "@vectorx/functions-framework": "0.1.2",
+    "@vectorx/ai-sdk": "0.3.0",
+    "@vectorx/functions-framework": "0.3.0",
+    "langfuse": "^3.38.4",
     "query-string": "^6.14.1",
     "zod": "^3.24.2"
   },