overload log for agents

Author: Andrei Bratu

src/humanloop.client.ts

@@ -8,6 +8,7 @@ import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
 import { AnthropicInstrumentation } from "@traceloop/instrumentation-anthropic";
 import { CohereInstrumentation } from "@traceloop/instrumentation-cohere";
 import { OpenAIInstrumentation } from "@traceloop/instrumentation-openai";
+import { Agents } from "api/resources/agents/client/Client";
 
 import { HumanloopClient as BaseHumanloopClient } from "./Client";
 import { ChatMessage } from "./api";
@@ -216,6 +217,7 @@ export class HumanloopClient extends BaseHumanloopClient {
     protected readonly _flows_overloaded: Flows;
     protected readonly _tools_overloaded: Tools;
     protected readonly _evaluators_overloaded: Evaluators;
+    protected readonly _agents_overloaded: Agents;
     protected readonly instrumentProviders: {
         OpenAI?: any;
         Anthropic?: any;
@@ -276,6 +278,8 @@ export class HumanloopClient extends BaseHumanloopClient {
 
         this._evaluators_overloaded = overloadLog(super.evaluators);
 
+        this._agents_overloaded = overloadLog(super.agents);
+
         this._evaluations = new ExtendedEvaluations(_options, this);
 
         // Initialize the tracer singleton
@@ -571,6 +575,10 @@ ${RESET}`,
         return this._evaluations;
     }
 
+    public get agents(): Agents {
+        return this._agents_overloaded;
+    }
+
     public get prompts(): Prompts {
         return this._prompts_overloaded;
     }

src/overload.ts

@@ -1,3 +1,4 @@
+import { Agents } from "api/resources/agents/client/Client";
 import {
     CreateEvaluatorLogRequest,
     FlowLogRequest,
@@ -12,7 +13,7 @@ import { Tools } from "./api/resources/tools/client/Client";
 import { getDecoratorContext, getEvaluationContext, getTraceId } from "./context";
 import { HumanloopRuntimeError } from "./error";
 
-export function overloadLog<T extends Flows | Prompts | Tools | Evaluators>(
+export function overloadLog<T extends Flows | Prompts | Tools | Evaluators | Agents>(
     client: T,
 ): T {
     const originalLog = client.log.bind(client);
@@ -35,7 +36,9 @@ export function overloadLog<T extends Flows | Prompts | Tools | Evaluators>(
                 ? Tools.RequestOptions
                 : T extends Evaluators
                   ? Evaluators.RequestOptions
-                  : never,
+                  : T extends Agents
+                    ? Agents.RequestOptions
+                    : never,
     ) => {
         const traceId = getTraceId();
         if (traceId !== undefined && client instanceof Flows) {