@vectorx/agent-runtime 0.0.0-beta-20251112071515

package/lib/agent.js

@@ -0,0 +1,785 @@
+"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 __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentRuntime = exports.AgentEventType = exports.AgentType = void 0;
+const ai_sdk_1 = require("@vectorx/ai-sdk");
+const functions_framework_1 = require("@vectorx/functions-framework");
+const schema_1 = require("./schema");
+const sse_sender_1 = require("./sse-sender");
+const agent_helper_1 = require("./utils/agent-helper");
+var AgentType;
+(function (AgentType) {
+    AgentType["AI_SDK_REQUEST_START"] = "AI_SDK_REQUEST_START";
+    AgentType["AI_SDK_REQUEST_SUCCESS"] = "AI_SDK_REQUEST_SUCCESS";
+    AgentType["AI_SDK_REQUEST_ERROR"] = "AI_SDK_REQUEST_ERROR";
+    AgentType["AI_SDK_CREATE_MODEL"] = "AI_SDK_CREATE_MODEL";
+    AgentType["AI_SDK_CREATE_RECORD_PAIR"] = "AI_SDK_CREATE_RECORD_PAIR";
+    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) {
+    AgentEventType["FRAMEWORK_EVENT"] = "FRAMEWORK_EVENT";
+    AgentEventType["HTTP_ACCESS"] = "HTTP_ACCESS";
+})(AgentEventType || (exports.AgentEventType = AgentEventType = {}));
+const safeJsonStringify = (obj) => {
+    try {
+        return JSON.stringify(obj);
+    }
+    catch (error) {
+        return JSON.stringify({ error: "Failed to stringify object" });
+    }
+};
+class AgentRuntime {
+    get agentTag() {
+        const url = this.context.httpContext.url;
+        return (0, agent_helper_1.parseAgentTag)(url);
+    }
+    get agentId() {
+        const url = this.context.httpContext.url;
+        return (0, agent_helper_1.parseAgentId)(url);
+    }
+    get version() {
+        return "0.0.1【TODO: 从 context header 中获取22222版本号】";
+    }
+    constructor(context) {
+        this.context = context;
+        this.sseSender = new sse_sender_1.SSESender(context);
+        this.request = context.request;
+        this.logger = context.logger;
+        const loggedRequest = this.createLoggedRequest(this.request, this.logger);
+        this.ai = (0, ai_sdk_1.createAi)({
+            request: loggedRequest,
+            getBaseUrl: () => this.context.baseUrl,
+            env: ai_sdk_1.AiSdkEnv.Cloud,
+        });
+    }
+    createLoggedRequest(originalRequest, logger) {
+        return {
+            get: (options) => __awaiter(this, void 0, void 0, function* () {
+                const startTime = Date.now();
+                const requestStartLog = safeJsonStringify({
+                    type: AgentType.AI_SDK_REQUEST_START,
+                    method: "GET",
+                    url: options.url,
+                    headers: options.headers,
+                    data: options.data,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestStartLog);
+                try {
+                    const response = yield originalRequest.get(options);
+                    const requestSuccessLog = safeJsonStringify({
+                        type: AgentType.AI_SDK_REQUEST_SUCCESS,
+                        method: "GET",
+                        url: options.url,
+                        status: response.statusCode,
+                        duration: `${Date.now() - startTime}ms`,
+                        eventId: this.context.eventID,
+                        timestamp: new Date().toISOString(),
+                    });
+                    logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestSuccessLog);
+                    return response;
+                }
+                catch (error) {
+                    const requestErrorLog = safeJsonStringify({
+                        type: AgentType.AI_SDK_REQUEST_ERROR,
+                        method: "GET",
+                        url: options.url,
+                        error: error instanceof Error ? error.message : "请求失败",
+                        stack: error instanceof Error ? error.stack : undefined,
+                        duration: `${Date.now() - startTime}ms`,
+                        eventId: this.context.eventID,
+                        timestamp: new Date().toISOString(),
+                    });
+                    logger.logAccesslog(functions_framework_1.LogLevel.ERROR, requestErrorLog);
+                    throw error;
+                }
+            }),
+            post: (options) => __awaiter(this, void 0, void 0, function* () {
+                const startTime = Date.now();
+                const requestStartLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_REQUEST_START,
+                    method: "POST",
+                    url: options.url,
+                    headers: options.headers,
+                    data: options.data,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestStartLog);
+                try {
+                    const response = yield originalRequest.post(options);
+                    const requestSuccessLog = safeJsonStringify({
+                        type: AgentEventType.FRAMEWORK_EVENT,
+                        scene: AgentType.AI_SDK_REQUEST_SUCCESS,
+                        method: "POST",
+                        url: options.url,
+                        status: response.statusCode,
+                        duration: `${Date.now() - startTime}ms`,
+                        eventId: this.context.eventID,
+                        timestamp: new Date().toISOString(),
+                    });
+                    logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestSuccessLog);
+                    return response;
+                }
+                catch (error) {
+                    const requestErrorLog = safeJsonStringify({
+                        type: AgentEventType.FRAMEWORK_EVENT,
+                        scene: AgentType.AI_SDK_REQUEST_ERROR,
+                        method: "POST",
+                        url: options.url,
+                        error: error instanceof Error ? error.message : "请求失败",
+                        stack: error instanceof Error ? error.stack : undefined,
+                        duration: `${Date.now() - startTime}ms`,
+                        eventId: this.context.eventID,
+                        timestamp: new Date().toISOString(),
+                    });
+                    logger.logAccesslog(functions_framework_1.LogLevel.ERROR, requestErrorLog);
+                    throw error;
+                }
+            }),
+            put: (options) => __awaiter(this, void 0, void 0, function* () {
+                const startTime = Date.now();
+                const requestStartLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_REQUEST_START,
+                    method: "PUT",
+                    url: options.url,
+                    headers: options.headers,
+                    data: options.data,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestStartLog);
+                try {
+                    const response = yield originalRequest.put(options);
+                    const requestSuccessLog = safeJsonStringify({
+                        type: AgentEventType.FRAMEWORK_EVENT,
+                        scene: AgentType.AI_SDK_REQUEST_SUCCESS,
+                        method: "PUT",
+                        url: options.url,
+                        status: response.statusCode,
+                        duration: `${Date.now() - startTime}ms`,
+                        eventId: this.context.eventID,
+                        timestamp: new Date().toISOString(),
+                    });
+                    logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestSuccessLog);
+                    return response;
+                }
+                catch (error) {
+                    const requestErrorLog = safeJsonStringify({
+                        type: AgentEventType.FRAMEWORK_EVENT,
+                        scene: AgentType.AI_SDK_REQUEST_ERROR,
+                        method: "PUT",
+                        url: options.url,
+                        error: error instanceof Error ? error.message : "请求失败",
+                        stack: error instanceof Error ? error.stack : undefined,
+                        duration: `${Date.now() - startTime}ms`,
+                        eventId: this.context.eventID,
+                        timestamp: new Date().toISOString(),
+                    });
+                    logger.logAccesslog(functions_framework_1.LogLevel.ERROR, requestErrorLog);
+                    throw error;
+                }
+            }),
+            upload: (options) => __awaiter(this, void 0, void 0, function* () {
+                const startTime = Date.now();
+                const requestStartLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_REQUEST_START,
+                    method: "UPLOAD",
+                    url: options.url,
+                    headers: options.headers,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestStartLog);
+                try {
+                    const response = yield originalRequest.upload(options);
+                    const requestSuccessLog = safeJsonStringify({
+                        type: AgentEventType.FRAMEWORK_EVENT,
+                        scene: AgentType.AI_SDK_REQUEST_SUCCESS,
+                        method: "UPLOAD",
+                        url: options.url,
+                        status: response.statusCode,
+                        duration: `${Date.now() - startTime}ms`,
+                        eventId: this.context.eventID,
+                        timestamp: new Date().toISOString(),
+                    });
+                    logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestSuccessLog);
+                    return response;
+                }
+                catch (error) {
+                    const requestErrorLog = safeJsonStringify({
+                        type: AgentEventType.FRAMEWORK_EVENT,
+                        scene: AgentType.AI_SDK_REQUEST_ERROR,
+                        method: "UPLOAD",
+                        url: options.url,
+                        error: error instanceof Error ? error.message : "请求失败",
+                        stack: error instanceof Error ? error.stack : undefined,
+                        duration: `${Date.now() - startTime}ms`,
+                        eventId: this.context.eventID,
+                        timestamp: new Date().toISOString(),
+                    });
+                    logger.logAccesslog(functions_framework_1.LogLevel.ERROR, requestErrorLog);
+                    throw error;
+                }
+            }),
+            download: (options) => __awaiter(this, void 0, void 0, function* () {
+                const startTime = Date.now();
+                const requestStartLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_REQUEST_START,
+                    method: "DOWNLOAD",
+                    url: options.url,
+                    headers: options.headers,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestStartLog);
+                try {
+                    const response = yield originalRequest.download(options);
+                    const requestSuccessLog = safeJsonStringify({
+                        type: AgentEventType.FRAMEWORK_EVENT,
+                        scene: AgentType.AI_SDK_REQUEST_SUCCESS,
+                        method: "DOWNLOAD",
+                        url: options.url,
+                        duration: `${Date.now() - startTime}ms`,
+                        eventId: this.context.eventID,
+                        timestamp: new Date().toISOString(),
+                    });
+                    logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestSuccessLog);
+                    return response;
+                }
+                catch (error) {
+                    const requestErrorLog = safeJsonStringify({
+                        type: AgentEventType.FRAMEWORK_EVENT,
+                        scene: AgentType.AI_SDK_REQUEST_ERROR,
+                        method: "DOWNLOAD",
+                        url: options.url,
+                        error: error instanceof Error ? error.message : "请求失败",
+                        stack: error instanceof Error ? error.stack : undefined,
+                        duration: `${Date.now() - startTime}ms`,
+                        eventId: this.context.eventID,
+                        timestamp: new Date().toISOString(),
+                    });
+                    logger.logAccesslog(functions_framework_1.LogLevel.ERROR, requestErrorLog);
+                    throw error;
+                }
+            }),
+            fetch: (options) => __awaiter(this, void 0, void 0, function* () {
+                const startTime = Date.now();
+                const requestStartLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_REQUEST_START,
+                    method: "FETCH",
+                    url: options.url,
+                    headers: options.headers,
+                    data: safeJsonStringify(options.body),
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestStartLog);
+                try {
+                    const response = yield originalRequest.fetch(options);
+                    const requestSuccessLog = safeJsonStringify({
+                        type: AgentEventType.FRAMEWORK_EVENT,
+                        scene: AgentType.AI_SDK_REQUEST_SUCCESS,
+                        method: "FETCH",
+                        url: options.url,
+                        status: response.statusCode,
+                        duration: `${Date.now() - startTime}ms`,
+                        eventId: this.context.eventID,
+                        timestamp: new Date().toISOString(),
+                    });
+                    logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestSuccessLog);
+                    return response;
+                }
+                catch (error) {
+                    const requestErrorLog = safeJsonStringify({
+                        type: AgentEventType.FRAMEWORK_EVENT,
+                        scene: AgentType.AI_SDK_REQUEST_ERROR,
+                        method: "FETCH",
+                        url: options.url,
+                        error: error instanceof Error ? error.message : "请求失败",
+                        stack: error instanceof Error ? error.stack : undefined,
+                        duration: `${Date.now() - startTime}ms`,
+                        eventId: this.context.eventID,
+                        timestamp: new Date().toISOString(),
+                    });
+                    logger.logAccesslog(functions_framework_1.LogLevel.ERROR, requestErrorLog);
+                    throw error;
+                }
+            }),
+        };
+    }
+    createModel(modelName) {
+        const createModelLog = safeJsonStringify({
+            type: AgentEventType.FRAMEWORK_EVENT,
+            scene: AgentType.AI_SDK_CREATE_MODEL,
+            modelName: modelName,
+            eventId: this.context.eventID,
+            timestamp: new Date().toISOString(),
+        });
+        this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, createModelLog);
+        return this.ai.createModel(modelName);
+    }
+    createRecordPair(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                const validatedParams = schema_1.createRecordPairParamsSchema.passthrough().parse(params);
+                const { conversation_id, user_input, assistant_output } = validatedParams;
+                const createRecordPairStartLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_CREATE_RECORD_PAIR,
+                    method: "POST",
+                    url: `${this.context.baseUrl}/conversation/save`,
+                    headers: {
+                        "Content-Type": "application/json",
+                    },
+                    data: {
+                        conversation_id,
+                        user_input: JSON.stringify(user_input),
+                        assistant_output: JSON.stringify(assistant_output),
+                    },
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, createRecordPairStartLog);
+                const response = yield this.request.post({
+                    headers: {
+                        "Content-Type": "application/json",
+                    },
+                    url: `${this.context.baseUrl}/conversation/save`,
+                    data: {
+                        conversation_id,
+                        user_input: JSON.stringify(user_input),
+                        assistant_output: JSON.stringify(assistant_output),
+                    },
+                });
+                const createRecordPairEndLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_CREATE_RECORD_PAIR,
+                    method: "POST",
+                    url: `${this.context.baseUrl}/conversation/save`,
+                    status: response.status,
+                    data: response.data,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, createRecordPairEndLog);
+                return response.data;
+            }
+            catch (error) {
+                const createRecordPairErrorLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_CREATE_RECORD_PAIR,
+                    method: "POST",
+                    url: `${this.context.baseUrl}/conversation/save`,
+                    error: error instanceof Error ? error.message : "创建对话对失败",
+                    stack: error instanceof Error ? error.stack : undefined,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.ERROR, createRecordPairErrorLog);
+                return {
+                    code: -1,
+                    msg: error instanceof Error ? error.message : "创建对话对失败",
+                };
+            }
+        });
+    }
+    getHistoryMessages(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                const validatedParams = schema_1.getHistoryMessagesParamsSchema.passthrough().parse(params);
+                const getHistoryMessagesStartLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_GET_HISTORY_MESSAGES,
+                    method: "GET",
+                    url: `${this.context.baseUrl}/conversation/history`,
+                    data: validatedParams,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, getHistoryMessagesStartLog);
+                const response = yield this.request.get({
+                    url: `${this.context.baseUrl}/conversation/history`,
+                    data: validatedParams,
+                });
+                const getHistoryMessagesEndLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_GET_HISTORY_MESSAGES,
+                    method: "GET",
+                    url: `${this.context.baseUrl}/conversation/history`,
+                    status: response.status,
+                    data: response.data,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, getHistoryMessagesEndLog);
+                return response.data;
+            }
+            catch (error) {
+                const getHistoryMessagesErrorLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_GET_HISTORY_MESSAGES,
+                    method: "GET",
+                    url: `${this.context.baseUrl}/conversation/history`,
+                    error: error instanceof Error ? error.message : "获取历史消息失败",
+                    stack: error instanceof Error ? error.stack : undefined,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.ERROR, getHistoryMessagesErrorLog);
+                return {
+                    code: -1,
+                    msg: error instanceof Error ? error.message : "获取历史消息失败",
+                    data: {
+                        message_list: [],
+                        conversation_id: "",
+                        participant_info_map: {},
+                        next_cursor: 0,
+                        has_more: false,
+                    },
+                };
+            }
+        });
+    }
+    getConversations() {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                const getConversationsStartLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_GET_CONVERSATIONS,
+                    method: "GET",
+                    url: `${this.context.baseUrl}/conversation/list`,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, getConversationsStartLog);
+                const response = yield this.request.get({
+                    url: `${this.context.baseUrl}/conversation/list`,
+                });
+                const getConversationsEndLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_GET_CONVERSATIONS,
+                    method: "GET",
+                    url: `${this.context.baseUrl}/conversation/list`,
+                    status: response.status,
+                    data: response.data,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, getConversationsEndLog);
+                return response.data;
+            }
+            catch (error) {
+                const getConversationsErrorLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_GET_CONVERSATIONS,
+                    method: "GET",
+                    url: `${this.context.baseUrl}/conversation/list`,
+                    error: error instanceof Error ? error.message : "获取会话列表失败",
+                    stack: error instanceof Error ? error.stack : undefined,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.ERROR, getConversationsErrorLog);
+                return {
+                    code: -1,
+                    msg: error instanceof Error ? error.message : "获取会话列表失败",
+                    data: {
+                        conversations: [],
+                    },
+                };
+            }
+        });
+    }
+    getAgentInfo() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const response = yield this.request.get({
+                url: `${this.context.baseUrl}/agent/info`,
+            });
+            return response.data;
+        });
+    }
+    knowledgeBaseRetrieve(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                const validatedParams = schema_1.knowledgeSearchInputSchema.passthrough().parse(params);
+                const knowledgeBaseRetrieveStartLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.Ai_SDK_POST_KNOWLEDGE_BASE,
+                    method: "POST",
+                    url: `${this.context.baseUrl}/rag/retrieve`,
+                    data: validatedParams,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, knowledgeBaseRetrieveStartLog);
+                const response = yield this.request.post({
+                    headers: {
+                        "Content-Type": "application/json",
+                    },
+                    url: `${this.context.baseUrl}/rag/retrieve`,
+                    data: validatedParams,
+                });
+                const knowledgeBaseRetrieveEndLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.Ai_SDK_POST_KNOWLEDGE_BASE,
+                    method: "POST",
+                    url: `${this.context.baseUrl}/rag/retrieve`,
+                    status: response.status,
+                    data: response.data,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.INFO, knowledgeBaseRetrieveEndLog);
+                return response.data;
+            }
+            catch (error) {
+                const knowledgeBaseRetrieveErrorLog = safeJsonStringify({
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.Ai_SDK_POST_KNOWLEDGE_BASE,
+                    method: "POST",
+                    url: `${this.context.baseUrl}/rag/retrieve`,
+                    error: error instanceof Error ? error.message : "知识库检索失败",
+                    stack: error instanceof Error ? error.stack : undefined,
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                });
+                this.logger.logAccesslog(functions_framework_1.LogLevel.ERROR, knowledgeBaseRetrieveErrorLog);
+                return {
+                    code: -1,
+                    msg: error instanceof Error ? error.message : "知识库检索失败",
+                    data: {
+                        knowledge_results: [],
+                        total_tokens: 0,
+                    },
+                };
+            }
+        });
+    }
+    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* () {
+            var _a, _b;
+            const fs = yield Promise.resolve().then(() => __importStar(require("fs")));
+            const path = yield Promise.resolve().then(() => __importStar(require("path")));
+            const os = yield Promise.resolve().then(() => __importStar(require("os")));
+            let tempFilePath = null;
+            console.log("==== input ====", input);
+            try {
+                const fileData = input.file;
+                const filename = fileData.name || fileData.originalFilename || fileData.newFilename || "unknown";
+                const purpose = input.purpose || "file-extract";
+                const mimeType = fileData.type || fileData.mimetype || "application/octet-stream";
+                console.log("==== filename ====", filename);
+                console.log("==== purpose ====", purpose);
+                console.log("==== mimeType ====", mimeType);
+                const fileBuffer = fileData.buffer || fileData;
+                if (!Buffer.isBuffer(fileBuffer)) {
+                    throw new Error("无法从上传的文件对象中提取文件内容：buffer 不存在或格式错误");
+                }
+                const tempDir = os.tmpdir();
+                const timestamp = Date.now();
+                const randomStr = Math.random().toString(36).substring(2, 15);
+                tempFilePath = path.join(tempDir, `upload_${timestamp}_${randomStr}_${filename}`);
+                fs.writeFileSync(tempFilePath, fileBuffer);
+                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, purpose, tempFilePath },
+                    eventId: this.context.eventID,
+                    timestamp: new Date().toISOString(),
+                }));
+                const token = yield this.ai.tokenManager.getValidToken();
+                const FormData = (yield Promise.resolve().then(() => __importStar(require("form-data")))).default;
+                const formData = new FormData();
+                formData.append("file", fs.createReadStream(tempFilePath), {
+                    filename,
+                    contentType: mimeType,
+                });
+                formData.append("purpose", purpose);
+                const response = yield this.request.fetch({
+                    url: "https://dashscope.aliyuncs.com/compatible-mode/v1/files",
+                    headers: Object.assign({ Authorization: `Bearer ${token}` }, formData.getHeaders()),
+                    method: "POST",
+                    body: 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.statusCode,
+                    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.statusCode,
+                        msg: ((_b = (_a = response.data) === null || _a === void 0 ? void 0 : _a.error) === null || _b === void 0 ? void 0 : _b.message) || "文件上传失败",
+                        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 : "文件上传失败",
+                };
+            }
+            finally {
+                if (tempFilePath) {
+                    try {
+                        fs.unlinkSync(tempFilePath);
+                    }
+                    catch (cleanupError) {
+                        this.logger.logAccesslog(functions_framework_1.LogLevel.WARN, safeJsonStringify({
+                            type: AgentEventType.FRAMEWORK_EVENT,
+                            scene: "AI_SDK_UPLOAD_FILE_CLEANUP",
+                            message: "临时文件清理失败",
+                            tempFilePath,
+                            error: cleanupError instanceof Error ? cleanupError.message : "未知错误",
+                            eventId: this.context.eventID,
+                            timestamp: new Date().toISOString(),
+                        }));
+                    }
+                }
+            }
+        });
+    }
+}
+exports.AgentRuntime = AgentRuntime;