@aigne/core 1.21.0 → 1.22.0

package/lib/esm/aigne/context.js

@@ -1,3 +1,4 @@
+import { SpanStatusCode, context, trace } from "@opentelemetry/api";
 import equal from "fast-deep-equal";
 import { Emitter } from "strict-event-emitter";
 import { v7 } from "uuid";
@@ -6,6 +7,7 @@ import { Agent, isAgentResponseDelta, isEmptyChunk, } from "../agents/agent.js";
 import { isTransferAgentOutput, transferAgentOutputKey, } from "../agents/types.js";
 import { UserAgent } from "../agents/user-agent.js";
 import { AgentResponseProgressStream } from "../utils/event-stream.js";
+import { logger } from "../utils/logger.js";
 import { promiseWithResolvers } from "../utils/promise.js";
 import { agentResponseStreamToObject, asyncGeneratorToReadableStream, mergeReadableStreams, onAgentResponseStreamEnd, } from "../utils/stream-utils.js";
 import { checkArguments, isEmpty, isNil, omit, } from "../utils/type-utils.js";
@@ -15,24 +17,47 @@ import { newEmptyContextUsage } from "./usage.js";
  * @hidden
  */
 export class AIGNEContext {
-    constructor(...[parent, ...args]) {
-        if (parent instanceof AIGNEContext) {
-            this.parentId = parent.id;
+    constructor(parent, { reset } = {}) {
+        const tracer = parent?.observer?.tracer;
+        if (parent instanceof AIGNEContext && !reset) {
             this.internal = parent.internal;
+            this.parentId = parent.id;
+            this.rootId = parent.rootId;
+            if (parent.span) {
+                const parentContext = trace.setSpan(context.active(), parent.span);
+                this.span = tracer?.startSpan("childAIGNEContext", undefined, parentContext);
+            }
+            else {
+                if (!process.env.AIGNE_OBSERVABILITY_DISABLED) {
+                    throw new Error("parent span is not set");
+                }
+            }
         }
         else {
-            this.internal = new AIGNEContextShared(parent, ...args);
+            this.internal = new AIGNEContextShared(parent);
+            this.span = tracer?.startSpan("AIGNEContext");
+            // 修改了 rootId 是否会之前的有影响？，之前为 this.id
+            this.rootId = this.span?.spanContext?.().traceId ?? v7();
         }
+        this.id = this.span?.spanContext()?.spanId ?? v7();
     }
+    id;
     parentId;
-    id = v7();
+    rootId;
+    span;
     internal;
+    get messageQueue() {
+        return this.internal.messageQueue;
+    }
     get model() {
         return this.internal.model;
     }
     get skills() {
         return this.internal.skills;
     }
+    get observer() {
+        return this.internal.observer;
+    }
     get limits() {
         return this.internal.limits;
     }
@@ -55,9 +80,7 @@ export class AIGNEContext {
         this.internal.memories = memories;
     }
     newContext({ reset } = {}) {
-        if (reset)
-            return new AIGNEContext(this, { userContext: {} });
-        return new AIGNEContext(this);
+        return new AIGNEContext(this, { reset });
     }
     invoke = ((agent, message, options) => {
         checkArguments("AIGNEContext.invoke", aigneContextInvokeArgsSchema, {
@@ -79,7 +102,7 @@ export class AIGNEContext {
                 activeAgent: agent,
             });
         }
-        const newContext = this.newContext();
+        const newContext = options?.newContext === false ? this : this.newContext();
         return Promise.resolve(newContext.internal.invoke(agent, message, newContext, options)).then(async (response) => {
             if (!options?.streaming) {
                 let { __activeAgent__: activeAgent, ...output } = await agentResponseStreamToObject(response);
@@ -156,8 +179,55 @@ export class AIGNEContext {
             timestamp: Date.now(),
         };
         const newArgs = [b, ...args.slice(1)];
+        this.trace(eventName, args, b);
         return this.internal.events.emit(eventName, ...newArgs);
     }
+    async trace(eventName, args, b) {
+        const span = this.span;
+        if (!span)
+            return;
+        try {
+            switch (eventName) {
+                case "agentStarted": {
+                    const { agent, input } = args[0];
+                    span.updateName(agent.name);
+                    span.setAttribute("custom.trace_id", this.rootId);
+                    span.setAttribute("custom.span_id", this.id);
+                    if (this.parentId) {
+                        span.setAttribute("custom.parent_id", this.parentId);
+                    }
+                    span.setAttribute("custom.started_at", b.timestamp);
+                    span.setAttribute("input", JSON.stringify(input));
+                    span.setAttribute("agentTag", agent.tag ?? "UnknownAgent");
+                    break;
+                }
+                case "agentSucceed": {
+                    const { output } = args[0];
+                    try {
+                        span.setAttribute("output", JSON.stringify(output));
+                    }
+                    catch (_e) {
+                        logger.error("parse output error", _e.message);
+                        span.setAttribute("output", JSON.stringify({}));
+                    }
+                    span.setStatus({ code: SpanStatusCode.OK, message: "Agent succeed" });
+                    span.end();
+                    await this.observer?.traceExporter?.forceFlush?.();
+                    break;
+                }
+                case "agentFailed": {
+                    const { error } = args[0];
+                    span.setStatus({ code: SpanStatusCode.ERROR, message: error.message });
+                    span.end();
+                    await this.observer?.traceExporter?.forceFlush?.();
+                    break;
+                }
+            }
+        }
+        catch (err) {
+            logger.error("AIGNEContext.trace observer error", { eventName, error: err });
+        }
+    }
     on(eventName, listener) {
         this.internal.events.on(eventName, listener);
         return this;
@@ -173,11 +243,10 @@ export class AIGNEContext {
 }
 class AIGNEContextShared {
     parent;
-    constructor(parent, overrides) {
+    span;
+    constructor(parent) {
         this.parent = parent;
         this.messageQueue = this.parent?.messageQueue ?? new MessageQueue();
-        this.userContext = overrides?.userContext ?? {};
-        this.memories = overrides?.memories ?? [];
     }
     messageQueue;
     events = new Emitter();
@@ -187,12 +256,15 @@ class AIGNEContextShared {
     get skills() {
         return this.parent?.skills;
     }
+    get observer() {
+        return this.parent?.observer;
+    }
     get limits() {
         return this.parent?.limits;
     }
     usage = newEmptyContextUsage();
-    userContext;
-    memories;
+    userContext = {};
+    memories = [];
     abortController = new AbortController();
     timer;
     initTimeout() {

package/lib/esm/loader/agent-yaml.d.ts

@@ -4,8 +4,8 @@ import { ProcessMode } from "../agents/team-agent.js";
 export declare function loadAgentFromYamlFile(path: string): Promise<{
     type: "mcp";
     url?: string | undefined;
-    args?: string[] | undefined;
     command?: string | undefined;
+    args?: string[] | undefined;
 } | {
     type: "team";
     name: string;

package/lib/esm/memory/memory.d.ts

@@ -21,6 +21,7 @@ export interface MemoryAgentOptions extends Partial<Pick<MemoryAgent, "recorder"
  * instead provides memory management capabilities to the system.
  */
 export declare class MemoryAgent extends Agent {
+    tag: string;
     /**
      * Creates a new MemoryAgent instance.
      */

package/lib/esm/memory/memory.js

@@ -12,6 +12,7 @@ export const newMemoryId = () => v7();
  * instead provides memory management capabilities to the system.
  */
 export class MemoryAgent extends Agent {
+    tag = "MemoryAgent";
     /**
      * Creates a new MemoryAgent instance.
      */

package/lib/esm/memory/recorder.d.ts

@@ -77,6 +77,7 @@ export declare const memoryRecorderOutputSchema: z.ZodObject<{
  * implementations of the process method to handle the actual storage logic.
  */
 export declare abstract class MemoryRecorder extends Agent<MemoryRecorderInput, MemoryRecorderOutput> {
+    tag: string;
     /**
      * Creates a new MemoryRecorder instance with predefined input and output schemas.
      *

package/lib/esm/memory/recorder.js

@@ -31,6 +31,7 @@ export const memoryRecorderOutputSchema = z.object({
  * implementations of the process method to handle the actual storage logic.
  */
 export class MemoryRecorder extends Agent {
+    tag = "MemoryRecorderAgent";
     /**
      * Creates a new MemoryRecorder instance with predefined input and output schemas.
      *

package/lib/esm/memory/retriever.d.ts

@@ -90,6 +90,7 @@ export declare const memoryRetrieverOutputSchema: z.ZodObject<{
  * implementations of the process method to handle the actual retrieval logic.
  */
 export declare abstract class MemoryRetriever extends Agent<MemoryRetrieverInput, MemoryRetrieverOutput> {
+    tag: string;
     /**
      * Creates a new MemoryRetriever instance with predefined input and output schemas.
      *

package/lib/esm/memory/retriever.js

@@ -32,6 +32,7 @@ export const memoryRetrieverOutputSchema = z.object({
  * implementations of the process method to handle the actual retrieval logic.
  */
 export class MemoryRetriever extends Agent {
+    tag = "MemoryRetrieverAgent";
     /**
      * Creates a new MemoryRetriever instance with predefined input and output schemas.
      *

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.21.0",
+  "version": "1.22.0",
   "description": "AIGNE core library for building AI-powered applications",
   "publishConfig": {
     "access": "public"
@@ -81,6 +81,7 @@
     "yaml": "^2.7.1",
     "zod": "^3.24.4",
     "zod-to-json-schema": "^3.24.5",
+    "@aigne/observability": "^0.1.0",
     "@aigne/platform-helpers": "^0.1.2"
   },
   "devDependencies": {