@zespan/sdk 1.0.1 → 1.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/index.d.mts +8 -29
- package/dist/index.d.ts +8 -29
- package/dist/index.js +12 -8
- package/dist/index.mjs +12 -8
- package/package.json +1 -1
package/dist/index.mjs
CHANGED
|
@@ -1,10 +1,14 @@
|
|
|
1
|
-
var
|
|
2
|
-
`)));try{let e=await this._post(n,0);e.ok?(this.queue.splice(0,t.length),this.c.debug&&console.log(`[lumiq] flushed ${t.length} event(s) \u2192 ${this.c.endpoint}`),this.consecutiveFlushFailures>0&&this.c.debug&&console.warn("[lumiq] ingest recovered",{recoveredBatchSize:t.length,queuedEvents:this.queue.length,priorFailures:this.consecutiveFlushFailures}),this.consecutiveFlushFailures=0):(this.consecutiveFlushFailures+=1,(this.c.debug||this.consecutiveFlushFailures===1||this.consecutiveFlushFailures%5===0)&&console.error("[lumiq] ingest flush failed; retaining events for retry",{retriable:e.retriable,status:e.status??null,error:e.error??null,queuedEvents:this.queue.length,failedBatchSize:t.length,consecutiveFailures:this.consecutiveFlushFailures}))}finally{this.flushing=!1}}async _post(t,n){let e=s=>new Promise(o=>setTimeout(o,s));try{let s=await fetch(this.c.endpoint,{method:"POST",body:t,headers:{"x-api-key":this.c.apiKey,"content-type":"application/x-ndjson","content-encoding":"gzip"}});if(s.ok){try{let o=await s.json();typeof o?.cv=="number"&&this.configClient&&this.configClient.checkVersion(o.cv)}catch{}return{ok:!0,retriable:!1}}return s.status===429&&n<3?(await e(Number(s.headers.get("retry-after")??5)*1e3),this._post(t,n+1)):s.status>=500&&n<3?(await e(2**n*1e3),this._post(t,n+1)):s.status===429||s.status>=500?(this.c.debug&&console.error("[lumiq] retryable ingest failure",{status:s.status,attempt:n}),{ok:!1,retriable:!0,status:s.status}):(console.error("[lumiq] non-retriable ingest failure",{status:s.status}),{ok:!1,retriable:!1,status:s.status})}catch(s){return n<3?(await e(2**n*1e3),this._post(t,n+1)):(this.c.debug&&console.error("[lumiq] ingest transport error",s),{ok:!1,retriable:!0,error:s instanceof Error?s.message:String(s)})}}};import{trace as Qt,SpanKind as _e,SpanStatusCode as ge,context as Xt}from"@opentelemetry/api";import{BatchSpanProcessor as Yt}from"@opentelemetry/sdk-trace-base";import{OTLPTraceExporter as en}from"@opentelemetry/exporter-trace-otlp-http";import{AsyncHooksContextManager as tn}from"@opentelemetry/context-async-hooks";import{W3CTraceContextPropagator as nn}from"@opentelemetry/core";var Re=null,It=null;function Ne(r){let t=r.serviceName??"zespan-sdk",n=new en({url:r.endpoint??"https://localhost:3001/v1/traces",headers:r.apiKey?{"x-api-key":r.apiKey}:void 0}),e=new Yt(n,{maxQueueSize:100,maxExportBatchSize:10,exportTimeoutMillis:3e4,scheduledDelayMillis:2e3}),{NodeTracerProvider:s}=st("@opentelemetry/sdk-trace-node"),o=new s({resource:{attributes:{"service.name":t,"service.version":"1.0.0","deployment.environment":r.environment??"production"}}});return o.addSpanProcessor(e),o.register({propagator:new nn}),Re=Qt.getTracer(t,"1.0.0"),It=new tn,Xt.setGlobalContextManager(It.enable()),{tracer:Re,provider:o}}function me(){if(!Re)throw new Error("OTel not initialized. Call initOTel() first.");return Re}function vt(r){return me().startSpan(r.name,{kind:r.kind??_e.INTERNAL,attributes:r.attributes??{}})}async function St(r,t,n){let s=me().startSpan(r,{kind:n?.kind??_e.INTERNAL,attributes:n?.attributes??{}});try{let o=await t(s);return s.setStatus({code:ge.OK}),s.end(),o}catch(o){throw s.setStatus({code:ge.ERROR,message:o.message}),s.recordException(o),s.end(),o}}import{createHash as sn}from"crypto";var fe=class{constructor(t){this.client=t}cache=new Map;cacheTTL=300*1e3;async get(t,n={}){let{version:e,label:s,cache:o=!0}=n;if(o){let m=this.getCacheKey(t,e,s),g=this.cache.get(m);if(g&&Date.now()-g.timestamp<this.cacheTTL)return g.prompt}let a=new URLSearchParams;e!==void 0&&a.set("version",e.toString()),s&&a.set("label",s);let d=`${this.client.baseURL}/v1/prompts/${encodeURIComponent(t)}?${a}`,i=await fetch(d,{headers:{"x-api-key":this.client.apiKey}});if(!i.ok)throw new Error(`Failed to fetch prompt: ${i.statusText}`);let l=await i.json(),p=sn("sha256").update(JSON.stringify(l.prompt)).digest("hex"),c={...l,promptHash:p};if(o){let m=this.getCacheKey(t,e,s);this.cache.set(m,{prompt:c,timestamp:Date.now()})}return c}async list(t){let n=new URLSearchParams;t&&n.set("name",t);let e=`${this.client.baseURL}/v1/prompts?${n}`,s=await fetch(e,{headers:{"x-api-key":this.client.apiKey}});if(!s.ok)throw new Error(`Failed to list prompts: ${s.statusText}`);return(await s.json()).prompts||[]}async create(t){let n=`${this.client.baseURL}/v1/prompts`,e=await fetch(n,{method:"POST",headers:{"Content-Type":"application/json","x-api-key":this.client.apiKey},body:JSON.stringify(t)});if(!e.ok)throw new Error(`Failed to create prompt: ${e.statusText}`);let s=await e.json();return this.invalidateCache(t.name),s}async updateLabels(t,n,e){let s=`${this.client.baseURL}/v1/prompts/${encodeURIComponent(t)}/versions/${n}/labels`,o=await fetch(s,{method:"PATCH",headers:{"Content-Type":"application/json","x-api-key":this.client.apiKey},body:JSON.stringify({labels:e})});if(!o.ok)throw new Error(`Failed to update labels: ${o.statusText}`);let a=await o.json();return this.invalidateCache(t),a}compile(t,n){if(t.type==="text"){let e=typeof t.prompt=="string"?t.prompt:t.prompt.content;return Object.entries(n).forEach(([s,o])=>{e=e.replace(new RegExp(`\\{\\{${s}\\}\\}`,"g"),o)}),e}return JSON.stringify(t.prompt)}clearCache(t){t?this.invalidateCache(t):this.cache.clear()}getCacheKey(t,n,e){return n!==void 0?`${t}:v${n}`:e?`${t}:label:${e}`:`${t}:latest`}invalidateCache(t){let n=[];for(let e of this.cache.keys())e.startsWith(`${t}:`)&&n.push(e);n.forEach(e=>this.cache.delete(e))}};var an="[REDACTED]",bt=r=>r.toLowerCase().replace(/[^a-z0-9]/g,""),on=r=>{let t=new Set;for(let n of r){let e=bt(String(n));e&&t.add(e)}return t},Me=(r,t)=>{if(Array.isArray(r))return r.map(s=>Me(s,t));if(!r||typeof r!="object")return r;let n=r,e={};for(let[s,o]of Object.entries(n)){if(t.has(bt(s))){e[s]=an;continue}e[s]=Me(o,t)}return e},wt=(r,t)=>{if(r==null)return"";let n=on(t);if(typeof r=="string")try{let e=JSON.parse(r);return e&&typeof e=="object"?JSON.stringify(Me(e,n)):r}catch{return r}return JSON.stringify(Me(r,n))};function Ct(){return{modelOverrides:new Map,cacheConfig:new Map,rateLimits:new Map,guardrailOverrides:new Map,samplingRate:1,promptVersions:new Map,fallbacks:new Map,retryPolicies:new Map,timeouts:new Map,concurrencyLimits:new Map,abTests:new Map}}function he(r,t){if(r.size!==t.size)return!1;for(let[n,e]of r){let s=t.get(n);if(typeof e=="object"&&e!==null&&typeof s=="object"&&s!==null){if(JSON.stringify(e)!==JSON.stringify(s))return!1}else if(s!==e)return!1}return!0}var qe=class{localVersion=0;config=Ct();listeners=new Map;fetching=!1;baseUrl;apiKey;projectId;constructor(t){this.baseUrl=t.baseUrl.replace(/\/+$/,""),this.apiKey=t.apiKey,this.projectId=t.projectId}checkVersion(t){t>this.localVersion&&!this.fetching&&(this.fetching=!0,this.fetchConfig().then(n=>{n&&(this.localVersion=t,this.applyFullConfig(n))}).catch(()=>{}).finally(()=>{this.fetching=!1}))}getModel(t,n){return this.config.modelOverrides.get(t)??n}getCacheConfig(t){return this.config.cacheConfig.get(t)??null}getSamplingRate(){return this.config.samplingRate}getRuntimeConfig(){return this.config}getLocalVersion(){return this.localVersion}getFallback(t){return this.config.fallbacks.get(t)??null}getRetryPolicy(t){return this.config.retryPolicies.get(t)??null}getTimeout(t){return this.config.timeouts.get(t)??null}getConcurrencyLimit(t){return this.config.concurrencyLimits.get(t)??null}getAbTest(t){return this.config.abTests.get(t)??null}on(t,n){let e=this.listeners.get(t)??[];e.push(n),this.listeners.set(t,e)}emit(t,n){let e=this.listeners.get(t);if(e)for(let s of e)try{s(n)}catch{}}async fetchConfig(){try{let t=`${this.baseUrl}/v1/sdk/config?projectId=${encodeURIComponent(this.projectId)}`,n=await fetch(t,{headers:{"x-api-key":this.apiKey}});return n.ok?(await n.json()).config??null:null}catch{return null}}applyFullConfig(t){let n=this.config,e=this.parseSnapshot(t);this.config=e,he(n.modelOverrides,e.modelOverrides)||this.emit("model_override",Object.fromEntries(e.modelOverrides)),n.samplingRate!==e.samplingRate&&this.emit("sampling_config",{samplingRate:e.samplingRate}),he(n.cacheConfig,e.cacheConfig)||this.emit("cache_config",Object.fromEntries(e.cacheConfig)),he(n.fallbacks,e.fallbacks)||this.emit("fallback_config",Object.fromEntries(e.fallbacks)),he(n.retryPolicies,e.retryPolicies)||this.emit("retry_config",Object.fromEntries(e.retryPolicies)),he(n.timeouts,e.timeouts)||this.emit("timeout_config",Object.fromEntries(e.timeouts)),he(n.abTests,e.abTests)||this.emit("ab_test_config",Object.fromEntries(e.abTests))}parseSnapshot(t){let n=Ct();for(let[e,s]of Object.entries(t))try{let o=typeof s=="string"?JSON.parse(s):s;if(e.startsWith("model_override.")){let a=e.slice(15),d=typeof o=="string"?o:o?.model??o?.value;typeof d=="string"&&n.modelOverrides.set(a,d)}else if(e.startsWith("cache.")){let a=e.slice(6);n.cacheConfig.set(a,{enabled:o?.enabled??!0,ttlSeconds:typeof o?.ttlSeconds=="number"?o.ttlSeconds:300,similarityThreshold:typeof o?.similarityThreshold=="number"?o.similarityThreshold:.85})}else if(e.startsWith("ratelimit.")){let a=e.slice(10);n.rateLimits.set(a,{maxRequests:typeof o?.maxRequests=="number"?o.maxRequests:100,windowSeconds:typeof o?.windowSeconds=="number"?o.windowSeconds:60})}else if(e.startsWith("guardrail.")&&!e.startsWith("guardrail_mode.")){let a=e.slice(10);n.guardrailOverrides.set(a,!!(o?.enabled??o))}else if(e.startsWith("sampling.")){let a=typeof o=="number"?o:o?.rate??o?.sampleRate;typeof a=="number"&&a>=0&&a<=1&&(n.samplingRate=a)}else if(e.startsWith("prompt_version.")){let a=e.slice(15),d=typeof o=="number"?o:o?.version;typeof d=="number"&&n.promptVersions.set(a,d)}else if(e.startsWith("fallback.")){let a=e.slice(9),d=typeof o?.model=="string"?o.model:null,i=Array.isArray(o?.conditions)?o.conditions.filter(l=>typeof l=="string"):[];d&&n.fallbacks.set(a,{model:d,conditions:i})}else if(e.startsWith("retry.")){let a=e.slice(6),d=typeof o?.enabled=="boolean"?o.enabled:!1,i=typeof o?.maxRetries=="number"?o.maxRetries:3,l=typeof o?.backoffMultiplier=="number"?o.backoffMultiplier:2,p=Array.isArray(o?.retryOnStatusCodes)?o.retryOnStatusCodes.filter(c=>typeof c=="number"):[];n.retryPolicies.set(a,{enabled:d,maxRetries:i,backoffMultiplier:l,retryOnStatusCodes:p})}else if(e.startsWith("timeout.")){let a=e.slice(8),d=typeof o?.timeoutMs=="number"?o.timeoutMs:typeof o=="number"?o:null;typeof d=="number"&&d>0&&n.timeouts.set(a,{timeoutMs:d})}else if(e.startsWith("concurrency.")){let a=e.slice(12),d=typeof o=="number"?o:typeof o?.limit=="number"?o.limit:null;typeof d=="number"&&d>0&&n.concurrencyLimits.set(a,d)}else if(e.startsWith("ab_test.")){let a=e.slice(8),d=typeof o?.candidateModel=="string"?o.candidateModel:null,i=typeof o?.splitPercentage=="number"?o.splitPercentage:null;d&&i!==null&&i>=0&&i<=100&&n.abTests.set(a,{candidateModel:d,splitPercentage:i})}}catch{}return n}};import{AsyncLocalStorage as rn}from"async_hooks";import{randomUUID as dn}from"crypto";var at=new rn,u=()=>dn(),G=()=>at.getStore();function W(r,t){return at.run(r,t)}function je(r,t){let n=G(),e={traceId:r.traceId??n?.traceId??u(),spanId:u(),userId:r.userId??n?.userId,sessionId:r.sessionId??n?.sessionId,tags:{...n?.tags,...r.tags},agentId:r.agentId??n?.agentId,agentName:r.agentName??n?.agentName,agentRole:r.agentRole??n?.agentRole,agentFramework:r.agentFramework??n?.agentFramework,parentAgentId:r.agentId?n?.agentId:n?.parentAgentId,conversationTurn:n?.conversationTurn,availableTools:n?.availableTools};return at.run(e,t)}function ot(r,t){return je(t??{},r)}var cn=ot,ln=je;function Ge(){try{return P()}catch{return null}}var it=new Set;function ie(r){it.add(r)}function Q(r){it.delete(r)}function De(){return it.size>0}var Pe=!1,Ee=!1,xt=!1,Se=null,be=null,rt=null,Ss=(async()=>{try{Se=await import("openai/resources/chat/completions")}catch{}try{be=await import("@anthropic-ai/sdk/resources/messages")}catch{}try{rt=await import("@google/generative-ai")}catch{}At()})();function Tt(r){async function t(n,e){let s=Ge();if(!s||De())return r.call(this,n,e);let o=performance.now(),a=G(),d=u(),i=a?.traceId??u(),l=n?.model??"unknown",p=n?.stream===!0;try{let c=await r.call(this,n,e);if(p){let m=c[Symbol.asyncIterator]?.bind(c);return m&&(c[Symbol.asyncIterator]=function(){let g=m(),f=0,S=0,b="",I;return{async next(){let{value:k,done:y}=await g.next();return y?(s.enqueue({event_id:u(),project_id:"",trace_id:i,span_id:d,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"openai",model:l,operation:"chat",latency_ms:performance.now()-o,ttft_ms:I,stream:!0,input_tokens:f,output_tokens:S,finish_reason:b,status:"success",span_kind:"llm",agent_id:a?.agentId,agent_name:a?.agentName,agent_role:a?.agentRole,environment:s.environment,sdk_version:"1.0.0"}),{value:k,done:!0}):(I||(I=performance.now()-o),k?.usage&&(f=k.usage.prompt_tokens??f,S=k.usage.completion_tokens??S),k?.choices?.[0]?.finish_reason&&(b=k.choices[0].finish_reason),{value:k,done:!1})},[Symbol.asyncIterator](){return this}}}),c}return s.enqueue({event_id:u(),project_id:"",trace_id:i,span_id:d,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"openai",model:c.model??l,operation:"chat",latency_ms:performance.now()-o,stream:!1,input_tokens:c.usage?.prompt_tokens??0,output_tokens:c.usage?.completion_tokens??0,cached_tokens:c.usage?.prompt_tokens_details?.cached_tokens??0,finish_reason:c.choices?.[0]?.finish_reason??"",status:"success",span_kind:"llm",agent_id:a?.agentId,agent_name:a?.agentName,agent_role:a?.agentRole,...s.storePrompts?{prompt_text:s.redactForStorage(n.messages),completion_text:s.redactForStorage(c.choices?.[0]?.message?.content??"")}:{},environment:s.environment,sdk_version:"1.0.0"}),c}catch(c){throw s.enqueue({event_id:u(),project_id:"",trace_id:i,span_id:d,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"openai",model:l,operation:"chat",latency_ms:performance.now()-o,stream:p,input_tokens:0,output_tokens:0,status:"error",error_code:String(c.status??"unknown"),error_message:String(c.message).slice(0,500),span_kind:"llm",environment:s.environment,sdk_version:"1.0.0"}),c}}return t.__lumiqAutoPatched=!0,t}function un(){if(!Pe&&Se)try{let r=Se.Completions;if(!r?.prototype)return;let t=r.prototype.create;if(typeof t=="function"&&!t.__lumiqAutoPatched){r.prototype.create=Tt(t),Pe=!0;return}if(typeof t!="function"){if(r.__lumiqAutoPatched){Pe=!0;return}let n=r;class e extends n{constructor(...a){if(super(...a),Object.prototype.hasOwnProperty.call(this,"create")&&!this.create?.__lumiqAutoPatched){let d=this.create;this.create=Tt(d)}}}e.__lumiqAutoPatched=!0;let s=Object.getOwnPropertyDescriptor(Se,"Completions");if(s&&(s.writable!==!1||s.configurable))try{Object.defineProperty(Se,"Completions",{value:e,writable:!0,enumerable:!0,configurable:!0}),Pe=!0}catch{}}}catch{}}function Ot(r){async function t(n,e){let s=Ge();if(!s||De())return r.call(this,n,e);let o=performance.now(),a=G(),d=u(),i=a?.traceId??u(),l=n?.model??"unknown",p=n?.stream===!0;try{let c=await r.call(this,n,e);if(p){let m=c[Symbol.asyncIterator]?.bind(c);return m&&(c[Symbol.asyncIterator]=function(){let g=m(),f=0,S=0,b="",I;return{async next(){let{value:k,done:y}=await g.next();return y?(s.enqueue({event_id:u(),project_id:"",trace_id:i,span_id:d,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:l,operation:"messages",latency_ms:performance.now()-o,ttft_ms:I,stream:!0,input_tokens:f,output_tokens:S,finish_reason:b,status:"success",span_kind:"llm",agent_id:a?.agentId,agent_name:a?.agentName,agent_role:a?.agentRole,environment:s.environment,sdk_version:"1.0.0"}),{value:k,done:!0}):(I||(I=performance.now()-o),k?.type==="message_delta"&&k?.usage&&(S=k.usage.output_tokens??S),k?.type==="message_start"&&k?.message?.usage&&(f=k.message.usage.input_tokens??f),k?.type==="message_delta"&&k?.delta?.stop_reason&&(b=k.delta.stop_reason),{value:k,done:!1})},[Symbol.asyncIterator](){return this}}}),c}return s.enqueue({event_id:u(),project_id:"",trace_id:i,span_id:d,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:c.model??l,operation:"messages",latency_ms:performance.now()-o,stream:!1,input_tokens:c.usage?.input_tokens??0,output_tokens:c.usage?.output_tokens??0,finish_reason:c.stop_reason??"",status:"success",span_kind:"llm",agent_id:a?.agentId,agent_name:a?.agentName,agent_role:a?.agentRole,...s.storePrompts?{prompt_text:s.redactForStorage(n.messages),completion_text:s.redactForStorage(c.content?.filter(m=>m.type==="text").map(m=>m.text).join("")??"")}:{},environment:s.environment,sdk_version:"1.0.0"}),c}catch(c){throw s.enqueue({event_id:u(),project_id:"",trace_id:i,span_id:d,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:l,operation:"messages",latency_ms:performance.now()-o,stream:p,input_tokens:0,output_tokens:0,status:"error",error_code:String(c.status??"unknown"),error_message:String(c.message).slice(0,500),span_kind:"llm",environment:s.environment,sdk_version:"1.0.0"}),c}}return t.__lumiqAutoPatched=!0,t}function pn(){if(!Ee&&be)try{let r=be.Messages;if(!r?.prototype)return;let t=r.prototype.create;if(typeof t=="function"&&!t.__lumiqAutoPatched){r.prototype.create=Ot(t),Ee=!0;return}if(typeof t!="function"){if(r.__lumiqAutoPatched){Ee=!0;return}let n=r;class e extends n{constructor(...a){if(super(...a),Object.prototype.hasOwnProperty.call(this,"create")&&!this.create?.__lumiqAutoPatched){let d=this.create;this.create=Ot(d)}}}e.__lumiqAutoPatched=!0;let s=Object.getOwnPropertyDescriptor(be,"Messages");if(s&&(s.writable!==!1||s.configurable))try{Object.defineProperty(be,"Messages",{value:e,writable:!0,enumerable:!0,configurable:!0}),Ee=!0}catch{}}}catch{}}function mn(){if(!xt&&rt)try{let e=function(s,o){let a=n.call(this,s,o),d=a.generateContent?.bind(a),i=a.generateContentStream?.bind(a);return!d||d.__lumiqAutoPatched||(a.generateContent=async l=>{let p=Ge();if(!p||De())return d(l);let c=performance.now(),m=G(),g=u(),f=m?.traceId??u(),S=s.model??"gemini-1.5-flash";try{let b=await d(l),I=b.response?.usageMetadata;return p.enqueue({event_id:u(),project_id:"",trace_id:f,span_id:g,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:S,operation:"generateContent",latency_ms:performance.now()-c,stream:!1,input_tokens:I?.promptTokenCount??0,output_tokens:I?.candidatesTokenCount??0,finish_reason:b.response?.candidates?.[0]?.finishReason??"",status:"success",span_kind:"llm",environment:p.environment,sdk_version:"1.0.0"}),b}catch(b){throw p.enqueue({event_id:u(),project_id:"",trace_id:f,span_id:g,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:S,operation:"generateContent",latency_ms:performance.now()-c,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_message:String(b.message).slice(0,500),span_kind:"llm",environment:p.environment,sdk_version:"1.0.0"}),b}},a.generateContent.__lumiqAutoPatched=!0,i&&(a.generateContentStream=async l=>{let p=Ge();if(!p||De())return i(l);let c=performance.now(),m=G(),g=u(),f=m?.traceId??u(),S=s.model??"gemini-1.5-flash",b=await i(l),I=b.stream,k=0,y=0,x;return b.stream=(async function*(){try{for await(let _ of I){x||(x=performance.now()-c);let E=_.usageMetadata;E&&(k=E.promptTokenCount??k,y=E.candidatesTokenCount??y),yield _}p.enqueue({event_id:u(),project_id:"",trace_id:f,span_id:g,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:S,operation:"generateContent",latency_ms:performance.now()-c,ttft_ms:x,stream:!0,input_tokens:k,output_tokens:y,status:"success",span_kind:"llm",environment:p.environment,sdk_version:"1.0.0"})}catch(_){throw p.enqueue({event_id:u(),project_id:"",trace_id:f,span_id:g,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:S,operation:"generateContent",latency_ms:performance.now()-c,ttft_ms:x,stream:!0,input_tokens:k,output_tokens:y,status:"error",error_message:String(_.message).slice(0,500),span_kind:"llm",environment:p.environment,sdk_version:"1.0.0"}),_}})(),b})),a};var r=e;let t=rt?.GoogleGenerativeAI;if(!t?.prototype?.getGenerativeModel||t.prototype.getGenerativeModel.__lumiqAutoPatched)return;let n=t.prototype.getGenerativeModel;e.__lumiqAutoPatched=!0,t.prototype.getGenerativeModel=e,xt=!0}catch{}}function At(){un(),pn(),mn()}function Fe(){At()}var we=class{apiKey;endpoint;baseURL;environment;storePrompts;redactKeys;sampleRate;debug;enableOTel;prompts;configClient;projectId;batch;_storePromptsDefault;_promptWarningShown=!1;constructor(t){if(!t.apiKey)throw new Error("Zespan: apiKey is required");t.apiKey.startsWith("zsp_")||console.warn("[zespan] API key should start with zsp_ \u2014 check your key"),this.apiKey=t.apiKey,this.baseURL=t.baseURL??"https://api.zespan.com",this.endpoint=t.endpoint??`${this.baseURL.replace(/\/+$/,"")}/v1/ingest`,this.environment=t.environment??"production",this._storePromptsDefault=t.storePrompts===void 0,this.storePrompts=t.storePrompts??!0,this.redactKeys=t.redactKeys??["password","secret","token","api_key"],this.sampleRate=t.sampleRate??1,this.debug=t.debug??!1,this.enableOTel=t.enableOTel??!1,this.projectId=t.projectId,this.prompts=new fe(this),t.enableZespanPilot!==!1&&t.projectId?this.configClient=new qe({baseUrl:this.baseURL,apiKey:this.apiKey,projectId:t.projectId}):this.configClient=null,this.batch=new Ae(this,{maxSize:t.batchSize??50,flushInterval:t.flushInterval??2e3},this.configClient??void 0),process.once("beforeExit",()=>this.batch.flush()),this.enableOTel&&Ne({endpoint:t.otelEndpoint,serviceName:t.otelServiceName,sampleRate:t.sampleRate,apiKey:this.apiKey,environment:this.environment}),t.autopatch!==!1&&Fe()}enqueue(t){Math.random()>this.sampleRate||(this.storePrompts&&this._storePromptsDefault&&!this._promptWarningShown&&(this._promptWarningShown=!0,console.warn("[zespan] Prompt and completion text is being captured by default. PII (emails, phone numbers, SSNs, credit cards) is automatically redacted server-side. To opt out: { storePrompts: false }")),this.batch.push(t))}redactForStorage(t){return wt(t,this.redactKeys)}resolveGuardrailsUrl(){let t=this.baseURL.replace(/\/+$/,"");return t.endsWith("/v1")?`${t}/guardrails/check`:`${t}/v1/guardrails/check`}async checkGuardrails(t){let n=await fetch(this.resolveGuardrailsUrl(),{method:"POST",headers:{"content-type":"application/json","x-api-key":this.apiKey},body:JSON.stringify(t)});if(!n.ok){let s="";try{s=await n.text()}catch{s=""}throw new Error(`Guardrails check failed (${n.status})${s?`: ${s.slice(0,400)}`:""}`)}let e=await n.json();return{allowed:!!e.allowed,results:Array.isArray(e.results)?e.results:[],modifiedText:typeof e.modifiedText=="string"?e.modifiedText:null}}flush(){return this.batch.flush()}createSpan(t){return me().startSpan(t.name,{kind:t.kind??_e.INTERNAL,attributes:t.attributes??{}})}async withSpan(t,n,e){let o=me().startSpan(t,{kind:e?.kind??_e.INTERNAL,attributes:e?.attributes??{}});try{let a=await n(o);return o.setStatus({code:ge.OK}),a}catch(a){throw o.setStatus({code:ge.ERROR,message:a.message}),o.recordException(a),a}finally{o.end()}}},dt=null,ct=r=>{dt=new we(r)},P=()=>{if(!dt)throw new Error("Call zespan.init() first");return dt};var gn=ct,_n=P;var J=class extends Error{phase;results;constructor(t,n){super(`Guardrail blocked ${t} content`),this.name="GuardrailBlockedError",this.phase=t,this.results=n}};function V(r,t){if(!r)return null;if(r===!0)return{pre:!0,post:!0,failClosed:!1,operation:t};let n=typeof r.pre=="boolean"||typeof r.post=="boolean";return{pre:n?!!r.pre:!0,post:n?!!r.post:!0,failClosed:!!r.failClosed,operation:r.operation??t}}function X(r,t){return t?.modifiedText?t.modifiedText:r}async function B(r,t,n){if(!t||n.phase==="pre"&&!t.pre||n.phase==="post"&&!t.post)return null;let e=n.text?.trim();if(!e)return null;try{let s=await r.checkGuardrails({...n,text:e.slice(0,5e4),operation:n.operation??t.operation});if(!s.allowed)throw new J(n.phase,s.results);return s}catch(s){if(s instanceof J||t.failClosed)throw s;return null}}function fn(r){let t={};try{let n=[],e=[];for(let s of r){let o=typeof s?.content=="string"?s.content:Array.isArray(s?.content)?s.content.map(a=>typeof a=="string"?a:a?.text??"").join(" "):"";s?.role==="system"&&n.push(o),e.push(o)}for(let s of n){try{let o=s.match(/You are the (\w+)Agent/i);if(o){t.agent_name=o[1];break}}catch{}try{let o=s.match(/Agent Name:\s*(\w+)/i);if(o){t.agent_name=o[1];break}}catch{}}for(let s of n)try{let o=s.match(/Role:\s*(orchestrator|specialist|planner|executor|critic)/i);if(o?.[1]){t.agent_role=o[1].toLowerCase();break}}catch{}try{r.length>10&&(t.has_memory=!0)}catch{}try{let s=e.join(" ").toLowerCase();(s.includes("think step by step")||s.includes("let's reason")||s.includes("step 1"))&&(t.has_planning=!0)}catch{}}catch{}return t}function se(r,t){let n=fn(r);try{!n.has_planning&&t&&/^o[13]/.test(t)&&(n.has_planning=!0)}catch{}return n}function ae(r){try{let t=r.filter(s=>s?.role==="user").length,n=r[r.length-1],e=typeof n?.role=="string"?n.role:"user";return{conversation_turn:t,message_role:e}}catch{return{conversation_turn:0,message_role:"user"}}}function oe(r){try{return r.map(t=>{try{return t?.type==="function"&&t?.function?{name:String(t.function.name??""),description:t.function.description?String(t.function.description):void 0,parameters_schema:t.function.parameters??void 0,source:"function"}:t?.name?{name:String(t.name),description:t.description?String(t.description):void 0,parameters_schema:t.input_schema??t.parameters??void 0,source:"function"}:null}catch{return null}}).filter(t=>t!==null)}catch{return[]}}function Le(r){try{let t=r?.[0]?.message?.tool_calls??[],n=[],e={};for(let s of t)try{let o=s?.function?.name;if(typeof o=="string"&&o){n.includes(o)||n.push(o);try{e[o]=typeof s.function.arguments=="string"?JSON.parse(s.function.arguments):s.function.arguments??{}}catch{e[o]=s.function?.arguments??{}}}}catch{}return{tools_used:n,tool_call_args:e}}catch{return{tools_used:[],tool_call_args:{}}}}function Ke(r){try{let t={};for(let s of r)try{let o=typeof s?.index=="number"?s.index:0;t[o]||(t[o]={name:"",argumentsStr:""}),s?.function?.name&&(t[o].name+=s.function.name),s?.function?.arguments&&(t[o].argumentsStr+=s.function.arguments)}catch{}let n=[],e={};for(let s of Object.values(t))try{let o=s.name;if(o&&!n.includes(o)){n.push(o);try{e[o]=s.argumentsStr?JSON.parse(s.argumentsStr):{}}catch{e[o]=s.argumentsStr}}}catch{}return{tools_used:n,tool_call_args:e}}catch{return{tools_used:[],tool_call_args:{}}}}import{createHash as yn}from"crypto";import{SpanKind as kn,SpanStatusCode as Ie,trace as In}from"@opentelemetry/api";async function ye(r,t){if(!t||!t.enabled)return r();let n;for(let e=0;e<=t.maxRetries;e++)try{return await r()}catch(s){if(n=s,t.retryOnStatusCodes.length>0&&!t.retryOnStatusCodes.includes(s?.status))throw s;if(e>=t.maxRetries)break;let o=Math.min(1e3*Math.pow(t.backoffMultiplier,e),3e4);await hn(o)}throw n}function hn(r){return new Promise(t=>setTimeout(t,r))}var lt=class{constructor(t){this.max=t;this.counter=t}counter;queue=[];acquire(){return this.counter>0?(this.counter--,Promise.resolve()):new Promise(t=>{this.queue.push(t)})}release(){this.queue.length>0?this.queue.shift()():this.counter++}async run(t){await this.acquire();try{return await t()}finally{this.release()}}},Rt=new Map;function ke(r,t){let n=Rt.get(r);return n||(n=new lt(t),Rt.set(r,n)),n}var $e=r=>yn("sha256").update(r).digest("hex");function vn(r,t,n="openai"){return{"gen_ai.system":n,"gen_ai.model":t,"gen_ai.operation.name":"chat","gen_ai.prompt":JSON.stringify(r.messages),"gen_ai.request.max_tokens":r.max_tokens,"gen_ai.request.temperature":r.temperature,"gen_ai.request.top_p":r.top_p}}async function*Sn(r,t){let{lumiq:n,t0:e,traceId:s,spanId:o,parent:a,params:d,otelSpan:i,model:l,agentCtx:p,autoDetected:c,conversation_turn:m,message_role:g,tool_definitions:f,provider:S,operation:b}=t,I=S??"openai",k,y=0,x=0,_=0,E="",v=l,T=[],h="",A=[];try{for await(let O of r){k||(k=performance.now()-e,n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:I,model:v,operation:b,latency_ms:k,ttft_ms:k,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:$e(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(d.messages)}:{},user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0",finish_reason:"stream_start",...p,span_kind:"llm",tool_definitions:f,conversation_turn:m,message_role:g,metadata:{auto_detected:Object.keys(c).length?c:void 0}}),i&&i.setAttribute("gen_ai.response.time_to_first_token",k));let q=O.choices?.[0]?.delta?.content;if(q&&(x++,h+=q),O.choices?.[0]?.finish_reason&&(E=O.choices[0].finish_reason),O.model&&(v=O.model),O.usage&&(y=Number(O.usage.prompt_tokens||y||0),x=Number(O.usage.completion_tokens||x||0),_=Number(O.usage.prompt_tokens_details?.cached_tokens||_||0)),O.choices?.[0]?.delta?.tool_calls)for(let $ of O.choices[0].delta.tool_calls)A.push($),$.function?.name&&!T.includes($.function.name)&&T.push($.function.name);yield O}let{tools_used:R,tool_call_args:w}=Ke(A),M=R.length>0?R:T,C=performance.now()-e;n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:I,model:v,operation:b,latency_ms:C,ttft_ms:k,stream:!0,input_tokens:y,output_tokens:x,cached_tokens:_,status:"success",finish_reason:E||"stream_end",tools_used:M,tool_call_args:Object.keys(w).length?w:void 0,prompt_hash:$e(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(d.messages),completion_text:n.redactForStorage(h)}:{},...p,span_kind:"llm",tool_definitions:f,conversation_turn:m,message_role:g,metadata:{usage:{prompt_tokens:y,completion_tokens:x,cached_tokens:_},auto_detected:Object.keys(c).length?c:void 0},user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0"}),i&&(i.setAttribute("gen_ai.usage.output_tokens",x),i.setAttribute("gen_ai.usage.total_tokens",x),i.setAttribute("gen_ai.response.finish_reason",E||"stream_end"),i.setAttribute("gen_ai.response.latency",C))}catch(R){throw n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:I,model:v,operation:b,latency_ms:performance.now()-e,ttft_ms:k,stream:!0,input_tokens:0,output_tokens:x,cost_usd:0,status:"error",error_code:String(R.status??"unknown"),error_message:String(R.message).slice(0,500),...p,span_kind:"llm",conversation_turn:m,message_role:g,metadata:{auto_detected:Object.keys(c).length?c:void 0},user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0"}),i&&(i.setStatus({code:Ie.ERROR,message:R.message}),i.recordException(R)),R}}function We(r,t={}){let n=r,e=n.chat.completions.create.bind(n.chat.completions),s=V(t.guardrails,"chat"),o=t.providerOverride??"openai",a=t.operation??"chat";return n.chat.completions.create=async(d,i)=>{let l=P(),p=performance.now(),c=G(),m=u(),g=c?.traceId??u(),f=d.model??"gpt-4",S;if(l.configClient){let w=l.configClient.getAbTest(a);if(w)Math.random()<w.splitPercentage/100?(f=w.candidateModel,S={ab_test:!0,ab_variant:"candidate"}):S={ab_test:!0,ab_variant:"control"},d={...d,model:f};else{let M=l.configClient.getModel(a,f);if(M!==f){let C=f;f=M,d={...d,model:M},l.enqueue({event_id:u(),project_id:"",trace_id:g,span_id:m,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:o,model:M,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:C,override_model:M},environment:l.environment,sdk_version:"1.0.0"})}}}let b=null,I=null,k=c?.agentId?{agent_id:c.agentId,agent_name:c.agentName,agent_role:c.agentRole,agent_framework:c.agentFramework,parent_agent_id:c.parentAgentId}:{},y=se(d.messages??[],d.model),{conversation_turn:x,message_role:_}=ae(d.messages??[]),E=d.tools?.length?oe(d.tools):void 0,v=null,T=l.enableOTel?In.getTracer("zespan-sdk"):null;T&&(v=T.startSpan(`chat.completions ${f}`,{kind:kn.CLIENT,attributes:vn(d,f,o)}));let h;if(l.configClient){let w=l.configClient.getTimeout(a);w&&(h=AbortSignal.timeout(w.timeoutMs))}let A=l.configClient?.getRetryPolicy(a)??null,R=l.configClient?.getConcurrencyLimit(a)??null;try{b=await B(l,s,{text:JSON.stringify(d.messages??[]),phase:"pre",traceId:g,spanId:m,model:f,operation:a,agentName:c?.agentName});let w=()=>ye(()=>W({traceId:g,spanId:m,userId:c?.userId,sessionId:c?.sessionId,tags:c?.tags},()=>e(d,h?{...i,signal:h}:i)),A),M=await(R!==null?ke(`${l.projectId??"default"}:${a}`,R).run(w):w());if(d.stream===!0)return Sn(M,{lumiq:l,t0:p,traceId:g,spanId:m,parent:c,params:d,otelSpan:v,model:f,agentCtx:k,autoDetected:y,conversation_turn:x,message_role:_,tool_definitions:E,provider:o,operation:a});let{prompt_tokens:O,completion_tokens:q,prompt_tokens_details:$}=M.usage??{},j=$?.cached_tokens??0,Z=M.model??d.model,D=M.usage?.completion_tokens_details?.reasoning_tokens,L=M.choices?.[0]?.message?.content||"";I=await B(l,s,{text:L,phase:"post",traceId:g,spanId:m,model:Z,operation:a,agentName:c?.agentName});let U=X(L,I);U!==L&&typeof M?.choices?.[0]?.message?.content=="string"&&(M.choices[0].message.content=U);let{tools_used:N,tool_call_args:z}=Le(M.choices??[]);return l.enqueue({event_id:u(),project_id:"",trace_id:g,span_id:m,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:o,model:Z,operation:a,latency_ms:performance.now()-p,stream:!1,input_tokens:O??0,output_tokens:q??0,cached_tokens:j,...D?{reasoning_tokens:D}:{},status:"success",finish_reason:M.choices?.[0]?.finish_reason??"",tools_used:N.length>0?N:M.choices?.[0]?.message?.tool_calls?.map(K=>K.function.name)??[],tool_call_args:Object.keys(z).length?z:void 0,prompt_hash:$e(JSON.stringify(d.messages)),...l.storePrompts?{prompt_text:l.redactForStorage(d.messages),completion_text:l.redactForStorage(U)}:{},...k,span_kind:"llm",tool_definitions:E,conversation_turn:x,message_role:_,metadata:{usage:M.usage??null,guardrails:{pre:b?{matched:b.results.length,modified:!!b.modifiedText}:null,post:I?{matched:I.results.length,modified:!!I.modifiedText}:null},auto_detected:Object.keys(y).length?y:void 0,...S??{}},user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:l.environment,sdk_version:"1.0.0"}),v&&(v.setAttribute("gen_ai.usage.input_tokens",O??0),v.setAttribute("gen_ai.usage.output_tokens",q??0),v.setAttribute("gen_ai.usage.total_tokens",(O??0)+(q??0)),v.setAttribute("gen_ai.usage.cached_tokens",j),v.setAttribute("gen_ai.response.finish_reason",M.choices?.[0]?.finish_reason??""),v.setAttribute("gen_ai.response.latency",performance.now()-p),v.setStatus({code:Ie.OK}),v.end()),M}catch(w){if(w?.name==="TimeoutError"||w?.code==="ABORT_ERR"||h&&h.aborted)throw l.enqueue({event_id:u(),project_id:"",trace_id:g,span_id:m,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:o,model:d.model,operation:a,latency_ms:performance.now()-p,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:"timeout",error_message:String(w.message).slice(0,500),...k,span_kind:"llm",conversation_turn:x,message_role:_,metadata:{auto_detected:Object.keys(y).length?y:void 0},user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:l.environment,sdk_version:"1.0.0"}),v&&(v.setStatus({code:Ie.ERROR,message:w.message}),v.recordException(w),v.end()),w;let C=l.configClient?.getFallback(a)??null;if(C&&C.model){let $=String(w.message??"")+String(w.status??"")+String(w.code??"");if(C.conditions.length===0||C.conditions.some(Z=>$.toLowerCase().includes(Z.toLowerCase()))){let Z=f,D={...d,model:C.model};l.enqueue({event_id:u(),project_id:"",trace_id:g,span_id:m,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:o,model:C.model,operation:a,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:Z,fallback_model:C.model,reason:String(w.message).slice(0,200)},environment:l.environment,sdk_version:"1.0.0"});try{let L=await W({traceId:g,spanId:m,userId:c?.userId,sessionId:c?.sessionId,tags:c?.tags},()=>e(D,i)),{prompt_tokens:U,completion_tokens:N,prompt_tokens_details:z}=L.usage??{},K=z?.cached_tokens??0,F=L.model??C.model;return l.enqueue({event_id:u(),project_id:"",trace_id:g,span_id:m,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:o,model:F,operation:a,latency_ms:performance.now()-p,stream:!1,input_tokens:U??0,output_tokens:N??0,cached_tokens:K,status:"success",finish_reason:L.choices?.[0]?.finish_reason??"",prompt_hash:$e(JSON.stringify(d.messages)),...k,span_kind:"llm",conversation_turn:x,message_role:_,metadata:{usage:L.usage??null,fallback_used:!0,original_model:Z,auto_detected:Object.keys(y).length?y:void 0},user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:l.environment,sdk_version:"1.0.0"}),v&&(v.setStatus({code:Ie.OK}),v.end()),L}catch(L){throw v&&(v.setStatus({code:Ie.ERROR,message:L.message}),v.recordException(L),v.end()),L}}}let O=w instanceof J,q=O?"error":w.status===429?"rate_limited":w.status===408?"timeout":"error";throw l.enqueue({event_id:u(),project_id:"",trace_id:g,span_id:m,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:o,model:d.model,operation:a,latency_ms:performance.now()-p,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:q,error_code:O?"guardrail_blocked":String(w.status??"unknown"),error_message:String(w.message).slice(0,500),...k,span_kind:"llm",conversation_turn:x,message_role:_,metadata:{auto_detected:Object.keys(y).length?y:void 0},user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:l.environment,sdk_version:"1.0.0"}),v&&(v.setStatus({code:Ie.ERROR,message:w.message}),v.recordException(w),v.end()),w}},r}import{createHash as bn}from"crypto";import{SpanKind as wn,SpanStatusCode as ve,trace as Cn}from"@opentelemetry/api";var Be=r=>bn("sha256").update(r).digest("hex");function xn(r){let t=Array.isArray(r?.messages)?[...r.messages]:[],n=r?.system;if(typeof n=="string"&&n.trim().length>0)return[{role:"system",content:n},...t];if(Array.isArray(n)&&n.length>0){let e=n.map(s=>typeof s=="string"?s:typeof s?.text=="string"?s.text:"").filter(s=>s.length>0).join(`
|
|
3
|
-
`);if(e.length>0)return[{role:"system",content:e},...t]}return t}function Tn(r,t){return{"gen_ai.system":"anthropic","gen_ai.model":t,"gen_ai.operation.name":"chat","gen_ai.prompt":JSON.stringify(r.messages),"gen_ai.request.max_tokens":r.max_tokens,"gen_ai.request.temperature":r.temperature}}async function*On(r,t){let{lumiq:n,t0:e,traceId:s,spanId:o,parent:a,params:d,otelSpan:i,agentCtx:l,tool_definitions:p,autoDetected:c,conversation_turn:m,message_role:g,operation:f,detectionMessages:S}=t,b,I=0,k=0,y=0,x=0,_="",E=d.model,v=[],T="";try{for await(let A of r){if(!b&&A.type==="content_block_start"&&(b=performance.now()-e,n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:E,operation:f,latency_ms:b,ttft_ms:b,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:Be(JSON.stringify(S)),...n.storePrompts?{prompt_text:n.redactForStorage(S)}:{},user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0",finish_reason:"stream_start",...l,span_kind:"llm",tool_definitions:p,conversation_turn:m,message_role:g,metadata:{auto_detected:Object.keys(c).length?c:void 0}}),i&&i.setAttribute("gen_ai.response.time_to_first_token",b)),A.type==="content_block_delta"&&A.delta?.type==="text_delta"&&(x++,T+=A.delta.text||""),A.type==="message_delta"&&A.delta?.stop_reason&&(_=A.delta.stop_reason),A.type==="message_delta"&&A.usage&&(I=Number(A.usage.input_tokens||I||0),x=Number(A.usage.output_tokens||x||0),k=Number(A.usage.cache_read_input_tokens||k||0)),A.type==="message_start"&&A.message?.usage&&(I=Number(A.message.usage.input_tokens||I||0),x=Number(A.message.usage.output_tokens||x||0),k=Number(A.message.usage.cache_read_input_tokens||k||0)),A.type==="content_block_start"&&A.content_block?.type==="tool_use"){let R=A.content_block.name;R&&!v.includes(R)&&v.push(R)}yield A}let h=performance.now()-e;n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:E,operation:f,latency_ms:h,ttft_ms:b,stream:!0,input_tokens:I,output_tokens:x,cached_tokens:k,reasoning_tokens:y,status:"success",finish_reason:_||"stream_end",tools_used:v,prompt_hash:Be(JSON.stringify(S)),...n.storePrompts?{prompt_text:n.redactForStorage(S),completion_text:n.redactForStorage(T)}:{},...l,span_kind:"llm",tool_definitions:p,conversation_turn:m,message_role:g,metadata:{usage:{input_tokens:I,output_tokens:x,cache_read_input_tokens:k,reasoning_tokens:y},auto_detected:Object.keys(c).length?c:void 0},user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0"}),i&&(i.setAttribute("gen_ai.usage.output_tokens",x),i.setAttribute("gen_ai.response.finish_reason",_||"stream_end"))}catch(h){throw n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:E,operation:f,latency_ms:performance.now()-e,ttft_ms:b,stream:!0,input_tokens:0,output_tokens:x,cost_usd:0,status:"error",error_code:String(h.status??h.type??"unknown"),error_message:String(h.message).slice(0,500),...l,span_kind:"llm",conversation_turn:m,message_role:g,metadata:{auto_detected:Object.keys(c).length?c:void 0},user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0"}),i&&(i.setStatus({code:ve.ERROR,message:h.message}),i.recordException(h)),h}}function Nt(r,t={}){let n=r,e=n.messages.create.bind(n.messages),s=V(t.guardrails,"chat"),o=t.operation??"chat";return n.messages.create=async(a,d)=>{let i=P(),l=performance.now(),p=G(),c=u(),m=p?.traceId??u(),g=a.model??"claude-3-5-sonnet-20241022",f;if(i.configClient){let C=i.configClient.getAbTest(o);if(C)Math.random()<C.splitPercentage/100?(g=C.candidateModel,f={ab_test:!0,ab_variant:"candidate"}):f={ab_test:!0,ab_variant:"control"},a={...a,model:g};else{let O=i.configClient.getModel(o,g);if(O!==g){let q=g;g=O,a={...a,model:O},i.enqueue({event_id:u(),project_id:"",trace_id:m,span_id:c,parent_span_id:p?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:O,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:q,override_model:O},environment:i.environment,sdk_version:"1.0.0"})}}}let S=null,b=null,I=xn(a),k=se(I,g),{conversation_turn:y,message_role:x}=ae(I),_=p?.agentName??k.agent_name,E=p?.agentRole??k.agent_role,v=p?.agentId||_||E?{agent_id:p?.agentId,agent_name:_,agent_role:E,agent_framework:p?.agentFramework,parent_agent_id:p?.parentAgentId}:{},T=a.tools?.length?oe(a.tools):void 0,h=null,A=i.enableOTel?Cn.getTracer("zespan-sdk"):null;A&&(h=A.startSpan(`messages.create ${g}`,{kind:wn.CLIENT,attributes:Tn(a,g)}));let R;if(i.configClient){let C=i.configClient.getTimeout(o);C&&(R=AbortSignal.timeout(C.timeoutMs))}let w=i.configClient?.getRetryPolicy(o)??null,M=i.configClient?.getConcurrencyLimit(o)??null;try{S=await B(i,s,{text:JSON.stringify(a.messages??[]),phase:"pre",traceId:m,spanId:c,model:g,operation:o,agentName:p?.agentName});let C=()=>ye(()=>W({traceId:m,spanId:c,userId:p?.userId,sessionId:p?.sessionId,tags:p?.tags,agentId:p?.agentId,agentName:_,agentRole:E,agentFramework:p?.agentFramework,parentAgentId:p?.parentAgentId,conversationTurn:y,availableTools:T},()=>e(a,R?{...d,signal:R}:d)),w),O=await(M!==null?ke(`${i.projectId??"default"}:${o}`,M).run(C):C());if(a.stream===!0)return On(O,{lumiq:i,t0:l,traceId:m,spanId:c,parent:p,params:a,otelSpan:h,agentCtx:v,tool_definitions:T,autoDetected:k,conversation_turn:y,message_role:x,operation:o,detectionMessages:I});let $=O.usage?.input_tokens??0,j=O.usage?.output_tokens??0,Z=O.usage?.cache_read_input_tokens??0,D=O.model??a.model,L=O.content?.find(F=>F.type==="text")?.text||"",U=(O.content??[]).filter(F=>F.type==="tool_use"),N=U.map(F=>F.name).filter(Boolean),z={};for(let F of U)F.name&&(z[F.name]=F.input??{});b=await B(i,s,{text:L,phase:"post",traceId:m,spanId:c,model:D,operation:o,agentName:p?.agentName});let K=X(L,b);if(K!==L){let F=O.content?.find(Y=>Y.type==="text");F&&typeof F.text=="string"&&(F.text=K)}return i.enqueue({event_id:u(),project_id:"",trace_id:m,span_id:c,parent_span_id:p?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:D,operation:o,latency_ms:performance.now()-l,stream:!1,input_tokens:$,output_tokens:j,cached_tokens:Z,status:"success",finish_reason:O.stop_reason??"",tools_used:N.length>0?N:O.content?.filter(F=>F.type==="tool_use").map(F=>F.name)??[],tool_call_args:Object.keys(z).length?z:void 0,prompt_hash:Be(JSON.stringify(I)),...i.storePrompts?{prompt_text:i.redactForStorage(I),completion_text:i.redactForStorage(K)}:{},...v,span_kind:"llm",tool_definitions:T,conversation_turn:y,message_role:x,metadata:{usage:O.usage??null,guardrails:{pre:S?{matched:S.results.length,modified:!!S.modifiedText}:null,post:b?{matched:b.results.length,modified:!!b.modifiedText}:null},auto_detected:Object.keys(k).length?k:void 0,...f??{}},user_id:p?.userId,session_id:p?.sessionId,tags:p?.tags,environment:i.environment,sdk_version:"1.0.0"}),h&&(h.setAttribute("gen_ai.usage.input_tokens",$),h.setAttribute("gen_ai.usage.output_tokens",j),h.setAttribute("gen_ai.usage.cached_tokens",Z),h.setAttribute("gen_ai.response.finish_reason",O.stop_reason??""),h.setAttribute("gen_ai.response.latency",performance.now()-l),h.setStatus({code:ve.OK}),h.end()),O}catch(C){if(C?.name==="TimeoutError"||C?.code==="ABORT_ERR"||R&&R.aborted)throw i.enqueue({event_id:u(),project_id:"",trace_id:m,span_id:c,parent_span_id:p?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:a.model,operation:o,latency_ms:performance.now()-l,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:"timeout",error_message:String(C.message).slice(0,500),...v,span_kind:"llm",conversation_turn:y,message_role:x,metadata:{auto_detected:Object.keys(k).length?k:void 0},user_id:p?.userId,session_id:p?.sessionId,tags:p?.tags,environment:i.environment,sdk_version:"1.0.0"}),h&&(h.setStatus({code:ve.ERROR,message:C.message}),h.recordException(C),h.end()),C;let q=i.configClient?.getFallback(o)??null;if(q&&q.model){let Z=String(C.message??"")+String(C.status??"")+String(C.code??"");if(q.conditions.length===0||q.conditions.some(L=>Z.toLowerCase().includes(L.toLowerCase()))){let L=g,U={...a,model:q.model};i.enqueue({event_id:u(),project_id:"",trace_id:m,span_id:c,parent_span_id:p?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:q.model,operation:o,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:L,fallback_model:q.model,reason:String(C.message).slice(0,200)},environment:i.environment,sdk_version:"1.0.0"});try{let N=await W({traceId:m,spanId:c,userId:p?.userId,sessionId:p?.sessionId,tags:p?.tags},()=>e(U,d)),z=N.usage?.input_tokens??0,K=N.usage?.output_tokens??0,F=N.usage?.cache_read_input_tokens??0,Y=N.model??q.model;return i.enqueue({event_id:u(),project_id:"",trace_id:m,span_id:c,parent_span_id:p?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:Y,operation:o,latency_ms:performance.now()-l,stream:!1,input_tokens:z,output_tokens:K,cached_tokens:F,status:"success",finish_reason:N.stop_reason??"",prompt_hash:Be(JSON.stringify(I)),...v,span_kind:"llm",conversation_turn:y,message_role:x,metadata:{usage:N.usage??null,fallback_used:!0,original_model:L,auto_detected:Object.keys(k).length?k:void 0},user_id:p?.userId,session_id:p?.sessionId,tags:p?.tags,environment:i.environment,sdk_version:"1.0.0"}),h&&(h.setStatus({code:ve.OK}),h.end()),N}catch(N){throw h&&(h.setStatus({code:ve.ERROR,message:N.message}),h.recordException(N),h.end()),N}}}let $=C instanceof J,j=$?"error":C.status===429||C.type==="overloaded_error"?"rate_limited":C.status===408?"timeout":"error";throw i.enqueue({event_id:u(),project_id:"",trace_id:m,span_id:c,parent_span_id:p?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:a.model,operation:o,latency_ms:performance.now()-l,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:j,error_code:$?"guardrail_blocked":String(C.status??C.type??"unknown"),error_message:String(C.message).slice(0,500),...v,span_kind:"llm",conversation_turn:y,message_role:x,metadata:{auto_detected:Object.keys(k).length?k:void 0},user_id:p?.userId,session_id:p?.sessionId,tags:p?.tags,environment:i.environment,sdk_version:"1.0.0"}),h&&(h.setStatus({code:ve.ERROR,message:C.message}),h.recordException(C),h.end()),C}},r}import{createHash as An}from"crypto";import{SpanKind as Ze,SpanStatusCode as re,trace as Je}from"@opentelemetry/api";var Ce=r=>An("sha256").update(r).digest("hex");function ze(r){let t=a=>{if(typeof a=="string")return a;if(Array.isArray(a))return a.map(d=>typeof d=="string"?d:typeof d?.text=="string"?d.text:"").filter(Boolean).join(`
|
|
4
|
-
`);if(a&&typeof a=="object"){if(typeof a.text=="string")return a.text;if(Array.isArray(a.parts))return a.parts.map(d=>typeof d?.text=="string"?d.text:"").filter(Boolean).join(`
|
|
5
|
-
`)}return""},n=r?.contents??r,e=[],s=r?.systemInstruction??r?.system_instruction,o=t(s);if(o.length>0&&e.push({role:"system",content:o}),typeof n=="string")return e.push({role:"user",content:n}),e;if(Array.isArray(n)){for(let a of n){if(typeof a=="string"){e.push({role:"user",content:a});continue}if(a&&typeof a=="object"){let d=typeof a.role=="string"?a.role:"user",i=t(a);i.length>0&&e.push({role:d,content:i})}}return e}if(n&&typeof n=="object"){let a=typeof n.role=="string"?n.role:"user",d=t(n);d.length>0&&e.push({role:a,content:d})}return e}function Ue(r,t){return{"gen_ai.system":"google","gen_ai.model":t,"gen_ai.operation.name":"chat","gen_ai.prompt":JSON.stringify(r),"gen_ai.request.max_tokens":r.generationConfig?.maxOutputTokens,"gen_ai.request.temperature":r.generationConfig?.temperature,"gen_ai.request.top_p":r.generationConfig?.topP}}function ut(r){try{if(typeof r?.text=="function"){let n=r.text();if(typeof n=="string"&&n.length>0)return n}}catch{}return(r?.candidates?.[0]?.content?.parts??[]).map(n=>typeof n?.text=="string"?n.text:"").filter(Boolean).join("")}async function*Mt(r,t){let{lumiq:n,t0:e,traceId:s,spanId:o,parent:a,params:d,modelName:i,otelSpan:l,streamResult:p,agentCtx:c,tool_definitions:m,autoDetected:g,conversation_turn:f,message_role:S,operation:b}=t,I,k=0,y=0,x=0,_=0,E="",v=[],T="";try{for await(let w of r){I||(I=performance.now()-e,n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:b,latency_ms:I,ttft_ms:I,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:Ce(JSON.stringify(d)),...n.storePrompts?{prompt_text:n.redactForStorage(d)}:{},user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0",finish_reason:"stream_start",...c,span_kind:"llm",tool_definitions:m,conversation_turn:f,message_role:S,metadata:{auto_detected:Object.keys(g).length?g:void 0}}),l&&l.setAttribute("gen_ai.response.time_to_first_token",I));let M=(typeof w?.text=="function"?w.text():w?.text)||w?.candidates?.[0]?.content?.parts?.map(q=>q?.text||"").join("")||"";M&&(k++,T+=M),w.candidates?.[0]?.finishReason&&(E=w.candidates[0].finishReason);let C=Array.isArray(w?.functionCalls)?w.functionCalls:w?.functionCalls?[w.functionCalls]:[],O=(w?.candidates?.[0]?.content?.parts??[]).map(q=>q?.functionCall).filter(Boolean);for(let q of[...C,...O])q?.name&&!v.includes(q.name)&&v.push(q.name);yield w}let h=await p?.response;h?.usageMetadata&&(y=Number(h.usageMetadata.promptTokenCount||y||0),k=Number(h.usageMetadata.candidatesTokenCount||k||0),x=Number(h.usageMetadata.cachedContentTokenCount||x||0),_=Number(h.usageMetadata.thoughtsTokenCount||_||0));let A=ut(h);A&&(T=A);let R=performance.now()-e;n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:b,latency_ms:R,ttft_ms:I,stream:!0,input_tokens:y,output_tokens:k,cached_tokens:x,reasoning_tokens:_,status:"success",finish_reason:E||"stream_end",tools_used:v,prompt_hash:Ce(JSON.stringify(d)),...n.storePrompts?{prompt_text:n.redactForStorage(d),completion_text:n.redactForStorage(T)}:{},...c,span_kind:"llm",tool_definitions:m,conversation_turn:f,message_role:S,metadata:{usage:h?.usageMetadata??{promptTokenCount:y,candidatesTokenCount:k,cachedContentTokenCount:x,thoughtsTokenCount:_},auto_detected:Object.keys(g).length?g:void 0},user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0"}),l&&(l.setAttribute("gen_ai.completion",T),l.setAttribute("gen_ai.usage.input_tokens",y),l.setAttribute("gen_ai.usage.output_tokens",k),l.setAttribute("gen_ai.usage.total_tokens",y+k),l.setAttribute("gen_ai.usage.cached_tokens",x),l.setAttribute("gen_ai.usage.reasoning_tokens",_),l.setAttribute("gen_ai.response.finish_reason",E||"stream_end"),l.setStatus({code:re.OK}),l.end())}catch(h){throw n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:b,latency_ms:performance.now()-e,ttft_ms:I,stream:!0,input_tokens:0,output_tokens:k,cost_usd:0,status:"error",error_code:String(h.status??"unknown"),error_message:String(h.message).slice(0,500),...c,span_kind:"llm",conversation_turn:f,message_role:S,metadata:{auto_detected:Object.keys(g).length?g:void 0},user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0"}),l&&(l.setStatus({code:re.ERROR,message:h.message}),l.recordException(h),l.end()),h}}function qt(r,t={}){let n=r,e=n.getGenerativeModel.bind(n),s=V(t.guardrails,"chat"),o=t.operation??"chat";return n.getGenerativeModel=(a,d)=>{let i=a.model,l=P(),p=!1,c=i,m;if(l.configClient){let I=l.configClient.getAbTest(o);if(I)Math.random()<I.splitPercentage/100?(c=i,i=I.candidateModel,a={...a,model:i},p=!0,m={ab_test:!0,ab_variant:"candidate"}):m={ab_test:!0,ab_variant:"control"};else{let k=l.configClient.getModel(o,i);k!==i&&(c=i,i=k,a={...a,model:k},p=!0)}}let g=e(a,d),f=g.generateContent.bind(g);g.generateContent=async(I,k)=>{let y=P(),x=performance.now(),_=G(),E=u(),v=_?.traceId??u(),T=null,h=null,A=ze(I),R=se(A,i),{conversation_turn:w,message_role:M}=ae(A),C=_?.agentName??R.agent_name,O=_?.agentRole??R.agent_role,q=_?.agentId||C||O?{agent_id:_?.agentId,agent_name:C,agent_role:O,agent_framework:_?.agentFramework,parent_agent_id:_?.parentAgentId}:{},$=I?.tools?.length?oe(I.tools):void 0,j=null,Z=y.enableOTel?Je.getTracer("zespan-sdk"):null;Z&&(j=Z.startSpan(`generateContent ${i}`,{kind:Ze.CLIENT,attributes:Ue(I,i)})),p&&y.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:E,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:c,override_model:i},environment:y.environment,sdk_version:"1.0.0"});try{T=await B(y,s,{text:JSON.stringify(I??""),phase:"pre",traceId:v,spanId:E,model:i,operation:o,agentName:_?.agentName});let D;if(y.configClient){let H=y.configClient.getTimeout(o);H&&(D=AbortSignal.timeout(H.timeoutMs))}let L=y.configClient?.getRetryPolicy(o)??null,U=y.configClient?.getConcurrencyLimit(o)??null,N=()=>ye(()=>W({traceId:v,spanId:E,userId:_?.userId,sessionId:_?.sessionId,tags:_?.tags,agentId:_?.agentId,agentName:C,agentRole:O,agentFramework:_?.agentFramework,parentAgentId:_?.parentAgentId,conversationTurn:w,availableTools:$},()=>f(I,D?{...k,signal:D}:k)),L),z=await(U!==null?ke(`${y.projectId??"default"}:${o}`,U).run(N):N()),K=z.response,F=K.usageMetadata?.promptTokenCount??0,Y=K.usageMetadata?.candidatesTokenCount??0,ee=K.usageMetadata?.cachedContentTokenCount??0,ue=K.usageMetadata?.thoughtsTokenCount??0,ce=ut(K),pe=(K.candidates?.[0]?.content?.parts??[]).filter(H=>H?.functionCall),te=pe.map(H=>H.functionCall.name).filter(Boolean),ne={};for(let H of pe)H.functionCall?.name&&(ne[H.functionCall.name]=H.functionCall.args??{});h=await B(y,s,{text:ce,phase:"post",traceId:v,spanId:E,model:i,operation:o,agentName:_?.agentName});let le=X(ce,h);return y.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:E,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:o,latency_ms:performance.now()-x,stream:!1,input_tokens:F,output_tokens:Y,cached_tokens:ee,reasoning_tokens:ue,status:"success",finish_reason:K.candidates?.[0]?.finishReason??"",tools_used:te,tool_call_args:Object.keys(ne).length?ne:void 0,prompt_hash:Ce(JSON.stringify(I)),...y.storePrompts?{prompt_text:y.redactForStorage(I),completion_text:y.redactForStorage(le)}:{},...q,span_kind:"llm",tool_definitions:$,conversation_turn:w,message_role:M,metadata:{usage:K.usageMetadata??null,guardrails:{pre:T?{matched:T.results.length,modified:!!T.modifiedText}:null,post:h?{matched:h.results.length,modified:!!h.modifiedText}:null},auto_detected:Object.keys(R).length?R:void 0,...m??{}},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:y.environment,sdk_version:"1.0.0"}),j&&(j.setAttribute("gen_ai.completion",le),j.setAttribute("gen_ai.usage.input_tokens",F),j.setAttribute("gen_ai.usage.output_tokens",Y),j.setAttribute("gen_ai.usage.total_tokens",F+Y),j.setAttribute("gen_ai.usage.cached_tokens",ee),j.setAttribute("gen_ai.usage.reasoning_tokens",ue),j.setAttribute("gen_ai.response.finish_reason",K.candidates?.[0]?.finishReason??""),j.setAttribute("gen_ai.response.latency",performance.now()-x),j.setStatus({code:re.OK}),j.end()),z}catch(D){let L=D.__signal;if(D?.name==="TimeoutError"||D?.code==="ABORT_ERR")throw y.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:E,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:o,latency_ms:performance.now()-x,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:"timeout",error_message:String(D.message).slice(0,500),...q,span_kind:"llm",conversation_turn:w,message_role:M,metadata:{auto_detected:Object.keys(R).length?R:void 0},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:y.environment,sdk_version:"1.0.0"}),j&&(j.setStatus({code:re.ERROR,message:D.message}),j.recordException(D),j.end()),D;let N=y.configClient?.getFallback(o)??null;if(N&&N.model){let F=String(D.message??"")+String(D.status??"")+String(D.code??"");if(N.conditions.length===0||N.conditions.some(ee=>F.toLowerCase().includes(ee.toLowerCase()))){let ee=i,ue={...a,model:N.model},ce=e(ue,d),pe=ce.generateContent.bind(ce);y.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:E,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:N.model,operation:o,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:ee,fallback_model:N.model,reason:String(D.message).slice(0,200)},environment:y.environment,sdk_version:"1.0.0"});try{let te=await W({traceId:v,spanId:E,userId:_?.userId,sessionId:_?.sessionId,tags:_?.tags},()=>pe(I,k)),ne=te.response,le=ne.usageMetadata?.promptTokenCount??0,H=ne.usageMetadata?.candidatesTokenCount??0,de=ne.usageMetadata?.cachedContentTokenCount??0;return y.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:E,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:N.model,operation:o,latency_ms:performance.now()-x,stream:!1,input_tokens:le,output_tokens:H,cached_tokens:de,status:"success",finish_reason:ne.candidates?.[0]?.finishReason??"",prompt_hash:Ce(JSON.stringify(I)),...q,span_kind:"llm",conversation_turn:w,message_role:M,metadata:{usage:ne.usageMetadata??null,fallback_used:!0,original_model:ee,auto_detected:Object.keys(R).length?R:void 0},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:y.environment,sdk_version:"1.0.0"}),j&&(j.setStatus({code:re.OK}),j.end()),te}catch(te){throw j&&(j.setStatus({code:re.ERROR,message:te.message}),j.recordException(te),j.end()),te}}}let z=D instanceof J,K=z?"error":D.status===429?"rate_limited":D.status===408?"timeout":"error";throw y.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:E,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:o,latency_ms:performance.now()-x,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:K,error_code:z?"guardrail_blocked":String(D.status??"unknown"),error_message:String(D.message).slice(0,500),...q,span_kind:"llm",conversation_turn:w,message_role:M,metadata:{auto_detected:Object.keys(R).length?R:void 0},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:y.environment,sdk_version:"1.0.0"}),j&&(j.setStatus({code:re.ERROR,message:D.message}),j.recordException(D),j.end()),D}};let S=g.generateContentStream.bind(g);g.generateContentStream=async(I,k)=>{let y=P(),x=performance.now(),_=G(),E=u(),v=_?.traceId??u(),T=ze(I),h=se(T,i),{conversation_turn:A,message_role:R}=ae(T),w=_?.agentName??h.agent_name,M=_?.agentRole??h.agent_role,C=_?.agentId||w||M?{agent_id:_?.agentId,agent_name:w,agent_role:M,agent_framework:_?.agentFramework,parent_agent_id:_?.parentAgentId}:{},O=I?.tools?.length?oe(I.tools):void 0,q=null,$=y.enableOTel?Je.getTracer("zespan-sdk"):null;$&&(q=$.startSpan(`generateContentStream ${i}`,{kind:Ze.CLIENT,attributes:Ue(I,i)}));try{let j=await W({traceId:v,spanId:E,userId:_?.userId,sessionId:_?.sessionId,tags:_?.tags,agentId:_?.agentId,agentName:w,agentRole:M,agentFramework:_?.agentFramework,parentAgentId:_?.parentAgentId,conversationTurn:A,availableTools:O},()=>S(I,k));return q&&q.setAttribute("gen_ai.request.stream",!0),j.stream=Mt(j.stream,{lumiq:y,t0:x,traceId:v,spanId:E,parent:_,params:I,modelName:i,otelSpan:q,streamResult:j,agentCtx:C,tool_definitions:O,autoDetected:h,conversation_turn:A,message_role:R,operation:o}),j}catch(j){let Z=j instanceof J,D=Z?"error":j.status===429?"rate_limited":j.status===408?"timeout":"error";throw y.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:E,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:o,latency_ms:performance.now()-x,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:D,error_code:Z?"guardrail_blocked":String(j.status??"unknown"),error_message:String(j.message).slice(0,500),...C,span_kind:"llm",conversation_turn:A,message_role:R,metadata:{auto_detected:Object.keys(h).length?h:void 0},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:y.environment,sdk_version:"1.0.0"}),q&&(q.setStatus({code:re.ERROR,message:j.message}),q.recordException(j),q.end()),j}};let b=g.startChat?.bind(g);return b&&(g.startChat=I=>{let k=b(I),y=k.sendMessage?.bind(k),x=k.sendMessageStream?.bind(k);return y&&(k.sendMessage=async(_,E)=>{let v=P(),T=performance.now(),h=G(),A=u(),R=h?.traceId??u(),w={...I||{},message:_},M=null,C=null,O=ze(w),q=se(O,i),{conversation_turn:$,message_role:j}=ae(O),Z=h?.agentName??q.agent_name,D=h?.agentRole??q.agent_role,L=h?.agentId||Z||D?{agent_id:h?.agentId,agent_name:Z,agent_role:D,agent_framework:h?.agentFramework,parent_agent_id:h?.parentAgentId}:{},U=I?.tools?.length?oe(I.tools):void 0,N=null,z=v.enableOTel?Je.getTracer("zespan-sdk"):null;z&&(N=z.startSpan(`sendMessage ${i}`,{kind:Ze.CLIENT,attributes:Ue(w,i)}));try{M=await B(v,s,{text:JSON.stringify(w),phase:"pre",traceId:R,spanId:A,model:i,operation:o,agentName:h?.agentName});let K=await W({traceId:R,spanId:A,userId:h?.userId,sessionId:h?.sessionId,tags:h?.tags,agentId:h?.agentId,agentName:Z,agentRole:D,agentFramework:h?.agentFramework,parentAgentId:h?.parentAgentId,conversationTurn:$,availableTools:U},()=>y(_,E)),F=K.response,Y=F.usageMetadata?.promptTokenCount??0,ee=F.usageMetadata?.candidatesTokenCount??0,ue=F.usageMetadata?.cachedContentTokenCount??0,ce=F.usageMetadata?.thoughtsTokenCount??0,pe=ut(F),te=(F.candidates?.[0]?.content?.parts??[]).filter(de=>de?.functionCall),ne=te.map(de=>de.functionCall.name).filter(Boolean),le={};for(let de of te)de.functionCall?.name&&(le[de.functionCall.name]=de.functionCall.args??{});C=await B(v,s,{text:pe,phase:"post",traceId:R,spanId:A,model:i,operation:o,agentName:h?.agentName});let H=X(pe,C);return v.enqueue({event_id:u(),project_id:"",trace_id:R,span_id:A,parent_span_id:h?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:o,latency_ms:performance.now()-T,stream:!1,input_tokens:Y,output_tokens:ee,cached_tokens:ue,reasoning_tokens:ce,status:"success",finish_reason:F.candidates?.[0]?.finishReason??"",tools_used:ne,tool_call_args:Object.keys(le).length?le:void 0,prompt_hash:Ce(JSON.stringify(w)),...v.storePrompts?{prompt_text:v.redactForStorage(w),completion_text:v.redactForStorage(H)}:{},...L,span_kind:"llm",tool_definitions:U,conversation_turn:$,message_role:j,metadata:{usage:F.usageMetadata??null,guardrails:{pre:M?{matched:M.results.length,modified:!!M.modifiedText}:null,post:C?{matched:C.results.length,modified:!!C.modifiedText}:null},auto_detected:Object.keys(q).length?q:void 0},user_id:h?.userId,session_id:h?.sessionId,tags:h?.tags,environment:v.environment,sdk_version:"1.0.0"}),N&&(N.setAttribute("gen_ai.completion",H),N.setAttribute("gen_ai.usage.input_tokens",Y),N.setAttribute("gen_ai.usage.output_tokens",ee),N.setAttribute("gen_ai.usage.total_tokens",Y+ee),N.setAttribute("gen_ai.usage.cached_tokens",ue),N.setAttribute("gen_ai.usage.reasoning_tokens",ce),N.setAttribute("gen_ai.response.finish_reason",F.candidates?.[0]?.finishReason??""),N.setAttribute("gen_ai.response.latency",performance.now()-T),N.setStatus({code:re.OK}),N.end()),K}catch(K){let F=K instanceof J,Y=F?"error":K.status===429?"rate_limited":K.status===408?"timeout":"error";throw v.enqueue({event_id:u(),project_id:"",trace_id:R,span_id:A,parent_span_id:h?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:o,latency_ms:performance.now()-T,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:Y,error_code:F?"guardrail_blocked":String(K.status??"unknown"),error_message:String(K.message).slice(0,500),...L,span_kind:"llm",conversation_turn:$,message_role:j,metadata:{auto_detected:Object.keys(q).length?q:void 0},user_id:h?.userId,session_id:h?.sessionId,tags:h?.tags,environment:v.environment,sdk_version:"1.0.0"}),N&&(N.setStatus({code:re.ERROR,message:K.message}),N.recordException(K),N.end()),K}}),x&&(k.sendMessageStream=async(_,E)=>{let v=P(),T=performance.now(),h=G(),A=u(),R=h?.traceId??u(),w={...I||{},message:_},M=ze(w),C=se(M,i),{conversation_turn:O,message_role:q}=ae(M),$=h?.agentName??C.agent_name,j=h?.agentRole??C.agent_role,Z=h?.agentId||$||j?{agent_id:h?.agentId,agent_name:$,agent_role:j,agent_framework:h?.agentFramework,parent_agent_id:h?.parentAgentId}:{},D=I?.tools?.length?oe(I.tools):void 0,L=null,U=v.enableOTel?Je.getTracer("zespan-sdk"):null;U&&(L=U.startSpan(`sendMessageStream ${i}`,{kind:Ze.CLIENT,attributes:Ue(w,i)}));try{let N=await W({traceId:R,spanId:A,userId:h?.userId,sessionId:h?.sessionId,tags:h?.tags,agentId:h?.agentId,agentName:$,agentRole:j,agentFramework:h?.agentFramework,parentAgentId:h?.parentAgentId,conversationTurn:O,availableTools:D},()=>x(_,E));return L&&L.setAttribute("gen_ai.request.stream",!0),N.stream=Mt(N.stream,{lumiq:v,t0:T,traceId:R,spanId:A,parent:h,params:w,modelName:i,otelSpan:L,streamResult:N,agentCtx:Z,tool_definitions:D,autoDetected:C,conversation_turn:O,message_role:q,operation:o}),N}catch(N){let z=N instanceof J,K=z?"error":N.status===429?"rate_limited":N.status===408?"timeout":"error";throw v.enqueue({event_id:u(),project_id:"",trace_id:R,span_id:A,parent_span_id:h?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:o,latency_ms:performance.now()-T,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:K,error_code:z?"guardrail_blocked":String(N.status??"unknown"),error_message:String(N.message).slice(0,500),...Z,span_kind:"llm",conversation_turn:O,message_role:q,metadata:{auto_detected:Object.keys(C).length?C:void 0},user_id:h?.userId,session_id:h?.sessionId,tags:h?.tags,environment:v.environment,sdk_version:"1.0.0"}),L&&(L.setStatus({code:re.ERROR,message:N.message}),L.recordException(N),L.end()),N}}),k}),g},r}import{createHash as Rn}from"crypto";import{SpanKind as Nn,SpanStatusCode as Ve,trace as Mn}from"@opentelemetry/api";var pt=r=>Rn("sha256").update(r).digest("hex");function qn(r,t,n,e){let s=Number(r?.usage?.cost);if(Number.isFinite(s))return{cost:s,source:"usage.cost"};let o=r?.headers?.get?.("x-ratelimit-cost"),a=Number.parseFloat(String(o??""));return Number.isFinite(a)?{cost:a,source:"x-ratelimit-cost"}:{cost:0,source:"computed"}}function jn(r,t){return{"gen_ai.system":"openrouter","gen_ai.model":t,"gen_ai.operation.name":"chat","gen_ai.prompt":JSON.stringify(r.messages),"gen_ai.request.max_tokens":r.max_tokens,"gen_ai.request.temperature":r.temperature,"gen_ai.request.top_p":r.top_p}}async function*Pn(r,t){let{lumiq:n,t0:e,traceId:s,spanId:o,parent:a,params:d,otelSpan:i,agentCtx:l,autoDetected:p,conversation_turn:c,message_role:m,tool_definitions:g}=t,f,S=0,b=0,I=0,k=0,y="",x=d.model,_=[],E="",v=[];try{for await(let w of r){if(f||(f=performance.now()-e,n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:x,operation:"chat",latency_ms:f,ttft_ms:f,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:pt(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(d.messages)}:{},user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0",finish_reason:"stream_start",...l,span_kind:"llm",tool_definitions:g,conversation_turn:c,message_role:m,metadata:{auto_detected:Object.keys(p).length?p:void 0}}),i&&i.setAttribute("gen_ai.response.time_to_first_token",f)),w.choices?.[0]?.delta?.content&&(S++,E+=w.choices[0].delta.content),w.choices?.[0]?.finish_reason&&(y=w.choices[0].finish_reason),w.model&&(x=w.model),w.usage){b=Number(w.usage.prompt_tokens||b||0),S=Number(w.usage.completion_tokens||S||0),I=Number(w.usage.prompt_tokens_details?.cached_tokens||I||0);let M=w.usage.completion_tokens_details||{};k=Number(M.reasoning_tokens||k||0)}if(w.choices?.[0]?.delta?.tool_calls)for(let M of w.choices[0].delta.tool_calls)v.push(M),M.function?.name&&!_.includes(M.function.name)&&_.push(M.function.name);yield w}let{tools_used:T,tool_call_args:h}=Ke(v),A=T.length>0?T:_,R=performance.now()-e;n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:x,operation:"chat",latency_ms:R,ttft_ms:f,stream:!0,input_tokens:b,output_tokens:S,cached_tokens:I,reasoning_tokens:k,status:"success",finish_reason:y||"stream_end",tools_used:A,tool_call_args:Object.keys(h).length?h:void 0,prompt_hash:pt(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(d.messages),completion_text:n.redactForStorage(E)}:{},...l,span_kind:"llm",tool_definitions:g,conversation_turn:c,message_role:m,metadata:{usage:{prompt_tokens:b,completion_tokens:S,cached_tokens:I,reasoning_tokens:k},auto_detected:Object.keys(p).length?p:void 0},user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0"}),i&&(i.setAttribute("gen_ai.completion",E),i.setAttribute("gen_ai.usage.input_tokens",b),i.setAttribute("gen_ai.usage.output_tokens",S),i.setAttribute("gen_ai.usage.total_tokens",b+S),i.setAttribute("gen_ai.usage.reasoning_tokens",k),i.setAttribute("gen_ai.response.finish_reason",y||"stream_end"),i.setStatus({code:Ve.OK}),i.end())}catch(T){throw n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:x,operation:"chat",latency_ms:performance.now()-e,ttft_ms:f,stream:!0,input_tokens:0,output_tokens:S,cost_usd:0,status:"error",error_code:String(T.status??"unknown"),error_message:String(T.message).slice(0,500),...l,span_kind:"llm",conversation_turn:c,message_role:m,metadata:{auto_detected:Object.keys(p).length?p:void 0},user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0"}),i&&(i.setStatus({code:Ve.ERROR,message:T.message}),i.recordException(T),i.end()),T}}function jt(r,t={}){let n=r.chat.completions.create,e=V(t.guardrails,"chat");return r.chat.completions.create=async(s,o)=>{let a=P(),d=performance.now(),i=G(),l=u(),p=i?.traceId??u(),c=s.model??"openrouter\u6A21\u578B",m=null,g=null;if(a.configClient){let _=a.configClient.getModel("chat",c);if(_!==c){let E=c;c=_,s={...s,model:_},a.enqueue({event_id:u(),project_id:"",trace_id:p,span_id:l,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:_,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:E,override_model:_},environment:a.environment,sdk_version:"1.0.0"})}}let f=i?.agentId?{agent_id:i.agentId,agent_name:i.agentName,agent_role:i.agentRole,agent_framework:i.agentFramework,parent_agent_id:i.parentAgentId}:{},S=se(s.messages??[],s.model),{conversation_turn:b,message_role:I}=ae(s.messages??[]),k=s.tools?.length?oe(s.tools):void 0,y=null,x=a.enableOTel?Mn.getTracer("zespan-sdk"):null;x&&(y=x.startSpan(`chat.completions ${c}`,{kind:Nn.CLIENT,attributes:jn(s,c)}));try{m=await B(a,e,{text:JSON.stringify(s.messages??[]),phase:"pre",traceId:p,spanId:l,model:c,operation:"chat"});let _=await W({traceId:p,spanId:l,userId:i?.userId,sessionId:i?.sessionId,tags:i?.tags},()=>n.call(r.chat.completions,s,o));if(s.stream===!0)return y&&y.setAttribute("gen_ai.request.stream",!0),Pn(_,{lumiq:a,t0:d,traceId:p,spanId:l,parent:i,params:s,otelSpan:y,agentCtx:f,autoDetected:S,conversation_turn:b,message_role:I,tool_definitions:k});let{prompt_tokens:v,completion_tokens:T,completion_tokens_details:h}=_.usage??{},A=h?.reasoning_tokens??0,R=_.model??s.model,{cost:w,source:M}=qn(_,R,v??0,T??0);_.pricing||(_.pricing={cost_usd:w,source:M});let C=_.choices?.[0]?.message?.content||"";g=await B(a,e,{text:C,phase:"post",traceId:p,spanId:l,model:R,operation:"chat"});let O=X(C,g);O!==C&&typeof _?.choices?.[0]?.message?.content=="string"&&(_.choices[0].message.content=O);let{tools_used:q,tool_call_args:$}=Le(_.choices??[]);return a.enqueue({event_id:u(),project_id:"",trace_id:p,span_id:l,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:R,operation:"chat",latency_ms:performance.now()-d,stream:!1,input_tokens:v??0,output_tokens:T??0,cached_tokens:0,cost_usd:w,status:"success",finish_reason:_.choices?.[0]?.finish_reason??"",tools_used:q.length>0?q:_.choices?.[0]?.message?.tool_calls?.map(j=>j.function.name)??[],tool_call_args:Object.keys($).length?$:void 0,prompt_hash:pt(JSON.stringify(s.messages)),...a.storePrompts?{prompt_text:a.redactForStorage(s.messages),completion_text:a.redactForStorage(O)}:{},...f,span_kind:"llm",tool_definitions:k,conversation_turn:b,message_role:I,metadata:{usage:_.usage??null,pricing:_.pricing??{cost_usd:w,source:M},guardrails:{pre:m?{matched:m.results.length,modified:!!m.modifiedText}:null,post:g?{matched:g.results.length,modified:!!g.modifiedText}:null},auto_detected:Object.keys(S).length?S:void 0},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:a.environment,sdk_version:"1.0.0"}),y&&(y.setAttribute("gen_ai.usage.input_tokens",v??0),y.setAttribute("gen_ai.usage.output_tokens",T??0),y.setAttribute("gen_ai.usage.total_tokens",(v??0)+(T??0)),y.setAttribute("gen_ai.usage.reasoning_tokens",A),y.setAttribute("gen_ai.completion",_.choices?.[0]?.message?.content||""),y.setAttribute("gen_ai.response.finish_reason",_.choices?.[0]?.finish_reason??""),y.setAttribute("gen_ai.response.latency",performance.now()-d),y.setStatus({code:Ve.OK}),y.end()),_}catch(_){let E=_ instanceof J,v=E?"error":_.status===429?"rate_limited":_.status===408?"timeout":"error";throw a.enqueue({event_id:u(),project_id:"",trace_id:p,span_id:l,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:s.model,operation:"chat",latency_ms:performance.now()-d,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:v,error_code:E?"guardrail_blocked":String(_.status??_.code??"unknown"),error_message:String(_.message).slice(0,500),...f,span_kind:"llm",conversation_turn:b,message_role:I,metadata:{auto_detected:Object.keys(S).length?S:void 0},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:a.environment,sdk_version:"1.0.0"}),y&&(y.setStatus({code:Ve.ERROR,message:_.message}),y.recordException(_),y.end()),_}},r}async function*Pt(r,t){let{lumiq:n,t0:e,traceId:s,spanId:o,parent:a,model:d,operation:i}=t,l=0,p=0,c=0,m="",g;try{for await(let f of r)g||(g=performance.now()-e),"metadata"in f&&f.metadata?.usage&&(l=f.metadata.usage.inputTokens??l,p=f.metadata.usage.outputTokens??p,c=f.metadata.usage.cacheReadInputTokens??c),"messageStop"in f&&f.messageStop?.stopReason&&(m=f.messageStop.stopReason),yield f;n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:d,operation:i,latency_ms:performance.now()-e,ttft_ms:g,stream:!0,input_tokens:l,output_tokens:p,cached_tokens:c,status:"success",finish_reason:m,user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0"})}catch(f){throw n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:d,operation:i,latency_ms:performance.now()-e,ttft_ms:g,stream:!0,input_tokens:0,output_tokens:0,status:"error",error_code:String(f.$metadata?.httpStatusCode??f.name??"unknown"),error_message:String(f.message).slice(0,500),user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags,environment:n.environment,sdk_version:"1.0.0"}),f}}function En(r,t={}){let n=r,e=t.operation??"chat",s=V(t.guardrails,e),o=n.converse?.bind(n);o&&(n.converse=async(i,...l)=>{let p=P(),c=performance.now(),m=G(),g=u(),f=m?.traceId??u(),S=i.modelId??"amazon.nova-lite-v1:0";p.configClient&&(S=p.configClient.getModel(e,S),S!==i.modelId&&(i={...i,modelId:S}));let b=JSON.stringify(i.messages??[]);await B(p,s,{text:b,phase:"pre",traceId:f,spanId:g,model:S,operation:e,agentName:m?.agentName});try{let I=await W({traceId:f,spanId:g,userId:m?.userId,sessionId:m?.sessionId,tags:m?.tags},()=>o(i,...l)),k=I.usage??{};return p.enqueue({event_id:u(),project_id:"",trace_id:f,span_id:g,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:S,operation:e,latency_ms:performance.now()-c,stream:!1,input_tokens:Number(k.inputTokens??0),output_tokens:Number(k.outputTokens??0),cached_tokens:Number(k.cacheReadInputTokens??0),status:"success",finish_reason:I.stopReason??"",user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:p.environment,sdk_version:"1.0.0"}),I}catch(I){throw p.enqueue({event_id:u(),project_id:"",trace_id:f,span_id:g,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:S,operation:e,latency_ms:performance.now()-c,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:String(I.$metadata?.httpStatusCode??I.name??"unknown"),error_message:String(I.message).slice(0,500),user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:p.environment,sdk_version:"1.0.0"}),I}});let a=n.converseStream?.bind(n);a&&(n.converseStream=async(i,...l)=>{let p=P(),c=performance.now(),m=G(),g=u(),f=m?.traceId??u(),S=i.modelId??"amazon.nova-lite-v1:0",b=await W({traceId:f,spanId:g,userId:m?.userId,sessionId:m?.sessionId,tags:m?.tags},()=>a(i,...l));return{...b,stream:Pt(b.stream,{lumiq:p,t0:c,traceId:f,spanId:g,parent:m,model:S,operation:e})}});let d=n.send?.bind(n);return d&&(n.send=async(i,...l)=>{if(!(i?.constructor?.name==="ConverseStreamCommand"||i?.$name==="ConverseStream"))return d(i,...l);let c=P(),m=performance.now(),g=G(),f=u(),S=g?.traceId??u(),b=i?.input?.modelId??"amazon.nova-lite-v1:0",I=await W({traceId:S,spanId:f,userId:g?.userId,sessionId:g?.sessionId,tags:g?.tags},()=>d(i,...l));return{...I,stream:Pt(I.stream,{lumiq:c,t0:m,traceId:S,spanId:f,parent:g,model:b,operation:e})}}),r}function Gn(r,t={}){let n=r,e=t.operation??"chat",s=V(t.guardrails,e),o=n.chat.complete.bind(n.chat);n.chat.complete=async(d,...i)=>{let l=P(),p=performance.now(),c=G(),m=u(),g=c?.traceId??u(),f=d.model??"mistral-small-latest";l.configClient&&(f=l.configClient.getModel(e,f),f!==d.model&&(d={...d,model:f}));let S=JSON.stringify(d.messages??[]);await B(l,s,{text:S,phase:"pre",traceId:g,spanId:m,model:f,operation:e});try{let b=await W({traceId:g,spanId:m,userId:c?.userId,sessionId:c?.sessionId,tags:c?.tags},()=>o(d,...i)),I=b.model??f,k=b.usage??{},y=Number(k.promptTokens??0),x=Number(k.completionTokens??0),_=b.choices?.[0]?.finishReason??"";return await B(l,s,{text:b.choices?.[0]?.message?.content??"",phase:"post",traceId:g,spanId:m,model:I,operation:e}),l.enqueue({event_id:u(),project_id:"",trace_id:g,span_id:m,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:I,operation:e,latency_ms:performance.now()-p,stream:!1,input_tokens:y,output_tokens:x,status:"success",finish_reason:_,user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:l.environment,sdk_version:"1.0.0"}),b}catch(b){throw l.enqueue({event_id:u(),project_id:"",trace_id:g,span_id:m,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:f,operation:e,latency_ms:performance.now()-p,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_message:String(b.message).slice(0,500),user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:l.environment,sdk_version:"1.0.0"}),b}};let a=n.chat.stream?.bind(n.chat);return a&&(n.chat.stream=async function*(d,...i){let l=P(),p=performance.now(),c=G(),m=u(),g=c?.traceId??u(),f=d.model??"mistral-small-latest";l.configClient&&(f=l.configClient.getModel(e,f),f!==d.model&&(d={...d,model:f}));let S=0,b=0,I="",k;try{let y=await W({traceId:g,spanId:m,userId:c?.userId,sessionId:c?.sessionId,tags:c?.tags},()=>a(d,...i));for await(let x of y)k||(k=performance.now()-p),x.data?.usage&&(S=x.data.usage.promptTokens??S,b=x.data.usage.completionTokens??b),x.data?.choices?.[0]?.finishReason&&(I=x.data.choices[0].finishReason),x.data?.model&&(f=x.data.model),yield x;l.enqueue({event_id:u(),project_id:"",trace_id:g,span_id:m,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:f,operation:e,latency_ms:performance.now()-p,ttft_ms:k,stream:!0,input_tokens:S,output_tokens:b,status:"success",finish_reason:I,user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:l.environment,sdk_version:"1.0.0"})}catch(y){throw l.enqueue({event_id:u(),project_id:"",trace_id:g,span_id:m,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:f,operation:e,latency_ms:performance.now()-p,ttft_ms:k,stream:!0,input_tokens:0,output_tokens:0,status:"error",error_message:String(y.message).slice(0,500),user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:l.environment,sdk_version:"1.0.0"}),y}}),r}function Dn(r,t={}){let n=r,e=n.chat.completions.create.bind(n.chat.completions),s=t.operation??"chat";return n.chat.completions.create=async(o,...a)=>{let d=P(),i=performance.now(),l=G(),p=u(),c=l?.traceId??u(),m=o.model??"llama-3.3-70b-versatile";d.configClient&&(m=d.configClient.getModel(s,m),m!==o.model&&(o={...o,model:m}));try{let g=await W({traceId:c,spanId:p,userId:l?.userId,sessionId:l?.sessionId,tags:l?.tags},()=>e(o,...a)),f=g.usage??{},S=Number(f.prompt_tokens??0),b=Number(f.completion_tokens??0),I=g.choices?.[0]?.finish_reason??"";return d.enqueue({event_id:u(),project_id:"",trace_id:c,span_id:p,parent_span_id:l?.spanId,timestamp:new Date().toISOString(),provider:"groq",model:m,operation:s,latency_ms:performance.now()-i,stream:!1,input_tokens:S,output_tokens:b,status:"success",finish_reason:I,user_id:l?.userId,session_id:l?.sessionId,tags:l?.tags,environment:d.environment,sdk_version:"1.0.0"}),g}catch(g){throw d.enqueue({event_id:u(),project_id:"",trace_id:c,span_id:p,parent_span_id:l?.spanId,timestamp:new Date().toISOString(),provider:"groq",model:m,operation:s,latency_ms:performance.now()-i,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_message:String(g.message).slice(0,500),user_id:l?.userId,session_id:l?.sessionId,tags:l?.tags,environment:d.environment,sdk_version:"1.0.0"}),g}},r}function Fn(r,t={}){let n=st("openai").default??st("openai"),e=new n({baseURL:r.baseURL,apiKey:r.apiKey??"litellm"});return We(e,{...t,providerOverride:"litellm"})}import{BaseCallbackHandler as Ln}from"@langchain/core/callbacks/base";import{createHash as Kn}from"crypto";var Et=r=>Kn("sha256").update(r).digest("hex"),Gt=["AgentExecutor","ReactAgent","ToolCallingAgent"],Dt=["DynamicStructuredTool","StructuredTool","DynamicTool","Tool"],xe=class extends Ln{name="ZespanCallbackHandler";guardrailSettings=V(void 0,"chat");runs=new Map;constructor(t={}){super(),this.guardrailSettings=V(t.guardrails,"chat")}detectProvider(t){let n=t.toLowerCase();return/gpt|o1-|o3-|davinci|turbo|whisper|dall-e/i.test(n)?"openai":/claude/i.test(n)?"anthropic":/gemini|palm|bard/i.test(n)?"google":/llama|codellama/i.test(n)?"meta":/mistral|mixtral/i.test(n)?"mistralai":/deepseek/i.test(n)?"deepseek":/command/i.test(n)?"cohere":/qwen/i.test(n)?"alibaba":"langchain"}extractToolCalls(t){try{let n=t?.generations?.[0];if(!Array.isArray(n))return[];let e=n[0]?.message;if(!e)return[];let s=e.tool_calls;if(Array.isArray(s)&&s.length>0)return s.map(d=>({name:d.name??"unknown",args:d.args??{},id:d.id}));let o=e.additional_kwargs?.tool_calls;if(Array.isArray(o)&&o.length>0)return o.map(d=>({name:d.function?.name??"unknown",args:d.function?.arguments?JSON.parse(d.function.arguments):{},id:d.id}));let a=e.additional_kwargs?.function_call;return a?[{name:a.name??"unknown",args:a.arguments?JSON.parse(a.arguments):{}}]:[]}catch{return[]}}detectConversationInfo(t,n){let e=n?._lumiq_message_count;if(typeof e=="number"&&e>0)return{turns:e,hasMemory:e>2};let s=n?.invocation_params?.messages;if(Array.isArray(s))return{turns:s.length,hasMemory:s.length>2};let a=(JSON.stringify(t).match(/"role"\s*:/g)||[]).length;return a>0?{turns:a,hasMemory:a>2}:{turns:t.length,hasMemory:!1}}detectAgentFromPrompts(t,n){try{let e=JSON.stringify(t)+JSON.stringify(n?.invocation_params?.messages??[])+JSON.stringify(n?._lumiq_structured_messages??[]),s=[/you are (?:a |an |the )?([A-Z][A-Za-z]*(?:[_\s][A-Z][A-Za-z]*){0,3}(?:Agent|Bot|Assistant|AI|Specialist|Expert|Analyzer|Manager|Coordinator))/i,/act as (?:a |an |the )?([A-Z][A-Za-z]*(?:[_\s][A-Z][A-Za-z]*){0,3}(?:Agent|Bot|Assistant|AI))/i,/agent[_\s]?name["'\s:=]+["']?([A-Za-z][A-Za-z_\-]+)/i,/role["'\s:=]+["']?([A-Za-z]+(?:Agent|Bot|Assistant))/i,/name["'\s:=]+["']?([A-Za-z]+Agent)/i];for(let o of s){let a=e.match(o);if(a?.[1])return a[1].trim().replace(/[\s]+/g,"_")}return}catch{return}}detectPlanning(t,n){try{let e=t+" "+JSON.stringify(n);if(![/\bstep\s+\d+[:\s]/i,/\bfirst,?\s+(?:I|we|let)/i,/\bthen,?\s+(?:I|we)/i,/\bfinally,?\s+(?:I|we)/i,/\blet me think/i,/\blet'?s break (?:this |it )?down/i,/\bthink(?:ing)? step by step/i,/\bmy (?:plan|approach|strategy)/i,/\bI'?ll start by/i,/<thinking>/i,/<scratchpad>/i,/\bchain.of.thought/i].some(l=>l.test(e)))return{detected:!1};let a=/(?:step\s+\d+[:\s]|^\d+[\.\)]\s+)(.+)/gim,d=[],i;for(;(i=a.exec(t))!==null&&(i[1]&&d.push(i[1].trim().slice(0,200)),!(d.length>=10)););return{detected:!0,steps:d.length>0?d:void 0}}catch{return{detected:!1}}}extractCompletion(t){let n=t?.generations;if(!Array.isArray(n)||n.length===0)return"";let e=n[0];if(!Array.isArray(e)||e.length===0)return"";let s=e[0];if(typeof s?.text=="string")return s.text;let o=s?.message?.content;return typeof o=="string"?o:Array.isArray(o)?o.map(a=>typeof a?.text=="string"?a.text:"").filter(Boolean).join(`
|
|
6
|
-
`):""}async handleChatModelStart(t,n,e,s,o,a,d,i){try{let l=n.map(c=>c.map(m=>{let g=m._getType?.()??m.role??"unknown",f=typeof m.content=="string"?m.content:JSON.stringify(m.content);return`${g}: ${f}`}).join(`
|
|
7
|
-
`)),p=n[0]?.map(c=>({role:c._getType?.()??c.role??"unknown",content:typeof c.content=="string"?c.content:JSON.stringify(c.content),tool_calls:c.tool_calls??c.additional_kwargs?.tool_calls}));await this.handleLLMStart(t,l,e,s,{...o,_lumiq_structured_messages:p,_lumiq_message_count:n[0]?.length??0})}catch{}}async handleLLMStart(t,n,e,s,o){try{let a=G(),d=P(),i=s?this.runs.get(s):void 0,l=u();ie(l);let p=i?.traceId??a?.traceId??u(),c=o?.invocation_params?.model??o?.invocation_params?.model_name??t.kwargs?.model??t.kwargs?.modelName??t.id?.[t.id.length-1]??"unknown",m=o?.invocation_params?.stream===!0,g=i?.spanKind==="agent"?i.agentId:void 0,f=i?.spanKind==="agent"?i.agentName:void 0;if(!f){let y=this.detectAgentFromPrompts(n,o);y&&(f=y,g=g??u())}let S=o?.invocation_params?.tools??o?.invocation_params?.functions??[],b=S.map(y=>y.function?.name??y.name??"unknown").filter(Boolean),I=this.detectConversationInfo(n,o),k=await B(d,this.guardrailSettings,{text:JSON.stringify(n??[]),phase:"pre",traceId:p,spanId:l,model:c,operation:"chat"});this.runs.set(e,{t0:performance.now(),traceId:p,spanId:l,parentSpanId:i?.spanId??a?.spanId,model:c,agentId:g,agentName:f,spanKind:"llm",isStreaming:m,parent:a,prompts:n,invocationParams:o?.invocation_params,preGuardrailResult:k,toolsAvailable:b,toolSchemas:d.storePrompts?S:void 0,conversationTurns:I.turns,hasMemory:I.hasMemory})}catch(a){if(a instanceof J)throw a}}async handleLLMNewToken(t,n,e){try{let s=this.runs.get(e);if(!s||s.ttft!==void 0)return;s.ttft=performance.now()-s.t0;let o=P();o.enqueue({event_id:u(),project_id:"",trace_id:s.traceId,span_id:s.spanId,parent_span_id:s.parentSpanId,timestamp:new Date().toISOString(),provider:this.detectProvider(s.model),model:s.model,operation:"chat",latency_ms:s.ttft,ttft_ms:s.ttft,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",span_kind:"llm",agent_id:s.agentId,agent_name:s.agentName,prompt_hash:Et(JSON.stringify(s.prompts)),...o.storePrompts?{prompt_text:o.redactForStorage(s.prompts)}:{},metadata:{invocation_params:s.invocationParams??null},user_id:s.parent?.userId,session_id:s.parent?.sessionId,tags:s.parent?.tags,environment:o.environment,sdk_version:"1.0.0",finish_reason:"stream_start"})}catch{}}async handleLLMEnd(t,n){try{let e=this.runs.get(n);if(!e)return;Q(e.spanId),this.runs.delete(n);let s=P(),o=t.llmOutput?.tokenUsage?.promptTokens??t.llmOutput?.usageMetadata?.promptTokenCount??0,a=t.llmOutput?.tokenUsage?.completionTokens??t.llmOutput?.usageMetadata?.candidatesTokenCount??0,d=performance.now()-e.t0,i=this.extractCompletion(t),l=this.extractToolCalls(t),p=t.llmOutput?.tokenUsage?.reasoningTokens??t.llmOutput?.usage?.completion_tokens_details?.reasoning_tokens??t.llmOutput?.reasoningTokens??t.llmOutput?.usageMetadata?.thoughtsTokenCount??0,c=t.llmOutput?.tokenUsage?.cachedTokens??t.llmOutput?.usage?.prompt_tokens_details?.cached_tokens??t.llmOutput?.usageMetadata?.cachedContentTokenCount??0,m=this.detectPlanning(i,e.prompts),g=t.generations?.[0]?.[0]?.generationInfo?.finish_reason??t.generations?.[0]?.[0]?.generationInfo?.finishReason??t.llmOutput?.finish_reason??(e.isStreaming?"stream_end":"stop"),f=await B(s,this.guardrailSettings,{text:i,phase:"post",traceId:e.traceId,spanId:e.spanId,model:e.model,operation:"chat"}),S=X(i,f);s.enqueue({event_id:u(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:this.detectProvider(e.model),model:e.model,operation:"chat",latency_ms:d,ttft_ms:e.ttft,stream:e.isStreaming??!1,input_tokens:o,output_tokens:a,status:"success",span_kind:"llm",agent_id:e.agentId,agent_name:e.agentName,finish_reason:g,prompt_hash:Et(JSON.stringify(e.prompts)),tools_defined:e.toolsAvailable?.length?e.toolsAvailable:void 0,tools_used:l.length>0?l.map(b=>b.name):void 0,tool_calls:l.length>0?l:void 0,conversation_turns:e.conversationTurns,has_memory:e.hasMemory,reasoning_tokens:p>0?p:void 0,has_reasoning:p>0?!0:void 0,cached_tokens:c>0?c:void 0,has_planning:m.detected||void 0,planning_steps:m.steps,...s.storePrompts?{prompt_text:s.redactForStorage(e.prompts),completion_text:s.redactForStorage(S)}:{},metadata:{usage:t?.llmOutput?.tokenUsage??null,llm_output:t?.llmOutput??null,invocation_params:e.invocationParams??null,guardrails:{pre:e.preGuardrailResult?{matched:e.preGuardrailResult.results.length,modified:!!e.preGuardrailResult.modifiedText}:null,post:f?{matched:f.results.length,modified:!!f.modifiedText}:null}},user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:"1.0.0"})}catch{}}async handleLLMError(t,n){try{let e=this.runs.get(n);if(!e)return;Q(e.spanId),this.runs.delete(n);let s=P();s.enqueue({event_id:u(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:this.detectProvider(e.model),model:e.model,operation:"chat",latency_ms:performance.now()-e.t0,ttft_ms:e.ttft,stream:e.isStreaming??!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",span_kind:"llm",agent_id:e.agentId,agent_name:e.agentName,error_code:t instanceof J?"guardrail_blocked":void 0,error_message:String(t.message).slice(0,1e3),user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:"1.0.0"})}catch{}}async handleChainStart(t,n,e,s){try{let o=G(),a=s?this.runs.get(s):void 0,d=u(),i=a?.traceId??o?.traceId??u(),l=Array.isArray(t?.id)?t.id:[],p=l.some(y=>Gt.includes(y)),c=!p&&l.some(y=>Dt.includes(y)),m=p?"agent":c?"tool":"general",g=p?u():void 0,f=p?l.find(y=>Gt.includes(y))??void 0:void 0,S=c?t.name??t.kwargs?.name??l.find(y=>Dt.includes(y)):void 0,b=c?JSON.stringify(n):void 0,I=a?.spanKind==="agent"?a.agentId:void 0,k=a?.spanKind==="agent"?a.agentName:void 0;this.runs.set(e,{t0:performance.now(),traceId:i,spanId:d,parentSpanId:a?.spanId??o?.spanId,model:"",agentId:g??a?.agentId,agentName:f??a?.agentName,spanKind:m,toolName:S,toolInput:b,prompts:[],parent:o,delegatedFromAgentId:I,delegatedFromAgentName:k})}catch{}}async handleChainEnd(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=P();if(e.spanKind==="tool"){let o=t?.output??t,a=typeof o=="string"?o:JSON.stringify(o);s.enqueue({event_id:u(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:`tool.${e.toolName??"unknown"}`,latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"success",span_kind:"tool",agent_id:e.agentId,agent_name:e.agentName,tools_used:e.toolName?[e.toolName]:[],tool_call_id:e.spanId,tool_call_args:e.toolInput?JSON.parse(e.toolInput):void 0,tool_call_result:a.slice(0,10240),stream:!1,finish_reason:"stop",user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:"1.0.0"})}else s.enqueue({event_id:u(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:"chain",latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"success",span_kind:e.spanKind,agent_id:e.agentId,agent_name:e.agentName,stream:!1,finish_reason:"stop",delegation_from_agent_id:e.delegatedFromAgentId,delegation_from_agent_name:e.delegatedFromAgentName,user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:"1.0.0"})}catch{}}async handleChainError(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=P();s.enqueue({event_id:u(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:e.spanKind==="tool"?`tool.${e.toolName??"unknown"}`:"chain",latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"error",span_kind:e.spanKind,agent_id:e.agentId,agent_name:e.agentName,tools_used:e.spanKind==="tool"&&e.toolName?[e.toolName]:void 0,tool_call_id:e.spanKind==="tool"?e.spanId:void 0,tool_call_args:e.spanKind==="tool"&&e.toolInput?JSON.parse(e.toolInput):void 0,error_message:String(t?.message??t).slice(0,1e3),stream:!1,delegation_from_agent_id:e.delegatedFromAgentId,delegation_from_agent_name:e.delegatedFromAgentName,user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:"1.0.0"})}catch{}}async handleToolStart(t,n,e,s){try{let o=G(),a=s?this.runs.get(s):void 0,d=u(),i=a?.traceId??o?.traceId??u(),l=t.id?.[t.id.length-1]??"unknown_tool",p=t.name||t.kwargs?.name||l;this.runs.set(e,{t0:performance.now(),traceId:i,spanId:d,parentSpanId:a?.spanId??o?.spanId,model:"",agentId:a?.agentId,agentName:a?.agentName,spanKind:"tool",toolName:p,toolInput:n,prompts:[],parent:o})}catch{}}async handleToolEnd(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=P(),a=(typeof t=="string"?t:JSON.stringify(t)).slice(0,10240);s.enqueue({event_id:u(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:`tool.${e.toolName??"unknown"}`,latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"success",span_kind:"tool",agent_id:e.agentId,agent_name:e.agentName,tools_used:e.toolName?[e.toolName]:[],tool_call_id:e.spanId,tool_call_args:e.toolInput?{input:e.toolInput}:void 0,tool_call_result:a,stream:!1,finish_reason:"stop",user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:"1.0.0"})}catch{}}async handleToolError(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=P();s.enqueue({event_id:u(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:`tool.${e.toolName??"unknown"}`,latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"error",span_kind:"tool",agent_id:e.agentId,agent_name:e.agentName,tools_used:e.toolName?[e.toolName]:[],error_message:String(t?.message??t).slice(0,1e3),stream:!1,user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:"1.0.0"})}catch{}}async handleAgentAction(t,n){try{let e=this.runs.get(n),s=G(),o=P(),a=e?.traceId??s?.traceId??u(),d=u(),i=[t?.tool??"",typeof t?.toolInput=="string"?t.toolInput:JSON.stringify(t?.toolInput??""),t?.log??""].join(" | ");o.enqueue({event_id:u(),project_id:"",trace_id:a,span_id:d,parent_span_id:e?.spanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:"planning",latency_ms:0,input_tokens:0,output_tokens:0,status:"success",span_kind:"planning",agent_id:e?.agentId,agent_name:e?.agentName,planning_steps:[i],stream:!1,user_id:e?.parent?.userId??s?.userId,session_id:e?.parent?.sessionId??s?.sessionId,tags:e?.parent?.tags??s?.tags,environment:o.environment,sdk_version:"1.0.0"})}catch{}}async handleRetrieverStart(t,n,e,s){try{let o=G(),a=s?this.runs.get(s):void 0,d=u(),i=a?.traceId??o?.traceId??u();this.runs.set(e,{t0:performance.now(),traceId:i,spanId:d,parentSpanId:a?.spanId??o?.spanId,model:"",agentId:a?.agentId,agentName:a?.agentName,spanKind:"retriever",query:n,prompts:[],parent:o})}catch{}}async handleRetrieverEnd(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=P();s.enqueue({event_id:u(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:"retriever",latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"success",span_kind:"retriever",agent_id:e.agentId,agent_name:e.agentName,stream:!1,finish_reason:"stop",metadata:{"lumiq.rag.contexts":Array.isArray(t)?t:[]},user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:"1.0.0"})}catch{}}async handleRetrieverError(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=P();s.enqueue({event_id:u(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:"retriever",latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"error",span_kind:"retriever",agent_id:e.agentId,agent_name:e.agentName,error_message:String(t?.message??t).slice(0,1e3),stream:!1,user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:"1.0.0"})}catch{}}},$n=xe;function Ft(r){let t=r?.model;return typeof t=="string"?t:t&&typeof t=="object"?t.model??t.name??"gemini-2.5-flash":"gemini-2.5-flash"}function Wn(r){let t=r??{};return{input:Number(t.promptTokenCount??t.inputTokenCount??0),output:Number(t.candidatesTokenCount??t.outputTokenCount??0),cached:Number(t.cachedContentTokenCount??0),reasoning:Number(t.thoughtsTokenCount??t.reasoningTokenCount??0)}}var He=class{agentStates=new Map;pendingModelCalls=new Map;toolCallStates=new Map;agentKey(t,n){return`${t}:${n}`}getParentAgentState(t,n){if(n)return this.agentStates.get(this.agentKey(t,n))}beforeAgentCallback=t=>{try{let n=t.invocationId,e=t.agentName,s=t.invocationContext?.agent?.parentAgent?.name,o=G(),a=this.getParentAgentState(n,s),d=a?.traceId??o?.traceId??u(),i=u(),l=a?.spanId??o?.spanId,p=u(),c=Ft(t.invocationContext?.agent),m=t.userId||o?.userId,g=t.sessionId||o?.sessionId;this.agentStates.set(this.agentKey(n,e),{t0:performance.now(),traceId:d,spanId:i,parentSpanId:l,agentId:p,agentName:e,model:c,userId:m,sessionId:g,parentAgentId:a?.agentId,parentAgentName:s})}catch{}};afterAgentCallback=t=>{try{let n=t.invocationId,e=t.agentName,s=this.agentKey(n,e),o=this.agentStates.get(s);if(!o)return;this.agentStates.delete(s);let a=!o.parentAgentName,d=P();d.enqueue({event_id:u(),project_id:"",trace_id:o.traceId,span_id:o.spanId,parent_span_id:o.parentSpanId,timestamp:new Date().toISOString(),provider:"google-adk",model:o.model,operation:`agent.${e}`,span_kind:"agent",agent_id:o.agentId,agent_name:e,agent_role:a?"coordinator":"specialist",agent_framework:"google-adk",parent_agent_id:o.parentAgentId,delegation_from_agent_id:o.parentAgentId,delegation_from_agent_name:o.parentAgentName,latency_ms:performance.now()-o.t0,stream:!1,input_tokens:0,output_tokens:0,status:"success",user_id:o.userId,session_id:o.sessionId,environment:d.environment,sdk_version:"1.0.0"})}catch{}};beforeModelCallback=({context:t,request:n})=>{try{let e=t.invocationId,s=t.agentName,o=this.agentStates.get(this.agentKey(e,s)),a=o?.traceId??u(),d=Ft(t.invocationContext?.agent);this.pendingModelCalls.set(this.agentKey(e,s),{t0:performance.now(),spanId:u(),traceId:a,parentSpanId:o?.spanId??"",model:d,agentId:o?.agentId,agentName:s,userId:o?.userId??t.userId,sessionId:o?.sessionId??t.sessionId})}catch{}};afterModelCallback=({context:t,response:n})=>{try{let e=t.invocationId,s=t.agentName,o=this.agentKey(e,s),a=this.pendingModelCalls.get(o);if(!a)return;this.pendingModelCalls.delete(o);let{input:d,output:i,cached:l,reasoning:p}=Wn(n?.usageMetadata),c=n?.content?.parts??[],m=c.filter(S=>typeof S?.text=="string"&&!S.thought).map(S=>S.text).join(""),g=c.filter(S=>S?.functionCall?.name).map(S=>S.functionCall.name),f=P();f.enqueue({event_id:u(),project_id:"",trace_id:a.traceId,span_id:a.spanId,parent_span_id:a.parentSpanId,timestamp:new Date().toISOString(),provider:"google",model:a.model,operation:"chat",span_kind:"llm",agent_id:a.agentId,agent_name:a.agentName,agent_framework:"google-adk",latency_ms:performance.now()-a.t0,stream:!1,input_tokens:d,output_tokens:i,cached_tokens:l||void 0,reasoning_tokens:p||void 0,finish_reason:n?.finishReason??"stop",tools_used:g.length?g:void 0,status:n?.errorCode?"error":"success",error_message:n?.errorMessage,...f.storePrompts&&m?{completion_text:f.redactForStorage(m)}:{},user_id:a.userId,session_id:a.sessionId,environment:f.environment,sdk_version:"1.0.0"})}catch{}};beforeToolCallback=({tool:t,args:n,context:e})=>{try{let s=e.invocationId,o=e.agentName,a=e.functionCallId??u(),d=this.agentStates.get(this.agentKey(s,o));this.toolCallStates.set(a,{t0:performance.now(),spanId:u(),traceId:d?.traceId??u(),parentSpanId:d?.spanId??"",toolName:t?.name??"unknown_tool",args:n??{},agentId:d?.agentId,agentName:o,userId:d?.userId??e.userId,sessionId:d?.sessionId??e.sessionId})}catch{}};afterToolCallback=({tool:t,args:n,context:e,response:s})=>{try{let o=e.functionCallId??"",a=this.toolCallStates.get(o);if(!a)return;this.toolCallStates.delete(o);let d=P();d.enqueue({event_id:u(),project_id:"",trace_id:a.traceId,span_id:a.spanId,parent_span_id:a.parentSpanId,timestamp:new Date().toISOString(),provider:"google-adk",model:"",operation:`tool.${a.toolName}`,span_kind:"tool",agent_id:a.agentId,agent_name:a.agentName,agent_framework:"google-adk",latency_ms:performance.now()-a.t0,stream:!1,input_tokens:0,output_tokens:0,status:"success",tools_used:[a.toolName],tool_call_id:o,tool_call_args:a.args,tool_call_result:JSON.stringify(s??{}).slice(0,10240),user_id:a.userId,session_id:a.sessionId,environment:d.environment,sdk_version:"1.0.0"})}catch{}};get callbacks(){return{beforeAgentCallback:this.beforeAgentCallback,afterAgentCallback:this.afterAgentCallback,beforeModelCallback:this.beforeModelCallback,afterModelCallback:this.afterModelCallback,beforeToolCallback:this.beforeToolCallback,afterToolCallback:this.afterToolCallback}}};import{createHash as Bn}from"crypto";var mt=r=>Bn("sha256").update(typeof r=="string"?r:JSON.stringify(r)).digest("hex");function Zn(r){return r?.model??r?.config?.model??r?.llmConfig?.model??r?.generativeModel?.model??"gemini-2.5-flash"}function Kt(r){let t=r?.usageMetadata??r?.usage_metadata??{},n=r?.usage??{};return{input:Number(t.inputTokenCount??t.promptTokenCount??n.input_tokens??n.prompt_tokens??0),output:Number(t.outputTokenCount??t.candidatesTokenCount??n.output_tokens??n.completion_tokens??0),cached:Number(t.cachedContentTokenCount??n.cached_tokens??0),reasoning:Number(t.reasoningTokenCount??t.thinking_token_count??n.reasoning_tokens??0)}}function Jn(r){if(typeof r?.text=="string")return r.text;if(typeof r?.output=="string")return r.output;let n=(r?.candidates?.[0]?.content?.parts??r?.content?.parts??[]).map(e=>typeof e?.text=="string"?e.text:"").filter(Boolean).join("");return n||(r&&typeof r=="object"?JSON.stringify(r):"")}function zn(r){let t=[];try{let n=r?.candidates?.[0]?.content?.parts??r?.content?.parts??[];for(let e of n)e?.functionCall?.name&&t.push({name:e.functionCall.name,args:e.functionCall.args??{}});for(let e of r?.toolCalls??[])e?.name&&t.push({name:e.name,args:e.args??e.arguments??{}})}catch{}return t}function $t(r){if(r instanceof J)return"guardrail_blocked";let t=String(r?.message??"").toLowerCase();if(/rate.?limit|quota|429/.test(t))return"rate_limited";if(/context.?length|too.?long|max.?token/.test(t))return"context_length";if(/timeout|timed?.?out/.test(t))return"timeout";if(/safety|blocked|harm/.test(t))return"content_filter"}function Qe(r,t={}){let n=r;if(typeof n?.run!="function")return r;let e=n.run.bind(n),s=V(t.guardrails,"agent"),o=t.agentName??n.name??"ADKAgent",a=t.agentRole??(n.subAgents?.length?"coordinator":"specialist"),d=Zn(n);return t.traceSubAgents!==!1&&Array.isArray(n.subAgents)&&(n.subAgents=n.subAgents.map(i=>Qe(i,{...t,agentRole:"specialist",traceSubAgents:!0}))),n.run=async(i,l)=>{let p=P(),c=performance.now(),m=G(),g=u(),f=u(),S=m?.traceId??u(),b=t.sessionId??m?.sessionId,I=typeof i=="string"?i:JSON.stringify(i??"");return W({traceId:S,spanId:f,userId:m?.userId,sessionId:b,tags:m?.tags,agentId:g,agentName:o,agentRole:a,agentFramework:"google-adk",parentAgentId:m?.agentId},async()=>{try{let k=await B(p,s,{text:I,phase:"pre",traceId:S,spanId:f,model:d,operation:"agent"}),y=await e(i,l),{input:x,output:_,cached:E,reasoning:v}=Kt(y),T=Jn(y),h=zn(y),A=y?.candidates?.[0]?.finishReason??y?.finish_reason,R=await B(p,s,{text:T,phase:"post",traceId:S,spanId:f,model:d,operation:"agent"}),w=X(T,R);return p.enqueue({event_id:u(),project_id:"",trace_id:S,span_id:f,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:d,operation:"agent",span_kind:"agent",agent_id:g,agent_name:o,agent_role:a,agent_framework:"google-adk",parent_agent_id:m?.agentId,delegation_from_agent_id:m?.agentId,delegation_from_agent_name:m?.agentName,latency_ms:performance.now()-c,stream:!1,input_tokens:x,output_tokens:_,cached_tokens:E||void 0,reasoning_tokens:v||void 0,status:"success",finish_reason:A,tools_used:h.length?h.map(M=>M.name):void 0,tool_call_args:h.length?Object.fromEntries(h.map(M=>[M.name,M.args])):void 0,prompt_hash:mt(I),...p.storePrompts?{prompt_text:p.redactForStorage(I),completion_text:p.redactForStorage(w)}:{},user_id:m?.userId,session_id:b,tags:m?.tags,environment:p.environment,sdk_version:"1.0.0",metadata:{guardrails:{pre:k?{matched:k.results.length,modified:!!k.modifiedText}:null,post:R?{matched:R.results.length,modified:!!R.modifiedText}:null}}}),y}catch(k){throw p.enqueue({event_id:u(),project_id:"",trace_id:S,span_id:f,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:d,operation:"agent",span_kind:"agent",agent_id:g,agent_name:o,agent_role:a,agent_framework:"google-adk",parent_agent_id:m?.agentId,latency_ms:performance.now()-c,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_code:$t(k),error_message:String(k?.message??k).slice(0,500),user_id:m?.userId,session_id:b,tags:m?.tags,environment:p.environment,sdk_version:"1.0.0"}),k}})},r}function gt(r,t={}){let n=r;if(typeof n?.runEphemeral=="function"){let e=n.runEphemeral.bind(n);n.runEphemeral=s=>Lt(e(s),s,t)}if(typeof n?.run=="function"){let e=n.run.bind(n);n.run=s=>Lt(e(s),s,t)}return r}async function*Lt(r,t,n){let e=P(),s=G(),o=s?.traceId??u(),a=u(),d=u(),i=performance.now(),l=n.agentName??"ADKAgent",p=n.agentRole??"coordinator",c=n.model??"gemini-2.5-flash",m=t?.userId??s?.userId,g=n.sessionId??t?.sessionId??s?.sessionId,f=(t?.newMessage?.parts??[]).map(v=>typeof v?.text=="string"?v.text:"").filter(Boolean).join(""),S=new Map,b=new Map,I=new Set,k="",y=0,x=0,_=0,E=0;ie(a);try{yield*W({traceId:o,spanId:a,userId:m,sessionId:g,tags:s?.tags,agentId:d,agentName:l,agentRole:p,agentFramework:"google-adk",parentAgentId:s?.agentId},async function*(){for await(let v of r){y++;let T=v.author??l;if(T!=="user"){T!==l&&!I.has(T)&&e.enqueue({event_id:u(),project_id:"",trace_id:o,span_id:u(),parent_span_id:a,timestamp:new Date().toISOString(),provider:"google-adk",model:c,operation:`handoff.${l}\u2192${T}`,span_kind:"handoff",agent_name:T,agent_role:"specialist",agent_framework:"google-adk",delegation_from_agent_id:d,delegation_from_agent_name:l,delegation_reason:`delegated to ${T}`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",user_id:m,session_id:g,tags:s?.tags,environment:e.environment,sdk_version:"1.0.0"}),I.add(T),S.has(T)||(S.set(T,{spanId:u(),agentId:u(),t0:performance.now()}),b.set(T,{input:0,output:0,cached:0,reasoning:0}));let h=S.get(T),A=Kt(v);if(A.input>0||A.output>0){let C=b.get(T);C.input+=A.input,C.output+=A.output,C.cached+=A.cached,C.reasoning+=A.reasoning,x+=A.input,_+=A.output,E+=A.cached}let R=v.content?.parts??[];for(let C of R){let O=C?.functionCall;if(!O?.name)continue;let q=u();e.enqueue({event_id:u(),project_id:"",trace_id:o,span_id:q,parent_span_id:h.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:c,operation:`tool.${O.name}`,span_kind:"tool",agent_id:h.agentId,agent_name:T,agent_framework:"google-adk",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",tools_used:[O.name],tool_call_id:q,tool_call_args:O.args??{},user_id:m,session_id:g,tags:s?.tags,environment:e.environment,sdk_version:"1.0.0"})}let w=R.filter(C=>typeof C?.text=="string"&&!C.thought).map(C=>C.text).join("");if(w&&(k+=w),typeof v.isFinalResponse=="function"?v.isFinalResponse():!!v.isFinalResponse||v.actions?.stateDelta!==void 0||v.turnComplete===!0){let C=R.map(q=>q?.functionCall?.name).filter(Boolean),O=b.get(T)??{input:0,output:0,cached:0,reasoning:0};e.enqueue({event_id:u(),project_id:"",trace_id:o,span_id:h.spanId,parent_span_id:a,timestamp:new Date().toISOString(),provider:"google-adk",model:c,operation:`agent.${T}`,span_kind:"agent",agent_id:h.agentId,agent_name:T,agent_role:T===l?p:"specialist",agent_framework:"google-adk",parent_agent_id:d,latency_ms:performance.now()-h.t0,stream:!1,input_tokens:O.input,output_tokens:O.output,cached_tokens:O.cached||void 0,reasoning_tokens:O.reasoning||void 0,status:"success",tools_used:C.length?C:void 0,prompt_hash:f?mt(f):void 0,...e.storePrompts&&w?{completion_text:e.redactForStorage(w)}:{},user_id:m,session_id:g,tags:s?.tags,environment:e.environment,sdk_version:"1.0.0"}),S.delete(T),b.delete(T)}}yield v}}),e.enqueue({event_id:u(),project_id:"",trace_id:o,span_id:a,parent_span_id:s?.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:c,operation:"agent.run",span_kind:"agent",agent_id:d,agent_name:l,agent_role:p,agent_framework:"google-adk",parent_agent_id:s?.agentId,latency_ms:performance.now()-i,stream:!1,input_tokens:x,output_tokens:_,cached_tokens:E||void 0,status:"success",prompt_hash:f?mt(f):void 0,...e.storePrompts?{prompt_text:e.redactForStorage(f),completion_text:e.redactForStorage(k)}:{},user_id:m,session_id:g,tags:s?.tags,environment:e.environment,sdk_version:"1.0.0",metadata:{eventCount:y,agentsThatRan:[...I]}}),Q(a)}catch(v){throw Q(a),e.enqueue({event_id:u(),project_id:"",trace_id:o,span_id:a,parent_span_id:s?.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:c,operation:"agent.run",span_kind:"agent",agent_id:d,agent_name:l,agent_role:p,agent_framework:"google-adk",parent_agent_id:s?.agentId,latency_ms:performance.now()-i,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_code:$t(v),error_message:String(v?.message??v).slice(0,500),user_id:m,session_id:g,tags:s?.tags,environment:e.environment,sdk_version:"1.0.0"}),v}}function Wt(r){let{coordinator:t,runner:n,agentName:e,agentRole:s,model:o,...a}=r,d=Qe(t,{...a,agentName:e??t?.name,agentRole:s??"coordinator",traceSubAgents:!0}),i=gt(n,{agentName:e??t?.name,agentRole:s??"coordinator",model:o});return{coordinator:d,runner:i}}var Bt=Qe;import{NodeTracerProvider as Un}from"@opentelemetry/sdk-trace-node";import{ExportResultCode as Zt}from"@opentelemetry/core";function Vn(){try{return P()}catch{return null}}var ft=class{export(t,n){let e=Vn();if(!e){n({code:Zt.SUCCESS});return}try{for(let s of t){let{name:o}=s;if(!o.startsWith("ai."))continue;let a=s.attributes,d=G(),i=s.spanContext().spanId,l=s.spanContext().traceId,p=s.duration[0]*1e3+s.duration[1]/1e6,c=new Date(s.startTime[0]*1e3+s.startTime[1]/1e6).toISOString(),m=s.status.code===2,g={event_id:u(),project_id:"",trace_id:d?.traceId??l,span_id:i,parent_span_id:d?.spanId,timestamp:c,provider:String(a["ai.model.provider"]??"unknown"),model:String(a["ai.model.id"]??"unknown"),latency_ms:p,stream:!1,status:m?"error":"success",environment:e.environment,sdk_version:"1.0.0"};o==="ai.generateText"||o==="ai.generateText.doGenerate"?(Q(i),e.enqueue({...g,operation:"generateText",input_tokens:Number(a["ai.usage.promptTokens"]??0),output_tokens:Number(a["ai.usage.completionTokens"]??0),finish_reason:String(a["ai.response.finishReason"]??""),span_kind:"llm",...e.storePrompts&&a["ai.prompt"]?{prompt_text:e.redactForStorage(String(a["ai.prompt"]))}:{},...e.storePrompts&&a["ai.response.text"]?{completion_text:e.redactForStorage(String(a["ai.response.text"]))}:{}})):o==="ai.streamText"||o==="ai.streamText.doStream"?(Q(i),e.enqueue({...g,operation:"streamText",stream:!0,input_tokens:Number(a["ai.usage.promptTokens"]??0),output_tokens:Number(a["ai.usage.completionTokens"]??0),finish_reason:String(a["ai.response.finishReason"]??""),ttft_ms:a["ai.response.msToFirstChunk"]!==void 0?Number(a["ai.response.msToFirstChunk"]):void 0,span_kind:"llm"})):o==="ai.generateObject"||o==="ai.generateObject.doGenerate"?(Q(i),e.enqueue({...g,operation:"generateObject",input_tokens:Number(a["ai.usage.promptTokens"]??0),output_tokens:Number(a["ai.usage.completionTokens"]??0),span_kind:"general"})):o==="ai.toolCall"?e.enqueue({...g,provider:"vercel-ai",model:"tool",operation:`tool.${String(a["ai.toolCall.name"]??"unknown")}`,input_tokens:0,output_tokens:0,span_kind:"tool",tools_used:a["ai.toolCall.name"]?[String(a["ai.toolCall.name"])]:[]}):(o==="ai.embed"||o==="ai.embedMany")&&e.enqueue({...g,operation:o==="ai.embed"?"embed":"embedMany",input_tokens:Number(a["ai.usage.tokens"]??0),output_tokens:0,span_kind:"embedding"})}}catch{}n({code:Zt.SUCCESS})}shutdown(){return Promise.resolve()}},ht=class{exporter;constructor(t){this.exporter=t}onStart(t,n){t.name?.startsWith("ai.")&&ie(t.spanContext().spanId)}onEnd(t){t.name.startsWith("ai.")&&Q(t.spanContext().spanId),this.exporter.export([t],()=>{})}forceFlush(){return Promise.resolve()}shutdown(){return Promise.resolve()}},Xe=null,_t=null;function yt(){return Xe||(Xe=new Un,Xe.addSpanProcessor(new ht(new ft)),_t=Xe.getTracer("zespan-vercel-ai")),{tracer:_t,telemetry:{isEnabled:!0,tracer:_t}}}function Jt(){return yt().telemetry}function Te(){try{return P()}catch{return null}}function Hn(r){let t=r.toLowerCase();return/gpt|o1-|o3-|davinci|turbo|whisper|dall-e/i.test(t)?"openai":/claude/i.test(t)?"anthropic":/gemini|palm|bard/i.test(t)?"google":/llama|codellama/i.test(t)?"meta":/mistral|mixtral/i.test(t)?"mistralai":/deepseek/i.test(t)?"deepseek":/command/i.test(t)?"cohere":/qwen/i.test(t)?"alibaba":"unknown"}var Ye=class{runs=new Map;guardrailSettings;constructor(t={}){this.guardrailSettings=V(t.guardrails,"chat")}_onLLMStart=async t=>{let n=t.id??u(),e=G(),s=u();ie(s);let o=t.detail?.llm?.metadata?.model_name??t.detail?.model??"unknown",a=e?.traceId??u(),d=JSON.stringify(t.detail?.messages??t.detail?.prompt??t.detail?.input??""),i;try{let l=Te();l&&d&&(i=await B(l,this.guardrailSettings,{text:d,phase:"pre",traceId:a,spanId:s,model:o,operation:"chat"}))}catch(l){if(l instanceof J)throw l}this.runs.set(n,{t0:performance.now(),traceId:a,spanId:s,parentSpanId:e?.spanId,model:o,preGuardrailResult:i,promptText:d})};_onLLMEnd=async t=>{let n=this.runs.get(t.id??"");if(!n)return;this.runs.delete(t.id),Q(n.spanId);let e=Te();if(!e)return;let s=t.detail?.response??t.detail?.chunk??{},o=s.usage??s.raw?.usage??{},a=(typeof s.text=="string"?s.text:null)??(typeof s.message?.content=="string"?s.message.content:null)??"",d=a,i;if(a)try{i=await B(e,this.guardrailSettings,{text:a,phase:"post",traceId:n.traceId,spanId:n.spanId,model:n.model,operation:"chat"}),d=X(a,i)}catch(p){p instanceof J||(i=void 0)}let l=n.preGuardrailResult;try{e.enqueue({event_id:u(),project_id:"",trace_id:n.traceId,span_id:n.spanId,parent_span_id:n.parentSpanId,timestamp:new Date().toISOString(),provider:Hn(n.model),model:n.model,operation:"chat",latency_ms:performance.now()-n.t0,ttft_ms:n.ttft,stream:!1,input_tokens:o.promptTokens??o.prompt_tokens??0,output_tokens:o.completionTokens??o.completion_tokens??0,finish_reason:s.finishReason??s.finish_reason??"",status:"success",span_kind:"llm",agent_id:G()?.agentId,agent_name:G()?.agentName,...e.storePrompts?{prompt_text:e.redactForStorage(n.promptText??""),completion_text:e.redactForStorage(d)}:{},metadata:{guardrails:{pre:l?{matched:l.results.length,modified:!!l.modifiedText}:null,post:i?{matched:i.results.length,modified:!!i.modifiedText}:null}},environment:e.environment,sdk_version:"1.0.0"})}catch{}};_onLLMStream=t=>{let n=this.runs.get(t.id??"");n&&n.ttft===void 0&&(n.ttft=performance.now()-n.t0)};_onToolCall=t=>{let n=Te();if(!n)return;let e=G(),s=t.detail?.toolCall?.name??t.detail?.name??"";try{n.enqueue({event_id:u(),project_id:"",trace_id:e?.traceId??u(),span_id:u(),parent_span_id:e?.spanId,timestamp:new Date().toISOString(),provider:"llamaindex",model:"",operation:`tool.${s||"unknown"}`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",span_kind:"tool",tools_used:s?[s]:[],agent_id:e?.agentId,agent_name:e?.agentName,environment:n.environment,sdk_version:"1.0.0"})}catch{}};_onToolResult=t=>{let n=Te();if(!n)return;let e=G(),s=t.detail?.toolCall?.name??t.detail?.name??"",o=t.detail?.result??t.detail?.toolResult??"";try{n.enqueue({event_id:u(),project_id:"",trace_id:e?.traceId??u(),span_id:u(),parent_span_id:e?.spanId,timestamp:new Date().toISOString(),provider:"llamaindex",model:"",operation:`tool.${s||"unknown"}`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",span_kind:"tool",tools_used:s?[s]:[],tool_call_result:String(o).slice(0,10240),agent_id:e?.agentId,agent_name:e?.agentName,environment:n.environment,sdk_version:"1.0.0"})}catch{}};_onAgentStart=t=>{let n=t.id??u(),e=G();this.runs.set(`agent-${n}`,{t0:performance.now(),traceId:e?.traceId??u(),spanId:u(),parentSpanId:e?.spanId,model:"agent",agentName:t.detail?.agentName??e?.agentName??""})};_onAgentEnd=t=>{let n=this.runs.get(`agent-${t.id??""}`);if(!n)return;this.runs.delete(`agent-${t.id}`);let e=Te();if(e)try{e.enqueue({event_id:u(),project_id:"",trace_id:n.traceId,span_id:n.spanId,parent_span_id:n.parentSpanId,timestamp:new Date().toISOString(),provider:"llamaindex",model:"agent",operation:"agent",latency_ms:performance.now()-n.t0,stream:!1,input_tokens:0,output_tokens:0,status:"success",span_kind:"agent",agent_name:n.agentName??t.detail?.agentName??"",environment:e.environment,sdk_version:"1.0.0"})}catch{}};attach(t){t.on("llm-start",this._onLLMStart),t.on("llm-end",this._onLLMEnd),t.on("llm-stream",this._onLLMStream),t.on("llm-tool-call",this._onToolCall),t.on("llm-tool-result",this._onToolResult),t.on("agent-start",this._onAgentStart),t.on("agent-end",this._onAgentEnd)}detach(t){t.off("llm-start",this._onLLMStart),t.off("llm-end",this._onLLMEnd),t.off("llm-stream",this._onLLMStream),t.off("llm-tool-call",this._onToolCall),t.off("llm-tool-result",this._onToolResult),t.off("agent-start",this._onAgentStart),t.off("agent-end",this._onAgentEnd)}};function zt(r){let t=P(),n=G(),e=n?.traceId??u(),s=u(),o=performance.now(),a=n?.agentId?{agent_id:n.agentId,agent_name:n.agentName,agent_role:n.agentRole,agent_framework:n.agentFramework,parent_agent_id:n.parentAgentId,span_kind:r.span_kind??"general"}:{},d={};return{span:{setEvalScore(l,p){d[l]=p},async end(l){let p=performance.now()-o;try{t.enqueue({event_id:u(),project_id:"",trace_id:e,span_id:s,parent_span_id:n?.spanId,timestamp:new Date().toISOString(),provider:r.provider??"custom",model:r.model??"custom",operation:"custom",latency_ms:p,stream:!1,input_tokens:l.input_tokens??0,output_tokens:l.output_tokens??0,cost_usd:l.cost_usd,status:l.status,error_message:l.error_message?.slice(0,500),environment:t.environment,eval_scores:Object.keys(d).length?d:void 0,sdk_version:"1.0.0",...a})}catch{}}},run:l=>W({traceId:e,spanId:s,userId:n?.userId,sessionId:n?.sessionId,tags:n?.tags,agentId:n?.agentId,agentName:n?.agentName,agentRole:n?.agentRole,agentFramework:n?.agentFramework,parentAgentId:n?.parentAgentId,conversationTurn:n?.conversationTurn,availableTools:n?.availableTools},l)}}var et=class{agentId;name;role;framework;tools;constructor(t,n){this.agentId=t,this.name=n.name,this.role=n.role??"specialist",this.framework=n.framework??"custom",this.tools=n.tools??[]}logPlan(t){let n=P(),e=G(),s={event_id:u(),project_id:"",trace_id:e?.traceId??u(),span_id:u(),parent_span_id:e?.spanId,timestamp:new Date().toISOString(),provider:"agent",model:"",operation:`${this.name}.plan`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",environment:n.environment,sdk_version:"1.0.0",agent_id:this.agentId,agent_name:this.name,agent_role:this.role,agent_framework:this.framework,span_kind:"planning",planning_steps:t,user_id:e?.userId,session_id:e?.sessionId,tags:e?.tags};n.enqueue(s)}async traceTool(t,n,e){let s=P(),o=G(),a=u(),d=o?.traceId??u(),i=performance.now(),l=this.tools.find(p=>p.name===t);try{let p=await W({traceId:d,spanId:a,userId:o?.userId,sessionId:o?.sessionId,tags:o?.tags,agentId:this.agentId,agentName:this.name,agentRole:this.role,agentFramework:this.framework},()=>e()),c=typeof p=="string"?p:JSON.stringify(p),m={event_id:u(),project_id:"",trace_id:d,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"tool",model:"",operation:`tool.${t}`,latency_ms:performance.now()-i,stream:!1,input_tokens:0,output_tokens:0,status:"success",environment:s.environment,sdk_version:"1.0.0",agent_id:this.agentId,agent_name:this.name,agent_role:this.role,agent_framework:this.framework,span_kind:"tool",tools_used:[t],tool_call_id:a,tool_call_args:n,tool_call_result:c.slice(0,10240),tool_definitions:l?[l]:void 0,user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags};return s.enqueue(m),p}catch(p){let c=p instanceof Error?p.message:String(p),m={event_id:u(),project_id:"",trace_id:d,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"tool",model:"",operation:`tool.${t}`,latency_ms:performance.now()-i,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_message:c.slice(0,500),environment:s.environment,sdk_version:"1.0.0",agent_id:this.agentId,agent_name:this.name,agent_role:this.role,agent_framework:this.framework,span_kind:"tool",tools_used:[t],tool_call_id:a,tool_call_args:n,user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags};throw s.enqueue(m),p}}delegateTo(t,n){let e=P(),s=G(),o={event_id:u(),project_id:"",trace_id:s?.traceId??u(),span_id:u(),parent_span_id:s?.spanId,timestamp:new Date().toISOString(),provider:"agent",model:"",operation:`${this.name}.delegate\u2192${t}`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",environment:e.environment,sdk_version:"1.0.0",agent_id:this.agentId,agent_name:this.name,agent_role:this.role,agent_framework:this.framework,span_kind:"handoff",delegation_reason:n,user_id:s?.userId,session_id:s?.sessionId,tags:s?.tags};e.enqueue(o)}};function Ut(r,t){let n=P(),e=G(),s=u(),o=e?.traceId??u(),a=u(),d={event_id:u(),project_id:"",trace_id:o,span_id:a,parent_span_id:e?.spanId,timestamp:new Date().toISOString(),provider:"agent",model:"",operation:`agent.${r.name}`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",environment:n.environment,sdk_version:"1.0.0",agent_id:s,agent_name:r.name,agent_role:r.role??"specialist",agent_framework:r.framework??"custom",parent_agent_id:e?.agentId,span_kind:"agent",tool_definitions:r.tools,delegation_reason:e?.agentId?"delegated":"root",user_id:e?.userId,session_id:e?.sessionId,tags:e?.tags,metadata:r.metadata};return n.enqueue(d),W({traceId:o,spanId:a,userId:e?.userId,sessionId:e?.sessionId,tags:e?.tags,agentId:s,agentName:r.name,agentRole:r.role??"specialist",agentFramework:r.framework??"custom",parentAgentId:e?.agentId,availableTools:r.tools},()=>t(new et(s,r)))}import{propagation as Qn}from"@opentelemetry/api";var kt=class{onStart(t,n){let e=Qn.getBaggage(n);e&&e.getAllEntries().forEach(([s,o])=>{t.setAttribute(`baggage.${s}`,o.value)})}onEnd(){}shutdown(){return Promise.resolve()}forceFlush(){return Promise.resolve()}};import{context as Vt,propagation as Oe}from"@opentelemetry/api";function tt(r,t,n){let e={};t&&(e["agent.delegation.reason"]=t),n&&(e["agent.task"]=n);let s=Vt.active();if(Object.keys(e).length>0){let o=Oe.getBaggage(s)??Oe.createBaggage();for(let[a,d]of Object.entries(e))o=o.setEntry(a,{value:d});s=Oe.setBaggage(s,o)}Oe.inject(s,r)}function nt(r){return Oe.extract(Vt.active(),r)}var Xn={OTEL_EXPORTER_OTLP_ENDPOINT:"http://localhost:4318",OTEL_SERVICE_NAME:"crewai-agent"};function Yn(){return`
|
|
1
|
+
var ot=(a=>typeof require<"u"?require:typeof Proxy<"u"?new Proxy(a,{get:(t,n)=>(typeof require<"u"?require:t)[n]}):a)(function(a){if(typeof require<"u")return require.apply(this,arguments);throw Error('Dynamic require of "'+a+'" is not supported')});import{gzipSync as Yt}from"zlib";var Re=class{constructor(t,n,e){this.c=t;this.cfg=n;this.configClient=e??null,this.maxQueueSize=n.maxQueueSize??1e4,this.timer=setInterval(()=>this.flush(),n.flushInterval),this.timer.unref&&this.timer.unref()}queue=[];timer=null;flushing=!1;consecutiveFlushFailures=0;configClient;maxQueueSize;push(t){this.queue.length>=this.maxQueueSize&&(this.queue.shift(),this.c.debug&&console.warn("[zespan] queue full, dropping oldest event")),this.queue.push(t),this.c.debug&&console.log(`[zespan] event queued: ${t.operation} span_kind=${t.span_kind??"llm"} trace=${t.trace_id?.slice(0,8)} tokens=${t.input_tokens}in/${t.output_tokens}out status=${t.status}`),this.queue.length>=this.cfg.maxSize&&this.flush()}async flush(){if(!this.queue.length||this.flushing)return;this.flushing=!0;let t=this.queue.slice(),n=Yt(Buffer.from(t.map(e=>JSON.stringify(e)).join(`
|
|
2
|
+
`)));try{let e=await this._post(n,0);e.ok?(this.queue.splice(0,t.length),this.c.debug&&console.log(`[zespan] flushed ${t.length} event(s) \u2192 ${this.c.endpoint}`),this.consecutiveFlushFailures>0&&this.c.debug&&console.warn("[zespan] ingest recovered",{recoveredBatchSize:t.length,queuedEvents:this.queue.length,priorFailures:this.consecutiveFlushFailures}),this.consecutiveFlushFailures=0):(this.consecutiveFlushFailures+=1,(this.c.debug||this.consecutiveFlushFailures===1||this.consecutiveFlushFailures%5===0)&&console.error("[zespan] ingest flush failed; retaining events for retry",{retriable:e.retriable,status:e.status??null,error:e.error??null,queuedEvents:this.queue.length,failedBatchSize:t.length,consecutiveFailures:this.consecutiveFlushFailures}))}finally{this.flushing=!1}}async _post(t,n){let e=s=>new Promise(r=>setTimeout(r,s));try{let s=await fetch(this.c.endpoint,{method:"POST",body:t,headers:{"x-api-key":this.c.apiKey,"content-type":"application/x-ndjson","content-encoding":"gzip"}});if(s.ok){try{let r=await s.json();typeof r?.cv=="number"&&this.configClient&&this.configClient.checkVersion(r.cv)}catch{}return{ok:!0,retriable:!1}}return s.status===429&&n<3?(await e(Number(s.headers.get("retry-after")??5)*1e3),this._post(t,n+1)):s.status>=500&&n<3?(await e(2**n*1e3),this._post(t,n+1)):s.status===429||s.status>=500?(this.c.debug&&console.error("[zespan] retryable ingest failure",{status:s.status,attempt:n}),{ok:!1,retriable:!0,status:s.status}):(console.error("[zespan] non-retriable ingest failure",{status:s.status}),{ok:!1,retriable:!1,status:s.status})}catch(s){return n<3?(await e(2**n*1e3),this._post(t,n+1)):(this.c.debug&&console.error("[zespan] ingest transport error",s),{ok:!1,retriable:!0,error:s instanceof Error?s.message:String(s)})}}};import{trace as en,SpanKind as ye,SpanStatusCode as he,context as tn}from"@opentelemetry/api";import{BatchSpanProcessor as nn}from"@opentelemetry/sdk-trace-base";import{OTLPTraceExporter as sn}from"@opentelemetry/exporter-trace-otlp-http";import{AsyncHooksContextManager as on}from"@opentelemetry/context-async-hooks";import{W3CTraceContextPropagator as rn}from"@opentelemetry/core";var Ne=null,vt=null;function Me(a){let t=a.serviceName??"zespan-sdk",n=new sn({url:a.endpoint??"https://localhost:3001/v1/traces",headers:a.apiKey?{"x-api-key":a.apiKey}:void 0}),e=new nn(n,{maxQueueSize:100,maxExportBatchSize:10,exportTimeoutMillis:3e4,scheduledDelayMillis:2e3}),{NodeTracerProvider:s}=ot("@opentelemetry/sdk-trace-node"),r=new s({resource:{attributes:{"service.name":t,"service.version":"1.0.0","deployment.environment":a.environment??"production"}}});return r.addSpanProcessor(e),r.register({propagator:new rn}),Ne=en.getTracer(t,"1.0.0"),vt=new on,tn.setGlobalContextManager(vt.enable()),{tracer:Ne,provider:r}}function fe(){if(!Ne)throw new Error("OTel not initialized. Call initOTel() first.");return Ne}function wt(a){return fe().startSpan(a.name,{kind:a.kind??ye.INTERNAL,attributes:a.attributes??{}})}async function bt(a,t,n){let s=fe().startSpan(a,{kind:n?.kind??ye.INTERNAL,attributes:n?.attributes??{}});try{let r=await t(s);return s.setStatus({code:he.OK}),s.end(),r}catch(r){throw s.setStatus({code:he.ERROR,message:r.message}),s.recordException(r),s.end(),r}}import{createHash as an}from"crypto";var ke=class{constructor(t){this.client=t}cache=new Map;cacheTTL=300*1e3;async get(t,n={}){let{version:e,label:s,cache:r=!0}=n;if(r){let m=this.getCacheKey(t,e,s),_=this.cache.get(m);if(_&&Date.now()-_.timestamp<this.cacheTTL)return _.prompt}let o=new URLSearchParams;e!==void 0&&o.set("version",e.toString()),s&&o.set("label",s);let c=`${this.client.baseURL}/v1/prompts/${encodeURIComponent(t)}?${o}`,i=await fetch(c,{headers:{"x-api-key":this.client.apiKey}});if(!i.ok)throw new Error(`Failed to fetch prompt: ${i.statusText}`);let l=await i.json(),u=an("sha256").update(JSON.stringify(l.prompt)).digest("hex"),d={...l,promptHash:u};if(r){let m=this.getCacheKey(t,e,s);this.cache.set(m,{prompt:d,timestamp:Date.now()})}return d}async list(t){let n=new URLSearchParams;t&&n.set("name",t);let e=`${this.client.baseURL}/v1/prompts?${n}`,s=await fetch(e,{headers:{"x-api-key":this.client.apiKey}});if(!s.ok)throw new Error(`Failed to list prompts: ${s.statusText}`);return(await s.json()).prompts||[]}async create(t){let n=`${this.client.baseURL}/v1/prompts`,e=await fetch(n,{method:"POST",headers:{"Content-Type":"application/json","x-api-key":this.client.apiKey},body:JSON.stringify(t)});if(!e.ok)throw new Error(`Failed to create prompt: ${e.statusText}`);let s=await e.json();return this.invalidateCache(t.name),s}async updateLabels(t,n,e){let s=`${this.client.baseURL}/v1/prompts/${encodeURIComponent(t)}/versions/${n}/labels`,r=await fetch(s,{method:"PATCH",headers:{"Content-Type":"application/json","x-api-key":this.client.apiKey},body:JSON.stringify({labels:e})});if(!r.ok)throw new Error(`Failed to update labels: ${r.statusText}`);let o=await r.json();return this.invalidateCache(t),o}compile(t,n){if(t.type==="text"){let e=typeof t.prompt=="string"?t.prompt:t.prompt.content;return Object.entries(n).forEach(([s,r])=>{e=e.replace(new RegExp(`\\{\\{${s}\\}\\}`,"g"),r)}),e}return JSON.stringify(t.prompt)}clearCache(t){t?this.invalidateCache(t):this.cache.clear()}getCacheKey(t,n,e){return n!==void 0?`${t}:v${n}`:e?`${t}:label:${e}`:`${t}:latest`}invalidateCache(t){let n=[];for(let e of this.cache.keys())e.startsWith(`${t}:`)&&n.push(e);n.forEach(e=>this.cache.delete(e))}};var cn="[REDACTED]",xt=a=>a.toLowerCase().replace(/[^a-z0-9]/g,""),dn=a=>{let t=new Set;for(let n of a){let e=xt(String(n));e&&t.add(e)}return t},Pe=(a,t)=>{if(Array.isArray(a))return a.map(s=>Pe(s,t));if(!a||typeof a!="object")return a;let n=a,e={};for(let[s,r]of Object.entries(n)){if(t.has(xt(s))){e[s]=cn;continue}e[s]=Pe(r,t)}return e},Ct=(a,t)=>{if(a==null)return"";let n=dn(t);if(typeof a=="string")try{let e=JSON.parse(a);return e&&typeof e=="object"?JSON.stringify(Pe(e,n)):a}catch{return a}return JSON.stringify(Pe(a,n))};function Tt(){return{modelOverrides:new Map,cacheConfig:new Map,rateLimits:new Map,guardrailOverrides:new Map,samplingRate:1,promptVersions:new Map,fallbacks:new Map,retryPolicies:new Map,timeouts:new Map,concurrencyLimits:new Map,abTests:new Map}}function Ie(a,t){if(a.size!==t.size)return!1;for(let[n,e]of a){let s=t.get(n);if(typeof e=="object"&&e!==null&&typeof s=="object"&&s!==null){if(JSON.stringify(e)!==JSON.stringify(s))return!1}else if(s!==e)return!1}return!0}var je=class{localVersion=0;config=Tt();listeners=new Map;fetching=!1;baseUrl;apiKey;projectId;constructor(t){this.baseUrl=t.baseUrl.replace(/\/+$/,""),this.apiKey=t.apiKey,this.projectId=t.projectId}checkVersion(t){t>this.localVersion&&!this.fetching&&(this.fetching=!0,this.fetchConfig().then(n=>{n&&(this.localVersion=t,this.applyFullConfig(n))}).catch(()=>{}).finally(()=>{this.fetching=!1}))}getModel(t,n){return this.config.modelOverrides.get(t)??n}getCacheConfig(t){return this.config.cacheConfig.get(t)??null}getSamplingRate(){return this.config.samplingRate}getRuntimeConfig(){return this.config}getLocalVersion(){return this.localVersion}getFallback(t){return this.config.fallbacks.get(t)??null}getRetryPolicy(t){return this.config.retryPolicies.get(t)??null}getTimeout(t){return this.config.timeouts.get(t)??null}getConcurrencyLimit(t){return this.config.concurrencyLimits.get(t)??null}getAbTest(t){return this.config.abTests.get(t)??null}on(t,n){let e=this.listeners.get(t)??[];e.push(n),this.listeners.set(t,e)}emit(t,n){let e=this.listeners.get(t);if(e)for(let s of e)try{s(n)}catch{}}async fetchConfig(){try{let t=`${this.baseUrl}/v1/sdk/config?projectId=${encodeURIComponent(this.projectId)}`,n=await fetch(t,{headers:{"x-api-key":this.apiKey}});return n.ok?(await n.json()).config??null:null}catch{return null}}applyFullConfig(t){let n=this.config,e=this.parseSnapshot(t);this.config=e,Ie(n.modelOverrides,e.modelOverrides)||this.emit("model_override",Object.fromEntries(e.modelOverrides)),n.samplingRate!==e.samplingRate&&this.emit("sampling_config",{samplingRate:e.samplingRate}),Ie(n.cacheConfig,e.cacheConfig)||this.emit("cache_config",Object.fromEntries(e.cacheConfig)),Ie(n.fallbacks,e.fallbacks)||this.emit("fallback_config",Object.fromEntries(e.fallbacks)),Ie(n.retryPolicies,e.retryPolicies)||this.emit("retry_config",Object.fromEntries(e.retryPolicies)),Ie(n.timeouts,e.timeouts)||this.emit("timeout_config",Object.fromEntries(e.timeouts)),Ie(n.abTests,e.abTests)||this.emit("ab_test_config",Object.fromEntries(e.abTests))}parseSnapshot(t){let n=Tt();for(let[e,s]of Object.entries(t))try{let r=typeof s=="string"?JSON.parse(s):s;if(e.startsWith("model_override.")){let o=e.slice(15),c=typeof r=="string"?r:r?.model??r?.value;typeof c=="string"&&n.modelOverrides.set(o,c)}else if(e.startsWith("cache.")){let o=e.slice(6);n.cacheConfig.set(o,{enabled:r?.enabled??!0,ttlSeconds:typeof r?.ttlSeconds=="number"?r.ttlSeconds:300,similarityThreshold:typeof r?.similarityThreshold=="number"?r.similarityThreshold:.85})}else if(e.startsWith("ratelimit.")){let o=e.slice(10);n.rateLimits.set(o,{maxRequests:typeof r?.maxRequests=="number"?r.maxRequests:100,windowSeconds:typeof r?.windowSeconds=="number"?r.windowSeconds:60})}else if(e.startsWith("guardrail.")&&!e.startsWith("guardrail_mode.")){let o=e.slice(10);n.guardrailOverrides.set(o,!!(r?.enabled??r))}else if(e.startsWith("sampling.")){let o=typeof r=="number"?r:r?.rate??r?.sampleRate;typeof o=="number"&&o>=0&&o<=1&&(n.samplingRate=o)}else if(e.startsWith("prompt_version.")){let o=e.slice(15),c=typeof r=="number"?r:r?.version;typeof c=="number"&&n.promptVersions.set(o,c)}else if(e.startsWith("fallback.")){let o=e.slice(9),c=typeof r?.model=="string"?r.model:null,i=Array.isArray(r?.conditions)?r.conditions.filter(l=>typeof l=="string"):[];c&&n.fallbacks.set(o,{model:c,conditions:i})}else if(e.startsWith("retry.")){let o=e.slice(6),c=typeof r?.enabled=="boolean"?r.enabled:!1,i=typeof r?.maxRetries=="number"?r.maxRetries:3,l=typeof r?.backoffMultiplier=="number"?r.backoffMultiplier:2,u=Array.isArray(r?.retryOnStatusCodes)?r.retryOnStatusCodes.filter(d=>typeof d=="number"):[];n.retryPolicies.set(o,{enabled:c,maxRetries:i,backoffMultiplier:l,retryOnStatusCodes:u})}else if(e.startsWith("timeout.")){let o=e.slice(8),c=typeof r?.timeoutMs=="number"?r.timeoutMs:typeof r=="number"?r:null;typeof c=="number"&&c>0&&n.timeouts.set(o,{timeoutMs:c})}else if(e.startsWith("concurrency.")){let o=e.slice(12),c=typeof r=="number"?r:typeof r?.limit=="number"?r.limit:null;typeof c=="number"&&c>0&&n.concurrencyLimits.set(o,c)}else if(e.startsWith("ab_test.")){let o=e.slice(8),c=typeof r?.candidateModel=="string"?r.candidateModel:null,i=typeof r?.splitPercentage=="number"?r.splitPercentage:null;c&&i!==null&&i>=0&&i<=100&&n.abTests.set(o,{candidateModel:c,splitPercentage:i})}}catch{}return n}};import{AsyncLocalStorage as ln}from"async_hooks";import{randomUUID as pn}from"crypto";var rt=new ln,p=()=>pn(),G=()=>rt.getStore();function W(a,t){return rt.run(a,t)}function at(a,t){let n=G(),e={traceId:a.traceId??n?.traceId??p(),spanId:p(),userId:a.userId??n?.userId,sessionId:a.sessionId??n?.sessionId,tags:{...n?.tags,...a.tags},agentId:a.agentId??n?.agentId,agentName:a.agentName??n?.agentName,agentRole:a.agentRole??n?.agentRole,agentFramework:a.agentFramework??n?.agentFramework,parentAgentId:a.agentId?n?.agentId:n?.parentAgentId,conversationTurn:n?.conversationTurn,availableTools:n?.availableTools};return rt.run(e,t)}function it(a,t){return at(t??{},a)}var un=it;var C="1.1.0";function De(){try{return D()}catch{return null}}var dt=new Set;function le(a){dt.add(a)}function te(a){dt.delete(a)}function Ge(){return dt.size>0}var Fe=!1,Ee=!1,Ot=!1,xe=null,Ce=null,ct=null,Cs=(async()=>{try{xe=await import("openai/resources/chat/completions")}catch{}try{Ce=await import("@anthropic-ai/sdk/resources/messages")}catch{}try{ct=await import("@google/generative-ai")}catch{}Nt()})();function At(a){async function t(n,e){let s=De();if(!s||Ge())return a.call(this,n,e);let r=performance.now(),o=G(),c=p(),i=o?.traceId??p(),l=n?.model??"unknown",u=n?.stream===!0;try{let d=await a.call(this,n,e);if(u){let m=d[Symbol.asyncIterator]?.bind(d);return m&&(d[Symbol.asyncIterator]=function(){let _=m(),y=0,x=0,b="",k;return{async next(){let{value:I,done:h}=await _.next();return h?(s.enqueue({event_id:p(),project_id:"",trace_id:i,span_id:c,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openai",model:l,operation:"chat",latency_ms:performance.now()-r,ttft_ms:k,stream:!0,input_tokens:y,output_tokens:x,finish_reason:b,status:"success",span_kind:"llm",agent_id:o?.agentId,agent_name:o?.agentName,agent_role:o?.agentRole,environment:s.environment,sdk_version:C}),{value:I,done:!0}):(k||(k=performance.now()-r),I?.usage&&(y=I.usage.prompt_tokens??y,x=I.usage.completion_tokens??x),I?.choices?.[0]?.finish_reason&&(b=I.choices[0].finish_reason),{value:I,done:!1})},[Symbol.asyncIterator](){return this}}}),d}return s.enqueue({event_id:p(),project_id:"",trace_id:i,span_id:c,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openai",model:d.model??l,operation:"chat",latency_ms:performance.now()-r,stream:!1,input_tokens:d.usage?.prompt_tokens??0,output_tokens:d.usage?.completion_tokens??0,cached_tokens:d.usage?.prompt_tokens_details?.cached_tokens??0,finish_reason:d.choices?.[0]?.finish_reason??"",status:"success",span_kind:"llm",agent_id:o?.agentId,agent_name:o?.agentName,agent_role:o?.agentRole,...s.storePrompts?{prompt_text:s.redactForStorage(n.messages),completion_text:s.redactForStorage(d.choices?.[0]?.message?.content??"")}:{},environment:s.environment,sdk_version:C}),d}catch(d){throw s.enqueue({event_id:p(),project_id:"",trace_id:i,span_id:c,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openai",model:l,operation:"chat",latency_ms:performance.now()-r,stream:u,input_tokens:0,output_tokens:0,status:"error",error_code:String(d.status??"unknown"),error_message:String(d.message).slice(0,500),span_kind:"llm",environment:s.environment,sdk_version:C}),d}}return t.__zespanAutoPatched=!0,t}function mn(){if(!Fe&&xe)try{let a=xe.Completions;if(!a?.prototype)return;let t=a.prototype.create;if(typeof t=="function"&&!t.__zespanAutoPatched){a.prototype.create=At(t),Fe=!0;return}if(typeof t!="function"){if(a.__zespanAutoPatched){Fe=!0;return}let n=a;class e extends n{constructor(...o){if(super(...o),Object.prototype.hasOwnProperty.call(this,"create")&&!this.create?.__zespanAutoPatched){let c=this.create;this.create=At(c)}}}e.__zespanAutoPatched=!0;let s=Object.getOwnPropertyDescriptor(xe,"Completions");if(s&&(s.writable!==!1||s.configurable))try{Object.defineProperty(xe,"Completions",{value:e,writable:!0,enumerable:!0,configurable:!0}),Fe=!0}catch{}}}catch{}}function Rt(a){async function t(n,e){let s=De();if(!s||Ge())return a.call(this,n,e);let r=performance.now(),o=G(),c=p(),i=o?.traceId??p(),l=n?.model??"unknown",u=n?.stream===!0;try{let d=await a.call(this,n,e);if(u){let m=d[Symbol.asyncIterator]?.bind(d);return m&&(d[Symbol.asyncIterator]=function(){let _=m(),y=0,x=0,b="",k;return{async next(){let{value:I,done:h}=await _.next();return h?(s.enqueue({event_id:p(),project_id:"",trace_id:i,span_id:c,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:l,operation:"messages",latency_ms:performance.now()-r,ttft_ms:k,stream:!0,input_tokens:y,output_tokens:x,finish_reason:b,status:"success",span_kind:"llm",agent_id:o?.agentId,agent_name:o?.agentName,agent_role:o?.agentRole,environment:s.environment,sdk_version:C}),{value:I,done:!0}):(k||(k=performance.now()-r),I?.type==="message_delta"&&I?.usage&&(x=I.usage.output_tokens??x),I?.type==="message_start"&&I?.message?.usage&&(y=I.message.usage.input_tokens??y),I?.type==="message_delta"&&I?.delta?.stop_reason&&(b=I.delta.stop_reason),{value:I,done:!1})},[Symbol.asyncIterator](){return this}}}),d}return s.enqueue({event_id:p(),project_id:"",trace_id:i,span_id:c,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:d.model??l,operation:"messages",latency_ms:performance.now()-r,stream:!1,input_tokens:d.usage?.input_tokens??0,output_tokens:d.usage?.output_tokens??0,finish_reason:d.stop_reason??"",status:"success",span_kind:"llm",agent_id:o?.agentId,agent_name:o?.agentName,agent_role:o?.agentRole,...s.storePrompts?{prompt_text:s.redactForStorage(n.messages),completion_text:s.redactForStorage(d.content?.filter(m=>m.type==="text").map(m=>m.text).join("")??"")}:{},environment:s.environment,sdk_version:C}),d}catch(d){throw s.enqueue({event_id:p(),project_id:"",trace_id:i,span_id:c,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:l,operation:"messages",latency_ms:performance.now()-r,stream:u,input_tokens:0,output_tokens:0,status:"error",error_code:String(d.status??"unknown"),error_message:String(d.message).slice(0,500),span_kind:"llm",environment:s.environment,sdk_version:C}),d}}return t.__zespanAutoPatched=!0,t}function gn(){if(!Ee&&Ce)try{let a=Ce.Messages;if(!a?.prototype)return;let t=a.prototype.create;if(typeof t=="function"&&!t.__zespanAutoPatched){a.prototype.create=Rt(t),Ee=!0;return}if(typeof t!="function"){if(a.__zespanAutoPatched){Ee=!0;return}let n=a;class e extends n{constructor(...o){if(super(...o),Object.prototype.hasOwnProperty.call(this,"create")&&!this.create?.__zespanAutoPatched){let c=this.create;this.create=Rt(c)}}}e.__zespanAutoPatched=!0;let s=Object.getOwnPropertyDescriptor(Ce,"Messages");if(s&&(s.writable!==!1||s.configurable))try{Object.defineProperty(Ce,"Messages",{value:e,writable:!0,enumerable:!0,configurable:!0}),Ee=!0}catch{}}}catch{}}function _n(){if(!Ot&&ct)try{let e=function(s,r){let o=n.call(this,s,r),c=o.generateContent?.bind(o),i=o.generateContentStream?.bind(o);return!c||c.__zespanAutoPatched||(o.generateContent=async l=>{let u=De();if(!u||Ge())return c(l);let d=performance.now(),m=G(),_=p(),y=m?.traceId??p(),x=s.model??"gemini-1.5-flash";try{let b=await c(l),k=b.response?.usageMetadata;return u.enqueue({event_id:p(),project_id:"",trace_id:y,span_id:_,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:x,operation:"generateContent",latency_ms:performance.now()-d,stream:!1,input_tokens:k?.promptTokenCount??0,output_tokens:k?.candidatesTokenCount??0,finish_reason:b.response?.candidates?.[0]?.finishReason??"",status:"success",span_kind:"llm",environment:u.environment,sdk_version:C}),b}catch(b){throw u.enqueue({event_id:p(),project_id:"",trace_id:y,span_id:_,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:x,operation:"generateContent",latency_ms:performance.now()-d,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_message:String(b.message).slice(0,500),span_kind:"llm",environment:u.environment,sdk_version:C}),b}},o.generateContent.__zespanAutoPatched=!0,i&&(o.generateContentStream=async l=>{let u=De();if(!u||Ge())return i(l);let d=performance.now(),m=G(),_=p(),y=m?.traceId??p(),x=s.model??"gemini-1.5-flash",b=await i(l),k=b.stream,I=0,h=0,T;return b.stream=(async function*(){try{for await(let g of k){T||(T=performance.now()-d);let A=g.usageMetadata;A&&(I=A.promptTokenCount??I,h=A.candidatesTokenCount??h),yield g}u.enqueue({event_id:p(),project_id:"",trace_id:y,span_id:_,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:x,operation:"generateContent",latency_ms:performance.now()-d,ttft_ms:T,stream:!0,input_tokens:I,output_tokens:h,status:"success",span_kind:"llm",environment:u.environment,sdk_version:C})}catch(g){throw u.enqueue({event_id:p(),project_id:"",trace_id:y,span_id:_,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:x,operation:"generateContent",latency_ms:performance.now()-d,ttft_ms:T,stream:!0,input_tokens:I,output_tokens:h,status:"error",error_message:String(g.message).slice(0,500),span_kind:"llm",environment:u.environment,sdk_version:C}),g}})(),b})),o};var a=e;let t=ct?.GoogleGenerativeAI;if(!t?.prototype?.getGenerativeModel||t.prototype.getGenerativeModel.__zespanAutoPatched)return;let n=t.prototype.getGenerativeModel;e.__zespanAutoPatched=!0,t.prototype.getGenerativeModel=e,Ot=!0}catch{}}function Nt(){mn(),gn(),_n()}function qe(){Nt()}var Ke=class{apiKey;endpoint;baseURL;environment;storePrompts;redactKeys;sampleRate;debug;enableOTel;prompts;configClient;projectId;batch;_storePromptsDefault;_promptWarningShown=!1;constructor(t){if(!t.apiKey)throw new Error("Zespan: apiKey is required");t.apiKey.startsWith("zsp_")||console.warn("[zespan] API key should start with zsp_ \u2014 check your key"),this.apiKey=t.apiKey,this.baseURL=t.baseURL??"https://api.zespan.com",this.endpoint=t.endpoint??`${this.baseURL.replace(/\/+$/,"")}/v1/ingest`,this.environment=t.environment??"production",this._storePromptsDefault=t.storePrompts===void 0,this.storePrompts=t.storePrompts??!0,this.redactKeys=t.redactKeys??["password","secret","token","api_key"],this.sampleRate=t.sampleRate??1,this.debug=t.debug??!1,this.enableOTel=t.enableOTel??!1,this.projectId=t.projectId,this.prompts=new ke(this),t.enableZespanPilot!==!1&&t.projectId?this.configClient=new je({baseUrl:this.baseURL,apiKey:this.apiKey,projectId:t.projectId}):this.configClient=null,this.batch=new Re(this,{maxSize:t.batchSize??50,flushInterval:t.flushInterval??2e3},this.configClient??void 0),process.once("beforeExit",()=>this.batch.flush()),this.enableOTel&&Me({endpoint:t.otelEndpoint,serviceName:t.otelServiceName,sampleRate:t.sampleRate,apiKey:this.apiKey,environment:this.environment}),t.autopatch!==!1&&qe()}enqueue(t){Math.random()>this.sampleRate||(this.storePrompts&&this._storePromptsDefault&&!this._promptWarningShown&&(this._promptWarningShown=!0,console.warn("[zespan] Prompt and completion text is being captured by default. PII (emails, phone numbers, SSNs, credit cards) is automatically redacted server-side. To opt out: { storePrompts: false }")),this.batch.push(t))}redactForStorage(t){return Ct(t,this.redactKeys)}resolveGuardrailsUrl(){let t=this.baseURL.replace(/\/+$/,"");return t.endsWith("/v1")?`${t}/guardrails/check`:`${t}/v1/guardrails/check`}async checkGuardrails(t){let n=await fetch(this.resolveGuardrailsUrl(),{method:"POST",headers:{"content-type":"application/json","x-api-key":this.apiKey},body:JSON.stringify(t)});if(!n.ok){let s="";try{s=await n.text()}catch{s=""}throw new Error(`Guardrails check failed (${n.status})${s?`: ${s.slice(0,400)}`:""}`)}let e=await n.json();return{allowed:!!e.allowed,results:Array.isArray(e.results)?e.results:[],modifiedText:typeof e.modifiedText=="string"?e.modifiedText:null}}flush(){return this.batch.flush()}createSpan(t){return fe().startSpan(t.name,{kind:t.kind??ye.INTERNAL,attributes:t.attributes??{}})}async withSpan(t,n,e){let r=fe().startSpan(t,{kind:e?.kind??ye.INTERNAL,attributes:e?.attributes??{}});try{let o=await n(r);return r.setStatus({code:he.OK}),o}catch(o){throw r.setStatus({code:he.ERROR,message:o.message}),r.recordException(o),o}finally{r.end()}}},lt=null,pt=a=>{lt=new Ke(a)},D=()=>{if(!lt)throw new Error("Call zespan.init() first");return lt},fn=pt;var V=class extends Error{phase;results;constructor(t,n){super(`Guardrail blocked ${t} content`),this.name="GuardrailBlockedError",this.phase=t,this.results=n}};function X(a,t){if(!a)return null;if(a===!0)return{pre:!0,post:!0,failClosed:!1,operation:t};let n=typeof a.pre=="boolean"||typeof a.post=="boolean";return{pre:n?!!a.pre:!0,post:n?!!a.post:!0,failClosed:!!a.failClosed,operation:a.operation??t}}function ne(a,t){return t?.modifiedText?t.modifiedText:a}async function Z(a,t,n){if(!t||n.phase==="pre"&&!t.pre||n.phase==="post"&&!t.post)return null;let e=n.text?.trim();if(!e)return null;try{let s=await a.checkGuardrails({...n,text:e.slice(0,5e4),operation:n.operation??t.operation});if(!s.allowed)throw new V(n.phase,s.results);return s}catch(s){if(s instanceof V||t.failClosed)throw s;return null}}function Y(a){if(!Array.isArray(a))return{systemPrompt:void 0,conversationMessages:[]};let t=[],n=[];for(let e of a)if((typeof e?.role=="string"?e.role:e?._getType?.()??"")==="system"){let r=typeof e.content=="string"?e.content:Array.isArray(e.content)?e.content.map(o=>typeof o=="string"?o:o?.text??"").filter(Boolean).join(""):JSON.stringify(e.content??"");r&&t.push(r)}else n.push(e);return{systemPrompt:t.length?t.join(`
|
|
3
|
+
`):void 0,conversationMessages:n}}function hn(a){let t={};try{let n=[],e=[];for(let s of a){let r=typeof s?.content=="string"?s.content:Array.isArray(s?.content)?s.content.map(o=>typeof o=="string"?o:o?.text??"").join(" "):"";s?.role==="system"&&n.push(r),e.push(r)}for(let s of n){try{let r=s.match(/You are the (\w+)Agent/i);if(r){t.agent_name=r[1];break}}catch{}try{let r=s.match(/Agent Name:\s*(\w+)/i);if(r){t.agent_name=r[1];break}}catch{}}for(let s of n)try{let r=s.match(/Role:\s*(orchestrator|specialist|planner|executor|critic)/i);if(r?.[1]){t.agent_role=r[1].toLowerCase();break}}catch{}try{a.length>10&&(t.has_memory=!0)}catch{}try{let s=e.join(" ").toLowerCase();(s.includes("think step by step")||s.includes("let's reason")||s.includes("step 1"))&&(t.has_planning=!0)}catch{}}catch{}return t}function re(a,t){let n=hn(a);try{!n.has_planning&&t&&/^o[13]/.test(t)&&(n.has_planning=!0)}catch{}return n}function ae(a){try{let t=a.filter(s=>s?.role==="user").length,n=a[a.length-1],e=typeof n?.role=="string"?n.role:"user";return{conversation_turn:t,message_role:e}}catch{return{conversation_turn:0,message_role:"user"}}}function ie(a){try{return a.map(t=>{try{return t?.type==="function"&&t?.function?{name:String(t.function.name??""),description:t.function.description?String(t.function.description):void 0,parameters_schema:t.function.parameters??void 0,source:"function"}:t?.name?{name:String(t.name),description:t.description?String(t.description):void 0,parameters_schema:t.input_schema??t.parameters??void 0,source:"function"}:null}catch{return null}}).filter(t=>t!==null)}catch{return[]}}function Le(a){try{let t=a?.[0]?.message?.tool_calls??[],n=[],e={};for(let s of t)try{let r=s?.function?.name;if(typeof r=="string"&&r){n.includes(r)||n.push(r);try{e[r]=typeof s.function.arguments=="string"?JSON.parse(s.function.arguments):s.function.arguments??{}}catch{e[r]=s.function?.arguments??{}}}}catch{}return{tools_used:n,tool_call_args:e}}catch{return{tools_used:[],tool_call_args:{}}}}function me(a){try{let t={};for(let s of a)try{let r=typeof s?.index=="number"?s.index:0;t[r]||(t[r]={name:"",argumentsStr:""}),s?.function?.name&&(t[r].name+=s.function.name),s?.function?.arguments&&(t[r].argumentsStr+=s.function.arguments)}catch{}let n=[],e={};for(let s of Object.values(t))try{let r=s.name;if(r&&!n.includes(r)){n.push(r);try{e[r]=s.argumentsStr?JSON.parse(s.argumentsStr):{}}catch{e[r]=s.argumentsStr}}}catch{}return{tools_used:n,tool_call_args:e}}catch{return{tools_used:[],tool_call_args:{}}}}import{createHash as kn}from"crypto";import{SpanKind as In,SpanStatusCode as we,trace as Sn}from"@opentelemetry/api";async function Se(a,t){if(!t||!t.enabled)return a();let n;for(let e=0;e<=t.maxRetries;e++)try{return await a()}catch(s){if(n=s,t.retryOnStatusCodes.length>0&&!t.retryOnStatusCodes.includes(s?.status))throw s;if(e>=t.maxRetries)break;let r=Math.min(1e3*Math.pow(t.backoffMultiplier,e),3e4);await yn(r)}throw n}function yn(a){return new Promise(t=>setTimeout(t,a))}var ut=class{constructor(t){this.max=t;this.counter=t}counter;queue=[];acquire(){return this.counter>0?(this.counter--,Promise.resolve()):new Promise(t=>{this.queue.push(t)})}release(){this.queue.length>0?this.queue.shift()():this.counter++}async run(t){await this.acquire();try{return await t()}finally{this.release()}}},Mt=new Map;function ve(a,t){let n=Mt.get(a);return n||(n=new ut(t),Mt.set(a,n)),n}var ze=a=>kn("sha256").update(a).digest("hex");function vn(a,t,n="openai"){return{"gen_ai.system":n,"gen_ai.model":t,"gen_ai.operation.name":"chat","gen_ai.prompt":JSON.stringify(a.messages),"gen_ai.request.max_tokens":a.max_tokens,"gen_ai.request.temperature":a.temperature,"gen_ai.request.top_p":a.top_p}}async function*wn(a,t){let{client:n,t0:e,traceId:s,spanId:r,parent:o,params:c,otelSpan:i,model:l,agentCtx:u,autoDetected:d,conversation_turn:m,message_role:_,tool_definitions:y,provider:x,operation:b}=t,k=x??"openai",I,h=0,T=0,g=0,A="",f=l,v=[],w="",j=[],{systemPrompt:O,conversationMessages:S}=Y(c.messages??[]);try{for await(let F of a){I||(I=performance.now()-e,n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:k,model:f,operation:b,latency_ms:I,ttft_ms:I,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:ze(JSON.stringify(c.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(S),...O?{system_prompt:n.redactForStorage(O)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C,finish_reason:"stream_start",...u,span_kind:"llm",tool_definitions:y,conversation_turn:m,message_role:_,metadata:{auto_detected:Object.keys(d).length?d:void 0}}),i&&i.setAttribute("gen_ai.response.time_to_first_token",I));let P=F.choices?.[0]?.delta?.content;if(P&&(T++,w+=P),F.choices?.[0]?.finish_reason&&(A=F.choices[0].finish_reason),F.model&&(f=F.model),F.usage&&(h=Number(F.usage.prompt_tokens||h||0),T=Number(F.usage.completion_tokens||T||0),g=Number(F.usage.prompt_tokens_details?.cached_tokens||g||0)),F.choices?.[0]?.delta?.tool_calls)for(let B of F.choices[0].delta.tool_calls)j.push(B),B.function?.name&&!v.includes(B.function.name)&&v.push(B.function.name);yield F}let{tools_used:M,tool_call_args:N}=me(j),q=M.length>0?M:v,R=performance.now()-e;n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:k,model:f,operation:b,latency_ms:R,ttft_ms:I,stream:!0,input_tokens:h,output_tokens:T,cached_tokens:g,status:"success",finish_reason:A||"stream_end",tools_used:q,tool_call_args:Object.keys(N).length?N:void 0,prompt_hash:ze(JSON.stringify(c.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(S),completion_text:n.redactForStorage(w),...O?{system_prompt:n.redactForStorage(O)}:{}}:{},...q.length>0&&w?{decision_rationale:w}:{},...u,span_kind:"llm",tool_definitions:y,conversation_turn:m,message_role:_,metadata:{usage:{prompt_tokens:h,completion_tokens:T,cached_tokens:g},auto_detected:Object.keys(d).length?d:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C}),i&&(i.setAttribute("gen_ai.usage.output_tokens",T),i.setAttribute("gen_ai.usage.total_tokens",T),i.setAttribute("gen_ai.response.finish_reason",A||"stream_end"),i.setAttribute("gen_ai.response.latency",R))}catch(M){throw n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:k,model:f,operation:b,latency_ms:performance.now()-e,ttft_ms:I,stream:!0,input_tokens:0,output_tokens:T,cost_usd:0,status:"error",error_code:String(M.status??"unknown"),error_message:String(M.message).slice(0,500),...u,span_kind:"llm",conversation_turn:m,message_role:_,metadata:{auto_detected:Object.keys(d).length?d:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C}),i&&(i.setStatus({code:we.ERROR,message:M.message}),i.recordException(M)),M}}function Be(a,t={}){let n=a,e=n.chat.completions.create.bind(n.chat.completions),s=X(t.guardrails,"chat"),r=t.providerOverride??"openai",o=t.operation??"chat";return n.chat.completions.create=async(c,i)=>{let l=D(),u=performance.now(),d=G(),m=p(),_=d?.traceId??p(),y=c.model??"gpt-4",x;if(l.configClient){let S=l.configClient.getAbTest(o);if(S)Math.random()<S.splitPercentage/100?(y=S.candidateModel,x={ab_test:!0,ab_variant:"candidate"}):x={ab_test:!0,ab_variant:"control"},c={...c,model:y};else{let M=l.configClient.getModel(o,y);if(M!==y){let N=y;y=M,c={...c,model:M},l.enqueue({event_id:p(),project_id:"",trace_id:_,span_id:m,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:r,model:M,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:N,override_model:M},environment:l.environment,sdk_version:C})}}}let b=null,k=null,I=d?.agentId?{agent_id:d.agentId,agent_name:d.agentName,agent_role:d.agentRole,agent_framework:d.agentFramework,parent_agent_id:d.parentAgentId}:{},h=re(c.messages??[],c.model),{conversation_turn:T,message_role:g}=ae(c.messages??[]),A=c.tools?.length?ie(c.tools):void 0,f=null,v=l.enableOTel?Sn.getTracer("zespan-sdk"):null;v&&(f=v.startSpan(`chat.completions ${y}`,{kind:In.CLIENT,attributes:vn(c,y,r)}));let w;if(l.configClient){let S=l.configClient.getTimeout(o);S&&(w=AbortSignal.timeout(S.timeoutMs))}let j=l.configClient?.getRetryPolicy(o)??null,O=l.configClient?.getConcurrencyLimit(o)??null;try{b=await Z(l,s,{text:JSON.stringify(c.messages??[]),phase:"pre",traceId:_,spanId:m,model:y,operation:o,agentName:d?.agentName});let S=()=>Se(()=>W({traceId:_,spanId:m,userId:d?.userId,sessionId:d?.sessionId,tags:d?.tags},()=>e(c,w?{...i,signal:w}:i)),j),M=await(O!==null?ve(`${l.projectId??"default"}:${o}`,O).run(S):S());if(c.stream===!0)return wn(M,{client:l,t0:u,traceId:_,spanId:m,parent:d,params:c,otelSpan:f,model:y,agentCtx:I,autoDetected:h,conversation_turn:T,message_role:g,tool_definitions:A,provider:r,operation:o});let{prompt_tokens:q,completion_tokens:R,prompt_tokens_details:F}=M.usage??{},P=F?.cached_tokens??0,B=M.model??c.model,L=M.usage?.completion_tokens_details?.reasoning_tokens,z=M.choices?.[0]?.message?.content||"";k=await Z(l,s,{text:z,phase:"post",traceId:_,spanId:m,model:B,operation:o,agentName:d?.agentName});let Q=ne(z,k);Q!==z&&typeof M?.choices?.[0]?.message?.content=="string"&&(M.choices[0].message.content=Q);let{tools_used:E,tool_call_args:H}=Le(M.choices??[]),{systemPrompt:K,conversationMessages:J}=Y(c.messages??[]);return l.enqueue({event_id:p(),project_id:"",trace_id:_,span_id:m,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:r,model:B,operation:o,latency_ms:performance.now()-u,stream:!1,input_tokens:q??0,output_tokens:R??0,cached_tokens:P,...L?{reasoning_tokens:L}:{},status:"success",finish_reason:M.choices?.[0]?.finish_reason??"",tools_used:E.length>0?E:M.choices?.[0]?.message?.tool_calls?.map(ee=>ee.function.name)??[],tool_call_args:Object.keys(H).length?H:void 0,prompt_hash:ze(JSON.stringify(c.messages)),...l.storePrompts?{prompt_text:l.redactForStorage(J),completion_text:l.redactForStorage(Q),...K?{system_prompt:l.redactForStorage(K)}:{}}:{},...E.length>0&&z?{decision_rationale:z}:{},...I,span_kind:"llm",tool_definitions:A,conversation_turn:T,message_role:g,metadata:{usage:M.usage??null,guardrails:{pre:b?{matched:b.results.length,modified:!!b.modifiedText}:null,post:k?{matched:k.results.length,modified:!!k.modifiedText}:null},auto_detected:Object.keys(h).length?h:void 0,...x??{}},user_id:d?.userId,session_id:d?.sessionId,tags:d?.tags,environment:l.environment,sdk_version:C}),f&&(f.setAttribute("gen_ai.usage.input_tokens",q??0),f.setAttribute("gen_ai.usage.output_tokens",R??0),f.setAttribute("gen_ai.usage.total_tokens",(q??0)+(R??0)),f.setAttribute("gen_ai.usage.cached_tokens",P),f.setAttribute("gen_ai.response.finish_reason",M.choices?.[0]?.finish_reason??""),f.setAttribute("gen_ai.response.latency",performance.now()-u),f.setStatus({code:we.OK}),f.end()),M}catch(S){if(S?.name==="TimeoutError"||S?.code==="ABORT_ERR"||w&&w.aborted)throw l.enqueue({event_id:p(),project_id:"",trace_id:_,span_id:m,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:r,model:c.model,operation:o,latency_ms:performance.now()-u,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:"timeout",error_message:String(S.message).slice(0,500),...I,span_kind:"llm",conversation_turn:T,message_role:g,metadata:{auto_detected:Object.keys(h).length?h:void 0},user_id:d?.userId,session_id:d?.sessionId,tags:d?.tags,environment:l.environment,sdk_version:C}),f&&(f.setStatus({code:we.ERROR,message:S.message}),f.recordException(S),f.end()),S;let N=l.configClient?.getFallback(o)??null;if(N&&N.model){let F=String(S.message??"")+String(S.status??"")+String(S.code??"");if(N.conditions.length===0||N.conditions.some(B=>F.toLowerCase().includes(B.toLowerCase()))){let B=y,L={...c,model:N.model};l.enqueue({event_id:p(),project_id:"",trace_id:_,span_id:m,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:r,model:N.model,operation:o,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:B,fallback_model:N.model,reason:String(S.message).slice(0,200)},environment:l.environment,sdk_version:C});try{let z=await W({traceId:_,spanId:m,userId:d?.userId,sessionId:d?.sessionId,tags:d?.tags},()=>e(L,i)),{prompt_tokens:Q,completion_tokens:E,prompt_tokens_details:H}=z.usage??{},K=H?.cached_tokens??0,J=z.model??N.model;return l.enqueue({event_id:p(),project_id:"",trace_id:_,span_id:m,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:r,model:J,operation:o,latency_ms:performance.now()-u,stream:!1,input_tokens:Q??0,output_tokens:E??0,cached_tokens:K,status:"success",finish_reason:z.choices?.[0]?.finish_reason??"",prompt_hash:ze(JSON.stringify(c.messages)),...I,span_kind:"llm",conversation_turn:T,message_role:g,metadata:{usage:z.usage??null,fallback_used:!0,original_model:B,auto_detected:Object.keys(h).length?h:void 0},user_id:d?.userId,session_id:d?.sessionId,tags:d?.tags,environment:l.environment,sdk_version:C}),f&&(f.setStatus({code:we.OK}),f.end()),z}catch(z){throw f&&(f.setStatus({code:we.ERROR,message:z.message}),f.recordException(z),f.end()),z}}}let q=S instanceof V,R=q?"error":S.status===429?"rate_limited":S.status===408?"timeout":"error";throw l.enqueue({event_id:p(),project_id:"",trace_id:_,span_id:m,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:r,model:c.model,operation:o,latency_ms:performance.now()-u,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:R,error_code:q?"guardrail_blocked":String(S.status??"unknown"),error_message:String(S.message).slice(0,500),...I,span_kind:"llm",conversation_turn:T,message_role:g,metadata:{auto_detected:Object.keys(h).length?h:void 0},user_id:d?.userId,session_id:d?.sessionId,tags:d?.tags,environment:l.environment,sdk_version:C}),f&&(f.setStatus({code:we.ERROR,message:S.message}),f.recordException(S),f.end()),S}},a}import{createHash as bn}from"crypto";import{SpanKind as xn,SpanStatusCode as be,trace as Cn}from"@opentelemetry/api";var $e=a=>bn("sha256").update(a).digest("hex");function Tn(a){let t=Array.isArray(a?.messages)?[...a.messages]:[],n=a?.system;if(typeof n=="string"&&n.trim().length>0)return[{role:"system",content:n},...t];if(Array.isArray(n)&&n.length>0){let e=n.map(s=>typeof s=="string"?s:typeof s?.text=="string"?s.text:"").filter(s=>s.length>0).join(`
|
|
4
|
+
`);if(e.length>0)return[{role:"system",content:e},...t]}return t}function On(a,t){return{"gen_ai.system":"anthropic","gen_ai.model":t,"gen_ai.operation.name":"chat","gen_ai.prompt":JSON.stringify(a.messages),"gen_ai.request.max_tokens":a.max_tokens,"gen_ai.request.temperature":a.temperature}}async function*An(a,t){let{client:n,t0:e,traceId:s,spanId:r,parent:o,params:c,otelSpan:i,agentCtx:l,tool_definitions:u,autoDetected:d,conversation_turn:m,message_role:_,operation:y,detectionMessages:x}=t,b,k=0,I=0,h=0,T=0,g="",A=c.model,f=[],v="",{systemPrompt:w,conversationMessages:j}=Y(x);try{for await(let S of a){if(!b&&S.type==="content_block_start"&&(b=performance.now()-e,n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:A,operation:y,latency_ms:b,ttft_ms:b,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:$e(JSON.stringify(x)),...n.storePrompts?{prompt_text:n.redactForStorage(j),...w?{system_prompt:n.redactForStorage(w)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C,finish_reason:"stream_start",...l,span_kind:"llm",tool_definitions:u,conversation_turn:m,message_role:_,metadata:{auto_detected:Object.keys(d).length?d:void 0}}),i&&i.setAttribute("gen_ai.response.time_to_first_token",b)),S.type==="content_block_delta"&&S.delta?.type==="text_delta"&&(T++,v+=S.delta.text||""),S.type==="message_delta"&&S.delta?.stop_reason&&(g=S.delta.stop_reason),S.type==="message_delta"&&S.usage&&(k=Number(S.usage.input_tokens||k||0),T=Number(S.usage.output_tokens||T||0),I=Number(S.usage.cache_read_input_tokens||I||0)),S.type==="message_start"&&S.message?.usage&&(k=Number(S.message.usage.input_tokens||k||0),T=Number(S.message.usage.output_tokens||T||0),I=Number(S.message.usage.cache_read_input_tokens||I||0)),S.type==="content_block_start"&&S.content_block?.type==="tool_use"){let M=S.content_block.name;M&&!f.includes(M)&&f.push(M)}yield S}let O=performance.now()-e;n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:A,operation:y,latency_ms:O,ttft_ms:b,stream:!0,input_tokens:k,output_tokens:T,cached_tokens:I,reasoning_tokens:h,status:"success",finish_reason:g||"stream_end",tools_used:f,prompt_hash:$e(JSON.stringify(x)),...n.storePrompts?{prompt_text:n.redactForStorage(j),completion_text:n.redactForStorage(v),...w?{system_prompt:n.redactForStorage(w)}:{}}:{},...l,span_kind:"llm",tool_definitions:u,conversation_turn:m,message_role:_,metadata:{usage:{input_tokens:k,output_tokens:T,cache_read_input_tokens:I,reasoning_tokens:h},auto_detected:Object.keys(d).length?d:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C}),i&&(i.setAttribute("gen_ai.usage.output_tokens",T),i.setAttribute("gen_ai.response.finish_reason",g||"stream_end"))}catch(O){throw n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:A,operation:y,latency_ms:performance.now()-e,ttft_ms:b,stream:!0,input_tokens:0,output_tokens:T,cost_usd:0,status:"error",error_code:String(O.status??O.type??"unknown"),error_message:String(O.message).slice(0,500),...l,span_kind:"llm",conversation_turn:m,message_role:_,metadata:{auto_detected:Object.keys(d).length?d:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C}),i&&(i.setStatus({code:be.ERROR,message:O.message}),i.recordException(O)),O}}function Pt(a,t={}){let n=a,e=n.messages.create.bind(n.messages),s=X(t.guardrails,"chat"),r=t.operation??"chat";return n.messages.create=async(o,c)=>{let i=D(),l=performance.now(),u=G(),d=p(),m=u?.traceId??p(),_=o.model??"claude-3-5-sonnet-20241022",y;if(i.configClient){let R=i.configClient.getAbTest(r);if(R)Math.random()<R.splitPercentage/100?(_=R.candidateModel,y={ab_test:!0,ab_variant:"candidate"}):y={ab_test:!0,ab_variant:"control"},o={...o,model:_};else{let F=i.configClient.getModel(r,_);if(F!==_){let P=_;_=F,o={...o,model:F},i.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:d,parent_span_id:u?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:F,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:P,override_model:F},environment:i.environment,sdk_version:C})}}}let x=null,b=null,k=Tn(o),{systemPrompt:I,conversationMessages:h}=Y(k),T=re(k,_),{conversation_turn:g,message_role:A}=ae(k),f=u?.agentName??T.agent_name,v=u?.agentRole??T.agent_role,w=u?.agentId||f||v?{agent_id:u?.agentId,agent_name:f,agent_role:v,agent_framework:u?.agentFramework,parent_agent_id:u?.parentAgentId}:{},j=o.tools?.length?ie(o.tools):void 0,O=null,S=i.enableOTel?Cn.getTracer("zespan-sdk"):null;S&&(O=S.startSpan(`messages.create ${_}`,{kind:xn.CLIENT,attributes:On(o,_)}));let M;if(i.configClient){let R=i.configClient.getTimeout(r);R&&(M=AbortSignal.timeout(R.timeoutMs))}let N=i.configClient?.getRetryPolicy(r)??null,q=i.configClient?.getConcurrencyLimit(r)??null;try{x=await Z(i,s,{text:JSON.stringify(o.messages??[]),phase:"pre",traceId:m,spanId:d,model:_,operation:r,agentName:u?.agentName});let R=()=>Se(()=>W({traceId:m,spanId:d,userId:u?.userId,sessionId:u?.sessionId,tags:u?.tags,agentId:u?.agentId,agentName:f,agentRole:v,agentFramework:u?.agentFramework,parentAgentId:u?.parentAgentId,conversationTurn:g,availableTools:j},()=>e(o,M?{...c,signal:M}:c)),N),F=await(q!==null?ve(`${i.projectId??"default"}:${r}`,q).run(R):R());if(o.stream===!0)return An(F,{client:i,t0:l,traceId:m,spanId:d,parent:u,params:o,otelSpan:O,agentCtx:w,tool_definitions:j,autoDetected:T,conversation_turn:g,message_role:A,operation:r,detectionMessages:k});let B=F.usage?.input_tokens??0,L=F.usage?.output_tokens??0,z=F.usage?.cache_read_input_tokens??0,Q=F.model??o.model,E=F.content?.find($=>$.type==="text")?.text||"",H=(F.content??[]).filter($=>$.type==="tool_use"),K=H.map($=>$.name).filter(Boolean),J={};for(let $ of H)$.name&&(J[$.name]=$.input??{});b=await Z(i,s,{text:E,phase:"post",traceId:m,spanId:d,model:Q,operation:r,agentName:u?.agentName});let ee=ne(E,b);if(ee!==E){let $=F.content?.find(oe=>oe.type==="text");$&&typeof $.text=="string"&&($.text=ee)}return i.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:d,parent_span_id:u?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:Q,operation:r,latency_ms:performance.now()-l,stream:!1,input_tokens:B,output_tokens:L,cached_tokens:z,status:"success",finish_reason:F.stop_reason??"",tools_used:K.length>0?K:F.content?.filter($=>$.type==="tool_use").map($=>$.name)??[],tool_call_args:Object.keys(J).length?J:void 0,prompt_hash:$e(JSON.stringify(k)),...i.storePrompts?{prompt_text:i.redactForStorage(h),completion_text:i.redactForStorage(ee),...I?{system_prompt:i.redactForStorage(I)}:{}}:{},...H.length>0&&E?{decision_rationale:E}:{},...w,span_kind:"llm",tool_definitions:j,conversation_turn:g,message_role:A,metadata:{usage:F.usage??null,guardrails:{pre:x?{matched:x.results.length,modified:!!x.modifiedText}:null,post:b?{matched:b.results.length,modified:!!b.modifiedText}:null},auto_detected:Object.keys(T).length?T:void 0,...y??{}},user_id:u?.userId,session_id:u?.sessionId,tags:u?.tags,environment:i.environment,sdk_version:C}),O&&(O.setAttribute("gen_ai.usage.input_tokens",B),O.setAttribute("gen_ai.usage.output_tokens",L),O.setAttribute("gen_ai.usage.cached_tokens",z),O.setAttribute("gen_ai.response.finish_reason",F.stop_reason??""),O.setAttribute("gen_ai.response.latency",performance.now()-l),O.setStatus({code:be.OK}),O.end()),F}catch(R){if(R?.name==="TimeoutError"||R?.code==="ABORT_ERR"||M&&M.aborted)throw i.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:d,parent_span_id:u?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:o.model,operation:r,latency_ms:performance.now()-l,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:"timeout",error_message:String(R.message).slice(0,500),...w,span_kind:"llm",conversation_turn:g,message_role:A,metadata:{auto_detected:Object.keys(T).length?T:void 0},user_id:u?.userId,session_id:u?.sessionId,tags:u?.tags,environment:i.environment,sdk_version:C}),O&&(O.setStatus({code:be.ERROR,message:R.message}),O.recordException(R),O.end()),R;let P=i.configClient?.getFallback(r)??null;if(P&&P.model){let z=String(R.message??"")+String(R.status??"")+String(R.code??"");if(P.conditions.length===0||P.conditions.some(E=>z.toLowerCase().includes(E.toLowerCase()))){let E=_,H={...o,model:P.model};i.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:d,parent_span_id:u?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:P.model,operation:r,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:E,fallback_model:P.model,reason:String(R.message).slice(0,200)},environment:i.environment,sdk_version:C});try{let K=await W({traceId:m,spanId:d,userId:u?.userId,sessionId:u?.sessionId,tags:u?.tags},()=>e(H,c)),J=K.usage?.input_tokens??0,ee=K.usage?.output_tokens??0,$=K.usage?.cache_read_input_tokens??0,oe=K.model??P.model;return i.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:d,parent_span_id:u?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:oe,operation:r,latency_ms:performance.now()-l,stream:!1,input_tokens:J,output_tokens:ee,cached_tokens:$,status:"success",finish_reason:K.stop_reason??"",prompt_hash:$e(JSON.stringify(k)),...w,span_kind:"llm",conversation_turn:g,message_role:A,metadata:{usage:K.usage??null,fallback_used:!0,original_model:E,auto_detected:Object.keys(T).length?T:void 0},user_id:u?.userId,session_id:u?.sessionId,tags:u?.tags,environment:i.environment,sdk_version:C}),O&&(O.setStatus({code:be.OK}),O.end()),K}catch(K){throw O&&(O.setStatus({code:be.ERROR,message:K.message}),O.recordException(K),O.end()),K}}}let B=R instanceof V,L=B?"error":R.status===429||R.type==="overloaded_error"?"rate_limited":R.status===408?"timeout":"error";throw i.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:d,parent_span_id:u?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:o.model,operation:r,latency_ms:performance.now()-l,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:L,error_code:B?"guardrail_blocked":String(R.status??R.type??"unknown"),error_message:String(R.message).slice(0,500),...w,span_kind:"llm",conversation_turn:g,message_role:A,metadata:{auto_detected:Object.keys(T).length?T:void 0},user_id:u?.userId,session_id:u?.sessionId,tags:u?.tags,environment:i.environment,sdk_version:C}),O&&(O.setStatus({code:be.ERROR,message:R.message}),O.recordException(R),O.end()),R}},a}import{createHash as Rn}from"crypto";import{SpanKind as We,SpanStatusCode as ce,trace as Je}from"@opentelemetry/api";var Te=a=>Rn("sha256").update(a).digest("hex");function Ze(a){let t=o=>{if(typeof o=="string")return o;if(Array.isArray(o))return o.map(c=>typeof c=="string"?c:typeof c?.text=="string"?c.text:"").filter(Boolean).join(`
|
|
5
|
+
`);if(o&&typeof o=="object"){if(typeof o.text=="string")return o.text;if(Array.isArray(o.parts))return o.parts.map(c=>typeof c?.text=="string"?c.text:"").filter(Boolean).join(`
|
|
6
|
+
`)}return""},n=a?.contents??a,e=[],s=a?.systemInstruction??a?.system_instruction,r=t(s);if(r.length>0&&e.push({role:"system",content:r}),typeof n=="string")return e.push({role:"user",content:n}),e;if(Array.isArray(n)){for(let o of n){if(typeof o=="string"){e.push({role:"user",content:o});continue}if(o&&typeof o=="object"){let c=typeof o.role=="string"?o.role:"user",i=t(o);i.length>0&&e.push({role:c,content:i})}}return e}if(n&&typeof n=="object"){let o=typeof n.role=="string"?n.role:"user",c=t(n);c.length>0&&e.push({role:o,content:c})}return e}function Ue(a,t){return{"gen_ai.system":"google","gen_ai.model":t,"gen_ai.operation.name":"chat","gen_ai.prompt":JSON.stringify(a),"gen_ai.request.max_tokens":a.generationConfig?.maxOutputTokens,"gen_ai.request.temperature":a.generationConfig?.temperature,"gen_ai.request.top_p":a.generationConfig?.topP}}function mt(a){try{if(typeof a?.text=="function"){let n=a.text();if(typeof n=="string"&&n.length>0)return n}}catch{}return(a?.candidates?.[0]?.content?.parts??[]).map(n=>typeof n?.text=="string"?n.text:"").filter(Boolean).join("")}async function*jt(a,t){let{client:n,t0:e,traceId:s,spanId:r,parent:o,params:c,modelName:i,otelSpan:l,streamResult:u,agentCtx:d,tool_definitions:m,autoDetected:_,conversation_turn:y,message_role:x,operation:b}=t,k,I=0,h=0,T=0,g=0,A="",f=[],v="";try{for await(let S of a){k||(k=performance.now()-e,n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:b,latency_ms:k,ttft_ms:k,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:Te(JSON.stringify(c)),...n.storePrompts?{prompt_text:n.redactForStorage(c)}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C,finish_reason:"stream_start",...d,span_kind:"llm",tool_definitions:m,conversation_turn:y,message_role:x,metadata:{auto_detected:Object.keys(_).length?_:void 0}}),l&&l.setAttribute("gen_ai.response.time_to_first_token",k));let M=(typeof S?.text=="function"?S.text():S?.text)||S?.candidates?.[0]?.content?.parts?.map(R=>R?.text||"").join("")||"";M&&(I++,v+=M),S.candidates?.[0]?.finishReason&&(A=S.candidates[0].finishReason);let N=Array.isArray(S?.functionCalls)?S.functionCalls:S?.functionCalls?[S.functionCalls]:[],q=(S?.candidates?.[0]?.content?.parts??[]).map(R=>R?.functionCall).filter(Boolean);for(let R of[...N,...q])R?.name&&!f.includes(R.name)&&f.push(R.name);yield S}let w=await u?.response;w?.usageMetadata&&(h=Number(w.usageMetadata.promptTokenCount||h||0),I=Number(w.usageMetadata.candidatesTokenCount||I||0),T=Number(w.usageMetadata.cachedContentTokenCount||T||0),g=Number(w.usageMetadata.thoughtsTokenCount||g||0));let j=mt(w);j&&(v=j);let O=performance.now()-e;n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:b,latency_ms:O,ttft_ms:k,stream:!0,input_tokens:h,output_tokens:I,cached_tokens:T,reasoning_tokens:g,status:"success",finish_reason:A||"stream_end",tools_used:f,prompt_hash:Te(JSON.stringify(c)),...n.storePrompts?{prompt_text:n.redactForStorage(c),completion_text:n.redactForStorage(v)}:{},...d,span_kind:"llm",tool_definitions:m,conversation_turn:y,message_role:x,metadata:{usage:w?.usageMetadata??{promptTokenCount:h,candidatesTokenCount:I,cachedContentTokenCount:T,thoughtsTokenCount:g},auto_detected:Object.keys(_).length?_:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C}),l&&(l.setAttribute("gen_ai.completion",v),l.setAttribute("gen_ai.usage.input_tokens",h),l.setAttribute("gen_ai.usage.output_tokens",I),l.setAttribute("gen_ai.usage.total_tokens",h+I),l.setAttribute("gen_ai.usage.cached_tokens",T),l.setAttribute("gen_ai.usage.reasoning_tokens",g),l.setAttribute("gen_ai.response.finish_reason",A||"stream_end"),l.setStatus({code:ce.OK}),l.end())}catch(w){throw n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:b,latency_ms:performance.now()-e,ttft_ms:k,stream:!0,input_tokens:0,output_tokens:I,cost_usd:0,status:"error",error_code:String(w.status??"unknown"),error_message:String(w.message).slice(0,500),...d,span_kind:"llm",conversation_turn:y,message_role:x,metadata:{auto_detected:Object.keys(_).length?_:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C}),l&&(l.setStatus({code:ce.ERROR,message:w.message}),l.recordException(w),l.end()),w}}function Ft(a,t={}){let n=a,e=n.getGenerativeModel.bind(n),s=X(t.guardrails,"chat"),r=t.operation??"chat";return n.getGenerativeModel=(o,c)=>{let i=o.model,l=D(),u=!1,d=i,m;if(l.configClient){let k=l.configClient.getAbTest(r);if(k)Math.random()<k.splitPercentage/100?(d=i,i=k.candidateModel,o={...o,model:i},u=!0,m={ab_test:!0,ab_variant:"candidate"}):m={ab_test:!0,ab_variant:"control"};else{let I=l.configClient.getModel(r,i);I!==i&&(d=i,i=I,o={...o,model:I},u=!0)}}let _=e(o,c),y=_.generateContent.bind(_);_.generateContent=async(k,I)=>{let h=D(),T=performance.now(),g=G(),A=p(),f=g?.traceId??p(),v=null,w=null,j=Ze(k),O=re(j,i),{conversation_turn:S,message_role:M}=ae(j),N=g?.agentName??O.agent_name,q=g?.agentRole??O.agent_role,R=g?.agentId||N||q?{agent_id:g?.agentId,agent_name:N,agent_role:q,agent_framework:g?.agentFramework,parent_agent_id:g?.parentAgentId}:{},F=k?.tools?.length?ie(k.tools):void 0,P=null,B=h.enableOTel?Je.getTracer("zespan-sdk"):null;B&&(P=B.startSpan(`generateContent ${i}`,{kind:We.CLIENT,attributes:Ue(k,i)})),u&&h.enqueue({event_id:p(),project_id:"",trace_id:f,span_id:A,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:d,override_model:i},environment:h.environment,sdk_version:C});try{v=await Z(h,s,{text:JSON.stringify(k??""),phase:"pre",traceId:f,spanId:A,model:i,operation:r,agentName:g?.agentName});let L;if(h.configClient){let U=h.configClient.getTimeout(r);U&&(L=AbortSignal.timeout(U.timeoutMs))}let z=h.configClient?.getRetryPolicy(r)??null,Q=h.configClient?.getConcurrencyLimit(r)??null,E=()=>Se(()=>W({traceId:f,spanId:A,userId:g?.userId,sessionId:g?.sessionId,tags:g?.tags,agentId:g?.agentId,agentName:N,agentRole:q,agentFramework:g?.agentFramework,parentAgentId:g?.parentAgentId,conversationTurn:S,availableTools:F},()=>y(k,L?{...I,signal:L}:I)),z),H=await(Q!==null?ve(`${h.projectId??"default"}:${r}`,Q).run(E):E()),K=H.response,J=K.usageMetadata?.promptTokenCount??0,ee=K.usageMetadata?.candidatesTokenCount??0,$=K.usageMetadata?.cachedContentTokenCount??0,oe=K.usageMetadata?.thoughtsTokenCount??0,ue=mt(K),ge=(K.candidates?.[0]?.content?.parts??[]).filter(U=>U?.functionCall),se=ge.map(U=>U.functionCall.name).filter(Boolean),de=(K.candidates?.[0]?.content?.parts??[]).filter(U=>U?.text&&!U?.functionCall).map(U=>U.text).join(""),pe={};for(let U of ge)U.functionCall?.name&&(pe[U.functionCall.name]=U.functionCall.args??{});w=await Z(h,s,{text:ue,phase:"post",traceId:f,spanId:A,model:i,operation:r,agentName:g?.agentName});let _e=ne(ue,w);return h.enqueue({event_id:p(),project_id:"",trace_id:f,span_id:A,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-T,stream:!1,input_tokens:J,output_tokens:ee,cached_tokens:$,reasoning_tokens:oe,status:"success",finish_reason:K.candidates?.[0]?.finishReason??"",tools_used:se,tool_call_args:Object.keys(pe).length?pe:void 0,prompt_hash:Te(JSON.stringify(k)),...h.storePrompts?{prompt_text:h.redactForStorage(k),completion_text:h.redactForStorage(_e)}:{},...se.length>0&&de?{decision_rationale:de}:{},...R,span_kind:"llm",tool_definitions:F,conversation_turn:S,message_role:M,metadata:{usage:K.usageMetadata??null,guardrails:{pre:v?{matched:v.results.length,modified:!!v.modifiedText}:null,post:w?{matched:w.results.length,modified:!!w.modifiedText}:null},auto_detected:Object.keys(O).length?O:void 0,...m??{}},user_id:g?.userId,session_id:g?.sessionId,tags:g?.tags,environment:h.environment,sdk_version:C}),P&&(P.setAttribute("gen_ai.completion",_e),P.setAttribute("gen_ai.usage.input_tokens",J),P.setAttribute("gen_ai.usage.output_tokens",ee),P.setAttribute("gen_ai.usage.total_tokens",J+ee),P.setAttribute("gen_ai.usage.cached_tokens",$),P.setAttribute("gen_ai.usage.reasoning_tokens",oe),P.setAttribute("gen_ai.response.finish_reason",K.candidates?.[0]?.finishReason??""),P.setAttribute("gen_ai.response.latency",performance.now()-T),P.setStatus({code:ce.OK}),P.end()),H}catch(L){let z=L.__signal;if(L?.name==="TimeoutError"||L?.code==="ABORT_ERR")throw h.enqueue({event_id:p(),project_id:"",trace_id:f,span_id:A,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-T,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:"timeout",error_message:String(L.message).slice(0,500),...R,span_kind:"llm",conversation_turn:S,message_role:M,metadata:{auto_detected:Object.keys(O).length?O:void 0},user_id:g?.userId,session_id:g?.sessionId,tags:g?.tags,environment:h.environment,sdk_version:C}),P&&(P.setStatus({code:ce.ERROR,message:L.message}),P.recordException(L),P.end()),L;let E=h.configClient?.getFallback(r)??null;if(E&&E.model){let J=String(L.message??"")+String(L.status??"")+String(L.code??"");if(E.conditions.length===0||E.conditions.some($=>J.toLowerCase().includes($.toLowerCase()))){let $=i,oe={...o,model:E.model},ue=e(oe,c),ge=ue.generateContent.bind(ue);h.enqueue({event_id:p(),project_id:"",trace_id:f,span_id:A,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google",model:E.model,operation:r,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:$,fallback_model:E.model,reason:String(L.message).slice(0,200)},environment:h.environment,sdk_version:C});try{let se=await W({traceId:f,spanId:A,userId:g?.userId,sessionId:g?.sessionId,tags:g?.tags},()=>ge(k,I)),de=se.response,pe=de.usageMetadata?.promptTokenCount??0,_e=de.usageMetadata?.candidatesTokenCount??0,U=de.usageMetadata?.cachedContentTokenCount??0;return h.enqueue({event_id:p(),project_id:"",trace_id:f,span_id:A,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google",model:E.model,operation:r,latency_ms:performance.now()-T,stream:!1,input_tokens:pe,output_tokens:_e,cached_tokens:U,status:"success",finish_reason:de.candidates?.[0]?.finishReason??"",prompt_hash:Te(JSON.stringify(k)),...R,span_kind:"llm",conversation_turn:S,message_role:M,metadata:{usage:de.usageMetadata??null,fallback_used:!0,original_model:$,auto_detected:Object.keys(O).length?O:void 0},user_id:g?.userId,session_id:g?.sessionId,tags:g?.tags,environment:h.environment,sdk_version:C}),P&&(P.setStatus({code:ce.OK}),P.end()),se}catch(se){throw P&&(P.setStatus({code:ce.ERROR,message:se.message}),P.recordException(se),P.end()),se}}}let H=L instanceof V,K=H?"error":L.status===429?"rate_limited":L.status===408?"timeout":"error";throw h.enqueue({event_id:p(),project_id:"",trace_id:f,span_id:A,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-T,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:K,error_code:H?"guardrail_blocked":String(L.status??"unknown"),error_message:String(L.message).slice(0,500),...R,span_kind:"llm",conversation_turn:S,message_role:M,metadata:{auto_detected:Object.keys(O).length?O:void 0},user_id:g?.userId,session_id:g?.sessionId,tags:g?.tags,environment:h.environment,sdk_version:C}),P&&(P.setStatus({code:ce.ERROR,message:L.message}),P.recordException(L),P.end()),L}};let x=_.generateContentStream.bind(_);_.generateContentStream=async(k,I)=>{let h=D(),T=performance.now(),g=G(),A=p(),f=g?.traceId??p(),v=Ze(k),w=re(v,i),{conversation_turn:j,message_role:O}=ae(v),S=g?.agentName??w.agent_name,M=g?.agentRole??w.agent_role,N=g?.agentId||S||M?{agent_id:g?.agentId,agent_name:S,agent_role:M,agent_framework:g?.agentFramework,parent_agent_id:g?.parentAgentId}:{},q=k?.tools?.length?ie(k.tools):void 0,R=null,F=h.enableOTel?Je.getTracer("zespan-sdk"):null;F&&(R=F.startSpan(`generateContentStream ${i}`,{kind:We.CLIENT,attributes:Ue(k,i)}));try{let P=await W({traceId:f,spanId:A,userId:g?.userId,sessionId:g?.sessionId,tags:g?.tags,agentId:g?.agentId,agentName:S,agentRole:M,agentFramework:g?.agentFramework,parentAgentId:g?.parentAgentId,conversationTurn:j,availableTools:q},()=>x(k,I));return R&&R.setAttribute("gen_ai.request.stream",!0),P.stream=jt(P.stream,{client:h,t0:T,traceId:f,spanId:A,parent:g,params:k,modelName:i,otelSpan:R,streamResult:P,agentCtx:N,tool_definitions:q,autoDetected:w,conversation_turn:j,message_role:O,operation:r}),P}catch(P){let B=P instanceof V,L=B?"error":P.status===429?"rate_limited":P.status===408?"timeout":"error";throw h.enqueue({event_id:p(),project_id:"",trace_id:f,span_id:A,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-T,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:L,error_code:B?"guardrail_blocked":String(P.status??"unknown"),error_message:String(P.message).slice(0,500),...N,span_kind:"llm",conversation_turn:j,message_role:O,metadata:{auto_detected:Object.keys(w).length?w:void 0},user_id:g?.userId,session_id:g?.sessionId,tags:g?.tags,environment:h.environment,sdk_version:C}),R&&(R.setStatus({code:ce.ERROR,message:P.message}),R.recordException(P),R.end()),P}};let b=_.startChat?.bind(_);return b&&(_.startChat=k=>{let I=b(k),h=I.sendMessage?.bind(I),T=I.sendMessageStream?.bind(I);return h&&(I.sendMessage=async(g,A)=>{let f=D(),v=performance.now(),w=G(),j=p(),O=w?.traceId??p(),S={...k||{},message:g},M=null,N=null,q=Ze(S),R=re(q,i),{conversation_turn:F,message_role:P}=ae(q),B=w?.agentName??R.agent_name,L=w?.agentRole??R.agent_role,z=w?.agentId||B||L?{agent_id:w?.agentId,agent_name:B,agent_role:L,agent_framework:w?.agentFramework,parent_agent_id:w?.parentAgentId}:{},Q=k?.tools?.length?ie(k.tools):void 0,E=null,H=f.enableOTel?Je.getTracer("zespan-sdk"):null;H&&(E=H.startSpan(`sendMessage ${i}`,{kind:We.CLIENT,attributes:Ue(S,i)}));try{M=await Z(f,s,{text:JSON.stringify(S),phase:"pre",traceId:O,spanId:j,model:i,operation:r,agentName:w?.agentName});let K=await W({traceId:O,spanId:j,userId:w?.userId,sessionId:w?.sessionId,tags:w?.tags,agentId:w?.agentId,agentName:B,agentRole:L,agentFramework:w?.agentFramework,parentAgentId:w?.parentAgentId,conversationTurn:F,availableTools:Q},()=>h(g,A)),J=K.response,ee=J.usageMetadata?.promptTokenCount??0,$=J.usageMetadata?.candidatesTokenCount??0,oe=J.usageMetadata?.cachedContentTokenCount??0,ue=J.usageMetadata?.thoughtsTokenCount??0,ge=mt(J),se=(J.candidates?.[0]?.content?.parts??[]).filter(U=>U?.functionCall),de=se.map(U=>U.functionCall.name).filter(Boolean),pe={};for(let U of se)U.functionCall?.name&&(pe[U.functionCall.name]=U.functionCall.args??{});N=await Z(f,s,{text:ge,phase:"post",traceId:O,spanId:j,model:i,operation:r,agentName:w?.agentName});let _e=ne(ge,N);return f.enqueue({event_id:p(),project_id:"",trace_id:O,span_id:j,parent_span_id:w?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-v,stream:!1,input_tokens:ee,output_tokens:$,cached_tokens:oe,reasoning_tokens:ue,status:"success",finish_reason:J.candidates?.[0]?.finishReason??"",tools_used:de,tool_call_args:Object.keys(pe).length?pe:void 0,prompt_hash:Te(JSON.stringify(S)),...f.storePrompts?{prompt_text:f.redactForStorage(S),completion_text:f.redactForStorage(_e)}:{},...z,span_kind:"llm",tool_definitions:Q,conversation_turn:F,message_role:P,metadata:{usage:J.usageMetadata??null,guardrails:{pre:M?{matched:M.results.length,modified:!!M.modifiedText}:null,post:N?{matched:N.results.length,modified:!!N.modifiedText}:null},auto_detected:Object.keys(R).length?R:void 0},user_id:w?.userId,session_id:w?.sessionId,tags:w?.tags,environment:f.environment,sdk_version:C}),E&&(E.setAttribute("gen_ai.completion",_e),E.setAttribute("gen_ai.usage.input_tokens",ee),E.setAttribute("gen_ai.usage.output_tokens",$),E.setAttribute("gen_ai.usage.total_tokens",ee+$),E.setAttribute("gen_ai.usage.cached_tokens",oe),E.setAttribute("gen_ai.usage.reasoning_tokens",ue),E.setAttribute("gen_ai.response.finish_reason",J.candidates?.[0]?.finishReason??""),E.setAttribute("gen_ai.response.latency",performance.now()-v),E.setStatus({code:ce.OK}),E.end()),K}catch(K){let J=K instanceof V,ee=J?"error":K.status===429?"rate_limited":K.status===408?"timeout":"error";throw f.enqueue({event_id:p(),project_id:"",trace_id:O,span_id:j,parent_span_id:w?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-v,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:ee,error_code:J?"guardrail_blocked":String(K.status??"unknown"),error_message:String(K.message).slice(0,500),...z,span_kind:"llm",conversation_turn:F,message_role:P,metadata:{auto_detected:Object.keys(R).length?R:void 0},user_id:w?.userId,session_id:w?.sessionId,tags:w?.tags,environment:f.environment,sdk_version:C}),E&&(E.setStatus({code:ce.ERROR,message:K.message}),E.recordException(K),E.end()),K}}),T&&(I.sendMessageStream=async(g,A)=>{let f=D(),v=performance.now(),w=G(),j=p(),O=w?.traceId??p(),S={...k||{},message:g},M=Ze(S),N=re(M,i),{conversation_turn:q,message_role:R}=ae(M),F=w?.agentName??N.agent_name,P=w?.agentRole??N.agent_role,B=w?.agentId||F||P?{agent_id:w?.agentId,agent_name:F,agent_role:P,agent_framework:w?.agentFramework,parent_agent_id:w?.parentAgentId}:{},L=k?.tools?.length?ie(k.tools):void 0,z=null,Q=f.enableOTel?Je.getTracer("zespan-sdk"):null;Q&&(z=Q.startSpan(`sendMessageStream ${i}`,{kind:We.CLIENT,attributes:Ue(S,i)}));try{let E=await W({traceId:O,spanId:j,userId:w?.userId,sessionId:w?.sessionId,tags:w?.tags,agentId:w?.agentId,agentName:F,agentRole:P,agentFramework:w?.agentFramework,parentAgentId:w?.parentAgentId,conversationTurn:q,availableTools:L},()=>T(g,A));return z&&z.setAttribute("gen_ai.request.stream",!0),E.stream=jt(E.stream,{client:f,t0:v,traceId:O,spanId:j,parent:w,params:S,modelName:i,otelSpan:z,streamResult:E,agentCtx:B,tool_definitions:L,autoDetected:N,conversation_turn:q,message_role:R,operation:r}),E}catch(E){let H=E instanceof V,K=H?"error":E.status===429?"rate_limited":E.status===408?"timeout":"error";throw f.enqueue({event_id:p(),project_id:"",trace_id:O,span_id:j,parent_span_id:w?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-v,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:K,error_code:H?"guardrail_blocked":String(E.status??"unknown"),error_message:String(E.message).slice(0,500),...B,span_kind:"llm",conversation_turn:q,message_role:R,metadata:{auto_detected:Object.keys(N).length?N:void 0},user_id:w?.userId,session_id:w?.sessionId,tags:w?.tags,environment:f.environment,sdk_version:C}),z&&(z.setStatus({code:ce.ERROR,message:E.message}),z.recordException(E),z.end()),E}}),I}),_},a}import{createHash as Nn}from"crypto";import{SpanKind as Mn,SpanStatusCode as Ve,trace as Pn}from"@opentelemetry/api";var gt=a=>Nn("sha256").update(a).digest("hex");function jn(a,t,n,e){let s=Number(a?.usage?.cost);if(Number.isFinite(s))return{cost:s,source:"usage.cost"};let r=a?.headers?.get?.("x-ratelimit-cost"),o=Number.parseFloat(String(r??""));return Number.isFinite(o)?{cost:o,source:"x-ratelimit-cost"}:{cost:0,source:"computed"}}function Fn(a,t){return{"gen_ai.system":"openrouter","gen_ai.model":t,"gen_ai.operation.name":"chat","gen_ai.prompt":JSON.stringify(a.messages),"gen_ai.request.max_tokens":a.max_tokens,"gen_ai.request.temperature":a.temperature,"gen_ai.request.top_p":a.top_p}}async function*En(a,t){let{client:n,t0:e,traceId:s,spanId:r,parent:o,params:c,otelSpan:i,agentCtx:l,autoDetected:u,conversation_turn:d,message_role:m,tool_definitions:_}=t,y,x=0,b=0,k=0,I=0,h="",T=c.model,g=[],A="",f=[],{systemPrompt:v,conversationMessages:w}=Y(c.messages??[]);try{for await(let N of a){if(y||(y=performance.now()-e,n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:T,operation:"chat",latency_ms:y,ttft_ms:y,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:gt(JSON.stringify(c.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(w),...v?{system_prompt:n.redactForStorage(v)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C,finish_reason:"stream_start",...l,span_kind:"llm",tool_definitions:_,conversation_turn:d,message_role:m,metadata:{auto_detected:Object.keys(u).length?u:void 0}}),i&&i.setAttribute("gen_ai.response.time_to_first_token",y)),N.choices?.[0]?.delta?.content&&(x++,A+=N.choices[0].delta.content),N.choices?.[0]?.finish_reason&&(h=N.choices[0].finish_reason),N.model&&(T=N.model),N.usage){b=Number(N.usage.prompt_tokens||b||0),x=Number(N.usage.completion_tokens||x||0),k=Number(N.usage.prompt_tokens_details?.cached_tokens||k||0);let q=N.usage.completion_tokens_details||{};I=Number(q.reasoning_tokens||I||0)}if(N.choices?.[0]?.delta?.tool_calls)for(let q of N.choices[0].delta.tool_calls)f.push(q),q.function?.name&&!g.includes(q.function.name)&&g.push(q.function.name);yield N}let{tools_used:j,tool_call_args:O}=me(f),S=j.length>0?j:g,M=performance.now()-e;n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:T,operation:"chat",latency_ms:M,ttft_ms:y,stream:!0,input_tokens:b,output_tokens:x,cached_tokens:k,reasoning_tokens:I,status:"success",finish_reason:h||"stream_end",tools_used:S,tool_call_args:Object.keys(O).length?O:void 0,prompt_hash:gt(JSON.stringify(c.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(w),completion_text:n.redactForStorage(A),...v?{system_prompt:n.redactForStorage(v)}:{}}:{},...l,span_kind:"llm",tool_definitions:_,conversation_turn:d,message_role:m,metadata:{usage:{prompt_tokens:b,completion_tokens:x,cached_tokens:k,reasoning_tokens:I},auto_detected:Object.keys(u).length?u:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C}),i&&(i.setAttribute("gen_ai.completion",A),i.setAttribute("gen_ai.usage.input_tokens",b),i.setAttribute("gen_ai.usage.output_tokens",x),i.setAttribute("gen_ai.usage.total_tokens",b+x),i.setAttribute("gen_ai.usage.reasoning_tokens",I),i.setAttribute("gen_ai.response.finish_reason",h||"stream_end"),i.setStatus({code:Ve.OK}),i.end())}catch(j){throw n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:T,operation:"chat",latency_ms:performance.now()-e,ttft_ms:y,stream:!0,input_tokens:0,output_tokens:x,cost_usd:0,status:"error",error_code:String(j.status??"unknown"),error_message:String(j.message).slice(0,500),...l,span_kind:"llm",conversation_turn:d,message_role:m,metadata:{auto_detected:Object.keys(u).length?u:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C}),i&&(i.setStatus({code:Ve.ERROR,message:j.message}),i.recordException(j),i.end()),j}}function Et(a,t={}){let n=a.chat.completions.create,e=X(t.guardrails,"chat");return a.chat.completions.create=async(s,r)=>{let o=D(),c=performance.now(),i=G(),l=p(),u=i?.traceId??p(),d=s.model??"openrouter\u6A21\u578B",m=null,_=null;if(o.configClient){let g=o.configClient.getModel("chat",d);if(g!==d){let A=d;d=g,s={...s,model:g},o.enqueue({event_id:p(),project_id:"",trace_id:u,span_id:l,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:g,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:A,override_model:g},environment:o.environment,sdk_version:C})}}let y=i?.agentId?{agent_id:i.agentId,agent_name:i.agentName,agent_role:i.agentRole,agent_framework:i.agentFramework,parent_agent_id:i.parentAgentId}:{},x=re(s.messages??[],s.model),{conversation_turn:b,message_role:k}=ae(s.messages??[]),I=s.tools?.length?ie(s.tools):void 0,h=null,T=o.enableOTel?Pn.getTracer("zespan-sdk"):null;T&&(h=T.startSpan(`chat.completions ${d}`,{kind:Mn.CLIENT,attributes:Fn(s,d)}));try{m=await Z(o,e,{text:JSON.stringify(s.messages??[]),phase:"pre",traceId:u,spanId:l,model:d,operation:"chat"});let g=await W({traceId:u,spanId:l,userId:i?.userId,sessionId:i?.sessionId,tags:i?.tags},()=>n.call(a.chat.completions,s,r));if(s.stream===!0)return h&&h.setAttribute("gen_ai.request.stream",!0),En(g,{client:o,t0:c,traceId:u,spanId:l,parent:i,params:s,otelSpan:h,agentCtx:y,autoDetected:x,conversation_turn:b,message_role:k,tool_definitions:I});let{prompt_tokens:f,completion_tokens:v,completion_tokens_details:w}=g.usage??{},j=w?.reasoning_tokens??0,O=g.model??s.model,{cost:S,source:M}=jn(g,O,f??0,v??0);g.pricing||(g.pricing={cost_usd:S,source:M});let N=g.choices?.[0]?.message?.content||"";_=await Z(o,e,{text:N,phase:"post",traceId:u,spanId:l,model:O,operation:"chat"});let q=ne(N,_);q!==N&&typeof g?.choices?.[0]?.message?.content=="string"&&(g.choices[0].message.content=q);let{tools_used:R,tool_call_args:F}=Le(g.choices??[]);return o.enqueue({event_id:p(),project_id:"",trace_id:u,span_id:l,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:O,operation:"chat",latency_ms:performance.now()-c,stream:!1,input_tokens:f??0,output_tokens:v??0,cached_tokens:0,cost_usd:S,status:"success",finish_reason:g.choices?.[0]?.finish_reason??"",tools_used:R.length>0?R:g.choices?.[0]?.message?.tool_calls?.map(P=>P.function.name)??[],tool_call_args:Object.keys(F).length?F:void 0,prompt_hash:gt(JSON.stringify(s.messages)),...o.storePrompts?(()=>{let{systemPrompt:P,conversationMessages:B}=Y(s.messages??[]);return{prompt_text:o.redactForStorage(B),completion_text:o.redactForStorage(q),...P?{system_prompt:o.redactForStorage(P)}:{}}})():{},...R.length>0&&N?{decision_rationale:N}:{},...y,span_kind:"llm",tool_definitions:I,conversation_turn:b,message_role:k,metadata:{usage:g.usage??null,pricing:g.pricing??{cost_usd:S,source:M},guardrails:{pre:m?{matched:m.results.length,modified:!!m.modifiedText}:null,post:_?{matched:_.results.length,modified:!!_.modifiedText}:null},auto_detected:Object.keys(x).length?x:void 0},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:o.environment,sdk_version:C}),h&&(h.setAttribute("gen_ai.usage.input_tokens",f??0),h.setAttribute("gen_ai.usage.output_tokens",v??0),h.setAttribute("gen_ai.usage.total_tokens",(f??0)+(v??0)),h.setAttribute("gen_ai.usage.reasoning_tokens",j),h.setAttribute("gen_ai.completion",g.choices?.[0]?.message?.content||""),h.setAttribute("gen_ai.response.finish_reason",g.choices?.[0]?.finish_reason??""),h.setAttribute("gen_ai.response.latency",performance.now()-c),h.setStatus({code:Ve.OK}),h.end()),g}catch(g){let A=g instanceof V,f=A?"error":g.status===429?"rate_limited":g.status===408?"timeout":"error";throw o.enqueue({event_id:p(),project_id:"",trace_id:u,span_id:l,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:s.model,operation:"chat",latency_ms:performance.now()-c,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:f,error_code:A?"guardrail_blocked":String(g.status??g.code??"unknown"),error_message:String(g.message).slice(0,500),...y,span_kind:"llm",conversation_turn:b,message_role:k,metadata:{auto_detected:Object.keys(x).length?x:void 0},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:o.environment,sdk_version:C}),h&&(h.setStatus({code:Ve.ERROR,message:g.message}),h.recordException(g),h.end()),g}},a}async function*Dt(a,t){let{client:n,t0:e,traceId:s,spanId:r,parent:o,model:c,operation:i,input:l}=t,u=0,d=0,m=0,_="",y,x="",b=[],k={},I="",h="",T=l?.messages??[],g=(l?.system??[]).map(f=>f.text??"").filter(Boolean),A=g.length?g.join(`
|
|
7
|
+
`):void 0;try{for await(let f of a){if(y||(y=performance.now()-e,n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:c,operation:i,latency_ms:y,ttft_ms:y,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",finish_reason:"stream_start",span_kind:"llm",...n.storePrompts?{prompt_text:n.redactForStorage(T),...A?{system_prompt:n.redactForStorage(A)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C})),"contentBlockStart"in f){let v=f.contentBlockStart?.start?.toolUse;v?.name&&(I=v.name,h="",b.includes(I)||b.push(I))}else if("contentBlockDelta"in f){let v=f.contentBlockDelta?.delta;v?.text&&(x+=v.text),v?.toolUse?.input&&(h+=v.toolUse.input)}else if("contentBlockStop"in f){if(I){try{k[I]=JSON.parse(h)}catch{k[I]=h}I="",h=""}}else"metadata"in f&&f.metadata?.usage?(u=f.metadata.usage.inputTokens??u,d=f.metadata.usage.outputTokens??d,m=f.metadata.usage.cacheReadInputTokens??m):"messageStop"in f&&f.messageStop?.stopReason&&(_=f.messageStop.stopReason);yield f}n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:c,operation:i,latency_ms:performance.now()-e,ttft_ms:y,stream:!0,input_tokens:u,output_tokens:d,cached_tokens:m,cost_usd:0,status:"success",finish_reason:_||"stream_end",span_kind:"llm",tools_used:b.length?b:void 0,tool_call_args:Object.keys(k).length?k:void 0,...b.length>0&&x?{decision_rationale:x}:{},...n.storePrompts?{prompt_text:n.redactForStorage(T),completion_text:n.redactForStorage(x),...A?{system_prompt:n.redactForStorage(A)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C})}catch(f){throw n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:c,operation:i,latency_ms:performance.now()-e,ttft_ms:y,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:String(f.$metadata?.httpStatusCode??f.name??"unknown"),error_message:String(f.message).slice(0,500),user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C}),f}}function Dn(a,t={}){let n=a,e=t.operation??"chat",s=X(t.guardrails,e),r=n.converse?.bind(n);r&&(n.converse=async(i,...l)=>{let u=D(),d=performance.now(),m=G(),_=p(),y=m?.traceId??p(),x=i.modelId??"amazon.nova-lite-v1:0";u.configClient&&(x=u.configClient.getModel(e,x),x!==i.modelId&&(i={...i,modelId:x}));let b=JSON.stringify(i.messages??[]);await Z(u,s,{text:b,phase:"pre",traceId:y,spanId:_,model:x,operation:e,agentName:m?.agentName});try{let k=await W({traceId:y,spanId:_,userId:m?.userId,sessionId:m?.sessionId,tags:m?.tags},()=>r(i,...l)),I=k.usage??{},h=k.output?.message?.content??[],T=h.filter(v=>v.text).map(v=>v.text).join(""),g=h.filter(v=>v.toolUse),A=g.map(v=>v.toolUse?.name).filter(Boolean),f={};for(let v of g)v.toolUse?.name&&(f[v.toolUse.name]=v.toolUse.input??{});return u.enqueue({event_id:p(),project_id:"",trace_id:y,span_id:_,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:x,operation:e,latency_ms:performance.now()-d,stream:!1,input_tokens:Number(I.inputTokens??0),output_tokens:Number(I.outputTokens??0),cached_tokens:Number(I.cacheReadInputTokens??0),status:"success",finish_reason:k.stopReason??"",span_kind:"llm",tools_used:A.length?A:void 0,tool_call_args:Object.keys(f).length?f:void 0,...u.storePrompts?{prompt_text:u.redactForStorage(i.messages??[]),completion_text:u.redactForStorage(T),...(()=>{let v=(i.system??[]).map(w=>w.text??"").filter(Boolean);return v.length?{system_prompt:u.redactForStorage(v.join(`
|
|
8
|
+
`))}:{}})()}:{},...A.length>0&&T?{decision_rationale:T}:{},user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:u.environment,sdk_version:C}),k}catch(k){throw u.enqueue({event_id:p(),project_id:"",trace_id:y,span_id:_,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:x,operation:e,latency_ms:performance.now()-d,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:String(k.$metadata?.httpStatusCode??k.name??"unknown"),error_message:String(k.message).slice(0,500),user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:u.environment,sdk_version:C}),k}});let o=n.converseStream?.bind(n);o&&(n.converseStream=async(i,...l)=>{let u=D(),d=performance.now(),m=G(),_=p(),y=m?.traceId??p(),x=i.modelId??"amazon.nova-lite-v1:0",b=await W({traceId:y,spanId:_,userId:m?.userId,sessionId:m?.sessionId,tags:m?.tags},()=>o(i,...l));return{...b,stream:Dt(b.stream,{client:u,t0:d,traceId:y,spanId:_,parent:m,model:x,operation:e,input:i})}});let c=n.send?.bind(n);return c&&(n.send=async(i,...l)=>{if(!(i?.constructor?.name==="ConverseStreamCommand"||i?.$name==="ConverseStream"))return c(i,...l);let d=D(),m=performance.now(),_=G(),y=p(),x=_?.traceId??p(),b=i?.input?.modelId??"amazon.nova-lite-v1:0",k=await W({traceId:x,spanId:y,userId:_?.userId,sessionId:_?.sessionId,tags:_?.tags},()=>c(i,...l));return{...k,stream:Dt(k.stream,{client:d,t0:m,traceId:x,spanId:y,parent:_,model:b,operation:e,input:i.input})}}),a}function Gn(a,t={}){let n=a,e=t.operation??"chat",s=X(t.guardrails,e),r=n.chat.complete.bind(n.chat);n.chat.complete=async(c,...i)=>{let l=D(),u=performance.now(),d=G(),m=p(),_=d?.traceId??p(),y=c.model??"mistral-small-latest";l.configClient&&(y=l.configClient.getModel(e,y),y!==c.model&&(c={...c,model:y}));let x=JSON.stringify(c.messages??[]);await Z(l,s,{text:x,phase:"pre",traceId:_,spanId:m,model:y,operation:e});try{let b=await W({traceId:_,spanId:m,userId:d?.userId,sessionId:d?.sessionId,tags:d?.tags},()=>r(c,...i)),k=b.model??y,I=b.usage??{},h=Number(I.promptTokens??0),T=Number(I.completionTokens??0),g=b.choices?.[0]?.finishReason??"",A=b.choices?.[0]?.message?.content??"",f=b.choices?.[0]?.message?.tool_calls??[],v=f.map(S=>S.function?.name).filter(Boolean),w={};for(let S of f)if(S.function?.name)try{w[S.function.name]=JSON.parse(S.function.arguments??"{}")}catch{w[S.function.name]={}}await Z(l,s,{text:A,phase:"post",traceId:_,spanId:m,model:k,operation:e});let{systemPrompt:j,conversationMessages:O}=Y(c.messages??[]);return l.enqueue({event_id:p(),project_id:"",trace_id:_,span_id:m,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:k,operation:e,latency_ms:performance.now()-u,stream:!1,input_tokens:h,output_tokens:T,status:"success",finish_reason:g,span_kind:"llm",tools_used:v.length?v:void 0,tool_call_args:Object.keys(w).length?w:void 0,...l.storePrompts?{prompt_text:l.redactForStorage(O),completion_text:l.redactForStorage(A),...j?{system_prompt:l.redactForStorage(j)}:{}}:{},...v.length>0&&A?{decision_rationale:A}:{},user_id:d?.userId,session_id:d?.sessionId,tags:d?.tags,environment:l.environment,sdk_version:C}),b}catch(b){throw l.enqueue({event_id:p(),project_id:"",trace_id:_,span_id:m,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:y,operation:e,latency_ms:performance.now()-u,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_message:String(b.message).slice(0,500),user_id:d?.userId,session_id:d?.sessionId,tags:d?.tags,environment:l.environment,sdk_version:C}),b}};let o=n.chat.stream?.bind(n.chat);return o&&(n.chat.stream=async function*(c,...i){let l=D(),u=performance.now(),d=G(),m=p(),_=d?.traceId??p(),y=c.model??"mistral-small-latest";l.configClient&&(y=l.configClient.getModel(e,y),y!==c.model&&(c={...c,model:y}));let x=0,b=0,k="",I,h="",T=[],g=[],{systemPrompt:A,conversationMessages:f}=Y(c.messages??[]);try{let v=await W({traceId:_,spanId:m,userId:d?.userId,sessionId:d?.sessionId,tags:d?.tags},()=>o(c,...i));for await(let S of v){I||(I=performance.now()-u,l.enqueue({event_id:p(),project_id:"",trace_id:_,span_id:m,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:y,operation:e,latency_ms:I,ttft_ms:I,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",finish_reason:"stream_start",span_kind:"llm",...l.storePrompts?{prompt_text:l.redactForStorage(f),...A?{system_prompt:l.redactForStorage(A)}:{}}:{},user_id:d?.userId,session_id:d?.sessionId,tags:d?.tags,environment:l.environment,sdk_version:C}));let M=S.data?.choices?.[0]?.delta?.content;if(M&&(b++,h+=M),S.data?.usage&&(x=S.data.usage.promptTokens??x,b=S.data.usage.completionTokens??b),S.data?.choices?.[0]?.finishReason&&(k=S.data.choices[0].finishReason),S.data?.model&&(y=S.data.model),S.data?.choices?.[0]?.delta?.tool_calls)for(let N of S.data.choices[0].delta.tool_calls)g.push(N),N.function?.name&&!T.includes(N.function.name)&&T.push(N.function.name);yield S}let{tools_used:w,tool_call_args:j}=me(g),O=w.length>0?w:T;l.enqueue({event_id:p(),project_id:"",trace_id:_,span_id:m,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:y,operation:e,latency_ms:performance.now()-u,ttft_ms:I,stream:!0,input_tokens:x,output_tokens:b,cost_usd:0,status:"success",finish_reason:k||"stream_end",span_kind:"llm",tools_used:O.length?O:void 0,tool_call_args:Object.keys(j).length?j:void 0,...O.length>0&&h?{decision_rationale:h}:{},...l.storePrompts?{prompt_text:l.redactForStorage(f),completion_text:l.redactForStorage(h),...A?{system_prompt:l.redactForStorage(A)}:{}}:{},user_id:d?.userId,session_id:d?.sessionId,tags:d?.tags,environment:l.environment,sdk_version:C})}catch(v){throw l.enqueue({event_id:p(),project_id:"",trace_id:_,span_id:m,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:y,operation:e,latency_ms:performance.now()-u,ttft_ms:I,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_message:String(v.message).slice(0,500),user_id:d?.userId,session_id:d?.sessionId,tags:d?.tags,environment:l.environment,sdk_version:C}),v}}),a}import{createHash as qn}from"crypto";var Gt=a=>qn("sha256").update(a).digest("hex");async function*Kn(a,t){let{client:n,t0:e,traceId:s,spanId:r,parent:o,params:c,model:i,operation:l}=t,u,d=0,m=0,_=0,y="",x="",b=i,k=[],I=[],{systemPrompt:h,conversationMessages:T}=Y(c.messages??[]);try{for await(let v of a){u||(u=performance.now()-e,n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"groq",model:b,operation:l,latency_ms:u,ttft_ms:u,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:Gt(JSON.stringify(c.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(T),...h?{system_prompt:n.redactForStorage(h)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C,finish_reason:"stream_start",span_kind:"llm"}));let w=v.choices?.[0]?.delta?.content;if(w&&(m++,x+=w),v.choices?.[0]?.finish_reason&&(y=v.choices[0].finish_reason),v.model&&(b=v.model),v.usage&&(d=Number(v.usage.prompt_tokens||d||0),m=Number(v.usage.completion_tokens||m||0),_=Number(v.usage.prompt_tokens_details?.cached_tokens||_||0)),v.choices?.[0]?.delta?.tool_calls)for(let j of v.choices[0].delta.tool_calls)I.push(j),j.function?.name&&!k.includes(j.function.name)&&k.push(j.function.name);yield v}let{tools_used:g,tool_call_args:A}=me(I),f=g.length>0?g:k;n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"groq",model:b,operation:l,latency_ms:performance.now()-e,ttft_ms:u,stream:!0,input_tokens:d,output_tokens:m,cached_tokens:_,status:"success",finish_reason:y||"stream_end",tools_used:f.length?f:void 0,tool_call_args:Object.keys(A).length?A:void 0,...f.length>0&&x?{decision_rationale:x}:{},prompt_hash:Gt(JSON.stringify(c.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(T),completion_text:n.redactForStorage(x),...h?{system_prompt:n.redactForStorage(h)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C,span_kind:"llm"})}catch(g){throw n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"groq",model:b,operation:l,latency_ms:performance.now()-e,ttft_ms:u,stream:!0,input_tokens:0,output_tokens:m,cost_usd:0,status:"error",error_message:String(g.message).slice(0,500),user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:C}),g}}function Ln(a,t={}){let n=a,e=n.chat.completions.create.bind(n.chat.completions),s=t.operation??"chat";return n.chat.completions.create=async(r,...o)=>{let c=D(),i=performance.now(),l=G(),u=p(),d=l?.traceId??p(),m=r.model??"llama-3.3-70b-versatile";c.configClient&&(m=c.configClient.getModel(s,m),m!==r.model&&(r={...r,model:m}));try{let _=await W({traceId:d,spanId:u,userId:l?.userId,sessionId:l?.sessionId,tags:l?.tags},()=>e(r,...o));if(r.stream)return Kn(_,{client:c,t0:i,traceId:d,spanId:u,parent:l,params:r,model:m,operation:s});let y=_.usage??{},x=Number(y.prompt_tokens??0),b=Number(y.completion_tokens??0),k=_.choices?.[0]?.finish_reason??"",I=_.choices?.[0]?.message?.content??"",h=_.choices?.[0]?.message?.tool_calls??[],T=h.map(v=>v.function?.name).filter(Boolean),g={};for(let v of h)if(v.function?.name)try{g[v.function.name]=JSON.parse(v.function.arguments??"{}")}catch{g[v.function.name]={}}let{systemPrompt:A,conversationMessages:f}=Y(r.messages??[]);return c.enqueue({event_id:p(),project_id:"",trace_id:d,span_id:u,parent_span_id:l?.spanId,timestamp:new Date().toISOString(),provider:"groq",model:m,operation:s,latency_ms:performance.now()-i,stream:!1,input_tokens:x,output_tokens:b,status:"success",finish_reason:k,span_kind:"llm",tools_used:T.length?T:void 0,tool_call_args:Object.keys(g).length?g:void 0,...c.storePrompts?{prompt_text:c.redactForStorage(f),completion_text:c.redactForStorage(I),...A?{system_prompt:c.redactForStorage(A)}:{}}:{},...T.length>0&&I?{decision_rationale:I}:{},user_id:l?.userId,session_id:l?.sessionId,tags:l?.tags,environment:c.environment,sdk_version:C}),_}catch(_){throw c.enqueue({event_id:p(),project_id:"",trace_id:d,span_id:u,parent_span_id:l?.spanId,timestamp:new Date().toISOString(),provider:"groq",model:m,operation:s,latency_ms:performance.now()-i,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_message:String(_.message).slice(0,500),user_id:l?.userId,session_id:l?.sessionId,tags:l?.tags,environment:c.environment,sdk_version:C}),_}},a}function zn(a,t={}){let n=ot("openai").default??ot("openai"),e=new n({baseURL:a.baseURL,apiKey:a.apiKey??"litellm"});return Be(e,{...t,providerOverride:"litellm"})}import{BaseCallbackHandler as Bn}from"@langchain/core/callbacks/base";import{createHash as $n}from"crypto";var qt=a=>$n("sha256").update(a).digest("hex"),Kt=["AgentExecutor","ReactAgent","ToolCallingAgent"],Lt=["DynamicStructuredTool","StructuredTool","DynamicTool","Tool"],He=class extends Bn{name="ZespanCallbackHandler";guardrailSettings=X(void 0,"chat");runs=new Map;constructor(t={}){super(),this.guardrailSettings=X(t.guardrails,"chat")}detectProvider(t){let n=t.toLowerCase();return/gpt|o1-|o3-|davinci|turbo|whisper|dall-e/i.test(n)?"openai":/claude/i.test(n)?"anthropic":/gemini|palm|bard/i.test(n)?"google":/llama|codellama/i.test(n)?"meta":/mistral|mixtral/i.test(n)?"mistralai":/deepseek/i.test(n)?"deepseek":/command/i.test(n)?"cohere":/qwen/i.test(n)?"alibaba":"langchain"}extractToolCalls(t){try{let n=t?.generations?.[0];if(!Array.isArray(n))return[];let e=n[0]?.message;if(!e)return[];let s=e.tool_calls;if(Array.isArray(s)&&s.length>0)return s.map(c=>({name:c.name??"unknown",args:c.args??{},id:c.id}));let r=e.additional_kwargs?.tool_calls;if(Array.isArray(r)&&r.length>0)return r.map(c=>({name:c.function?.name??"unknown",args:c.function?.arguments?JSON.parse(c.function.arguments):{},id:c.id}));let o=e.additional_kwargs?.function_call;return o?[{name:o.name??"unknown",args:o.arguments?JSON.parse(o.arguments):{}}]:[]}catch{return[]}}detectConversationInfo(t,n){let e=n?._zespan_message_count;if(typeof e=="number"&&e>0)return{turns:e,hasMemory:e>2};let s=n?.invocation_params?.messages;if(Array.isArray(s))return{turns:s.length,hasMemory:s.length>2};let o=(JSON.stringify(t).match(/"role"\s*:/g)||[]).length;return o>0?{turns:o,hasMemory:o>2}:{turns:t.length,hasMemory:!1}}detectAgentFromPrompts(t,n){try{let e=JSON.stringify(t)+JSON.stringify(n?.invocation_params?.messages??[])+JSON.stringify(n?._zespan_structured_messages??[]),s=[/you are (?:a |an |the )?([A-Z][A-Za-z]*(?:[_\s][A-Z][A-Za-z]*){0,3}(?:Agent|Bot|Assistant|AI|Specialist|Expert|Analyzer|Manager|Coordinator))/i,/act as (?:a |an |the )?([A-Z][A-Za-z]*(?:[_\s][A-Z][A-Za-z]*){0,3}(?:Agent|Bot|Assistant|AI))/i,/agent[_\s]?name["'\s:=]+["']?([A-Za-z][A-Za-z_\-]+)/i,/role["'\s:=]+["']?([A-Za-z]+(?:Agent|Bot|Assistant))/i,/name["'\s:=]+["']?([A-Za-z]+Agent)/i];for(let r of s){let o=e.match(r);if(o?.[1])return o[1].trim().replace(/[\s]+/g,"_")}return}catch{return}}detectPlanning(t,n){try{let e=t+" "+JSON.stringify(n);if(![/\bstep\s+\d+[:\s]/i,/\bfirst,?\s+(?:I|we|let)/i,/\bthen,?\s+(?:I|we)/i,/\bfinally,?\s+(?:I|we)/i,/\blet me think/i,/\blet'?s break (?:this |it )?down/i,/\bthink(?:ing)? step by step/i,/\bmy (?:plan|approach|strategy)/i,/\bI'?ll start by/i,/<thinking>/i,/<scratchpad>/i,/\bchain.of.thought/i].some(l=>l.test(e)))return{detected:!1};let o=/(?:step\s+\d+[:\s]|^\d+[\.\)]\s+)(.+)/gim,c=[],i;for(;(i=o.exec(t))!==null&&(i[1]&&c.push(i[1].trim().slice(0,200)),!(c.length>=10)););return{detected:!0,steps:c.length>0?c:void 0}}catch{return{detected:!1}}}extractCompletion(t){let n=t?.generations;if(!Array.isArray(n)||n.length===0)return"";let e=n[0];if(!Array.isArray(e)||e.length===0)return"";let s=e[0];if(typeof s?.text=="string")return s.text;let r=s?.message?.content;return typeof r=="string"?r:Array.isArray(r)?r.map(o=>typeof o?.text=="string"?o.text:"").filter(Boolean).join(`
|
|
9
|
+
`):""}async handleChatModelStart(t,n,e,s,r,o,c,i){try{let l=n.map(d=>d.map(m=>{let _=m._getType?.()??m.role??"unknown",y=typeof m.content=="string"?m.content:JSON.stringify(m.content);return`${_}: ${y}`}).join(`
|
|
10
|
+
`)),u=n[0]?.map(d=>({role:d._getType?.()??d.role??"unknown",content:typeof d.content=="string"?d.content:JSON.stringify(d.content),tool_calls:d.tool_calls??d.additional_kwargs?.tool_calls}));await this.handleLLMStart(t,l,e,s,{...r,_zespan_structured_messages:u,_zespan_message_count:n[0]?.length??0})}catch{}}async handleLLMStart(t,n,e,s,r){try{let o=G(),c=D(),i=s?this.runs.get(s):void 0,l=p();le(l);let u=i?.traceId??o?.traceId??p(),d=r?.invocation_params?.model??r?.invocation_params?.model_name??t.kwargs?.model??t.kwargs?.modelName??t.id?.[t.id.length-1]??"unknown",m=r?.invocation_params?.stream===!0,_=i?.spanKind==="agent"?i.agentId:void 0,y=i?.spanKind==="agent"?i.agentName:void 0;if(!y){let h=this.detectAgentFromPrompts(n,r);h&&(y=h,_=_??p())}let x=r?.invocation_params?.tools??r?.invocation_params?.functions??[],b=x.map(h=>h.function?.name??h.name??"unknown").filter(Boolean),k=this.detectConversationInfo(n,r),I=await Z(c,this.guardrailSettings,{text:JSON.stringify(n??[]),phase:"pre",traceId:u,spanId:l,model:d,operation:"chat"});this.runs.set(e,{t0:performance.now(),traceId:u,spanId:l,parentSpanId:i?.spanId??o?.spanId,model:d,agentId:_,agentName:y,spanKind:"llm",isStreaming:m,parent:o,prompts:n,structuredMessages:r?._zespan_structured_messages,invocationParams:r?.invocation_params,preGuardrailResult:I,toolsAvailable:b,toolSchemas:c.storePrompts?x:void 0,conversationTurns:k.turns,hasMemory:k.hasMemory})}catch(o){if(o instanceof V)throw o}}async handleLLMNewToken(t,n,e){try{let s=this.runs.get(e);if(!s||s.ttft!==void 0)return;s.ttft=performance.now()-s.t0;let r=D();r.enqueue({event_id:p(),project_id:"",trace_id:s.traceId,span_id:s.spanId,parent_span_id:s.parentSpanId,timestamp:new Date().toISOString(),provider:this.detectProvider(s.model),model:s.model,operation:"chat",latency_ms:s.ttft,ttft_ms:s.ttft,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",span_kind:"llm",agent_id:s.agentId,agent_name:s.agentName,prompt_hash:qt(JSON.stringify(s.prompts)),...r.storePrompts?(()=>{let{systemPrompt:o,conversationMessages:c}=Y(s.structuredMessages??[]);return{prompt_text:r.redactForStorage(c.length?c:s.prompts),...o?{system_prompt:r.redactForStorage(o)}:{}}})():{},metadata:{invocation_params:s.invocationParams??null},user_id:s.parent?.userId,session_id:s.parent?.sessionId,tags:s.parent?.tags,environment:r.environment,sdk_version:C,finish_reason:"stream_start"})}catch{}}async handleLLMEnd(t,n){try{let e=this.runs.get(n);if(!e)return;te(e.spanId),this.runs.delete(n);let s=D(),r=t.llmOutput?.tokenUsage?.promptTokens??t.llmOutput?.usageMetadata?.promptTokenCount??0,o=t.llmOutput?.tokenUsage?.completionTokens??t.llmOutput?.usageMetadata?.candidatesTokenCount??0,c=performance.now()-e.t0,i=this.extractCompletion(t),l=this.extractToolCalls(t),u=t.llmOutput?.tokenUsage?.reasoningTokens??t.llmOutput?.usage?.completion_tokens_details?.reasoning_tokens??t.llmOutput?.reasoningTokens??t.llmOutput?.usageMetadata?.thoughtsTokenCount??0,d=t.llmOutput?.tokenUsage?.cachedTokens??t.llmOutput?.usage?.prompt_tokens_details?.cached_tokens??t.llmOutput?.usageMetadata?.cachedContentTokenCount??0,m=this.detectPlanning(i,e.prompts),_=t.generations?.[0]?.[0]?.generationInfo?.finish_reason??t.generations?.[0]?.[0]?.generationInfo?.finishReason??t.llmOutput?.finish_reason??(e.isStreaming?"stream_end":"stop"),y=await Z(s,this.guardrailSettings,{text:i,phase:"post",traceId:e.traceId,spanId:e.spanId,model:e.model,operation:"chat"}),x=ne(i,y);s.enqueue({event_id:p(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:this.detectProvider(e.model),model:e.model,operation:"chat",latency_ms:c,ttft_ms:e.ttft,stream:e.isStreaming??!1,input_tokens:r,output_tokens:o,status:"success",span_kind:"llm",agent_id:e.agentId,agent_name:e.agentName,finish_reason:_,prompt_hash:qt(JSON.stringify(e.prompts)),tools_defined:e.toolsAvailable?.length?e.toolsAvailable:void 0,tools_used:l.length>0?l.map(b=>b.name):void 0,tool_calls:l.length>0?l:void 0,conversation_turns:e.conversationTurns,has_memory:e.hasMemory,reasoning_tokens:u>0?u:void 0,has_reasoning:u>0?!0:void 0,cached_tokens:d>0?d:void 0,has_planning:m.detected||void 0,planning_steps:m.steps,...s.storePrompts?(()=>{let{systemPrompt:b,conversationMessages:k}=Y(e.structuredMessages??[]);return{prompt_text:s.redactForStorage(k.length?k:e.prompts),completion_text:s.redactForStorage(x),...b?{system_prompt:s.redactForStorage(b)}:{}}})():{},metadata:{usage:t?.llmOutput?.tokenUsage??null,llm_output:t?.llmOutput??null,invocation_params:e.invocationParams??null,guardrails:{pre:e.preGuardrailResult?{matched:e.preGuardrailResult.results.length,modified:!!e.preGuardrailResult.modifiedText}:null,post:y?{matched:y.results.length,modified:!!y.modifiedText}:null}},user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:C})}catch{}}async handleLLMError(t,n){try{let e=this.runs.get(n);if(!e)return;te(e.spanId),this.runs.delete(n);let s=D();s.enqueue({event_id:p(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:this.detectProvider(e.model),model:e.model,operation:"chat",latency_ms:performance.now()-e.t0,ttft_ms:e.ttft,stream:e.isStreaming??!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",span_kind:"llm",agent_id:e.agentId,agent_name:e.agentName,error_code:t instanceof V?"guardrail_blocked":void 0,error_message:String(t.message).slice(0,1e3),user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:C})}catch{}}async handleChainStart(t,n,e,s){try{let r=G(),o=s?this.runs.get(s):void 0,c=p(),i=o?.traceId??r?.traceId??p(),l=Array.isArray(t?.id)?t.id:[],u=l.some(h=>Kt.includes(h)),d=!u&&l.some(h=>Lt.includes(h)),m=u?"agent":d?"tool":"general",_=u?p():void 0,y=u?l.find(h=>Kt.includes(h))??void 0:void 0,x=d?t.name??t.kwargs?.name??l.find(h=>Lt.includes(h)):void 0,b=d?JSON.stringify(n):void 0,k=o?.spanKind==="agent"?o.agentId:void 0,I=o?.spanKind==="agent"?o.agentName:void 0;this.runs.set(e,{t0:performance.now(),traceId:i,spanId:c,parentSpanId:o?.spanId??r?.spanId,model:"",agentId:_??o?.agentId,agentName:y??o?.agentName,spanKind:m,toolName:x,toolInput:b,prompts:[],parent:r,delegatedFromAgentId:k,delegatedFromAgentName:I})}catch{}}async handleChainEnd(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=D();if(e.spanKind==="tool"){let r=t?.output??t,o=typeof r=="string"?r:JSON.stringify(r);s.enqueue({event_id:p(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:`tool.${e.toolName??"unknown"}`,latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"success",span_kind:"tool",agent_id:e.agentId,agent_name:e.agentName,tools_used:e.toolName?[e.toolName]:[],tool_call_id:e.spanId,tool_call_args:e.toolInput?JSON.parse(e.toolInput):void 0,tool_call_result:o.slice(0,10240),stream:!1,finish_reason:"stop",user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:C})}else s.enqueue({event_id:p(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:"chain",latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"success",span_kind:e.spanKind,agent_id:e.agentId,agent_name:e.agentName,stream:!1,finish_reason:"stop",delegation_from_agent_id:e.delegatedFromAgentId,delegation_from_agent_name:e.delegatedFromAgentName,user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:C})}catch{}}async handleChainError(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=D();s.enqueue({event_id:p(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:e.spanKind==="tool"?`tool.${e.toolName??"unknown"}`:"chain",latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"error",span_kind:e.spanKind,agent_id:e.agentId,agent_name:e.agentName,tools_used:e.spanKind==="tool"&&e.toolName?[e.toolName]:void 0,tool_call_id:e.spanKind==="tool"?e.spanId:void 0,tool_call_args:e.spanKind==="tool"&&e.toolInput?JSON.parse(e.toolInput):void 0,error_message:String(t?.message??t).slice(0,1e3),stream:!1,delegation_from_agent_id:e.delegatedFromAgentId,delegation_from_agent_name:e.delegatedFromAgentName,user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:C})}catch{}}async handleToolStart(t,n,e,s){try{let r=G(),o=s?this.runs.get(s):void 0,c=p(),i=o?.traceId??r?.traceId??p(),l=t.id?.[t.id.length-1]??"unknown_tool",u=t.name||t.kwargs?.name||l;this.runs.set(e,{t0:performance.now(),traceId:i,spanId:c,parentSpanId:o?.spanId??r?.spanId,model:"",agentId:o?.agentId,agentName:o?.agentName,spanKind:"tool",toolName:u,toolInput:n,prompts:[],parent:r})}catch{}}async handleToolEnd(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=D(),o=(typeof t=="string"?t:JSON.stringify(t)).slice(0,10240);s.enqueue({event_id:p(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:`tool.${e.toolName??"unknown"}`,latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"success",span_kind:"tool",agent_id:e.agentId,agent_name:e.agentName,tools_used:e.toolName?[e.toolName]:[],tool_call_id:e.spanId,tool_call_args:e.toolInput?{input:e.toolInput}:void 0,tool_call_result:o,stream:!1,finish_reason:"stop",user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:C})}catch{}}async handleToolError(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=D();s.enqueue({event_id:p(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:`tool.${e.toolName??"unknown"}`,latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"error",span_kind:"tool",agent_id:e.agentId,agent_name:e.agentName,tools_used:e.toolName?[e.toolName]:[],error_message:String(t?.message??t).slice(0,1e3),stream:!1,user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:C})}catch{}}async handleAgentAction(t,n){try{let e=this.runs.get(n),s=G(),r=D(),o=e?.traceId??s?.traceId??p(),c=p(),i=[t?.tool??"",typeof t?.toolInput=="string"?t.toolInput:JSON.stringify(t?.toolInput??""),t?.log??""].join(" | ");r.enqueue({event_id:p(),project_id:"",trace_id:o,span_id:c,parent_span_id:e?.spanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:"planning",latency_ms:0,input_tokens:0,output_tokens:0,status:"success",span_kind:"planning",agent_id:e?.agentId,agent_name:e?.agentName,planning_steps:[i],...t?.log?{decision_rationale:t.log}:{},stream:!1,user_id:e?.parent?.userId??s?.userId,session_id:e?.parent?.sessionId??s?.sessionId,tags:e?.parent?.tags??s?.tags,environment:r.environment,sdk_version:C})}catch{}}async handleRetrieverStart(t,n,e,s){try{let r=G(),o=s?this.runs.get(s):void 0,c=p(),i=o?.traceId??r?.traceId??p();this.runs.set(e,{t0:performance.now(),traceId:i,spanId:c,parentSpanId:o?.spanId??r?.spanId,model:"",agentId:o?.agentId,agentName:o?.agentName,spanKind:"retriever",query:n,prompts:[],parent:r})}catch{}}async handleRetrieverEnd(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=D();s.enqueue({event_id:p(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:"retriever",latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"success",span_kind:"retriever",agent_id:e.agentId,agent_name:e.agentName,stream:!1,finish_reason:"stop",metadata:{"zespan.rag.contexts":Array.isArray(t)?t:[]},user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:C})}catch{}}async handleRetrieverError(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=D();s.enqueue({event_id:p(),project_id:"",trace_id:e.traceId,span_id:e.spanId,parent_span_id:e.parentSpanId,timestamp:new Date().toISOString(),provider:"langchain",model:"",operation:"retriever",latency_ms:performance.now()-e.t0,input_tokens:0,output_tokens:0,status:"error",span_kind:"retriever",agent_id:e.agentId,agent_name:e.agentName,error_message:String(t?.message??t).slice(0,1e3),stream:!1,user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:C})}catch{}}};function zt(a){let t=a?.model;return typeof t=="string"?t:t&&typeof t=="object"?t.model??t.name??"gemini-2.5-flash":"gemini-2.5-flash"}function Wn(a){let t=a??{};return{input:Number(t.promptTokenCount??t.inputTokenCount??0),output:Number(t.candidatesTokenCount??t.outputTokenCount??0),cached:Number(t.cachedContentTokenCount??0),reasoning:Number(t.thoughtsTokenCount??t.reasoningTokenCount??0)}}var Qe=class{agentStates=new Map;pendingModelCalls=new Map;toolCallStates=new Map;agentKey(t,n){return`${t}:${n}`}getParentAgentState(t,n){if(n)return this.agentStates.get(this.agentKey(t,n))}beforeAgentCallback=t=>{try{let n=t.invocationId,e=t.agentName,s=t.invocationContext?.agent?.parentAgent?.name,r=G(),o=this.getParentAgentState(n,s),c=o?.traceId??r?.traceId??p(),i=p(),l=o?.spanId??r?.spanId,u=p(),d=zt(t.invocationContext?.agent),m=t.userId||r?.userId,_=t.sessionId||r?.sessionId;this.agentStates.set(this.agentKey(n,e),{t0:performance.now(),traceId:c,spanId:i,parentSpanId:l,agentId:u,agentName:e,model:d,userId:m,sessionId:_,parentAgentId:o?.agentId,parentAgentName:s})}catch{}};afterAgentCallback=t=>{try{let n=t.invocationId,e=t.agentName,s=this.agentKey(n,e),r=this.agentStates.get(s);if(!r)return;this.agentStates.delete(s);let o=!r.parentAgentName,c=D();c.enqueue({event_id:p(),project_id:"",trace_id:r.traceId,span_id:r.spanId,parent_span_id:r.parentSpanId,timestamp:new Date().toISOString(),provider:"google-adk",model:r.model,operation:`agent.${e}`,span_kind:"agent",agent_id:r.agentId,agent_name:e,agent_role:o?"coordinator":"specialist",agent_framework:"google-adk",parent_agent_id:r.parentAgentId,delegation_from_agent_id:r.parentAgentId,delegation_from_agent_name:r.parentAgentName,latency_ms:performance.now()-r.t0,stream:!1,input_tokens:0,output_tokens:0,status:"success",user_id:r.userId,session_id:r.sessionId,environment:c.environment,sdk_version:C})}catch{}};beforeModelCallback=({context:t,request:n})=>{try{let e=t.invocationId,s=t.agentName,r=this.agentStates.get(this.agentKey(e,s)),o=r?.traceId??p(),c=zt(t.invocationContext?.agent),i=n?.config?.systemInstruction,l;if(i){let m=typeof i=="string"?i:(i?.parts??[]).filter(_=>_?.text).map(_=>_.text).join("");m&&(l=m)}let u,d=n?.contents??[];if(d.length){let m=d.map(_=>{let y=_?.role??"user",x=(_?.parts??[]).filter(b=>typeof b?.text=="string"&&!b.thought).map(b=>b.text).join("");return x?`${y}: ${x}`:""}).filter(Boolean).join(`
|
|
11
|
+
`);m&&(u=m)}this.pendingModelCalls.set(this.agentKey(e,s),{t0:performance.now(),spanId:p(),traceId:o,parentSpanId:r?.spanId??"",model:c,agentId:r?.agentId,agentName:s,userId:r?.userId??t.userId,sessionId:r?.sessionId??t.sessionId,promptText:u,systemPrompt:l})}catch{}};afterModelCallback=({context:t,response:n})=>{try{let e=t.invocationId,s=t.agentName,r=this.agentKey(e,s),o=this.pendingModelCalls.get(r);if(!o)return;this.pendingModelCalls.delete(r);let{input:c,output:i,cached:l,reasoning:u}=Wn(n?.usageMetadata),d=n?.content?.parts??[],m=d.filter(x=>typeof x?.text=="string"&&!x.thought).map(x=>x.text).join(""),_=d.filter(x=>x?.functionCall?.name).map(x=>x.functionCall.name),y=D();y.enqueue({event_id:p(),project_id:"",trace_id:o.traceId,span_id:o.spanId,parent_span_id:o.parentSpanId,timestamp:new Date().toISOString(),provider:"google",model:o.model,operation:"chat",span_kind:"llm",agent_id:o.agentId,agent_name:o.agentName,agent_framework:"google-adk",latency_ms:performance.now()-o.t0,stream:!1,input_tokens:c,output_tokens:i,cached_tokens:l||void 0,reasoning_tokens:u||void 0,finish_reason:n?.finishReason??"stop",tools_used:_.length?_:void 0,status:n?.errorCode?"error":"success",error_message:n?.errorMessage,...y.storePrompts&&o.systemPrompt?{system_prompt:y.redactForStorage(o.systemPrompt)}:{},...y.storePrompts&&o.promptText?{prompt_text:y.redactForStorage(o.promptText)}:{},...y.storePrompts&&m?{completion_text:y.redactForStorage(m)}:{},user_id:o.userId,session_id:o.sessionId,environment:y.environment,sdk_version:C})}catch{}};beforeToolCallback=({tool:t,args:n,context:e})=>{try{let s=e.invocationId,r=e.agentName,o=e.functionCallId??p(),c=this.agentStates.get(this.agentKey(s,r));this.toolCallStates.set(o,{t0:performance.now(),spanId:p(),traceId:c?.traceId??p(),parentSpanId:c?.spanId??"",toolName:t?.name??"unknown_tool",args:n??{},agentId:c?.agentId,agentName:r,userId:c?.userId??e.userId,sessionId:c?.sessionId??e.sessionId})}catch{}};afterToolCallback=({tool:t,args:n,context:e,response:s})=>{try{let r=e.functionCallId??"",o=this.toolCallStates.get(r);if(!o)return;this.toolCallStates.delete(r);let c=D();c.enqueue({event_id:p(),project_id:"",trace_id:o.traceId,span_id:o.spanId,parent_span_id:o.parentSpanId,timestamp:new Date().toISOString(),provider:"google-adk",model:"",operation:`tool.${o.toolName}`,span_kind:"tool",agent_id:o.agentId,agent_name:o.agentName,agent_framework:"google-adk",latency_ms:performance.now()-o.t0,stream:!1,input_tokens:0,output_tokens:0,status:"success",tools_used:[o.toolName],tool_call_id:r,tool_call_args:o.args,tool_call_result:JSON.stringify(s??{}).slice(0,10240),user_id:o.userId,session_id:o.sessionId,environment:c.environment,sdk_version:C})}catch{}};get callbacks(){return{beforeAgentCallback:this.beforeAgentCallback,afterAgentCallback:this.afterAgentCallback,beforeModelCallback:this.beforeModelCallback,afterModelCallback:this.afterModelCallback,beforeToolCallback:this.beforeToolCallback,afterToolCallback:this.afterToolCallback}}};import{createHash as Jn}from"crypto";var _t=a=>Jn("sha256").update(typeof a=="string"?a:JSON.stringify(a)).digest("hex");function Zn(a){return a?.model??a?.config?.model??a?.llmConfig?.model??a?.generativeModel?.model??"gemini-2.5-flash"}function $t(a){let t=a?.usageMetadata??a?.usage_metadata??{},n=a?.usage??{};return{input:Number(t.inputTokenCount??t.promptTokenCount??n.input_tokens??n.prompt_tokens??0),output:Number(t.outputTokenCount??t.candidatesTokenCount??n.output_tokens??n.completion_tokens??0),cached:Number(t.cachedContentTokenCount??n.cached_tokens??0),reasoning:Number(t.reasoningTokenCount??t.thinking_token_count??n.reasoning_tokens??0)}}function Un(a){if(typeof a?.text=="string")return a.text;if(typeof a?.output=="string")return a.output;let n=(a?.candidates?.[0]?.content?.parts??a?.content?.parts??[]).map(e=>typeof e?.text=="string"?e.text:"").filter(Boolean).join("");return n||(a&&typeof a=="object"?JSON.stringify(a):"")}function Vn(a){let t=[];try{let n=a?.candidates?.[0]?.content?.parts??a?.content?.parts??[];for(let e of n)e?.functionCall?.name&&t.push({name:e.functionCall.name,args:e.functionCall.args??{}});for(let e of a?.toolCalls??[])e?.name&&t.push({name:e.name,args:e.args??e.arguments??{}})}catch{}return t}function Wt(a){if(a instanceof V)return"guardrail_blocked";let t=String(a?.message??"").toLowerCase();if(/rate.?limit|quota|429/.test(t))return"rate_limited";if(/context.?length|too.?long|max.?token/.test(t))return"context_length";if(/timeout|timed?.?out/.test(t))return"timeout";if(/safety|blocked|harm/.test(t))return"content_filter"}function Xe(a,t={}){let n=a;if(typeof n?.run!="function")return a;let e=n.run.bind(n),s=X(t.guardrails,"agent"),r=t.agentName??n.name??"ADKAgent",o=t.agentRole??(n.subAgents?.length?"coordinator":"specialist"),c=Zn(n);return t.traceSubAgents!==!1&&Array.isArray(n.subAgents)&&(n.subAgents=n.subAgents.map(i=>Xe(i,{...t,agentRole:"specialist",traceSubAgents:!0}))),n.run=async(i,l)=>{let u=D(),d=performance.now(),m=G(),_=p(),y=p(),x=m?.traceId??p(),b=t.sessionId??m?.sessionId,k=typeof i=="string"?i:JSON.stringify(i??"");return W({traceId:x,spanId:y,userId:m?.userId,sessionId:b,tags:m?.tags,agentId:_,agentName:r,agentRole:o,agentFramework:"google-adk",parentAgentId:m?.agentId},async()=>{try{let I=await Z(u,s,{text:k,phase:"pre",traceId:x,spanId:y,model:c,operation:"agent"}),h=await e(i,l),{input:T,output:g,cached:A,reasoning:f}=$t(h),v=Un(h),w=Vn(h),j=h?.candidates?.[0]?.finishReason??h?.finish_reason,O=await Z(u,s,{text:v,phase:"post",traceId:x,spanId:y,model:c,operation:"agent"}),S=ne(v,O);return u.enqueue({event_id:p(),project_id:"",trace_id:x,span_id:y,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:c,operation:"agent",span_kind:"agent",agent_id:_,agent_name:r,agent_role:o,agent_framework:"google-adk",parent_agent_id:m?.agentId,delegation_from_agent_id:m?.agentId,delegation_from_agent_name:m?.agentName,latency_ms:performance.now()-d,stream:!1,input_tokens:T,output_tokens:g,cached_tokens:A||void 0,reasoning_tokens:f||void 0,status:"success",finish_reason:j,tools_used:w.length?w.map(M=>M.name):void 0,tool_call_args:w.length?Object.fromEntries(w.map(M=>[M.name,M.args])):void 0,prompt_hash:_t(k),...u.storePrompts?{prompt_text:u.redactForStorage(k),completion_text:u.redactForStorage(S)}:{},...w.length>0&&v?{decision_rationale:v}:{},user_id:m?.userId,session_id:b,tags:m?.tags,environment:u.environment,sdk_version:C,metadata:{guardrails:{pre:I?{matched:I.results.length,modified:!!I.modifiedText}:null,post:O?{matched:O.results.length,modified:!!O.modifiedText}:null}}}),h}catch(I){throw u.enqueue({event_id:p(),project_id:"",trace_id:x,span_id:y,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:c,operation:"agent",span_kind:"agent",agent_id:_,agent_name:r,agent_role:o,agent_framework:"google-adk",parent_agent_id:m?.agentId,latency_ms:performance.now()-d,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_code:Wt(I),error_message:String(I?.message??I).slice(0,500),user_id:m?.userId,session_id:b,tags:m?.tags,environment:u.environment,sdk_version:C}),I}})},a}function ft(a,t={}){let n=a;if(typeof n?.runEphemeral=="function"){let e=n.runEphemeral.bind(n);n.runEphemeral=s=>Bt(e(s),s,t)}if(typeof n?.run=="function"){let e=n.run.bind(n);n.run=s=>Bt(e(s),s,t)}return a}async function*Bt(a,t,n){let e=D(),s=G(),r=s?.traceId??p(),o=p(),c=p(),i=performance.now(),l=n.agentName??"ADKAgent",u=n.agentRole??"coordinator",d=n.model??"gemini-2.5-flash",m=t?.userId??s?.userId,_=n.sessionId??t?.sessionId??s?.sessionId,y=(t?.newMessage?.parts??[]).map(f=>typeof f?.text=="string"?f.text:"").filter(Boolean).join(""),x=new Map,b=new Map,k=new Set,I="",h=0,T=0,g=0,A=0;le(o);try{yield*W({traceId:r,spanId:o,userId:m,sessionId:_,tags:s?.tags,agentId:c,agentName:l,agentRole:u,agentFramework:"google-adk",parentAgentId:s?.agentId},async function*(){for await(let f of a){h++;let v=f.author??l;if(v!=="user"){v!==l&&!k.has(v)&&e.enqueue({event_id:p(),project_id:"",trace_id:r,span_id:p(),parent_span_id:o,timestamp:new Date().toISOString(),provider:"google-adk",model:d,operation:`handoff.${l}\u2192${v}`,span_kind:"handoff",agent_name:v,agent_role:"specialist",agent_framework:"google-adk",delegation_from_agent_id:c,delegation_from_agent_name:l,delegation_reason:`delegated to ${v}`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",user_id:m,session_id:_,tags:s?.tags,environment:e.environment,sdk_version:C}),k.add(v),x.has(v)||(x.set(v,{spanId:p(),agentId:p(),t0:performance.now()}),b.set(v,{input:0,output:0,cached:0,reasoning:0}));let w=x.get(v),j=$t(f);if(j.input>0||j.output>0){let N=b.get(v);N.input+=j.input,N.output+=j.output,N.cached+=j.cached,N.reasoning+=j.reasoning,T+=j.input,g+=j.output,A+=j.cached}let O=f.content?.parts??[];for(let N of O){let q=N?.functionCall;if(!q?.name)continue;let R=p();e.enqueue({event_id:p(),project_id:"",trace_id:r,span_id:R,parent_span_id:w.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:d,operation:`tool.${q.name}`,span_kind:"tool",agent_id:w.agentId,agent_name:v,agent_framework:"google-adk",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",tools_used:[q.name],tool_call_id:R,tool_call_args:q.args??{},user_id:m,session_id:_,tags:s?.tags,environment:e.environment,sdk_version:C})}let S=O.filter(N=>typeof N?.text=="string"&&!N.thought).map(N=>N.text).join("");if(S&&(I+=S),typeof f.isFinalResponse=="function"?f.isFinalResponse():!!f.isFinalResponse||f.actions?.stateDelta!==void 0||f.turnComplete===!0){let N=O.map(F=>F?.functionCall?.name).filter(Boolean),q=O.filter(F=>typeof F?.text=="string"&&!F.thought).map(F=>F.text).join(""),R=b.get(v)??{input:0,output:0,cached:0,reasoning:0};e.enqueue({event_id:p(),project_id:"",trace_id:r,span_id:w.spanId,parent_span_id:o,timestamp:new Date().toISOString(),provider:"google-adk",model:d,operation:`agent.${v}`,span_kind:"agent",agent_id:w.agentId,agent_name:v,agent_role:v===l?u:"specialist",agent_framework:"google-adk",parent_agent_id:c,latency_ms:performance.now()-w.t0,stream:!1,input_tokens:R.input,output_tokens:R.output,cached_tokens:R.cached||void 0,reasoning_tokens:R.reasoning||void 0,status:"success",tools_used:N.length?N:void 0,prompt_hash:y?_t(y):void 0,...e.storePrompts&&S?{completion_text:e.redactForStorage(S)}:{},...N.length>0&&q?{decision_rationale:q}:{},user_id:m,session_id:_,tags:s?.tags,environment:e.environment,sdk_version:C}),x.delete(v),b.delete(v)}}yield f}}),e.enqueue({event_id:p(),project_id:"",trace_id:r,span_id:o,parent_span_id:s?.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:d,operation:"agent.run",span_kind:"agent",agent_id:c,agent_name:l,agent_role:u,agent_framework:"google-adk",parent_agent_id:s?.agentId,latency_ms:performance.now()-i,stream:!1,input_tokens:T,output_tokens:g,cached_tokens:A||void 0,status:"success",prompt_hash:y?_t(y):void 0,...e.storePrompts?{prompt_text:e.redactForStorage(y),completion_text:e.redactForStorage(I)}:{},user_id:m,session_id:_,tags:s?.tags,environment:e.environment,sdk_version:C,metadata:{eventCount:h,agentsThatRan:[...k]}}),te(o)}catch(f){throw te(o),e.enqueue({event_id:p(),project_id:"",trace_id:r,span_id:o,parent_span_id:s?.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:d,operation:"agent.run",span_kind:"agent",agent_id:c,agent_name:l,agent_role:u,agent_framework:"google-adk",parent_agent_id:s?.agentId,latency_ms:performance.now()-i,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_code:Wt(f),error_message:String(f?.message??f).slice(0,500),user_id:m,session_id:_,tags:s?.tags,environment:e.environment,sdk_version:C}),f}}function Jt(a){let{coordinator:t,runner:n,agentName:e,agentRole:s,model:r,...o}=a,c=Xe(t,{...o,agentName:e??t?.name,agentRole:s??"coordinator",traceSubAgents:!0}),i=ft(n,{agentName:e??t?.name,agentRole:s??"coordinator",model:r});return{coordinator:c,runner:i}}var Zt=Xe;import{NodeTracerProvider as Hn}from"@opentelemetry/sdk-trace-node";import{ExportResultCode as Ut}from"@opentelemetry/core";function Qn(){try{return D()}catch{return null}}var yt=class{export(t,n){let e=Qn();if(!e){n({code:Ut.SUCCESS});return}try{for(let s of t){let{name:r}=s;if(!r.startsWith("ai."))continue;let o=s.attributes,c=G(),i=s.spanContext().spanId,l=s.spanContext().traceId,u=s.duration[0]*1e3+s.duration[1]/1e6,d=new Date(s.startTime[0]*1e3+s.startTime[1]/1e6).toISOString(),m=s.status.code===2,_={event_id:p(),project_id:"",trace_id:c?.traceId??l,span_id:i,parent_span_id:c?.spanId,timestamp:d,provider:String(o["ai.model.provider"]??"unknown"),model:String(o["ai.model.id"]??"unknown"),latency_ms:u,stream:!1,status:m?"error":"success",environment:e.environment,sdk_version:C};r==="ai.generateText"||r==="ai.generateText.doGenerate"?(te(i),e.enqueue({..._,operation:"generateText",input_tokens:Number(o["ai.usage.promptTokens"]??0),output_tokens:Number(o["ai.usage.completionTokens"]??0),finish_reason:String(o["ai.response.finishReason"]??""),span_kind:"llm",...e.storePrompts&&o["ai.prompt"]?{prompt_text:e.redactForStorage(String(o["ai.prompt"]))}:{},...e.storePrompts&&o["ai.response.text"]?{completion_text:e.redactForStorage(String(o["ai.response.text"]))}:{}})):r==="ai.streamText"||r==="ai.streamText.doStream"?(te(i),e.enqueue({..._,operation:"streamText",stream:!0,input_tokens:Number(o["ai.usage.promptTokens"]??0),output_tokens:Number(o["ai.usage.completionTokens"]??0),finish_reason:String(o["ai.response.finishReason"]??""),ttft_ms:o["ai.response.msToFirstChunk"]!==void 0?Number(o["ai.response.msToFirstChunk"]):void 0,span_kind:"llm"})):r==="ai.generateObject"||r==="ai.generateObject.doGenerate"?(te(i),e.enqueue({..._,operation:"generateObject",input_tokens:Number(o["ai.usage.promptTokens"]??0),output_tokens:Number(o["ai.usage.completionTokens"]??0),span_kind:"general"})):r==="ai.toolCall"?e.enqueue({..._,provider:"vercel-ai",model:"tool",operation:`tool.${String(o["ai.toolCall.name"]??"unknown")}`,input_tokens:0,output_tokens:0,span_kind:"tool",tools_used:o["ai.toolCall.name"]?[String(o["ai.toolCall.name"])]:[]}):(r==="ai.embed"||r==="ai.embedMany")&&e.enqueue({..._,operation:r==="ai.embed"?"embed":"embedMany",input_tokens:Number(o["ai.usage.tokens"]??0),output_tokens:0,span_kind:"embedding"})}}catch{}n({code:Ut.SUCCESS})}shutdown(){return Promise.resolve()}},kt=class{exporter;constructor(t){this.exporter=t}onStart(t,n){t.name?.startsWith("ai.")&&le(t.spanContext().spanId)}onEnd(t){t.name.startsWith("ai.")&&te(t.spanContext().spanId),this.exporter.export([t],()=>{})}forceFlush(){return Promise.resolve()}shutdown(){return Promise.resolve()}},Ye=null,ht=null;function It(){return Ye||(Ye=new Hn,Ye.addSpanProcessor(new kt(new yt)),ht=Ye.getTracer("zespan-vercel-ai")),{tracer:ht,telemetry:{isEnabled:!0,tracer:ht}}}function Vt(){return It().telemetry}function Oe(){try{return D()}catch{return null}}function Xn(a){let t=a.toLowerCase();return/gpt|o1-|o3-|davinci|turbo|whisper|dall-e/i.test(t)?"openai":/claude/i.test(t)?"anthropic":/gemini|palm|bard/i.test(t)?"google":/llama|codellama/i.test(t)?"meta":/mistral|mixtral/i.test(t)?"mistralai":/deepseek/i.test(t)?"deepseek":/command/i.test(t)?"cohere":/qwen/i.test(t)?"alibaba":"unknown"}var et=class{runs=new Map;guardrailSettings;constructor(t={}){this.guardrailSettings=X(t.guardrails,"chat")}_onLLMStart=async t=>{let n=t.id??p(),e=G(),s=p();le(s);let r=t.detail?.llm?.metadata?.model_name??t.detail?.model??"unknown",o=e?.traceId??p(),c=JSON.stringify(t.detail?.messages??t.detail?.prompt??t.detail?.input??""),i;try{let l=Oe();l&&c&&(i=await Z(l,this.guardrailSettings,{text:c,phase:"pre",traceId:o,spanId:s,model:r,operation:"chat"}))}catch(l){if(l instanceof V)throw l}this.runs.set(n,{t0:performance.now(),traceId:o,spanId:s,parentSpanId:e?.spanId,model:r,preGuardrailResult:i,promptText:c})};_onLLMEnd=async t=>{let n=this.runs.get(t.id??"");if(!n)return;this.runs.delete(t.id),te(n.spanId);let e=Oe();if(!e)return;let s=t.detail?.response??t.detail?.chunk??{},r=s.usage??s.raw?.usage??{},o=(typeof s.text=="string"?s.text:null)??(typeof s.message?.content=="string"?s.message.content:null)??"",c=o,i;if(o)try{i=await Z(e,this.guardrailSettings,{text:o,phase:"post",traceId:n.traceId,spanId:n.spanId,model:n.model,operation:"chat"}),c=ne(o,i)}catch(u){u instanceof V||(i=void 0)}let l=n.preGuardrailResult;try{e.enqueue({event_id:p(),project_id:"",trace_id:n.traceId,span_id:n.spanId,parent_span_id:n.parentSpanId,timestamp:new Date().toISOString(),provider:Xn(n.model),model:n.model,operation:"chat",latency_ms:performance.now()-n.t0,ttft_ms:n.ttft,stream:!1,input_tokens:r.promptTokens??r.prompt_tokens??0,output_tokens:r.completionTokens??r.completion_tokens??0,finish_reason:s.finishReason??s.finish_reason??"",status:"success",span_kind:"llm",agent_id:G()?.agentId,agent_name:G()?.agentName,...e.storePrompts?{prompt_text:e.redactForStorage(n.promptText??""),completion_text:e.redactForStorage(c)}:{},metadata:{guardrails:{pre:l?{matched:l.results.length,modified:!!l.modifiedText}:null,post:i?{matched:i.results.length,modified:!!i.modifiedText}:null}},environment:e.environment,sdk_version:C})}catch{}};_onLLMStream=t=>{let n=this.runs.get(t.id??"");n&&n.ttft===void 0&&(n.ttft=performance.now()-n.t0)};_onToolCall=t=>{let n=Oe();if(!n)return;let e=G(),s=t.detail?.toolCall?.name??t.detail?.name??"";try{n.enqueue({event_id:p(),project_id:"",trace_id:e?.traceId??p(),span_id:p(),parent_span_id:e?.spanId,timestamp:new Date().toISOString(),provider:"llamaindex",model:"",operation:`tool.${s||"unknown"}`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",span_kind:"tool",tools_used:s?[s]:[],agent_id:e?.agentId,agent_name:e?.agentName,environment:n.environment,sdk_version:C})}catch{}};_onToolResult=t=>{let n=Oe();if(!n)return;let e=G(),s=t.detail?.toolCall?.name??t.detail?.name??"",r=t.detail?.result??t.detail?.toolResult??"";try{n.enqueue({event_id:p(),project_id:"",trace_id:e?.traceId??p(),span_id:p(),parent_span_id:e?.spanId,timestamp:new Date().toISOString(),provider:"llamaindex",model:"",operation:`tool.${s||"unknown"}`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",span_kind:"tool",tools_used:s?[s]:[],tool_call_result:String(r).slice(0,10240),agent_id:e?.agentId,agent_name:e?.agentName,environment:n.environment,sdk_version:C})}catch{}};_onAgentStart=t=>{let n=t.id??p(),e=G();this.runs.set(`agent-${n}`,{t0:performance.now(),traceId:e?.traceId??p(),spanId:p(),parentSpanId:e?.spanId,model:"agent",agentName:t.detail?.agentName??e?.agentName??""})};_onAgentEnd=t=>{let n=this.runs.get(`agent-${t.id??""}`);if(!n)return;this.runs.delete(`agent-${t.id}`);let e=Oe();if(e)try{e.enqueue({event_id:p(),project_id:"",trace_id:n.traceId,span_id:n.spanId,parent_span_id:n.parentSpanId,timestamp:new Date().toISOString(),provider:"llamaindex",model:"agent",operation:"agent",latency_ms:performance.now()-n.t0,stream:!1,input_tokens:0,output_tokens:0,status:"success",span_kind:"agent",agent_name:n.agentName??t.detail?.agentName??"",environment:e.environment,sdk_version:C})}catch{}};attach(t){t.on("llm-start",this._onLLMStart),t.on("llm-end",this._onLLMEnd),t.on("llm-stream",this._onLLMStream),t.on("llm-tool-call",this._onToolCall),t.on("llm-tool-result",this._onToolResult),t.on("agent-start",this._onAgentStart),t.on("agent-end",this._onAgentEnd)}detach(t){t.off("llm-start",this._onLLMStart),t.off("llm-end",this._onLLMEnd),t.off("llm-stream",this._onLLMStream),t.off("llm-tool-call",this._onToolCall),t.off("llm-tool-result",this._onToolResult),t.off("agent-start",this._onAgentStart),t.off("agent-end",this._onAgentEnd)}};function Ht(a){let t=D(),n=G(),e=n?.traceId??p(),s=p(),r=performance.now(),o=n?.agentId?{agent_id:n.agentId,agent_name:n.agentName,agent_role:n.agentRole,agent_framework:n.agentFramework,parent_agent_id:n.parentAgentId,span_kind:a.span_kind??"general"}:{},c={};return{span:{setEvalScore(l,u){c[l]=u},async end(l){let u=performance.now()-r;try{t.enqueue({event_id:p(),project_id:"",trace_id:e,span_id:s,parent_span_id:n?.spanId,timestamp:new Date().toISOString(),provider:a.provider??"custom",model:a.model??"custom",operation:"custom",latency_ms:u,stream:!1,input_tokens:l.input_tokens??0,output_tokens:l.output_tokens??0,cost_usd:l.cost_usd,status:l.status,error_message:l.error_message?.slice(0,500),environment:t.environment,eval_scores:Object.keys(c).length?c:void 0,sdk_version:C,...o})}catch{}}},run:l=>W({traceId:e,spanId:s,userId:n?.userId,sessionId:n?.sessionId,tags:n?.tags,agentId:n?.agentId,agentName:n?.agentName,agentRole:n?.agentRole,agentFramework:n?.agentFramework,parentAgentId:n?.parentAgentId,conversationTurn:n?.conversationTurn,availableTools:n?.availableTools},l)}}var tt=class{agentId;name;role;framework;tools;constructor(t,n){this.agentId=t,this.name=n.name,this.role=n.role??"specialist",this.framework=n.framework??"custom",this.tools=n.tools??[]}logPlan(t){let n=D(),e=G(),s={event_id:p(),project_id:"",trace_id:e?.traceId??p(),span_id:p(),parent_span_id:e?.spanId,timestamp:new Date().toISOString(),provider:"agent",model:"",operation:`${this.name}.plan`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",environment:n.environment,sdk_version:C,agent_id:this.agentId,agent_name:this.name,agent_role:this.role,agent_framework:this.framework,span_kind:"planning",planning_steps:t,user_id:e?.userId,session_id:e?.sessionId,tags:e?.tags};n.enqueue(s)}async traceTool(t,n,e){let s=D(),r=G(),o=p(),c=r?.traceId??p(),i=performance.now(),l=this.tools.find(u=>u.name===t);try{let u=await W({traceId:c,spanId:o,userId:r?.userId,sessionId:r?.sessionId,tags:r?.tags,agentId:this.agentId,agentName:this.name,agentRole:this.role,agentFramework:this.framework},()=>e()),d=typeof u=="string"?u:JSON.stringify(u),m={event_id:p(),project_id:"",trace_id:c,span_id:o,parent_span_id:r?.spanId,timestamp:new Date().toISOString(),provider:"tool",model:"",operation:`tool.${t}`,latency_ms:performance.now()-i,stream:!1,input_tokens:0,output_tokens:0,status:"success",environment:s.environment,sdk_version:C,agent_id:this.agentId,agent_name:this.name,agent_role:this.role,agent_framework:this.framework,span_kind:"tool",tools_used:[t],tool_call_id:o,tool_call_args:n,tool_call_result:d.slice(0,10240),tool_definitions:l?[l]:void 0,user_id:r?.userId,session_id:r?.sessionId,tags:r?.tags};return s.enqueue(m),u}catch(u){let d=u instanceof Error?u.message:String(u),m={event_id:p(),project_id:"",trace_id:c,span_id:o,parent_span_id:r?.spanId,timestamp:new Date().toISOString(),provider:"tool",model:"",operation:`tool.${t}`,latency_ms:performance.now()-i,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_message:d.slice(0,500),environment:s.environment,sdk_version:C,agent_id:this.agentId,agent_name:this.name,agent_role:this.role,agent_framework:this.framework,span_kind:"tool",tools_used:[t],tool_call_id:o,tool_call_args:n,user_id:r?.userId,session_id:r?.sessionId,tags:r?.tags};throw s.enqueue(m),u}}delegateTo(t,n){let e=D(),s=G(),r={event_id:p(),project_id:"",trace_id:s?.traceId??p(),span_id:p(),parent_span_id:s?.spanId,timestamp:new Date().toISOString(),provider:"agent",model:"",operation:`${this.name}.delegate\u2192${t}`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",environment:e.environment,sdk_version:C,agent_id:this.agentId,agent_name:this.name,agent_role:this.role,agent_framework:this.framework,span_kind:"handoff",delegation_reason:n,user_id:s?.userId,session_id:s?.sessionId,tags:s?.tags};e.enqueue(r)}};function Qt(a,t){let n=D(),e=G(),s=p(),r=e?.traceId??p(),o=p(),c={event_id:p(),project_id:"",trace_id:r,span_id:o,parent_span_id:e?.spanId,timestamp:new Date().toISOString(),provider:"agent",model:"",operation:`agent.${a.name}`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",environment:n.environment,sdk_version:C,agent_id:s,agent_name:a.name,agent_role:a.role??"specialist",agent_framework:a.framework??"custom",parent_agent_id:e?.agentId,span_kind:"agent",tool_definitions:a.tools,delegation_reason:e?.agentId?"delegated":"root",user_id:e?.userId,session_id:e?.sessionId,tags:e?.tags,metadata:a.metadata};return n.enqueue(c),W({traceId:r,spanId:o,userId:e?.userId,sessionId:e?.sessionId,tags:e?.tags,agentId:s,agentName:a.name,agentRole:a.role??"specialist",agentFramework:a.framework??"custom",parentAgentId:e?.agentId,availableTools:a.tools},()=>t(new tt(s,a)))}import{propagation as Yn}from"@opentelemetry/api";var St=class{onStart(t,n){let e=Yn.getBaggage(n);e&&e.getAllEntries().forEach(([s,r])=>{t.setAttribute(`baggage.${s}`,r.value)})}onEnd(){}shutdown(){return Promise.resolve()}forceFlush(){return Promise.resolve()}};import{context as Xt,propagation as Ae}from"@opentelemetry/api";function nt(a,t,n){let e={};t&&(e["agent.delegation.reason"]=t),n&&(e["agent.task"]=n);let s=Xt.active();if(Object.keys(e).length>0){let r=Ae.getBaggage(s)??Ae.createBaggage();for(let[o,c]of Object.entries(e))r=r.setEntry(o,{value:c});s=Ae.setBaggage(s,r)}Ae.inject(s,a)}function st(a){return Ae.extract(Xt.active(),a)}var es={OTEL_EXPORTER_OTLP_ENDPOINT:"http://localhost:4318",OTEL_SERVICE_NAME:"crewai-agent"};function ts(){return`
|
|
8
12
|
pip install openinference-instrumentation-crewai opentelemetry-exporter-otlp-proto-grpc
|
|
9
13
|
|
|
10
14
|
from openinference.instrumentation.crewai import CrewAIInstrumentor
|
|
@@ -17,4 +21,4 @@ provider = TracerProvider()
|
|
|
17
21
|
provider.add_span_processor(BatchSpanProcessor(OTLPSpanExporter(endpoint="http://localhost:4318")))
|
|
18
22
|
trace.set_tracer_provider(provider)
|
|
19
23
|
CrewAIInstrumentor().instrument()
|
|
20
|
-
`.trim()}function
|
|
24
|
+
`.trim()}function ns(a,t){let n={};return nt(n,t),{...a,_zespan_trace:n}}function ss(a){let t=a._zespan_trace??{};return st(t)}function os(a){return{OTEL_EXPORTER_OTLP_ENDPOINT:a,OTEL_EXPORTER_OTLP_PROTOCOL:"http/protobuf",OTEL_SERVICE_NAME:"pydantic-ai-agent",LOGFIRE_SEND_TO_LOGFIRE:"false"}}var pa={init:pt,getClient:D,getZespanClient:D,wrapOpenAI:Be,wrapAnthropic:Pt,wrapGoogle:Ft,wrapOpenRouter:Et,wrapADKAgent:Zt,wrapADKRunner:ft,instrumentADK:Jt,startSpan:Ht,withContext:at,withTrace:it,withAgent:Qt,initOTel:Me,getTracer:fe,createSpan:wt,withSpan:bt,PromptClient:ke,CallbackHandler:He,ADKCallbackHandler:Qe,autopatch:qe,markFrameworkActive:le,markFrameworkInactive:te,instrumentVercelAI:It,getZespanVercelTelemetry:Vt,ZespanLlamaIndexHandler:et};export{tt as AgentContext,St as BaggageSpanProcessor,es as CREWAI_OTEL_ENV,V as GuardrailBlockedError,ke as PromptClient,ye as SpanKind,he as SpanStatusCode,Qe as ZespanADKCallbackHandler,He as ZespanCallbackHandler,Ke as ZespanClient,et as ZespanLlamaIndexHandler,ns as attachTraceToAutoGenMessage,qe as autopatch,wt as createSpan,st as extractAgentContext,st as extractAutoGenContext,ss as extractTraceFromAutoGenMessage,ts as getCrewAIInstrumentationGuide,os as getPydanticAIConfig,fe as getTracer,D as getZespanClient,Vt as getZespanVercelTelemetry,fn as init,Me as initOTel,pt as initZespan,nt as injectAgentContext,nt as injectAutoGenContext,Jt as instrumentADK,It as instrumentVercelAI,le as markFrameworkActive,te as markFrameworkInactive,Ht as startSpan,Qt as withAgent,bt as withSpan,un as withZespan,at as withZespanContext,it as withZespanTrace,Xe as wrapADKAgent,ft as wrapADKRunner,Pt as wrapAnthropic,Dn as wrapBedrock,Ft as wrapGoogle,Ln as wrapGroq,zn as wrapLiteLLM,Gn as wrapMistral,Be as wrapOpenAI,Et as wrapOpenRouter,Zt as wrapZespanADKAgent,pa as zespan};
|