@looopy-ai/core 2.1.1 → 2.1.3

package/dist/core/agent.js

@@ -1,5 +1,6 @@
 import { catchError, concat, filter, Observable, of, tap } from 'rxjs';
 import { createTaskStatusEvent } from '../events';
+import { isChildTaskEvent } from '../events/utils';
 import { addMessagesCompactedEvent, addMessagesLoadedEvent, completeAgentInitializeSpan, completeAgentTurnSpan, failAgentInitializeSpan, failAgentTurnSpan, setResumeAttributes, setTurnCountAttribute, startAgentInitializeSpan, startAgentTurnSpan, } from '../observability/spans';
 import { serializeError } from '../utils/error';
 import { getLogger } from './logger';
@@ -178,9 +179,12 @@ export class Agent {
                         maxIterations: 5,
                         stopOnToolError: false,
                     }, messages).pipe(tap(async (event) => {
+                        if (isChildTaskEvent(event))
+                            return;
                         switch (event.kind) {
                             case 'content-complete':
                                 if (event.content || event.toolCalls) {
+                                    logger.debug({ event }, 'Saving content-complete to message store');
                                     await this.config.messageStore.append(this.config.contextId, [
                                         {
                                             role: 'assistant',
@@ -198,6 +202,7 @@ export class Agent {
                                 }
                                 break;
                             case 'tool-complete': {
+                                logger.debug({ event }, 'Saving tool-complete to message store');
                                 const message = {
                                     role: 'tool',
                                     content: JSON.stringify({
@@ -207,12 +212,11 @@ export class Agent {
                                     }),
                                     toolCallId: event.toolCallId,
                                 };
-                                logger.debug({ message }, 'Saving tool message to message store');
                                 await this.config.messageStore.append(this.config.contextId, [message]);
                                 break;
                             }
                             case 'internal:tool-message':
-                                logger.debug({ message: event.message }, 'Saving internal tool message to message store');
+                                logger.debug({ event }, 'Saving internal:tool-message to message store');
                                 await this.config.messageStore.append(this.config.contextId, [event.message]);
                                 break;
                             default:

package/dist/core/loop.js

@@ -1,5 +1,6 @@
 import { concat, EMPTY, mergeMap, of, reduce, shareReplay } from 'rxjs';
 import { createTaskCompleteEvent, createTaskCreatedEvent, createTaskStatusEvent } from '../events';
+import { isChildTaskEvent } from '../events/utils';
 import { startAgentLoopSpan } from '../observability/spans';
 import { recursiveMerge } from '../utils/recursive-merge';
 import { runIteration } from './iteration';
@@ -37,8 +38,8 @@ export const runLoop = (context, config, history) => {
     }, state.messages), (state, { events }) => ({
         ...state,
         messages: [...state.messages, ...eventsToMessages(events)],
-    }), (e) => e.kind === 'content-complete' && e.finishReason !== 'tool_calls').pipe(shareReplay({ refCount: true }));
-    const finalSummary$ = merged$.pipe(reduce((last, e) => (e.kind === 'content-complete' ? e : last), null), mergeMap((last) => {
+    }), (e) => !isChildTaskEvent(e) && e.kind === 'content-complete' && e.finishReason !== 'tool_calls').pipe(shareReplay({ refCount: true }));
+    const finalSummary$ = merged$.pipe(reduce((last, e) => (!isChildTaskEvent(e) && e.kind === 'content-complete' ? e : last), null), mergeMap((last) => {
         if (!last)
             return EMPTY;
         return of(createTaskCompleteEvent({

package/dist/core/tools.js

@@ -1,4 +1,5 @@
 import { catchError, concat, defer, mergeMap, of, tap } from 'rxjs';
+import { isChildTaskEvent } from '../events/utils';
 import { startToolExecuteSpan } from '../observability/spans';
 import { toolErrorEvent } from '../tools/tool-result-events';
 export const runToolCall = (context, toolCall) => {
@@ -42,6 +43,8 @@ export const runToolCall = (context, toolCall) => {
             try {
                 logger.trace({ providerName: provider.name }, 'Executing tool');
                 return provider.execute(toolCallInput, context).pipe(tap((event) => {
+                    if (isChildTaskEvent(event))
+                        return;
                     if (event.kind !== 'tool-complete') {
                         return;
                     }

package/dist/events/utils.d.ts

@@ -1,4 +1,4 @@
-import type { AuthCompletedEvent, AuthRequiredEvent, AuthType, ContentCompleteEvent, ContentDeltaEvent, ContextAnyEvent, ContextEvent, DatasetWriteEvent, DataWriteEvent, FileWriteEvent, FinishReason, InputProvider, InputReceivedEvent, InputRequiredEvent, InputType, InternalCheckpointEvent, InternalLLMCallEvent, InternalThoughtProcessEvent, JSONSchema, SubtaskCreatedEvent, TaskCompleteEvent, TaskCreatedEvent, TaskInitiator, TaskStatus, TaskStatusEvent, ThoughtStreamEvent, ThoughtType, ThoughtVerbosity, ToolCompleteEvent, ToolProgressEvent, ToolStartEvent } from '../types/event';
+import type { AnyEvent, AuthCompletedEvent, AuthRequiredEvent, AuthType, ContentCompleteEvent, ContentDeltaEvent, ContextAnyEvent, ContextEvent, DatasetWriteEvent, DataWriteEvent, FileWriteEvent, FinishReason, InputProvider, InputReceivedEvent, InputRequiredEvent, InputType, InternalCheckpointEvent, InternalLLMCallEvent, InternalThoughtProcessEvent, JSONSchema, SubtaskCreatedEvent, TaskCompleteEvent, TaskCreatedEvent, TaskInitiator, TaskStatus, TaskStatusEvent, ThoughtStreamEvent, ThoughtType, ThoughtVerbosity, ToolCompleteEvent, ToolProgressEvent, ToolStartEvent } from '../types/event';
 export declare function generateEventId(): string;
 export interface CreateTaskCreatedEventOptions {
     contextId: string;
@@ -248,3 +248,4 @@ export declare function filterByContextId(events: ContextAnyEvent[], contextId:
 export declare function filterByKind<K extends ContextAnyEvent['kind']>(events: ContextAnyEvent[], kind: K): Extract<ContextAnyEvent, {
     kind: K;
 }>[];
+export declare const isChildTaskEvent: (event: AnyEvent) => boolean;

package/dist/events/utils.js

@@ -261,3 +261,6 @@ export function filterByContextId(events, contextId) {
 export function filterByKind(events, kind) {
     return events.filter((event) => event.kind === kind);
 }
+export const isChildTaskEvent = (event) => {
+    return !!event.parentTaskId;
+};

package/dist/observability/spans/agent-turn.js

@@ -1,5 +1,6 @@
 import { context, SpanStatusCode, trace } from '@opentelemetry/api';
 import { tap } from 'rxjs/internal/operators/tap';
+import { isChildTaskEvent } from '../../events/utils';
 import { SpanAttributes } from '../tracing';
 const safeName = (name) => name.replace(/[^a-zA-Z0-9_-]+/g, '-');
 export const startAgentTurnSpan = (params) => {
@@ -20,6 +21,8 @@ export const startAgentTurnSpan = (params) => {
         traceContext,
         tapFinish: tap({
             next: (event) => {
+                if (isChildTaskEvent(event))
+                    return;
                 switch (event.kind) {
                     case 'task-complete':
                         if (event.content) {

package/dist/observability/spans/iteration.js

@@ -1,5 +1,6 @@
 import { SpanStatusCode, trace } from '@opentelemetry/api';
 import { tap } from 'rxjs/internal/operators/tap';
+import { isChildTaskEvent } from '../../events/utils';
 import { SpanAttributes, SpanNames } from '../tracing';
 export const startLoopIterationSpan = (context, iteration) => {
     const logger = context.logger;
@@ -20,6 +21,8 @@ export const startLoopIterationSpan = (context, iteration) => {
         traceContext,
         tapFinish: tap({
             next: (event) => {
+                if (isChildTaskEvent(event))
+                    return;
                 if (event.kind === 'content-complete') {
                     if (event.content) {
                         span.setAttribute(SpanAttributes.OUTPUT, event.content);

package/dist/observability/spans/llm-call.js

@@ -1,5 +1,6 @@
 import { SpanStatusCode, trace } from '@opentelemetry/api';
 import { tap } from 'rxjs/internal/operators/tap';
+import { isChildTaskEvent } from '../../events/utils';
 import { SpanAttributes, SpanNames } from '../tracing';
 export const startLLMCallSpan = (context, systemPrompt, messages, tools) => {
     const tracer = trace.getTracer('looopy');
@@ -20,6 +21,8 @@ export const startLLMCallSpan = (context, systemPrompt, messages, tools) => {
         traceContext,
         tapFinish: tap({
             next: (event) => {
+                if (isChildTaskEvent(event))
+                    return;
                 switch (event.kind) {
                     case 'content-complete':
                         if (event.content) {

package/dist/observability/spans/loop.js

@@ -1,5 +1,6 @@
 import { SpanStatusCode, trace } from '@opentelemetry/api';
 import { tap } from 'rxjs/internal/operators/tap';
+import { isChildTaskEvent } from '../../events/utils';
 import { SpanAttributes, SpanNames } from '../tracing';
 export const startAgentLoopSpan = (params) => {
     const tracer = trace.getTracer('looopy');
@@ -18,6 +19,8 @@ export const startAgentLoopSpan = (params) => {
         traceContext,
         tapFinish: tap({
             next: (event) => {
+                if (isChildTaskEvent(event))
+                    return;
                 switch (event.kind) {
                     case 'content-complete':
                         if (event.content) {

package/dist/observability/spans/tool.js

@@ -1,5 +1,6 @@
 import { context as otelContext, SpanStatusCode, trace } from '@opentelemetry/api';
 import { tap } from 'rxjs';
+import { isChildTaskEvent } from '../../events/utils';
 import { SpanAttributes, SpanNames } from '../tracing';
 export const startToolExecuteSpan = (context, toolStart) => {
     const tracer = trace.getTracer('looopy');
@@ -19,7 +20,7 @@ export const startToolExecuteSpan = (context, toolStart) => {
         traceContext,
         tapFinish: tap({
             next: (event) => {
-                if (!isToolCompleteEvent(event)) {
+                if (isChildTaskEvent(event) || !isToolCompleteEvent(event)) {
                     return;
                 }
                 try {

package/dist/tools/agent-tool-provider.js

@@ -1,4 +1,5 @@
 import { consumeSSEStream } from '@geee-be/sse-stream-parser';
+import { trace } from '@opentelemetry/api';
 import { Observable } from 'rxjs';
 import z from 'zod';
 import { getLogger } from '../core';
@@ -96,6 +97,8 @@ export class AgentToolProvider {
                     subscriber.complete();
                     return;
                 }
+                const span = trace.getSpan(context.parentContext);
+                span?.updateName(`agent.invoke[${toolCall.function.name}]`);
                 const res = await fetch(`${this.card.url}/invocations?qualifier=DEFAULT`, {
                     method: 'POST',
                     headers: {

package/dist/types/event.d.ts

@@ -315,6 +315,9 @@ export type ContextEvent<T> = T & {
     contextId: string;
     taskId: string;
 };
+export type ChildTaskEvent<T> = T & {
+    parentTaskId: string;
+};
 export type ContextAnyEvent = ContextEvent<AnyEvent>;
 export type ExternalEvent = Exclude<AnyEvent, InternalDebugEvent>;
 export type DebugEvent = InternalDebugEvent;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@looopy-ai/core",
-  "version": "2.1.1",
+  "version": "2.1.3",
   "description": "RxJS-based AI agent framework",
   "keywords": [
     "agent",