@vectorx/agent-runtime 0.3.0 → 0.4.0

package/lib/agent-runner.d.ts

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

package/lib/agent-runner.js

@@ -30,6 +30,7 @@ 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),
@@ -107,5 +108,17 @@ class AgentDriver {
             }
         });
     }
+    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/lib/agent-types.d.ts

@@ -100,3 +100,19 @@ export interface QueryTasksData {
 }
 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/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, QueryTasksResponse } 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 {
@@ -36,4 +36,5 @@ export declare class AgentRuntime {
     getAgentInfo(): Promise<GetAgentInfoResponse>;
     knowledgeBaseRetrieve(params: KnowledgeSearchInput): Promise<KnowledgeBaseRetrieveResponse>;
     queryTasks(task_id?: string): Promise<QueryTasksResponse>;
+    uploadFile(input: any): Promise<UploadFileResponse>;
 }

package/lib/agent.js

@@ -638,5 +638,81 @@ class AgentRuntime {
             }
         });
     }
+    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/lib/schema/index.d.ts

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

@@ -19,3 +19,4 @@ __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/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/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vectorx/agent-runtime",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "description": "Cloud AI agent runtime",
   "main": "lib/index.js",
   "files": [
@@ -20,8 +20,8 @@
     "node": ">=18.0.0"
   },
   "dependencies": {
-    "@vectorx/ai-sdk": "0.3.0",
-    "@vectorx/functions-framework": "0.3.0",
+    "@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"