@vectorx/agent-runtime 0.6.1 → 0.7.0

package/lib/agent.d.ts

@@ -1,4 +1,4 @@
-import { type AI, type models } from "@vectorx/ai-sdk";
+import { type AI, models as aiModels, type models } from "@vectorx/ai-sdk";
 import { type RcbContext } from "@vectorx/functions-framework";
 import type { CreateRecordPairResponse, GetAgentInfoResponse, GetConversationsResponse, GetHistoryMessagesResponse, KnowledgeBaseRetrieveResponse, QueryTasksResponse, UploadFileResponse } from "./agent-types";
 import { type CreateRecordPairParams, type GetHistoryMessagesParams, type KnowledgeSearchInput } from "./schema";
@@ -24,12 +24,16 @@ export declare class AgentRuntime {
     protected ai: AI;
     protected request: RcbContext["request"];
     protected logger: RcbContext["logger"];
+    private currentModelName?;
     get agentTag(): string;
     get agentId(): string;
     get version(): string;
     constructor(context: RcbContext);
+    private isModelRequest;
+    private getModelNameFromUrl;
+    private updateModelApmData;
     private createLoggedRequest;
-    protected createModel(modelName: models.ModelName): models.ReActModel;
+    protected createModel(modelName: models.ModelName): aiModels.ReActModel;
     protected createRecordPair(params: CreateRecordPairParams): Promise<CreateRecordPairResponse>;
     protected getHistoryMessages(params: GetHistoryMessagesParams): Promise<GetHistoryMessagesResponse>;
     protected getConversations(): Promise<GetConversationsResponse>;

package/lib/agent.js

@@ -97,12 +97,72 @@ class AgentRuntime {
             env: ai_sdk_1.AiSdkEnv.Cloud,
         });
     }
+    isModelRequest(url) {
+        if (!url)
+            return false;
+        return ai_sdk_1.models.isModelRequestUrl(url);
+    }
+    getModelNameFromUrl(url) {
+        if (!url)
+            return undefined;
+        try {
+            const urlObj = new URL(url);
+            const fullPath = urlObj.pathname;
+            for (const [modelName, { domain, paths }] of ai_sdk_1.models.MODEL_REQUEST_PATTERNS) {
+                if (domain !== "*" && urlObj.origin !== domain) {
+                    continue;
+                }
+                for (const path of paths) {
+                    if (fullPath.includes(path) || fullPath.endsWith(path)) {
+                        return modelName;
+                    }
+                }
+            }
+            return undefined;
+        }
+        catch (_a) {
+            for (const [modelName, { paths }] of ai_sdk_1.models.MODEL_REQUEST_PATTERNS) {
+                for (const path of paths) {
+                    if (url.includes(path)) {
+                        return modelName;
+                    }
+                }
+            }
+            return undefined;
+        }
+    }
+    updateModelApmData(url, statusCode, rt) {
+        if (!this.isModelRequest(url)) {
+            (0, functions_framework_1.updateApmMeasurementData)({
+                kit_model_name: undefined,
+                kit_model_service_url: undefined,
+                kit_model_status_code: undefined,
+                kit_model_rt: undefined,
+            });
+            return;
+        }
+        const modelName = this.currentModelName || this.getModelNameFromUrl(url);
+        const updates = {
+            kit_model_service_url: url,
+        };
+        if (modelName) {
+            updates.kit_model_name = modelName;
+        }
+        if (statusCode !== undefined) {
+            updates.kit_model_status_code = statusCode;
+        }
+        if (rt !== undefined && rt > 0) {
+            updates.kit_model_rt = rt;
+        }
+        (0, functions_framework_1.updateApmMeasurementData)(updates);
+    }
     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,
+                    type: AgentEventType.FRAMEWORK_EVENT,
+                    scene: AgentType.AI_SDK_REQUEST_START,
                     method: "GET",
                     url: options.url,
                     headers: options.headers,
@@ -113,12 +173,14 @@ class AgentRuntime {
                 logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestStartLog);
                 try {
                     const response = yield originalRequest.get(options);
+                    const duration = `${Date.now() - startTime}ms`;
                     const requestSuccessLog = safeJsonStringify({
-                        type: AgentType.AI_SDK_REQUEST_SUCCESS,
+                        type: AgentEventType.FRAMEWORK_EVENT,
+                        scene: AgentType.AI_SDK_REQUEST_SUCCESS,
                         method: "GET",
                         url: options.url,
                         status: response.statusCode,
-                        duration: `${Date.now() - startTime}ms`,
+                        duration: duration,
                         eventId: this.context.eventID,
                         timestamp: new Date().toISOString(),
                     });
@@ -126,13 +188,15 @@ class AgentRuntime {
                     return response;
                 }
                 catch (error) {
+                    const duration = `${Date.now() - startTime}ms`;
                     const requestErrorLog = safeJsonStringify({
-                        type: AgentType.AI_SDK_REQUEST_ERROR,
+                        type: AgentEventType.FRAMEWORK_EVENT,
+                        scene: 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`,
+                        duration: duration,
                         eventId: this.context.eventID,
                         timestamp: new Date().toISOString(),
                     });
@@ -155,13 +219,14 @@ class AgentRuntime {
                 logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestStartLog);
                 try {
                     const response = yield originalRequest.post(options);
+                    const duration = `${Date.now() - startTime}ms`;
                     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`,
+                        duration: duration,
                         eventId: this.context.eventID,
                         timestamp: new Date().toISOString(),
                     });
@@ -169,6 +234,7 @@ class AgentRuntime {
                     return response;
                 }
                 catch (error) {
+                    const duration = `${Date.now() - startTime}ms`;
                     const requestErrorLog = safeJsonStringify({
                         type: AgentEventType.FRAMEWORK_EVENT,
                         scene: AgentType.AI_SDK_REQUEST_ERROR,
@@ -176,7 +242,7 @@ class AgentRuntime {
                         url: options.url,
                         error: error instanceof Error ? error.message : "请求失败",
                         stack: error instanceof Error ? error.stack : undefined,
-                        duration: `${Date.now() - startTime}ms`,
+                        duration: duration,
                         eventId: this.context.eventID,
                         timestamp: new Date().toISOString(),
                     });
@@ -199,13 +265,14 @@ class AgentRuntime {
                 logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestStartLog);
                 try {
                     const response = yield originalRequest.put(options);
+                    const duration = `${Date.now() - startTime}ms`;
                     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`,
+                        duration: duration,
                         eventId: this.context.eventID,
                         timestamp: new Date().toISOString(),
                     });
@@ -213,6 +280,7 @@ class AgentRuntime {
                     return response;
                 }
                 catch (error) {
+                    const duration = `${Date.now() - startTime}ms`;
                     const requestErrorLog = safeJsonStringify({
                         type: AgentEventType.FRAMEWORK_EVENT,
                         scene: AgentType.AI_SDK_REQUEST_ERROR,
@@ -220,7 +288,7 @@ class AgentRuntime {
                         url: options.url,
                         error: error instanceof Error ? error.message : "请求失败",
                         stack: error instanceof Error ? error.stack : undefined,
-                        duration: `${Date.now() - startTime}ms`,
+                        duration: duration,
                         eventId: this.context.eventID,
                         timestamp: new Date().toISOString(),
                     });
@@ -242,13 +310,14 @@ class AgentRuntime {
                 logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestStartLog);
                 try {
                     const response = yield originalRequest.upload(options);
+                    const duration = `${Date.now() - startTime}ms`;
                     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`,
+                        duration: duration,
                         eventId: this.context.eventID,
                         timestamp: new Date().toISOString(),
                     });
@@ -256,6 +325,7 @@ class AgentRuntime {
                     return response;
                 }
                 catch (error) {
+                    const duration = `${Date.now() - startTime}ms`;
                     const requestErrorLog = safeJsonStringify({
                         type: AgentEventType.FRAMEWORK_EVENT,
                         scene: AgentType.AI_SDK_REQUEST_ERROR,
@@ -263,7 +333,7 @@ class AgentRuntime {
                         url: options.url,
                         error: error instanceof Error ? error.message : "请求失败",
                         stack: error instanceof Error ? error.stack : undefined,
-                        duration: `${Date.now() - startTime}ms`,
+                        duration: duration,
                         eventId: this.context.eventID,
                         timestamp: new Date().toISOString(),
                     });
@@ -285,12 +355,15 @@ class AgentRuntime {
                 logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestStartLog);
                 try {
                     const response = yield originalRequest.download(options);
+                    const duration = `${Date.now() - startTime}ms`;
+                    const statusCode = response === null || response === void 0 ? void 0 : response.statusCode;
                     const requestSuccessLog = safeJsonStringify({
                         type: AgentEventType.FRAMEWORK_EVENT,
                         scene: AgentType.AI_SDK_REQUEST_SUCCESS,
                         method: "DOWNLOAD",
                         url: options.url,
-                        duration: `${Date.now() - startTime}ms`,
+                        status: statusCode,
+                        duration: duration,
                         eventId: this.context.eventID,
                         timestamp: new Date().toISOString(),
                     });
@@ -298,6 +371,7 @@ class AgentRuntime {
                     return response;
                 }
                 catch (error) {
+                    const duration = `${Date.now() - startTime}ms`;
                     const requestErrorLog = safeJsonStringify({
                         type: AgentEventType.FRAMEWORK_EVENT,
                         scene: AgentType.AI_SDK_REQUEST_ERROR,
@@ -305,7 +379,7 @@ class AgentRuntime {
                         url: options.url,
                         error: error instanceof Error ? error.message : "请求失败",
                         stack: error instanceof Error ? error.stack : undefined,
-                        duration: `${Date.now() - startTime}ms`,
+                        duration: duration,
                         eventId: this.context.eventID,
                         timestamp: new Date().toISOString(),
                     });
@@ -328,20 +402,33 @@ class AgentRuntime {
                 logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestStartLog);
                 try {
                     const response = yield originalRequest.fetch(options);
-                    const requestSuccessLog = safeJsonStringify({
+                    const rt = Date.now() - startTime;
+                    const duration = `${rt}ms`;
+                    const isSuccess = (response === null || response === void 0 ? void 0 : response.statusCode) >= 200 && (response === null || response === void 0 ? void 0 : response.statusCode) < 300;
+                    const logData = {
                         type: AgentEventType.FRAMEWORK_EVENT,
-                        scene: AgentType.AI_SDK_REQUEST_SUCCESS,
+                        scene: isSuccess ? AgentType.AI_SDK_REQUEST_SUCCESS : AgentType.AI_SDK_REQUEST_ERROR,
                         method: "FETCH",
                         url: options.url,
-                        status: response.statusCode,
-                        duration: `${Date.now() - startTime}ms`,
+                        duration: duration,
                         eventId: this.context.eventID,
                         timestamp: new Date().toISOString(),
-                    });
-                    logger.logAccesslog(functions_framework_1.LogLevel.INFO, requestSuccessLog);
+                    };
+                    if (isSuccess) {
+                        logData.status = response.statusCode;
+                    }
+                    else {
+                        logData.error = `Request failed with status code: ${response === null || response === void 0 ? void 0 : response.statusCode}`;
+                    }
+                    this.updateModelApmData(options.url, response === null || response === void 0 ? void 0 : response.statusCode, rt);
+                    const requestLog = safeJsonStringify(logData);
+                    logger.logAccesslog(isSuccess ? functions_framework_1.LogLevel.INFO : functions_framework_1.LogLevel.ERROR, requestLog);
                     return response;
                 }
                 catch (error) {
+                    const rt = Date.now() - startTime;
+                    const duration = `${rt}ms`;
+                    this.updateModelApmData(options.url, undefined, rt);
                     const requestErrorLog = safeJsonStringify({
                         type: AgentEventType.FRAMEWORK_EVENT,
                         scene: AgentType.AI_SDK_REQUEST_ERROR,
@@ -349,7 +436,7 @@ class AgentRuntime {
                         url: options.url,
                         error: error instanceof Error ? error.message : "请求失败",
                         stack: error instanceof Error ? error.stack : undefined,
-                        duration: `${Date.now() - startTime}ms`,
+                        duration: duration,
                         eventId: this.context.eventID,
                         timestamp: new Date().toISOString(),
                     });
@@ -360,6 +447,10 @@ class AgentRuntime {
         };
     }
     createModel(modelName) {
+        this.currentModelName = modelName;
+        (0, functions_framework_1.updateApmMeasurementData)({
+            kit_model_name: modelName,
+        });
         const createModelLog = safeJsonStringify({
             type: AgentEventType.FRAMEWORK_EVENT,
             scene: AgentType.AI_SDK_CREATE_MODEL,

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

@@ -7,12 +7,12 @@ export declare const uploadFileInputSchema: z.ZodObject<{
 }, "strip", z.ZodTypeAny, {
     filename?: string;
     purpose?: string;
-    fileContent?: string;
     mimeType?: string;
+    fileContent?: string;
 }, {
     filename?: string;
     purpose?: string;
-    fileContent?: string;
     mimeType?: string;
+    fileContent?: string;
 }>;
 export type UploadFileInputSchema = z.infer<typeof uploadFileInputSchema>;

package/lib/schema/message.schema.d.ts

@@ -918,25 +918,46 @@ export declare const sendMessageInputSchema: z.ZodObject<{
         content?: string | any[] | Record<string, any>;
         name?: string;
     }>]>, "many">;
-    files: z.ZodOptional<z.ZodArray<z.ZodObject<{
-        id: z.ZodString;
-        filename: z.ZodString;
-        purpose: z.ZodString;
-        bytes: z.ZodNumber;
-        created_at: z.ZodNumber;
+    files: z.ZodOptional<z.ZodArray<z.ZodUnion<[z.ZodObject<{
+        id: z.ZodOptional<z.ZodString>;
+        filename: z.ZodOptional<z.ZodString>;
+        purpose: z.ZodOptional<z.ZodString>;
+        bytes: z.ZodOptional<z.ZodNumber>;
+        created_at: z.ZodOptional<z.ZodNumber>;
     }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
-        id: z.ZodString;
-        filename: z.ZodString;
-        purpose: z.ZodString;
-        bytes: z.ZodNumber;
-        created_at: z.ZodNumber;
+        id: z.ZodOptional<z.ZodString>;
+        filename: z.ZodOptional<z.ZodString>;
+        purpose: z.ZodOptional<z.ZodString>;
+        bytes: z.ZodOptional<z.ZodNumber>;
+        created_at: z.ZodOptional<z.ZodNumber>;
     }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
-        id: z.ZodString;
-        filename: z.ZodString;
-        purpose: z.ZodString;
-        bytes: z.ZodNumber;
-        created_at: z.ZodNumber;
-    }, z.ZodTypeAny, "passthrough">>, "many">>;
+        id: z.ZodOptional<z.ZodString>;
+        filename: z.ZodOptional<z.ZodString>;
+        purpose: z.ZodOptional<z.ZodString>;
+        bytes: z.ZodOptional<z.ZodNumber>;
+        created_at: z.ZodOptional<z.ZodNumber>;
+    }, z.ZodTypeAny, "passthrough">>, z.ZodObject<{
+        url: z.ZodOptional<z.ZodString>;
+        type: z.ZodOptional<z.ZodString>;
+        size: z.ZodOptional<z.ZodNumber>;
+        mimeType: z.ZodOptional<z.ZodString>;
+        fileSource: z.ZodOptional<z.ZodString>;
+        filename: z.ZodOptional<z.ZodString>;
+    }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+        url: z.ZodOptional<z.ZodString>;
+        type: z.ZodOptional<z.ZodString>;
+        size: z.ZodOptional<z.ZodNumber>;
+        mimeType: z.ZodOptional<z.ZodString>;
+        fileSource: z.ZodOptional<z.ZodString>;
+        filename: z.ZodOptional<z.ZodString>;
+    }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+        url: z.ZodOptional<z.ZodString>;
+        type: z.ZodOptional<z.ZodString>;
+        size: z.ZodOptional<z.ZodNumber>;
+        mimeType: z.ZodOptional<z.ZodString>;
+        fileSource: z.ZodOptional<z.ZodString>;
+        filename: z.ZodOptional<z.ZodString>;
+    }, z.ZodTypeAny, "passthrough">>]>, "many">>;
 }, "strip", z.ZodTypeAny, {
     msg?: string;
     history?: ({
@@ -978,13 +999,20 @@ export declare const sendMessageInputSchema: z.ZodObject<{
         content?: string | any[] | Record<string, any>;
         name?: string;
     })[];
-    files?: z.objectOutputType<{
-        id: z.ZodString;
-        filename: z.ZodString;
-        purpose: z.ZodString;
-        bytes: z.ZodNumber;
-        created_at: z.ZodNumber;
-    }, z.ZodTypeAny, "passthrough">[];
+    files?: (z.objectOutputType<{
+        id: z.ZodOptional<z.ZodString>;
+        filename: z.ZodOptional<z.ZodString>;
+        purpose: z.ZodOptional<z.ZodString>;
+        bytes: z.ZodOptional<z.ZodNumber>;
+        created_at: z.ZodOptional<z.ZodNumber>;
+    }, z.ZodTypeAny, "passthrough"> | z.objectOutputType<{
+        url: z.ZodOptional<z.ZodString>;
+        type: z.ZodOptional<z.ZodString>;
+        size: z.ZodOptional<z.ZodNumber>;
+        mimeType: z.ZodOptional<z.ZodString>;
+        fileSource: z.ZodOptional<z.ZodString>;
+        filename: z.ZodOptional<z.ZodString>;
+    }, z.ZodTypeAny, "passthrough">)[];
 }, {
     msg?: string;
     history?: ({
@@ -1026,13 +1054,20 @@ export declare const sendMessageInputSchema: z.ZodObject<{
         content?: string | any[] | Record<string, any>;
         name?: string;
     })[];
-    files?: z.objectInputType<{
-        id: z.ZodString;
-        filename: z.ZodString;
-        purpose: z.ZodString;
-        bytes: z.ZodNumber;
-        created_at: z.ZodNumber;
-    }, z.ZodTypeAny, "passthrough">[];
+    files?: (z.objectInputType<{
+        id: z.ZodOptional<z.ZodString>;
+        filename: z.ZodOptional<z.ZodString>;
+        purpose: z.ZodOptional<z.ZodString>;
+        bytes: z.ZodOptional<z.ZodNumber>;
+        created_at: z.ZodOptional<z.ZodNumber>;
+    }, z.ZodTypeAny, "passthrough"> | z.objectInputType<{
+        url: z.ZodOptional<z.ZodString>;
+        type: z.ZodOptional<z.ZodString>;
+        size: z.ZodOptional<z.ZodNumber>;
+        mimeType: z.ZodOptional<z.ZodString>;
+        fileSource: z.ZodOptional<z.ZodString>;
+        filename: z.ZodOptional<z.ZodString>;
+    }, z.ZodTypeAny, "passthrough">)[];
 }>;
 export type TextContent = z.infer<typeof textContentSchema>;
 export type ImageContent = z.infer<typeof imageContentSchema>;

package/lib/schema/message.schema.js

@@ -66,14 +66,26 @@ exports.sendMessageInputSchema = zod_1.z.object({
     msg: zod_1.z.string(),
     history: exports.messageHistorySchema,
     files: zod_1.z
-        .array(zod_1.z
-        .object({
-        id: zod_1.z.string(),
-        filename: zod_1.z.string(),
-        purpose: zod_1.z.string(),
-        bytes: zod_1.z.number(),
-        created_at: zod_1.z.number(),
-    })
-        .passthrough())
+        .array(zod_1.z.union([
+        zod_1.z
+            .object({
+            id: zod_1.z.string().optional(),
+            filename: zod_1.z.string().optional(),
+            purpose: zod_1.z.string().optional(),
+            bytes: zod_1.z.number().optional(),
+            created_at: zod_1.z.number().optional(),
+        })
+            .passthrough(),
+        zod_1.z
+            .object({
+            url: zod_1.z.string().optional(),
+            type: zod_1.z.string().optional(),
+            size: zod_1.z.number().optional(),
+            mimeType: zod_1.z.string().optional(),
+            fileSource: zod_1.z.string().optional(),
+            filename: zod_1.z.string().optional(),
+        })
+            .passthrough(),
+    ]))
         .optional(),
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vectorx/agent-runtime",
-  "version": "0.6.1",
+  "version": "0.7.0",
   "description": "Cloud AI agent runtime",
   "main": "lib/index.js",
   "files": [
@@ -20,8 +20,8 @@
     "node": ">=18.0.0"
   },
   "dependencies": {
-    "@vectorx/ai-sdk": "0.6.1",
-    "@vectorx/functions-framework": "0.6.1",
+    "@vectorx/ai-sdk": "0.7.0",
+    "@vectorx/functions-framework": "0.7.0",
     "form-data": "^4.0.0",
     "langfuse": "^3.38.4",
     "query-string": "^6.14.1",