@zespan/sdk 1.1.0 → 1.3.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 +9 -11
- package/dist/index.d.ts +9 -11
- package/dist/index.js +11 -10
- package/dist/index.mjs +11 -10
- package/package.json +6 -1
package/dist/index.mjs
CHANGED
|
@@ -1,13 +1,14 @@
|
|
|
1
|
-
var ot=(
|
|
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(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("[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 Xt,SpanKind as fe,SpanStatusCode as _e,context as Yt}from"@opentelemetry/api";import{BatchSpanProcessor as en}from"@opentelemetry/sdk-trace-base";import{OTLPTraceExporter as tn}from"@opentelemetry/exporter-trace-otlp-http";import{AsyncHooksContextManager as nn}from"@opentelemetry/context-async-hooks";import{W3CTraceContextPropagator as sn}from"@opentelemetry/core";var Ne=null,vt=null;function Me(a){let t=a.serviceName??"zespan-sdk",n=new tn({url:a.endpoint??"https://localhost:3001/v1/traces",headers:a.apiKey?{"x-api-key":a.apiKey}:void 0}),e=new en(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 sn}),Ne=Xt.getTracer(t,"1.0.0"),vt=new nn,Yt.setGlobalContextManager(vt.enable()),{tracer:Ne,provider:r}}function ge(){if(!Ne)throw new Error("OTel not initialized. Call initOTel() first.");return Ne}function St(a){return ge().startSpan(a.name,{kind:a.kind??fe.INTERNAL,attributes:a.attributes??{}})}async function wt(a,t,n){let s=ge().startSpan(a,{kind:n?.kind??fe.INTERNAL,attributes:n?.attributes??{}});try{let r=await t(s);return s.setStatus({code:_e.OK}),s.end(),r}catch(r){throw s.setStatus({code:_e.ERROR,message:r.message}),s.recordException(r),s.end(),r}}import{createHash as on}from"crypto";var he=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),g=this.cache.get(m);if(g&&Date.now()-g.timestamp<this.cacheTTL)return g.prompt}let o=new URLSearchParams;e!==void 0&&o.set("version",e.toString()),s&&o.set("label",s);let d=`${this.client.baseURL}/v1/prompts/${encodeURIComponent(t)}?${o}`,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=on("sha256").update(JSON.stringify(l.prompt)).digest("hex"),c={...l,promptHash:p};if(r){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`,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 rn="[REDACTED]",bt=a=>a.toLowerCase().replace(/[^a-z0-9]/g,""),an=a=>{let t=new Set;for(let n of a){let e=bt(String(n));e&&t.add(e)}return t},qe=(a,t)=>{if(Array.isArray(a))return a.map(s=>qe(s,t));if(!a||typeof a!="object")return a;let n=a,e={};for(let[s,r]of Object.entries(n)){if(t.has(bt(s))){e[s]=rn;continue}e[s]=qe(r,t)}return e},xt=(a,t)=>{if(a==null)return"";let n=an(t);if(typeof a=="string")try{let e=JSON.parse(a);return e&&typeof e=="object"?JSON.stringify(qe(e,n)):a}catch{return a}return JSON.stringify(qe(a,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 ye(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 Pe=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,ye(n.modelOverrides,e.modelOverrides)||this.emit("model_override",Object.fromEntries(e.modelOverrides)),n.samplingRate!==e.samplingRate&&this.emit("sampling_config",{samplingRate:e.samplingRate}),ye(n.cacheConfig,e.cacheConfig)||this.emit("cache_config",Object.fromEntries(e.cacheConfig)),ye(n.fallbacks,e.fallbacks)||this.emit("fallback_config",Object.fromEntries(e.fallbacks)),ye(n.retryPolicies,e.retryPolicies)||this.emit("retry_config",Object.fromEntries(e.retryPolicies)),ye(n.timeouts,e.timeouts)||this.emit("timeout_config",Object.fromEntries(e.timeouts)),ye(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 r=typeof s=="string"?JSON.parse(s):s;if(e.startsWith("model_override.")){let o=e.slice(15),d=typeof r=="string"?r:r?.model??r?.value;typeof d=="string"&&n.modelOverrides.set(o,d)}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),d=typeof r=="number"?r:r?.version;typeof d=="number"&&n.promptVersions.set(o,d)}else if(e.startsWith("fallback.")){let o=e.slice(9),d=typeof r?.model=="string"?r.model:null,i=Array.isArray(r?.conditions)?r.conditions.filter(l=>typeof l=="string"):[];d&&n.fallbacks.set(o,{model:d,conditions:i})}else if(e.startsWith("retry.")){let o=e.slice(6),d=typeof r?.enabled=="boolean"?r.enabled:!1,i=typeof r?.maxRetries=="number"?r.maxRetries:3,l=typeof r?.backoffMultiplier=="number"?r.backoffMultiplier:2,p=Array.isArray(r?.retryOnStatusCodes)?r.retryOnStatusCodes.filter(c=>typeof c=="number"):[];n.retryPolicies.set(o,{enabled:d,maxRetries:i,backoffMultiplier:l,retryOnStatusCodes:p})}else if(e.startsWith("timeout.")){let o=e.slice(8),d=typeof r?.timeoutMs=="number"?r.timeoutMs:typeof r=="number"?r:null;typeof d=="number"&&d>0&&n.timeouts.set(o,{timeoutMs:d})}else if(e.startsWith("concurrency.")){let o=e.slice(12),d=typeof r=="number"?r:typeof r?.limit=="number"?r.limit:null;typeof d=="number"&&d>0&&n.concurrencyLimits.set(o,d)}else if(e.startsWith("ab_test.")){let o=e.slice(8),d=typeof r?.candidateModel=="string"?r.candidateModel:null,i=typeof r?.splitPercentage=="number"?r.splitPercentage:null;d&&i!==null&&i>=0&&i<=100&&n.abTests.set(o,{candidateModel:d,splitPercentage:i})}}catch{}return n}};import{AsyncLocalStorage as dn}from"async_hooks";import{randomUUID as cn}from"crypto";var rt=new dn,u=()=>cn(),E=()=>rt.getStore();function W(a,t){return rt.run(a,t)}function je(a,t){let n=E(),e={traceId:a.traceId??n?.traceId??u(),spanId:u(),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 at(a,t){return je(t??{},a)}var ln=at,un=je;function Ge(){try{return j()}catch{return null}}var dt=new Set;function ce(a){dt.add(a)}function ee(a){dt.delete(a)}function De(){return dt.size>0}var Fe=!1,Ee=!1,Tt=!1,we=null,be=null,it=null,ws=(async()=>{try{we=await import("openai/resources/chat/completions")}catch{}try{be=await import("@anthropic-ai/sdk/resources/messages")}catch{}try{it=await import("@google/generative-ai")}catch{}Rt()})();function Ot(a){async function t(n,e){let s=Ge();if(!s||De())return a.call(this,n,e);let r=performance.now(),o=E(),d=u(),i=o?.traceId??u(),l=n?.model??"unknown",p=n?.stream===!0;try{let c=await a.call(this,n,e);if(p){let m=c[Symbol.asyncIterator]?.bind(c);return m&&(c[Symbol.asyncIterator]=function(){let g=m(),f=0,b=0,w="",y;return{async next(){let{value:k,done:h}=await g.next();return h?(s.enqueue({event_id:u(),project_id:"",trace_id:i,span_id:d,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openai",model:l,operation:"chat",latency_ms:performance.now()-r,ttft_ms:y,stream:!0,input_tokens:f,output_tokens:b,finish_reason:w,status:"success",span_kind:"llm",agent_id:o?.agentId,agent_name:o?.agentName,agent_role:o?.agentRole,environment:s.environment,sdk_version:"1.0.0"}),{value:k,done:!0}):(y||(y=performance.now()-r),k?.usage&&(f=k.usage.prompt_tokens??f,b=k.usage.completion_tokens??b),k?.choices?.[0]?.finish_reason&&(w=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:o?.spanId,timestamp:new Date().toISOString(),provider:"openai",model:c.model??l,operation:"chat",latency_ms:performance.now()-r,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:o?.agentId,agent_name:o?.agentName,agent_role:o?.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:o?.spanId,timestamp:new Date().toISOString(),provider:"openai",model:l,operation:"chat",latency_ms:performance.now()-r,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(!Fe&&we)try{let a=we.Completions;if(!a?.prototype)return;let t=a.prototype.create;if(typeof t=="function"&&!t.__lumiqAutoPatched){a.prototype.create=Ot(t),Fe=!0;return}if(typeof t!="function"){if(a.__lumiqAutoPatched){Fe=!0;return}let n=a;class e extends n{constructor(...o){if(super(...o),Object.prototype.hasOwnProperty.call(this,"create")&&!this.create?.__lumiqAutoPatched){let d=this.create;this.create=Ot(d)}}}e.__lumiqAutoPatched=!0;let s=Object.getOwnPropertyDescriptor(we,"Completions");if(s&&(s.writable!==!1||s.configurable))try{Object.defineProperty(we,"Completions",{value:e,writable:!0,enumerable:!0,configurable:!0}),Fe=!0}catch{}}}catch{}}function At(a){async function t(n,e){let s=Ge();if(!s||De())return a.call(this,n,e);let r=performance.now(),o=E(),d=u(),i=o?.traceId??u(),l=n?.model??"unknown",p=n?.stream===!0;try{let c=await a.call(this,n,e);if(p){let m=c[Symbol.asyncIterator]?.bind(c);return m&&(c[Symbol.asyncIterator]=function(){let g=m(),f=0,b=0,w="",y;return{async next(){let{value:k,done:h}=await g.next();return h?(s.enqueue({event_id:u(),project_id:"",trace_id:i,span_id:d,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:l,operation:"messages",latency_ms:performance.now()-r,ttft_ms:y,stream:!0,input_tokens:f,output_tokens:b,finish_reason:w,status:"success",span_kind:"llm",agent_id:o?.agentId,agent_name:o?.agentName,agent_role:o?.agentRole,environment:s.environment,sdk_version:"1.0.0"}),{value:k,done:!0}):(y||(y=performance.now()-r),k?.type==="message_delta"&&k?.usage&&(b=k.usage.output_tokens??b),k?.type==="message_start"&&k?.message?.usage&&(f=k.message.usage.input_tokens??f),k?.type==="message_delta"&&k?.delta?.stop_reason&&(w=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:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:c.model??l,operation:"messages",latency_ms:performance.now()-r,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:o?.agentId,agent_name:o?.agentName,agent_role:o?.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:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:l,operation:"messages",latency_ms:performance.now()-r,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 mn(){if(!Ee&&be)try{let a=be.Messages;if(!a?.prototype)return;let t=a.prototype.create;if(typeof t=="function"&&!t.__lumiqAutoPatched){a.prototype.create=At(t),Ee=!0;return}if(typeof t!="function"){if(a.__lumiqAutoPatched){Ee=!0;return}let n=a;class e extends n{constructor(...o){if(super(...o),Object.prototype.hasOwnProperty.call(this,"create")&&!this.create?.__lumiqAutoPatched){let d=this.create;this.create=At(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 gn(){if(!Tt&&it)try{let e=function(s,r){let o=n.call(this,s,r),d=o.generateContent?.bind(o),i=o.generateContentStream?.bind(o);return!d||d.__lumiqAutoPatched||(o.generateContent=async l=>{let p=Ge();if(!p||De())return d(l);let c=performance.now(),m=E(),g=u(),f=m?.traceId??u(),b=s.model??"gemini-1.5-flash";try{let w=await d(l),y=w.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:b,operation:"generateContent",latency_ms:performance.now()-c,stream:!1,input_tokens:y?.promptTokenCount??0,output_tokens:y?.candidatesTokenCount??0,finish_reason:w.response?.candidates?.[0]?.finishReason??"",status:"success",span_kind:"llm",environment:p.environment,sdk_version:"1.0.0"}),w}catch(w){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:b,operation:"generateContent",latency_ms:performance.now()-c,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_message:String(w.message).slice(0,500),span_kind:"llm",environment:p.environment,sdk_version:"1.0.0"}),w}},o.generateContent.__lumiqAutoPatched=!0,i&&(o.generateContentStream=async l=>{let p=Ge();if(!p||De())return i(l);let c=performance.now(),m=E(),g=u(),f=m?.traceId??u(),b=s.model??"gemini-1.5-flash",w=await i(l),y=w.stream,k=0,h=0,C;return w.stream=(async function*(){try{for await(let _ of y){C||(C=performance.now()-c);let N=_.usageMetadata;N&&(k=N.promptTokenCount??k,h=N.candidatesTokenCount??h),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:b,operation:"generateContent",latency_ms:performance.now()-c,ttft_ms:C,stream:!0,input_tokens:k,output_tokens:h,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:b,operation:"generateContent",latency_ms:performance.now()-c,ttft_ms:C,stream:!0,input_tokens:k,output_tokens:h,status:"error",error_message:String(_.message).slice(0,500),span_kind:"llm",environment:p.environment,sdk_version:"1.0.0"}),_}})(),w})),o};var a=e;let t=it?.GoogleGenerativeAI;if(!t?.prototype?.getGenerativeModel||t.prototype.getGenerativeModel.__lumiqAutoPatched)return;let n=t.prototype.getGenerativeModel;e.__lumiqAutoPatched=!0,t.prototype.getGenerativeModel=e,Tt=!0}catch{}}function Rt(){pn(),mn(),gn()}function Le(){Rt()}var xe=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 he(this),t.enableZespanPilot!==!1&&t.projectId?this.configClient=new Pe({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&&Le()}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 xt(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 ge().startSpan(t.name,{kind:t.kind??fe.INTERNAL,attributes:t.attributes??{}})}async withSpan(t,n,e){let r=ge().startSpan(t,{kind:e?.kind??fe.INTERNAL,attributes:e?.attributes??{}});try{let o=await n(r);return r.setStatus({code:_e.OK}),o}catch(o){throw r.setStatus({code:_e.ERROR,message:o.message}),r.recordException(o),o}finally{r.end()}}},ct=null,lt=a=>{ct=new xe(a)},j=()=>{if(!ct)throw new Error("Call zespan.init() first");return ct};var _n=lt,fn=j;var U=class extends Error{phase;results;constructor(t,n){super(`Guardrail blocked ${t} content`),this.name="GuardrailBlockedError",this.phase=t,this.results=n}};function Q(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 te(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 U(n.phase,s.results);return s}catch(s){if(s instanceof U||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 oe(a,t){let n=hn(a);try{!n.has_planning&&t&&/^o[13]/.test(t)&&(n.has_planning=!0)}catch{}return n}function re(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 ae(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 Ke(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 $e(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 ve,trace as vn}from"@opentelemetry/api";async function ke(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()}}},Nt=new Map;function Ie(a,t){let n=Nt.get(a);return n||(n=new ut(t),Nt.set(a,n)),n}var Be=a=>kn("sha256").update(a).digest("hex");function Sn(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{lumiq:n,t0:e,traceId:s,spanId:r,parent:o,params:d,otelSpan:i,model:l,agentCtx:p,autoDetected:c,conversation_turn:m,message_role:g,tool_definitions:f,provider:b,operation:w}=t,y=b??"openai",k,h=0,C=0,_=0,N="",v=l,x=[],I="",F=[],{systemPrompt:T,conversationMessages:S}=Y(d.messages??[]);try{for await(let q of a){k||(k=performance.now()-e,n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:y,model:v,operation:w,latency_ms:k,ttft_ms:k,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:Be(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(S),...T?{system_prompt:n.redactForStorage(T)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.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 R=q.choices?.[0]?.delta?.content;if(R&&(C++,I+=R),q.choices?.[0]?.finish_reason&&(N=q.choices[0].finish_reason),q.model&&(v=q.model),q.usage&&(h=Number(q.usage.prompt_tokens||h||0),C=Number(q.usage.completion_tokens||C||0),_=Number(q.usage.prompt_tokens_details?.cached_tokens||_||0)),q.choices?.[0]?.delta?.tool_calls)for(let $ of q.choices[0].delta.tool_calls)F.push($),$.function?.name&&!x.includes($.function.name)&&x.push($.function.name);yield q}let{tools_used:M,tool_call_args:A}=$e(F),G=M.length>0?M:x,O=performance.now()-e;n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:y,model:v,operation:w,latency_ms:O,ttft_ms:k,stream:!0,input_tokens:h,output_tokens:C,cached_tokens:_,status:"success",finish_reason:N||"stream_end",tools_used:G,tool_call_args:Object.keys(A).length?A:void 0,prompt_hash:Be(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(S),completion_text:n.redactForStorage(I),...T?{system_prompt:n.redactForStorage(T)}:{}}:{},...G.length>0&&I?{decision_rationale:I}:{},...p,span_kind:"llm",tool_definitions:f,conversation_turn:m,message_role:g,metadata:{usage:{prompt_tokens:h,completion_tokens:C,cached_tokens:_},auto_detected:Object.keys(c).length?c:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:"1.0.0"}),i&&(i.setAttribute("gen_ai.usage.output_tokens",C),i.setAttribute("gen_ai.usage.total_tokens",C),i.setAttribute("gen_ai.response.finish_reason",N||"stream_end"),i.setAttribute("gen_ai.response.latency",O))}catch(M){throw n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:y,model:v,operation:w,latency_ms:performance.now()-e,ttft_ms:k,stream:!0,input_tokens:0,output_tokens:C,cost_usd:0,status:"error",error_code:String(M.status??"unknown"),error_message:String(M.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:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:"1.0.0"}),i&&(i.setStatus({code:ve.ERROR,message:M.message}),i.recordException(M)),M}}function We(a,t={}){let n=a,e=n.chat.completions.create.bind(n.chat.completions),s=Q(t.guardrails,"chat"),r=t.providerOverride??"openai",o=t.operation??"chat";return n.chat.completions.create=async(d,i)=>{let l=j(),p=performance.now(),c=E(),m=u(),g=c?.traceId??u(),f=d.model??"gpt-4",b;if(l.configClient){let S=l.configClient.getAbTest(o);if(S)Math.random()<S.splitPercentage/100?(f=S.candidateModel,b={ab_test:!0,ab_variant:"candidate"}):b={ab_test:!0,ab_variant:"control"},d={...d,model:f};else{let M=l.configClient.getModel(o,f);if(M!==f){let A=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:r,model:M,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:A,override_model:M},environment:l.environment,sdk_version:"1.0.0"})}}}let w=null,y=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}:{},h=oe(d.messages??[],d.model),{conversation_turn:C,message_role:_}=re(d.messages??[]),N=d.tools?.length?ae(d.tools):void 0,v=null,x=l.enableOTel?vn.getTracer("zespan-sdk"):null;x&&(v=x.startSpan(`chat.completions ${f}`,{kind:In.CLIENT,attributes:Sn(d,f,r)}));let I;if(l.configClient){let S=l.configClient.getTimeout(o);S&&(I=AbortSignal.timeout(S.timeoutMs))}let F=l.configClient?.getRetryPolicy(o)??null,T=l.configClient?.getConcurrencyLimit(o)??null;try{w=await Z(l,s,{text:JSON.stringify(d.messages??[]),phase:"pre",traceId:g,spanId:m,model:f,operation:o,agentName:c?.agentName});let S=()=>ke(()=>W({traceId:g,spanId:m,userId:c?.userId,sessionId:c?.sessionId,tags:c?.tags},()=>e(d,I?{...i,signal:I}:i)),F),M=await(T!==null?Ie(`${l.projectId??"default"}:${o}`,T).run(S):S());if(d.stream===!0)return wn(M,{lumiq:l,t0:p,traceId:g,spanId:m,parent:c,params:d,otelSpan:v,model:f,agentCtx:k,autoDetected:h,conversation_turn:C,message_role:_,tool_definitions:N,provider:r,operation:o});let{prompt_tokens:G,completion_tokens:O,prompt_tokens_details:q}=M.usage??{},R=q?.cached_tokens??0,$=M.model??d.model,L=M.usage?.completion_tokens_details?.reasoning_tokens,K=M.choices?.[0]?.message?.content||"";y=await Z(l,s,{text:K,phase:"post",traceId:g,spanId:m,model:$,operation:o,agentName:c?.agentName});let H=te(K,y);H!==K&&typeof M?.choices?.[0]?.message?.content=="string"&&(M.choices[0].message.content=H);let{tools_used:P,tool_call_args:V}=Ke(M.choices??[]),{systemPrompt:D,conversationMessages:J}=Y(d.messages??[]);return l.enqueue({event_id:u(),project_id:"",trace_id:g,span_id:m,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:r,model:$,operation:o,latency_ms:performance.now()-p,stream:!1,input_tokens:G??0,output_tokens:O??0,cached_tokens:R,...L?{reasoning_tokens:L}:{},status:"success",finish_reason:M.choices?.[0]?.finish_reason??"",tools_used:P.length>0?P:M.choices?.[0]?.message?.tool_calls?.map(X=>X.function.name)??[],tool_call_args:Object.keys(V).length?V:void 0,prompt_hash:Be(JSON.stringify(d.messages)),...l.storePrompts?{prompt_text:l.redactForStorage(J),completion_text:l.redactForStorage(H),...D?{system_prompt:l.redactForStorage(D)}:{}}:{},...P.length>0&&K?{decision_rationale:K}:{},...k,span_kind:"llm",tool_definitions:N,conversation_turn:C,message_role:_,metadata:{usage:M.usage??null,guardrails:{pre:w?{matched:w.results.length,modified:!!w.modifiedText}:null,post:y?{matched:y.results.length,modified:!!y.modifiedText}:null},auto_detected:Object.keys(h).length?h:void 0,...b??{}},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",G??0),v.setAttribute("gen_ai.usage.output_tokens",O??0),v.setAttribute("gen_ai.usage.total_tokens",(G??0)+(O??0)),v.setAttribute("gen_ai.usage.cached_tokens",R),v.setAttribute("gen_ai.response.finish_reason",M.choices?.[0]?.finish_reason??""),v.setAttribute("gen_ai.response.latency",performance.now()-p),v.setStatus({code:ve.OK}),v.end()),M}catch(S){if(S?.name==="TimeoutError"||S?.code==="ABORT_ERR"||I&&I.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:r,model:d.model,operation:o,latency_ms:performance.now()-p,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:"timeout",error_message:String(S.message).slice(0,500),...k,span_kind:"llm",conversation_turn:C,message_role:_,metadata:{auto_detected:Object.keys(h).length?h: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:ve.ERROR,message:S.message}),v.recordException(S),v.end()),S;let A=l.configClient?.getFallback(o)??null;if(A&&A.model){let q=String(S.message??"")+String(S.status??"")+String(S.code??"");if(A.conditions.length===0||A.conditions.some($=>q.toLowerCase().includes($.toLowerCase()))){let $=f,L={...d,model:A.model};l.enqueue({event_id:u(),project_id:"",trace_id:g,span_id:m,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:r,model:A.model,operation:o,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:$,fallback_model:A.model,reason:String(S.message).slice(0,200)},environment:l.environment,sdk_version:"1.0.0"});try{let K=await W({traceId:g,spanId:m,userId:c?.userId,sessionId:c?.sessionId,tags:c?.tags},()=>e(L,i)),{prompt_tokens:H,completion_tokens:P,prompt_tokens_details:V}=K.usage??{},D=V?.cached_tokens??0,J=K.model??A.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:r,model:J,operation:o,latency_ms:performance.now()-p,stream:!1,input_tokens:H??0,output_tokens:P??0,cached_tokens:D,status:"success",finish_reason:K.choices?.[0]?.finish_reason??"",prompt_hash:Be(JSON.stringify(d.messages)),...k,span_kind:"llm",conversation_turn:C,message_role:_,metadata:{usage:K.usage??null,fallback_used:!0,original_model:$,auto_detected:Object.keys(h).length?h: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:ve.OK}),v.end()),K}catch(K){throw v&&(v.setStatus({code:ve.ERROR,message:K.message}),v.recordException(K),v.end()),K}}}let G=S instanceof U,O=G?"error":S.status===429?"rate_limited":S.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:r,model:d.model,operation:o,latency_ms:performance.now()-p,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:O,error_code:G?"guardrail_blocked":String(S.status??"unknown"),error_message:String(S.message).slice(0,500),...k,span_kind:"llm",conversation_turn:C,message_role:_,metadata:{auto_detected:Object.keys(h).length?h: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:ve.ERROR,message:S.message}),v.recordException(S),v.end()),S}},a}import{createHash as bn}from"crypto";import{SpanKind as xn,SpanStatusCode as Se,trace as Cn}from"@opentelemetry/api";var Je=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{lumiq:n,t0:e,traceId:s,spanId:r,parent:o,params:d,otelSpan:i,agentCtx:l,tool_definitions:p,autoDetected:c,conversation_turn:m,message_role:g,operation:f,detectionMessages:b}=t,w,y=0,k=0,h=0,C=0,_="",N=d.model,v=[],x="",{systemPrompt:I,conversationMessages:F}=Y(b);try{for await(let S of a){if(!w&&S.type==="content_block_start"&&(w=performance.now()-e,n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:N,operation:f,latency_ms:w,ttft_ms:w,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:Je(JSON.stringify(b)),...n.storePrompts?{prompt_text:n.redactForStorage(F),...I?{system_prompt:n.redactForStorage(I)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.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",w)),S.type==="content_block_delta"&&S.delta?.type==="text_delta"&&(C++,x+=S.delta.text||""),S.type==="message_delta"&&S.delta?.stop_reason&&(_=S.delta.stop_reason),S.type==="message_delta"&&S.usage&&(y=Number(S.usage.input_tokens||y||0),C=Number(S.usage.output_tokens||C||0),k=Number(S.usage.cache_read_input_tokens||k||0)),S.type==="message_start"&&S.message?.usage&&(y=Number(S.message.usage.input_tokens||y||0),C=Number(S.message.usage.output_tokens||C||0),k=Number(S.message.usage.cache_read_input_tokens||k||0)),S.type==="content_block_start"&&S.content_block?.type==="tool_use"){let M=S.content_block.name;M&&!v.includes(M)&&v.push(M)}yield S}let T=performance.now()-e;n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:N,operation:f,latency_ms:T,ttft_ms:w,stream:!0,input_tokens:y,output_tokens:C,cached_tokens:k,reasoning_tokens:h,status:"success",finish_reason:_||"stream_end",tools_used:v,prompt_hash:Je(JSON.stringify(b)),...n.storePrompts?{prompt_text:n.redactForStorage(F),completion_text:n.redactForStorage(x),...I?{system_prompt:n.redactForStorage(I)}:{}}:{},...l,span_kind:"llm",tool_definitions:p,conversation_turn:m,message_role:g,metadata:{usage:{input_tokens:y,output_tokens:C,cache_read_input_tokens:k,reasoning_tokens:h},auto_detected:Object.keys(c).length?c:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:"1.0.0"}),i&&(i.setAttribute("gen_ai.usage.output_tokens",C),i.setAttribute("gen_ai.response.finish_reason",_||"stream_end"))}catch(T){throw n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:N,operation:f,latency_ms:performance.now()-e,ttft_ms:w,stream:!0,input_tokens:0,output_tokens:C,cost_usd:0,status:"error",error_code:String(T.status??T.type??"unknown"),error_message:String(T.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:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:"1.0.0"}),i&&(i.setStatus({code:Se.ERROR,message:T.message}),i.recordException(T)),T}}function Mt(a,t={}){let n=a,e=n.messages.create.bind(n.messages),s=Q(t.guardrails,"chat"),r=t.operation??"chat";return n.messages.create=async(o,d)=>{let i=j(),l=performance.now(),p=E(),c=u(),m=p?.traceId??u(),g=o.model??"claude-3-5-sonnet-20241022",f;if(i.configClient){let O=i.configClient.getAbTest(r);if(O)Math.random()<O.splitPercentage/100?(g=O.candidateModel,f={ab_test:!0,ab_variant:"candidate"}):f={ab_test:!0,ab_variant:"control"},o={...o,model:g};else{let q=i.configClient.getModel(r,g);if(q!==g){let R=g;g=q,o={...o,model:q},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,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:R,override_model:q},environment:i.environment,sdk_version:"1.0.0"})}}}let b=null,w=null,y=Tn(o),{systemPrompt:k,conversationMessages:h}=Y(y),C=oe(y,g),{conversation_turn:_,message_role:N}=re(y),v=p?.agentName??C.agent_name,x=p?.agentRole??C.agent_role,I=p?.agentId||v||x?{agent_id:p?.agentId,agent_name:v,agent_role:x,agent_framework:p?.agentFramework,parent_agent_id:p?.parentAgentId}:{},F=o.tools?.length?ae(o.tools):void 0,T=null,S=i.enableOTel?Cn.getTracer("zespan-sdk"):null;S&&(T=S.startSpan(`messages.create ${g}`,{kind:xn.CLIENT,attributes:On(o,g)}));let M;if(i.configClient){let O=i.configClient.getTimeout(r);O&&(M=AbortSignal.timeout(O.timeoutMs))}let A=i.configClient?.getRetryPolicy(r)??null,G=i.configClient?.getConcurrencyLimit(r)??null;try{b=await Z(i,s,{text:JSON.stringify(o.messages??[]),phase:"pre",traceId:m,spanId:c,model:g,operation:r,agentName:p?.agentName});let O=()=>ke(()=>W({traceId:m,spanId:c,userId:p?.userId,sessionId:p?.sessionId,tags:p?.tags,agentId:p?.agentId,agentName:v,agentRole:x,agentFramework:p?.agentFramework,parentAgentId:p?.parentAgentId,conversationTurn:_,availableTools:F},()=>e(o,M?{...d,signal:M}:d)),A),q=await(G!==null?Ie(`${i.projectId??"default"}:${r}`,G).run(O):O());if(o.stream===!0)return An(q,{lumiq:i,t0:l,traceId:m,spanId:c,parent:p,params:o,otelSpan:T,agentCtx:I,tool_definitions:F,autoDetected:C,conversation_turn:_,message_role:N,operation:r,detectionMessages:y});let $=q.usage?.input_tokens??0,L=q.usage?.output_tokens??0,K=q.usage?.cache_read_input_tokens??0,H=q.model??o.model,P=q.content?.find(B=>B.type==="text")?.text||"",V=(q.content??[]).filter(B=>B.type==="tool_use"),D=V.map(B=>B.name).filter(Boolean),J={};for(let B of V)B.name&&(J[B.name]=B.input??{});w=await Z(i,s,{text:P,phase:"post",traceId:m,spanId:c,model:H,operation:r,agentName:p?.agentName});let X=te(P,w);if(X!==P){let B=q.content?.find(se=>se.type==="text");B&&typeof B.text=="string"&&(B.text=X)}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:H,operation:r,latency_ms:performance.now()-l,stream:!1,input_tokens:$,output_tokens:L,cached_tokens:K,status:"success",finish_reason:q.stop_reason??"",tools_used:D.length>0?D:q.content?.filter(B=>B.type==="tool_use").map(B=>B.name)??[],tool_call_args:Object.keys(J).length?J:void 0,prompt_hash:Je(JSON.stringify(y)),...i.storePrompts?{prompt_text:i.redactForStorage(h),completion_text:i.redactForStorage(X),...k?{system_prompt:i.redactForStorage(k)}:{}}:{},...V.length>0&&P?{decision_rationale:P}:{},...I,span_kind:"llm",tool_definitions:F,conversation_turn:_,message_role:N,metadata:{usage:q.usage??null,guardrails:{pre:b?{matched:b.results.length,modified:!!b.modifiedText}:null,post:w?{matched:w.results.length,modified:!!w.modifiedText}:null},auto_detected:Object.keys(C).length?C:void 0,...f??{}},user_id:p?.userId,session_id:p?.sessionId,tags:p?.tags,environment:i.environment,sdk_version:"1.0.0"}),T&&(T.setAttribute("gen_ai.usage.input_tokens",$),T.setAttribute("gen_ai.usage.output_tokens",L),T.setAttribute("gen_ai.usage.cached_tokens",K),T.setAttribute("gen_ai.response.finish_reason",q.stop_reason??""),T.setAttribute("gen_ai.response.latency",performance.now()-l),T.setStatus({code:Se.OK}),T.end()),q}catch(O){if(O?.name==="TimeoutError"||O?.code==="ABORT_ERR"||M&&M.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: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(O.message).slice(0,500),...I,span_kind:"llm",conversation_turn:_,message_role:N,metadata:{auto_detected:Object.keys(C).length?C:void 0},user_id:p?.userId,session_id:p?.sessionId,tags:p?.tags,environment:i.environment,sdk_version:"1.0.0"}),T&&(T.setStatus({code:Se.ERROR,message:O.message}),T.recordException(O),T.end()),O;let R=i.configClient?.getFallback(r)??null;if(R&&R.model){let K=String(O.message??"")+String(O.status??"")+String(O.code??"");if(R.conditions.length===0||R.conditions.some(P=>K.toLowerCase().includes(P.toLowerCase()))){let P=g,V={...o,model:R.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:R.model,operation:r,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:P,fallback_model:R.model,reason:String(O.message).slice(0,200)},environment:i.environment,sdk_version:"1.0.0"});try{let D=await W({traceId:m,spanId:c,userId:p?.userId,sessionId:p?.sessionId,tags:p?.tags},()=>e(V,d)),J=D.usage?.input_tokens??0,X=D.usage?.output_tokens??0,B=D.usage?.cache_read_input_tokens??0,se=D.model??R.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:se,operation:r,latency_ms:performance.now()-l,stream:!1,input_tokens:J,output_tokens:X,cached_tokens:B,status:"success",finish_reason:D.stop_reason??"",prompt_hash:Je(JSON.stringify(y)),...I,span_kind:"llm",conversation_turn:_,message_role:N,metadata:{usage:D.usage??null,fallback_used:!0,original_model:P,auto_detected:Object.keys(C).length?C:void 0},user_id:p?.userId,session_id:p?.sessionId,tags:p?.tags,environment:i.environment,sdk_version:"1.0.0"}),T&&(T.setStatus({code:Se.OK}),T.end()),D}catch(D){throw T&&(T.setStatus({code:Se.ERROR,message:D.message}),T.recordException(D),T.end()),D}}}let $=O instanceof U,L=$?"error":O.status===429||O.type==="overloaded_error"?"rate_limited":O.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:o.model,operation:r,latency_ms:performance.now()-l,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:L,error_code:$?"guardrail_blocked":String(O.status??O.type??"unknown"),error_message:String(O.message).slice(0,500),...I,span_kind:"llm",conversation_turn:_,message_role:N,metadata:{auto_detected:Object.keys(C).length?C:void 0},user_id:p?.userId,session_id:p?.sessionId,tags:p?.tags,environment:i.environment,sdk_version:"1.0.0"}),T&&(T.setStatus({code:Se.ERROR,message:O.message}),T.recordException(O),T.end()),O}},a}import{createHash as Rn}from"crypto";import{SpanKind as Ze,SpanStatusCode as ie,trace as ze}from"@opentelemetry/api";var Ce=a=>Rn("sha256").update(a).digest("hex");function Ue(a){let t=o=>{if(typeof o=="string")return o;if(Array.isArray(o))return o.map(d=>typeof d=="string"?d:typeof d?.text=="string"?d.text:"").filter(Boolean).join(`
|
|
1
|
+
var ot=(r=>typeof require<"u"?require:typeof Proxy<"u"?new Proxy(r,{get:(t,n)=>(typeof require<"u"?require:t)[n]}):r)(function(r){if(typeof require<"u")return require.apply(this,arguments);throw Error('Dynamic require of "'+r+'" is not supported')});import{gzipSync as rn}from"zlib";var Me=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=rn(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(a=>setTimeout(a,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 a=await s.json();typeof a?.cv=="number"&&this.configClient&&this.configClient.checkVersion(a.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 cn,SpanKind as ve,SpanStatusCode as Se,context as dn}from"@opentelemetry/api";import{BatchSpanProcessor as ln}from"@opentelemetry/sdk-trace-base";import{OTLPTraceExporter as un}from"@opentelemetry/exporter-trace-otlp-http";import{AsyncHooksContextManager as pn}from"@opentelemetry/context-async-hooks";import{W3CTraceContextPropagator as mn}from"@opentelemetry/core";var Pe=null,bt=null;function je(r){let t=r.serviceName??"zespan-sdk",n=new un({url:r.endpoint??"https://localhost:3001/v1/traces",headers:r.apiKey?{"x-api-key":r.apiKey}:void 0}),e=new ln(n,{maxQueueSize:100,maxExportBatchSize:10,exportTimeoutMillis:3e4,scheduledDelayMillis:2e3}),{NodeTracerProvider:s}=ot("@opentelemetry/sdk-trace-node"),a=new s({resource:{attributes:{"service.name":t,"service.version":"1.0.0","deployment.environment":r.environment??"production"}}});return a.addSpanProcessor(e),a.register({propagator:new mn}),Pe=cn.getTracer(t,"1.0.0"),bt=new pn,dn.setGlobalContextManager(bt.enable()),{tracer:Pe,provider:a}}function ye(){if(!Pe)throw new Error("OTel not initialized. Call initOTel() first.");return Pe}function Ct(r){return ye().startSpan(r.name,{kind:r.kind??ve.INTERNAL,attributes:r.attributes??{}})}async function xt(r,t,n){let s=ye().startSpan(r,{kind:n?.kind??ve.INTERNAL,attributes:n?.attributes??{}});try{let a=await t(s);return s.setStatus({code:Se.OK}),s.end(),a}catch(a){throw s.setStatus({code:Se.ERROR,message:a.message}),s.recordException(a),s.end(),a}}import{createHash as gn}from"crypto";var we=class{constructor(t){this.client=t}cache=new Map;cacheTTL=300*1e3;async get(t,n={}){let{version:e,label:s,cache:a=!0}=n;if(a){let g=this.getCacheKey(t,e,s),m=this.cache.get(g);if(m&&Date.now()-m.timestamp<this.cacheTTL)return m.prompt}let o=new URLSearchParams;e!==void 0&&o.set("version",e.toString()),s&&o.set("label",s);let d=`${this.client.baseURL}/v1/prompts/${encodeURIComponent(t)}?${o}`,c=await fetch(d,{headers:{"x-api-key":this.client.apiKey}});if(!c.ok)throw new Error(`Failed to fetch prompt: ${c.statusText}`);let u=await c.json(),l=gn("sha256").update(JSON.stringify(u.prompt)).digest("hex"),i={...u,promptHash:l};if(a){let g=this.getCacheKey(t,e,s);this.cache.set(g,{prompt:i,timestamp:Date.now()})}return i}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`,a=await fetch(s,{method:"PATCH",headers:{"Content-Type":"application/json","x-api-key":this.client.apiKey},body:JSON.stringify({labels:e})});if(!a.ok)throw new Error(`Failed to update labels: ${a.statusText}`);let o=await a.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,a])=>{e=e.replace(new RegExp(`\\{\\{${s}\\}\\}`,"g"),a)}),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 _n="[REDACTED]",Tt=r=>r.toLowerCase().replace(/[^a-z0-9]/g,""),fn=r=>{let t=new Set;for(let n of r){let e=Tt(String(n));e&&t.add(e)}return t},De=(r,t)=>{if(Array.isArray(r))return r.map(s=>De(s,t));if(!r||typeof r!="object")return r;let n=r,e={};for(let[s,a]of Object.entries(n)){if(t.has(Tt(s))){e[s]=_n;continue}e[s]=De(a,t)}return e},Ot=(r,t)=>{if(r==null)return"";let n=fn(t);if(typeof r=="string")try{let e=JSON.parse(r);return e&&typeof e=="object"?JSON.stringify(De(e,n)):r}catch{return r}return JSON.stringify(De(r,n))};function At(){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 be(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 Fe=class{localVersion=0;config=At();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,be(n.modelOverrides,e.modelOverrides)||this.emit("model_override",Object.fromEntries(e.modelOverrides)),n.samplingRate!==e.samplingRate&&this.emit("sampling_config",{samplingRate:e.samplingRate}),be(n.cacheConfig,e.cacheConfig)||this.emit("cache_config",Object.fromEntries(e.cacheConfig)),be(n.fallbacks,e.fallbacks)||this.emit("fallback_config",Object.fromEntries(e.fallbacks)),be(n.retryPolicies,e.retryPolicies)||this.emit("retry_config",Object.fromEntries(e.retryPolicies)),be(n.timeouts,e.timeouts)||this.emit("timeout_config",Object.fromEntries(e.timeouts)),be(n.abTests,e.abTests)||this.emit("ab_test_config",Object.fromEntries(e.abTests))}parseSnapshot(t){let n=At();for(let[e,s]of Object.entries(t))try{let a=typeof s=="string"?JSON.parse(s):s;if(e.startsWith("model_override.")){let o=e.slice(15),d=typeof a=="string"?a:a?.model??a?.value;typeof d=="string"&&n.modelOverrides.set(o,d)}else if(e.startsWith("cache.")){let o=e.slice(6);n.cacheConfig.set(o,{enabled:a?.enabled??!0,ttlSeconds:typeof a?.ttlSeconds=="number"?a.ttlSeconds:300,similarityThreshold:typeof a?.similarityThreshold=="number"?a.similarityThreshold:.85})}else if(e.startsWith("ratelimit.")){let o=e.slice(10);n.rateLimits.set(o,{maxRequests:typeof a?.maxRequests=="number"?a.maxRequests:100,windowSeconds:typeof a?.windowSeconds=="number"?a.windowSeconds:60})}else if(e.startsWith("guardrail.")&&!e.startsWith("guardrail_mode.")){let o=e.slice(10);n.guardrailOverrides.set(o,!!(a?.enabled??a))}else if(e.startsWith("sampling.")){let o=typeof a=="number"?a:a?.rate??a?.sampleRate;typeof o=="number"&&o>=0&&o<=1&&(n.samplingRate=o)}else if(e.startsWith("prompt_version.")){let o=e.slice(15),d=typeof a=="number"?a:a?.version;typeof d=="number"&&n.promptVersions.set(o,d)}else if(e.startsWith("fallback.")){let o=e.slice(9),d=typeof a?.model=="string"?a.model:null,c=Array.isArray(a?.conditions)?a.conditions.filter(u=>typeof u=="string"):[];d&&n.fallbacks.set(o,{model:d,conditions:c})}else if(e.startsWith("retry.")){let o=e.slice(6),d=typeof a?.enabled=="boolean"?a.enabled:!1,c=typeof a?.maxRetries=="number"?a.maxRetries:3,u=typeof a?.backoffMultiplier=="number"?a.backoffMultiplier:2,l=Array.isArray(a?.retryOnStatusCodes)?a.retryOnStatusCodes.filter(i=>typeof i=="number"):[];n.retryPolicies.set(o,{enabled:d,maxRetries:c,backoffMultiplier:u,retryOnStatusCodes:l})}else if(e.startsWith("timeout.")){let o=e.slice(8),d=typeof a?.timeoutMs=="number"?a.timeoutMs:typeof a=="number"?a:null;typeof d=="number"&&d>0&&n.timeouts.set(o,{timeoutMs:d})}else if(e.startsWith("concurrency.")){let o=e.slice(12),d=typeof a=="number"?a:typeof a?.limit=="number"?a.limit:null;typeof d=="number"&&d>0&&n.concurrencyLimits.set(o,d)}else if(e.startsWith("ab_test.")){let o=e.slice(8),d=typeof a?.candidateModel=="string"?a.candidateModel:null,c=typeof a?.splitPercentage=="number"?a.splitPercentage:null;d&&c!==null&&c>=0&&c<=100&&n.abTests.set(o,{candidateModel:d,splitPercentage:c})}}catch{}return n}};import{AsyncLocalStorage as hn}from"async_hooks";import{randomUUID as yn}from"crypto";var at=new hn,p=()=>yn(),F=()=>at.getStore();function W(r,t){return at.run(r,t)}function rt(r,t){let n=F(),e={traceId:r.traceId??n?.traceId??p(),spanId:p(),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 it(r,t){return rt(t??{},r)}var kn=it;var x="1.1.0";function ke(){try{return E()}catch{return null}}var lt=new Set;function le(r){lt.add(r)}function se(r){lt.delete(r)}function Ie(){return lt.size>0}var Ee=!1,Ge=!1,Rt=!1,Nt=!1,Te=null,Oe=null,ct=null,dt=null,qs=(async()=>{try{Te=await import("openai/resources/chat/completions")}catch{}try{Oe=await import("@anthropic-ai/sdk/resources/messages")}catch{}try{ct=await import("@google/generative-ai")}catch{}try{dt=await import("@google/genai")}catch{}jt()})();function Mt(r){async function t(n,e){let s=ke();if(!s||Ie())return r.call(this,n,e);let a=performance.now(),o=F(),d=p(),c=o?.traceId??p(),u=n?.model??"unknown",l=n?.stream===!0;try{let i=await r.call(this,n,e);if(l){let g=i[Symbol.asyncIterator]?.bind(i);return g&&(i[Symbol.asyncIterator]=function(){let m=g(),h=0,w=0,y="",f;return{async next(){let{value:I,done:k}=await m.next();return k?(s.enqueue({event_id:p(),project_id:"",trace_id:c,span_id:d,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openai",model:u,operation:"chat",latency_ms:performance.now()-a,ttft_ms:f,stream:!0,input_tokens:h,output_tokens:w,finish_reason:y,status:"success",span_kind:"llm",agent_id:o?.agentId,agent_name:o?.agentName,agent_role:o?.agentRole,environment:s.environment,sdk_version:x}),{value:I,done:!0}):(f||(f=performance.now()-a),I?.usage&&(h=I.usage.prompt_tokens??h,w=I.usage.completion_tokens??w),I?.choices?.[0]?.finish_reason&&(y=I.choices[0].finish_reason),{value:I,done:!1})},[Symbol.asyncIterator](){return this}}}),i}return s.enqueue({event_id:p(),project_id:"",trace_id:c,span_id:d,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openai",model:i.model??u,operation:"chat",latency_ms:performance.now()-a,stream:!1,input_tokens:i.usage?.prompt_tokens??0,output_tokens:i.usage?.completion_tokens??0,cached_tokens:i.usage?.prompt_tokens_details?.cached_tokens??0,finish_reason:i.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(i.choices?.[0]?.message?.content??"")}:{},environment:s.environment,sdk_version:x}),i}catch(i){throw s.enqueue({event_id:p(),project_id:"",trace_id:c,span_id:d,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openai",model:u,operation:"chat",latency_ms:performance.now()-a,stream:l,input_tokens:0,output_tokens:0,status:"error",error_code:String(i.status??"unknown"),error_message:String(i.message).slice(0,500),span_kind:"llm",environment:s.environment,sdk_version:x}),i}}return t.__zespanAutoPatched=!0,t}function In(){if(!Ee&&Te)try{let r=Te.Completions;if(!r?.prototype)return;let t=r.prototype.create;if(typeof t=="function"&&!t.__zespanAutoPatched){r.prototype.create=Mt(t),Ee=!0;return}if(typeof t!="function"){if(r.__zespanAutoPatched){Ee=!0;return}let n=r;class e extends n{constructor(...o){if(super(...o),Object.prototype.hasOwnProperty.call(this,"create")&&!this.create?.__zespanAutoPatched){let d=this.create;this.create=Mt(d)}}}e.__zespanAutoPatched=!0;let s=Object.getOwnPropertyDescriptor(Te,"Completions");if(s&&(s.writable!==!1||s.configurable))try{Object.defineProperty(Te,"Completions",{value:e,writable:!0,enumerable:!0,configurable:!0}),Ee=!0}catch{}}}catch{}}function Pt(r){async function t(n,e){let s=ke();if(!s||Ie())return r.call(this,n,e);let a=performance.now(),o=F(),d=p(),c=o?.traceId??p(),u=n?.model??"unknown",l=n?.stream===!0;try{let i=await r.call(this,n,e);if(l){let g=i[Symbol.asyncIterator]?.bind(i);return g&&(i[Symbol.asyncIterator]=function(){let m=g(),h=0,w=0,y="",f;return{async next(){let{value:I,done:k}=await m.next();return k?(s.enqueue({event_id:p(),project_id:"",trace_id:c,span_id:d,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:u,operation:"messages",latency_ms:performance.now()-a,ttft_ms:f,stream:!0,input_tokens:h,output_tokens:w,finish_reason:y,status:"success",span_kind:"llm",agent_id:o?.agentId,agent_name:o?.agentName,agent_role:o?.agentRole,environment:s.environment,sdk_version:x}),{value:I,done:!0}):(f||(f=performance.now()-a),I?.type==="message_delta"&&I?.usage&&(w=I.usage.output_tokens??w),I?.type==="message_start"&&I?.message?.usage&&(h=I.message.usage.input_tokens??h),I?.type==="message_delta"&&I?.delta?.stop_reason&&(y=I.delta.stop_reason),{value:I,done:!1})},[Symbol.asyncIterator](){return this}}}),i}return s.enqueue({event_id:p(),project_id:"",trace_id:c,span_id:d,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:i.model??u,operation:"messages",latency_ms:performance.now()-a,stream:!1,input_tokens:i.usage?.input_tokens??0,output_tokens:i.usage?.output_tokens??0,finish_reason:i.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(i.content?.filter(g=>g.type==="text").map(g=>g.text).join("")??"")}:{},environment:s.environment,sdk_version:x}),i}catch(i){throw s.enqueue({event_id:p(),project_id:"",trace_id:c,span_id:d,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:u,operation:"messages",latency_ms:performance.now()-a,stream:l,input_tokens:0,output_tokens:0,status:"error",error_code:String(i.status??"unknown"),error_message:String(i.message).slice(0,500),span_kind:"llm",environment:s.environment,sdk_version:x}),i}}return t.__zespanAutoPatched=!0,t}function Sn(){if(!Ge&&Oe)try{let r=Oe.Messages;if(!r?.prototype)return;let t=r.prototype.create;if(typeof t=="function"&&!t.__zespanAutoPatched){r.prototype.create=Pt(t),Ge=!0;return}if(typeof t!="function"){if(r.__zespanAutoPatched){Ge=!0;return}let n=r;class e extends n{constructor(...o){if(super(...o),Object.prototype.hasOwnProperty.call(this,"create")&&!this.create?.__zespanAutoPatched){let d=this.create;this.create=Pt(d)}}}e.__zespanAutoPatched=!0;let s=Object.getOwnPropertyDescriptor(Oe,"Messages");if(s&&(s.writable!==!1||s.configurable))try{Object.defineProperty(Oe,"Messages",{value:e,writable:!0,enumerable:!0,configurable:!0}),Ge=!0}catch{}}}catch{}}function vn(){if(!Rt&&ct)try{let e=function(s,a){let o=n.call(this,s,a),d=o.generateContent?.bind(o),c=o.generateContentStream?.bind(o);return!d||d.__zespanAutoPatched||(o.generateContent=async u=>{let l=ke();if(!l||Ie())return d(u);let i=performance.now(),g=F(),m=p(),h=g?.traceId??p(),w=s.model??"gemini-1.5-flash";try{let y=await d(u),f=y.response?.usageMetadata;return l.enqueue({event_id:p(),project_id:"",trace_id:h,span_id:m,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google",model:w,operation:"generateContent",latency_ms:performance.now()-i,stream:!1,input_tokens:f?.promptTokenCount??0,output_tokens:f?.candidatesTokenCount??0,finish_reason:y.response?.candidates?.[0]?.finishReason??"",status:"success",span_kind:"llm",environment:l.environment,sdk_version:x}),y}catch(y){throw l.enqueue({event_id:p(),project_id:"",trace_id:h,span_id:m,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google",model:w,operation:"generateContent",latency_ms:performance.now()-i,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_message:String(y.message).slice(0,500),span_kind:"llm",environment:l.environment,sdk_version:x}),y}},o.generateContent.__zespanAutoPatched=!0,c&&(o.generateContentStream=async u=>{let l=ke();if(!l||Ie())return c(u);let i=performance.now(),g=F(),m=p(),h=g?.traceId??p(),w=s.model??"gemini-1.5-flash",y=await c(u),f=y.stream,I=0,k=0,A;return y.stream=(async function*(){try{for await(let _ of f){A||(A=performance.now()-i);let R=_.usageMetadata;R&&(I=R.promptTokenCount??I,k=R.candidatesTokenCount??k),yield _}l.enqueue({event_id:p(),project_id:"",trace_id:h,span_id:m,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google",model:w,operation:"generateContent",latency_ms:performance.now()-i,ttft_ms:A,stream:!0,input_tokens:I,output_tokens:k,status:"success",span_kind:"llm",environment:l.environment,sdk_version:x})}catch(_){throw l.enqueue({event_id:p(),project_id:"",trace_id:h,span_id:m,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google",model:w,operation:"generateContent",latency_ms:performance.now()-i,ttft_ms:A,stream:!0,input_tokens:I,output_tokens:k,status:"error",error_message:String(_.message).slice(0,500),span_kind:"llm",environment:l.environment,sdk_version:x}),_}})(),y})),o};var r=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,Rt=!0}catch{}}function wn(){if(!Nt&&dt)try{let r=dt?.Models;if(!r?.prototype||r.prototype.generateContent?.__zespanAutoPatched)return;let t=r.prototype.generateContent,n=r.prototype.generateContentStream,e=r.prototype.embedContent;r.prototype.generateContent=async function(s){let a=ke();if(!a||Ie())return t.call(this,s);let o=performance.now(),d=F(),c=p(),u=d?.traceId??p(),l=s?.model??"gemini";try{let i=await t.call(this,s),g=i?.usageMetadata,m=(i?.candidates?.[0]?.content?.parts??[]).filter(h=>h?.inlineData);return a.enqueue({event_id:p(),project_id:"",trace_id:u,span_id:c,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:"google",model:l,operation:"generateContent",latency_ms:performance.now()-o,stream:!1,input_tokens:Number(g?.promptTokenCount||0),output_tokens:Number(g?.candidatesTokenCount||0),cached_tokens:Number(g?.cachedContentTokenCount||0),reasoning_tokens:Number(g?.thoughtsTokenCount||0),finish_reason:i?.candidates?.[0]?.finishReason??"",status:"success",span_kind:m.length>0?"image_gen":"llm",...m.length>0?{metadata:{num_images:m.length}}:{},agent_id:d?.agentId,agent_name:d?.agentName,agent_role:d?.agentRole,environment:a.environment,sdk_version:x}),i}catch(i){throw a.enqueue({event_id:p(),project_id:"",trace_id:u,span_id:c,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:"google",model:l,operation:"generateContent",latency_ms:performance.now()-o,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_code:String(i.status??i.code??"unknown"),error_message:String(i.message).slice(0,500),span_kind:"llm",environment:a.environment,sdk_version:x}),i}},r.prototype.generateContent.__zespanAutoPatched=!0,n&&(r.prototype.generateContentStream=async function(s){let a=ke();if(!a||Ie())return n.call(this,s);let o=performance.now(),d=F(),c=p(),u=d?.traceId??p(),l=s?.model??"gemini",i=await n.call(this,s);return(async function*(){let g=0,m=0,h=0,w=0,y,f="";try{for await(let I of i){y||(y=performance.now()-o);let k=I?.usageMetadata;k&&(g=Number(k.promptTokenCount||g),m=Number(k.candidatesTokenCount||m),h=Number(k.cachedContentTokenCount||h),w=Number(k.thoughtsTokenCount||w)),I?.candidates?.[0]?.finishReason&&(f=I.candidates[0].finishReason),yield I}a.enqueue({event_id:p(),project_id:"",trace_id:u,span_id:c,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:"google",model:l,operation:"generateContent",latency_ms:performance.now()-o,ttft_ms:y,stream:!0,input_tokens:g,output_tokens:m,cached_tokens:h,reasoning_tokens:w,finish_reason:f||"stream_end",status:"success",span_kind:"llm",agent_id:d?.agentId,agent_name:d?.agentName,agent_role:d?.agentRole,environment:a.environment,sdk_version:x})}catch(I){throw a.enqueue({event_id:p(),project_id:"",trace_id:u,span_id:c,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:"google",model:l,operation:"generateContent",latency_ms:performance.now()-o,ttft_ms:y,stream:!0,input_tokens:g,output_tokens:m,status:"error",error_code:String(I.status??I.code??"unknown"),error_message:String(I.message).slice(0,500),span_kind:"llm",environment:a.environment,sdk_version:x}),I}})()},r.prototype.generateContentStream.__zespanAutoPatched=!0),e&&(r.prototype.embedContent=async function(s){let a=ke();if(!a||Ie())return e.call(this,s);let o=performance.now(),d=F(),c=p(),u=d?.traceId??p(),l=s?.model??"gemini-embedding";try{let i=await e.call(this,s),g=i?.usageMetadata;return a.enqueue({event_id:p(),project_id:"",trace_id:u,span_id:c,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:"google",model:l,operation:"embedding",latency_ms:performance.now()-o,stream:!1,input_tokens:Number(g?.promptTokenCount||0),output_tokens:0,status:"success",span_kind:"embedding",agent_id:d?.agentId,agent_name:d?.agentName,agent_role:d?.agentRole,environment:a.environment,sdk_version:x}),i}catch(i){throw a.enqueue({event_id:p(),project_id:"",trace_id:u,span_id:c,parent_span_id:d?.spanId,timestamp:new Date().toISOString(),provider:"google",model:l,operation:"embedding",latency_ms:performance.now()-o,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_code:String(i.status??i.code??"unknown"),error_message:String(i.message).slice(0,500),span_kind:"embedding",environment:a.environment,sdk_version:x}),i}},r.prototype.embedContent.__zespanAutoPatched=!0),Nt=!0}catch{}}function jt(){In(),Sn(),vn(),wn()}function qe(){jt()}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 we(this),t.enableZespanPilot!==!1&&t.projectId?this.configClient=new Fe({baseUrl:this.baseURL,apiKey:this.apiKey,projectId:t.projectId}):this.configClient=null,this.batch=new Me(this,{maxSize:t.batchSize??50,flushInterval:t.flushInterval??2e3},this.configClient??void 0),process.once("beforeExit",()=>this.batch.flush()),this.enableOTel&&je({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 Ot(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 ye().startSpan(t.name,{kind:t.kind??ve.INTERNAL,attributes:t.attributes??{}})}async withSpan(t,n,e){let a=ye().startSpan(t,{kind:e?.kind??ve.INTERNAL,attributes:e?.attributes??{}});try{let o=await n(a);return a.setStatus({code:Se.OK}),o}catch(o){throw a.setStatus({code:Se.ERROR,message:o.message}),a.recordException(o),o}finally{a.end()}}},ut=null,pt=r=>{ut=new Ke(r)},E=()=>{if(!ut)throw new Error("Call zespan.init() first");return ut},bn=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 ee(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 ne(r,t){return t?.modifiedText?t.modifiedText:r}async function Z(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 V(n.phase,s.results);return s}catch(s){if(s instanceof V||t.failClosed)throw s;return null}}function te(r){if(!Array.isArray(r))return{systemPrompt:void 0,conversationMessages:[]};let t=[],n=[];for(let e of r)if((typeof e?.role=="string"?e.role:e?._getType?.()??"")==="system"){let a=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??"");a&&t.push(a)}else n.push(e);return{systemPrompt:t.length?t.join(`
|
|
3
|
+
`):void 0,conversationMessages:n}}function Cn(r){let t={};try{let n=[],e=[];for(let s of r){let a=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(a),e.push(a)}for(let s of n){try{let a=s.match(/You are the (\w+)Agent/i);if(a){t.agent_name=a[1];break}}catch{}try{let a=s.match(/Agent Name:\s*(\w+)/i);if(a){t.agent_name=a[1];break}}catch{}}for(let s of n)try{let a=s.match(/Role:\s*(orchestrator|specialist|planner|executor|critic)/i);if(a?.[1]){t.agent_role=a[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 oe(r,t){let n=Cn(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 re(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 a=s?.function?.name;if(typeof a=="string"&&a){n.includes(a)||n.push(a);try{e[a]=typeof s.function.arguments=="string"?JSON.parse(s.function.arguments):s.function.arguments??{}}catch{e[a]=s.function?.arguments??{}}}}catch{}return{tools_used:n,tool_call_args:e}}catch{return{tools_used:[],tool_call_args:{}}}}function me(r){try{let t={};for(let s of r)try{let a=typeof s?.index=="number"?s.index:0;t[a]||(t[a]={name:"",argumentsStr:""}),s?.function?.name&&(t[a].name+=s.function.name),s?.function?.arguments&&(t[a].argumentsStr+=s.function.arguments)}catch{}let n=[],e={};for(let s of Object.values(t))try{let a=s.name;if(a&&!n.includes(a)){n.push(a);try{e[a]=s.argumentsStr?JSON.parse(s.argumentsStr):{}}catch{e[a]=s.argumentsStr}}}catch{}return{tools_used:n,tool_call_args:e}}catch{return{tools_used:[],tool_call_args:{}}}}import{createHash as Tn}from"crypto";import{SpanKind as On,SpanStatusCode as Ce,trace as An}from"@opentelemetry/api";async function ge(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 a=Math.min(1e3*Math.pow(t.backoffMultiplier,e),3e4);await xn(a)}throw n}function xn(r){return new Promise(t=>setTimeout(t,r))}var mt=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()}}},Dt=new Map;function _e(r,t){let n=Dt.get(r);return n||(n=new mt(t),Dt.set(r,n)),n}var ze=r=>Tn("sha256").update(r).digest("hex");function Rn(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*Nn(r,t){let{client:n,t0:e,traceId:s,spanId:a,parent:o,params:d,otelSpan:c,model:u,agentCtx:l,autoDetected:i,conversation_turn:g,message_role:m,tool_definitions:h,provider:w,operation:y}=t,f=w??"openai",I,k=0,A=0,_=0,R="",S=u,b=[],C="",j=[],{systemPrompt:O,conversationMessages:v}=te(d.messages??[]);try{for await(let D of r){I||(I=performance.now()-e,n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:f,model:S,operation:y,latency_ms:I,ttft_ms:I,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:ze(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(v),...O?{system_prompt:n.redactForStorage(O)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x,finish_reason:"stream_start",...l,span_kind:"llm",tool_definitions:h,conversation_turn:g,message_role:m,metadata:{auto_detected:Object.keys(i).length?i:void 0}}),c&&c.setAttribute("gen_ai.response.time_to_first_token",I));let M=D.choices?.[0]?.delta?.content;if(M&&(A++,C+=M),D.choices?.[0]?.finish_reason&&(R=D.choices[0].finish_reason),D.model&&(S=D.model),D.usage&&(k=Number(D.usage.prompt_tokens||k||0),A=Number(D.usage.completion_tokens||A||0),_=Number(D.usage.prompt_tokens_details?.cached_tokens||_||0)),D.choices?.[0]?.delta?.tool_calls)for(let B of D.choices[0].delta.tool_calls)j.push(B),B.function?.name&&!b.includes(B.function.name)&&b.push(B.function.name);yield D}let{tools_used:P,tool_call_args:N}=me(j),q=P.length>0?P:b,T=performance.now()-e;n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:f,model:S,operation:y,latency_ms:T,ttft_ms:I,stream:!0,input_tokens:k,output_tokens:A,cached_tokens:_,status:"success",finish_reason:R||"stream_end",tools_used:q,tool_call_args:Object.keys(N).length?N:void 0,prompt_hash:ze(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(v),completion_text:n.redactForStorage(C),...O?{system_prompt:n.redactForStorage(O)}:{}}:{},...q.length>0&&C?{decision_rationale:C}:{},...l,span_kind:"llm",tool_definitions:h,conversation_turn:g,message_role:m,metadata:{usage:{prompt_tokens:k,completion_tokens:A,cached_tokens:_},auto_detected:Object.keys(i).length?i:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x}),c&&(c.setAttribute("gen_ai.usage.output_tokens",A),c.setAttribute("gen_ai.usage.total_tokens",A),c.setAttribute("gen_ai.response.finish_reason",R||"stream_end"),c.setAttribute("gen_ai.response.latency",T))}catch(P){throw n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:f,model:S,operation:y,latency_ms:performance.now()-e,ttft_ms:I,stream:!0,input_tokens:0,output_tokens:A,cost_usd:0,status:"error",error_code:String(P.status??"unknown"),error_message:String(P.message).slice(0,500),...l,span_kind:"llm",conversation_turn:g,message_role:m,metadata:{auto_detected:Object.keys(i).length?i:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x}),c&&(c.setStatus({code:Ce.ERROR,message:P.message}),c.recordException(P)),P}}function Be(r,t={}){let n=r,e=n.chat.completions.create.bind(n.chat.completions),s=ee(t.guardrails,"chat"),a=t.providerOverride??"openai",o=t.operation??"chat";return n.chat.completions.create=async(d,c)=>{let u=E(),l=performance.now(),i=F(),g=p(),m=i?.traceId??p(),h=d.model??"gpt-4",w;if(u.configClient){let v=u.configClient.getAbTest(o);if(v)Math.random()<v.splitPercentage/100?(h=v.candidateModel,w={ab_test:!0,ab_variant:"candidate"}):w={ab_test:!0,ab_variant:"control"},d={...d,model:h};else{let P=u.configClient.getModel(o,h);if(P!==h){let N=h;h=P,d={...d,model:P},u.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:g,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:a,model:P,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:N,override_model:P},environment:u.environment,sdk_version:x})}}}let y=null,f=null,I=i?.agentId?{agent_id:i.agentId,agent_name:i.agentName,agent_role:i.agentRole,agent_framework:i.agentFramework,parent_agent_id:i.parentAgentId}:{},k=oe(d.messages??[],d.model),{conversation_turn:A,message_role:_}=ae(d.messages??[]),R=d.tools?.length?re(d.tools):void 0,S=null,b=u.enableOTel?An.getTracer("zespan-sdk"):null;b&&(S=b.startSpan(`chat.completions ${h}`,{kind:On.CLIENT,attributes:Rn(d,h,a)}));let C;if(u.configClient){let v=u.configClient.getTimeout(o);v&&(C=AbortSignal.timeout(v.timeoutMs))}let j=u.configClient?.getRetryPolicy(o)??null,O=u.configClient?.getConcurrencyLimit(o)??null;try{y=await Z(u,s,{text:JSON.stringify(d.messages??[]),phase:"pre",traceId:m,spanId:g,model:h,operation:o,agentName:i?.agentName});let v=()=>ge(()=>W({traceId:m,spanId:g,userId:i?.userId,sessionId:i?.sessionId,tags:i?.tags},()=>e(d,C?{...c,signal:C}:c)),j),P=await(O!==null?_e(`${u.projectId??"default"}:${o}`,O).run(v):v());if(d.stream===!0)return Nn(P,{client:u,t0:l,traceId:m,spanId:g,parent:i,params:d,otelSpan:S,model:h,agentCtx:I,autoDetected:k,conversation_turn:A,message_role:_,tool_definitions:R,provider:a,operation:o});let{prompt_tokens:q,completion_tokens:T,prompt_tokens_details:D}=P.usage??{},M=D?.cached_tokens??0,B=P.model??d.model,K=P.usage?.completion_tokens_details?.reasoning_tokens,z=P.choices?.[0]?.message?.content||"";f=await Z(u,s,{text:z,phase:"post",traceId:m,spanId:g,model:B,operation:o,agentName:i?.agentName});let U=ne(z,f);U!==z&&typeof P?.choices?.[0]?.message?.content=="string"&&(P.choices[0].message.content=U);let{tools_used:G,tool_call_args:H}=Le(P.choices??[]),{systemPrompt:L,conversationMessages:J}=te(d.messages??[]);return u.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:g,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:a,model:B,operation:o,latency_ms:performance.now()-l,stream:!1,input_tokens:q??0,output_tokens:T??0,cached_tokens:M,...K?{reasoning_tokens:K}:{},status:"success",finish_reason:P.choices?.[0]?.finish_reason??"",tools_used:G.length>0?G:P.choices?.[0]?.message?.tool_calls?.map(X=>X.function.name)??[],tool_call_args:Object.keys(H).length?H:void 0,prompt_hash:ze(JSON.stringify(d.messages)),...u.storePrompts?{prompt_text:u.redactForStorage(J),completion_text:u.redactForStorage(U),...L?{system_prompt:u.redactForStorage(L)}:{}}:{},...G.length>0&&z?{decision_rationale:z}:{},...I,span_kind:"llm",tool_definitions:R,conversation_turn:A,message_role:_,metadata:{usage:P.usage??null,guardrails:{pre:y?{matched:y.results.length,modified:!!y.modifiedText}:null,post:f?{matched:f.results.length,modified:!!f.modifiedText}:null},auto_detected:Object.keys(k).length?k:void 0,...w??{}},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:u.environment,sdk_version:x}),S&&(S.setAttribute("gen_ai.usage.input_tokens",q??0),S.setAttribute("gen_ai.usage.output_tokens",T??0),S.setAttribute("gen_ai.usage.total_tokens",(q??0)+(T??0)),S.setAttribute("gen_ai.usage.cached_tokens",M),S.setAttribute("gen_ai.response.finish_reason",P.choices?.[0]?.finish_reason??""),S.setAttribute("gen_ai.response.latency",performance.now()-l),S.setStatus({code:Ce.OK}),S.end()),P}catch(v){if(v?.name==="TimeoutError"||v?.code==="ABORT_ERR"||C&&C.aborted)throw u.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:g,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:a,model:d.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(v.message).slice(0,500),...I,span_kind:"llm",conversation_turn:A,message_role:_,metadata:{auto_detected:Object.keys(k).length?k:void 0},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:u.environment,sdk_version:x}),S&&(S.setStatus({code:Ce.ERROR,message:v.message}),S.recordException(v),S.end()),v;let N=u.configClient?.getFallback(o)??null;if(N&&N.model){let D=String(v.message??"")+String(v.status??"")+String(v.code??"");if(N.conditions.length===0||N.conditions.some(B=>D.toLowerCase().includes(B.toLowerCase()))){let B=h,K={...d,model:N.model};u.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:g,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:a,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(v.message).slice(0,200)},environment:u.environment,sdk_version:x});try{let z=await W({traceId:m,spanId:g,userId:i?.userId,sessionId:i?.sessionId,tags:i?.tags},()=>e(K,c)),{prompt_tokens:U,completion_tokens:G,prompt_tokens_details:H}=z.usage??{},L=H?.cached_tokens??0,J=z.model??N.model;return u.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:g,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:a,model:J,operation:o,latency_ms:performance.now()-l,stream:!1,input_tokens:U??0,output_tokens:G??0,cached_tokens:L,status:"success",finish_reason:z.choices?.[0]?.finish_reason??"",prompt_hash:ze(JSON.stringify(d.messages)),...I,span_kind:"llm",conversation_turn:A,message_role:_,metadata:{usage:z.usage??null,fallback_used:!0,original_model:B,auto_detected:Object.keys(k).length?k:void 0},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:u.environment,sdk_version:x}),S&&(S.setStatus({code:Ce.OK}),S.end()),z}catch(z){throw S&&(S.setStatus({code:Ce.ERROR,message:z.message}),S.recordException(z),S.end()),z}}}let q=v instanceof V,T=q?"error":v.status===429?"rate_limited":v.status===408?"timeout":"error";throw u.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:g,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:a,model:d.model,operation:o,latency_ms:performance.now()-l,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:T,error_code:q?"guardrail_blocked":String(v.status??"unknown"),error_message:String(v.message).slice(0,500),...I,span_kind:"llm",conversation_turn:A,message_role:_,metadata:{auto_detected:Object.keys(k).length?k:void 0},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:u.environment,sdk_version:x}),S&&(S.setStatus({code:Ce.ERROR,message:v.message}),S.recordException(v),S.end()),v}},r}import{createHash as Mn}from"crypto";import{SpanKind as Pn,SpanStatusCode as xe,trace as jn}from"@opentelemetry/api";var $e=r=>Mn("sha256").update(r).digest("hex");function Dn(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(`
|
|
4
|
+
`);if(e.length>0)return[{role:"system",content:e},...t]}return t}function Fn(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*En(r,t){let{client:n,t0:e,traceId:s,spanId:a,parent:o,params:d,otelSpan:c,agentCtx:u,tool_definitions:l,autoDetected:i,conversation_turn:g,message_role:m,operation:h,detectionMessages:w}=t,y,f=0,I=0,k=0,A=0,_="",R=d.model,S=[],b="",{systemPrompt:C,conversationMessages:j}=te(w);try{for await(let v of r){if(!y&&v.type==="content_block_start"&&(y=performance.now()-e,n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:R,operation:h,latency_ms:y,ttft_ms:y,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:$e(JSON.stringify(w)),...n.storePrompts?{prompt_text:n.redactForStorage(j),...C?{system_prompt:n.redactForStorage(C)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x,finish_reason:"stream_start",...u,span_kind:"llm",tool_definitions:l,conversation_turn:g,message_role:m,metadata:{auto_detected:Object.keys(i).length?i:void 0}}),c&&c.setAttribute("gen_ai.response.time_to_first_token",y)),v.type==="content_block_delta"&&v.delta?.type==="text_delta"&&(A++,b+=v.delta.text||""),v.type==="message_delta"&&v.delta?.stop_reason&&(_=v.delta.stop_reason),v.type==="message_delta"&&v.usage&&(f=Number(v.usage.input_tokens||f||0),A=Number(v.usage.output_tokens||A||0),I=Number(v.usage.cache_read_input_tokens||I||0)),v.type==="message_start"&&v.message?.usage&&(f=Number(v.message.usage.input_tokens||f||0),A=Number(v.message.usage.output_tokens||A||0),I=Number(v.message.usage.cache_read_input_tokens||I||0)),v.type==="content_block_start"&&v.content_block?.type==="tool_use"){let P=v.content_block.name;P&&!S.includes(P)&&S.push(P)}yield v}let O=performance.now()-e;n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:R,operation:h,latency_ms:O,ttft_ms:y,stream:!0,input_tokens:f,output_tokens:A,cached_tokens:I,reasoning_tokens:k,status:"success",finish_reason:_||"stream_end",tools_used:S,prompt_hash:$e(JSON.stringify(w)),...n.storePrompts?{prompt_text:n.redactForStorage(j),completion_text:n.redactForStorage(b),...C?{system_prompt:n.redactForStorage(C)}:{}}:{},...u,span_kind:"llm",tool_definitions:l,conversation_turn:g,message_role:m,metadata:{usage:{input_tokens:f,output_tokens:A,cache_read_input_tokens:I,reasoning_tokens:k},auto_detected:Object.keys(i).length?i:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x}),c&&(c.setAttribute("gen_ai.usage.output_tokens",A),c.setAttribute("gen_ai.response.finish_reason",_||"stream_end"))}catch(O){throw n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:R,operation:h,latency_ms:performance.now()-e,ttft_ms:y,stream:!0,input_tokens:0,output_tokens:A,cost_usd:0,status:"error",error_code:String(O.status??O.type??"unknown"),error_message:String(O.message).slice(0,500),...u,span_kind:"llm",conversation_turn:g,message_role:m,metadata:{auto_detected:Object.keys(i).length?i:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x}),c&&(c.setStatus({code:xe.ERROR,message:O.message}),c.recordException(O)),O}}function Ft(r,t={}){let n=r,e=n.messages.create.bind(n.messages),s=ee(t.guardrails,"chat"),a=t.operation??"chat";return n.messages.create=async(o,d)=>{let c=E(),u=performance.now(),l=F(),i=p(),g=l?.traceId??p(),m=o.model??"claude-3-5-sonnet-20241022",h;if(c.configClient){let T=c.configClient.getAbTest(a);if(T)Math.random()<T.splitPercentage/100?(m=T.candidateModel,h={ab_test:!0,ab_variant:"candidate"}):h={ab_test:!0,ab_variant:"control"},o={...o,model:m};else{let D=c.configClient.getModel(a,m);if(D!==m){let M=m;m=D,o={...o,model:D},c.enqueue({event_id:p(),project_id:"",trace_id:g,span_id:i,parent_span_id:l?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:D,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:M,override_model:D},environment:c.environment,sdk_version:x})}}}let w=null,y=null,f=Dn(o),{systemPrompt:I,conversationMessages:k}=te(f),A=oe(f,m),{conversation_turn:_,message_role:R}=ae(f),S=l?.agentName??A.agent_name,b=l?.agentRole??A.agent_role,C=l?.agentId||S||b?{agent_id:l?.agentId,agent_name:S,agent_role:b,agent_framework:l?.agentFramework,parent_agent_id:l?.parentAgentId}:{},j=o.tools?.length?re(o.tools):void 0,O=null,v=c.enableOTel?jn.getTracer("zespan-sdk"):null;v&&(O=v.startSpan(`messages.create ${m}`,{kind:Pn.CLIENT,attributes:Fn(o,m)}));let P;if(c.configClient){let T=c.configClient.getTimeout(a);T&&(P=AbortSignal.timeout(T.timeoutMs))}let N=c.configClient?.getRetryPolicy(a)??null,q=c.configClient?.getConcurrencyLimit(a)??null;try{w=await Z(c,s,{text:JSON.stringify(o.messages??[]),phase:"pre",traceId:g,spanId:i,model:m,operation:a,agentName:l?.agentName});let T=()=>ge(()=>W({traceId:g,spanId:i,userId:l?.userId,sessionId:l?.sessionId,tags:l?.tags,agentId:l?.agentId,agentName:S,agentRole:b,agentFramework:l?.agentFramework,parentAgentId:l?.parentAgentId,conversationTurn:_,availableTools:j},()=>e(o,P?{...d,signal:P}:d)),N),D=await(q!==null?_e(`${c.projectId??"default"}:${a}`,q).run(T):T());if(o.stream===!0)return En(D,{client:c,t0:u,traceId:g,spanId:i,parent:l,params:o,otelSpan:O,agentCtx:C,tool_definitions:j,autoDetected:A,conversation_turn:_,message_role:R,operation:a,detectionMessages:f});let B=D.usage?.input_tokens??0,K=D.usage?.output_tokens??0,z=D.usage?.cache_read_input_tokens??0,U=D.model??o.model,G=D.content?.find($=>$.type==="text")?.text||"",H=(D.content??[]).filter($=>$.type==="tool_use"),L=H.map($=>$.name).filter(Boolean),J={};for(let $ of H)$.name&&(J[$.name]=$.input??{});y=await Z(c,s,{text:G,phase:"post",traceId:g,spanId:i,model:U,operation:a,agentName:l?.agentName});let X=ne(G,y);if(X!==G){let $=D.content?.find(Y=>Y.type==="text");$&&typeof $.text=="string"&&($.text=X)}return c.enqueue({event_id:p(),project_id:"",trace_id:g,span_id:i,parent_span_id:l?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:U,operation:a,latency_ms:performance.now()-u,stream:!1,input_tokens:B,output_tokens:K,cached_tokens:z,status:"success",finish_reason:D.stop_reason??"",tools_used:L.length>0?L:D.content?.filter($=>$.type==="tool_use").map($=>$.name)??[],tool_call_args:Object.keys(J).length?J:void 0,prompt_hash:$e(JSON.stringify(f)),...c.storePrompts?{prompt_text:c.redactForStorage(k),completion_text:c.redactForStorage(X),...I?{system_prompt:c.redactForStorage(I)}:{}}:{},...H.length>0&&G?{decision_rationale:G}:{},...C,span_kind:"llm",tool_definitions:j,conversation_turn:_,message_role:R,metadata:{usage:D.usage??null,guardrails:{pre:w?{matched:w.results.length,modified:!!w.modifiedText}:null,post:y?{matched:y.results.length,modified:!!y.modifiedText}:null},auto_detected:Object.keys(A).length?A:void 0,...h??{}},user_id:l?.userId,session_id:l?.sessionId,tags:l?.tags,environment:c.environment,sdk_version:x}),O&&(O.setAttribute("gen_ai.usage.input_tokens",B),O.setAttribute("gen_ai.usage.output_tokens",K),O.setAttribute("gen_ai.usage.cached_tokens",z),O.setAttribute("gen_ai.response.finish_reason",D.stop_reason??""),O.setAttribute("gen_ai.response.latency",performance.now()-u),O.setStatus({code:xe.OK}),O.end()),D}catch(T){if(T?.name==="TimeoutError"||T?.code==="ABORT_ERR"||P&&P.aborted)throw c.enqueue({event_id:p(),project_id:"",trace_id:g,span_id:i,parent_span_id:l?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:o.model,operation:a,latency_ms:performance.now()-u,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:"timeout",error_message:String(T.message).slice(0,500),...C,span_kind:"llm",conversation_turn:_,message_role:R,metadata:{auto_detected:Object.keys(A).length?A:void 0},user_id:l?.userId,session_id:l?.sessionId,tags:l?.tags,environment:c.environment,sdk_version:x}),O&&(O.setStatus({code:xe.ERROR,message:T.message}),O.recordException(T),O.end()),T;let M=c.configClient?.getFallback(a)??null;if(M&&M.model){let z=String(T.message??"")+String(T.status??"")+String(T.code??"");if(M.conditions.length===0||M.conditions.some(G=>z.toLowerCase().includes(G.toLowerCase()))){let G=m,H={...o,model:M.model};c.enqueue({event_id:p(),project_id:"",trace_id:g,span_id:i,parent_span_id:l?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:M.model,operation:a,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:G,fallback_model:M.model,reason:String(T.message).slice(0,200)},environment:c.environment,sdk_version:x});try{let L=await W({traceId:g,spanId:i,userId:l?.userId,sessionId:l?.sessionId,tags:l?.tags},()=>e(H,d)),J=L.usage?.input_tokens??0,X=L.usage?.output_tokens??0,$=L.usage?.cache_read_input_tokens??0,Y=L.model??M.model;return c.enqueue({event_id:p(),project_id:"",trace_id:g,span_id:i,parent_span_id:l?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:Y,operation:a,latency_ms:performance.now()-u,stream:!1,input_tokens:J,output_tokens:X,cached_tokens:$,status:"success",finish_reason:L.stop_reason??"",prompt_hash:$e(JSON.stringify(f)),...C,span_kind:"llm",conversation_turn:_,message_role:R,metadata:{usage:L.usage??null,fallback_used:!0,original_model:G,auto_detected:Object.keys(A).length?A:void 0},user_id:l?.userId,session_id:l?.sessionId,tags:l?.tags,environment:c.environment,sdk_version:x}),O&&(O.setStatus({code:xe.OK}),O.end()),L}catch(L){throw O&&(O.setStatus({code:xe.ERROR,message:L.message}),O.recordException(L),O.end()),L}}}let B=T instanceof V,K=B?"error":T.status===429||T.type==="overloaded_error"?"rate_limited":T.status===408?"timeout":"error";throw c.enqueue({event_id:p(),project_id:"",trace_id:g,span_id:i,parent_span_id:l?.spanId,timestamp:new Date().toISOString(),provider:"anthropic",model:o.model,operation:a,latency_ms:performance.now()-u,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:K,error_code:B?"guardrail_blocked":String(T.status??T.type??"unknown"),error_message:String(T.message).slice(0,500),...C,span_kind:"llm",conversation_turn:_,message_role:R,metadata:{auto_detected:Object.keys(A).length?A:void 0},user_id:l?.userId,session_id:l?.sessionId,tags:l?.tags,environment:c.environment,sdk_version:x}),O&&(O.setStatus({code:xe.ERROR,message:T.message}),O.recordException(T),O.end()),T}},r}import{createHash as Gn}from"crypto";import{SpanKind as We,SpanStatusCode as ce,trace as Je}from"@opentelemetry/api";var Ae=r=>Gn("sha256").update(r).digest("hex");function Ze(r){let t=o=>{if(typeof o=="string")return o;if(Array.isArray(o))return o.map(d=>typeof d=="string"?d:typeof d?.text=="string"?d.text:"").filter(Boolean).join(`
|
|
5
5
|
`);if(o&&typeof o=="object"){if(typeof o.text=="string")return o.text;if(Array.isArray(o.parts))return o.parts.map(d=>typeof d?.text=="string"?d.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 d=typeof o.role=="string"?o.role:"user",i=t(o);i.length>0&&e.push({role:d,content:i})}}return e}if(n&&typeof n=="object"){let o=typeof n.role=="string"?n.role:"user",d=t(n);d.length>0&&e.push({role:o,content:d})}return e}function Ve(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 pt(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*qt(a,t){let{lumiq:n,t0:e,traceId:s,spanId:r,parent:o,params:d,modelName:i,otelSpan:l,streamResult:p,agentCtx:c,tool_definitions:m,autoDetected:g,conversation_turn:f,message_role:b,operation:w}=t,y,k=0,h=0,C=0,_=0,N="",v=[],x="";try{for await(let S of a){y||(y=performance.now()-e,n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:w,latency_ms:y,ttft_ms:y,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:o?.userId,session_id:o?.sessionId,tags:o?.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:b,metadata:{auto_detected:Object.keys(g).length?g:void 0}}),l&&l.setAttribute("gen_ai.response.time_to_first_token",y));let M=(typeof S?.text=="function"?S.text():S?.text)||S?.candidates?.[0]?.content?.parts?.map(O=>O?.text||"").join("")||"";M&&(k++,x+=M),S.candidates?.[0]?.finishReason&&(N=S.candidates[0].finishReason);let A=Array.isArray(S?.functionCalls)?S.functionCalls:S?.functionCalls?[S.functionCalls]:[],G=(S?.candidates?.[0]?.content?.parts??[]).map(O=>O?.functionCall).filter(Boolean);for(let O of[...A,...G])O?.name&&!v.includes(O.name)&&v.push(O.name);yield S}let I=await p?.response;I?.usageMetadata&&(h=Number(I.usageMetadata.promptTokenCount||h||0),k=Number(I.usageMetadata.candidatesTokenCount||k||0),C=Number(I.usageMetadata.cachedContentTokenCount||C||0),_=Number(I.usageMetadata.thoughtsTokenCount||_||0));let F=pt(I);F&&(x=F);let T=performance.now()-e;n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:w,latency_ms:T,ttft_ms:y,stream:!0,input_tokens:h,output_tokens:k,cached_tokens:C,reasoning_tokens:_,status:"success",finish_reason:N||"stream_end",tools_used:v,prompt_hash:Ce(JSON.stringify(d)),...n.storePrompts?{prompt_text:n.redactForStorage(d),completion_text:n.redactForStorage(x)}:{},...c,span_kind:"llm",tool_definitions:m,conversation_turn:f,message_role:b,metadata:{usage:I?.usageMetadata??{promptTokenCount:h,candidatesTokenCount:k,cachedContentTokenCount:C,thoughtsTokenCount:_},auto_detected:Object.keys(g).length?g:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:"1.0.0"}),l&&(l.setAttribute("gen_ai.completion",x),l.setAttribute("gen_ai.usage.input_tokens",h),l.setAttribute("gen_ai.usage.output_tokens",k),l.setAttribute("gen_ai.usage.total_tokens",h+k),l.setAttribute("gen_ai.usage.cached_tokens",C),l.setAttribute("gen_ai.usage.reasoning_tokens",_),l.setAttribute("gen_ai.response.finish_reason",N||"stream_end"),l.setStatus({code:ie.OK}),l.end())}catch(I){throw n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:w,latency_ms:performance.now()-e,ttft_ms:y,stream:!0,input_tokens:0,output_tokens:k,cost_usd:0,status:"error",error_code:String(I.status??"unknown"),error_message:String(I.message).slice(0,500),...c,span_kind:"llm",conversation_turn:f,message_role:b,metadata:{auto_detected:Object.keys(g).length?g:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:"1.0.0"}),l&&(l.setStatus({code:ie.ERROR,message:I.message}),l.recordException(I),l.end()),I}}function Pt(a,t={}){let n=a,e=n.getGenerativeModel.bind(n),s=Q(t.guardrails,"chat"),r=t.operation??"chat";return n.getGenerativeModel=(o,d)=>{let i=o.model,l=j(),p=!1,c=i,m;if(l.configClient){let y=l.configClient.getAbTest(r);if(y)Math.random()<y.splitPercentage/100?(c=i,i=y.candidateModel,o={...o,model:i},p=!0,m={ab_test:!0,ab_variant:"candidate"}):m={ab_test:!0,ab_variant:"control"};else{let k=l.configClient.getModel(r,i);k!==i&&(c=i,i=k,o={...o,model:k},p=!0)}}let g=e(o,d),f=g.generateContent.bind(g);g.generateContent=async(y,k)=>{let h=j(),C=performance.now(),_=E(),N=u(),v=_?.traceId??u(),x=null,I=null,F=Ue(y),T=oe(F,i),{conversation_turn:S,message_role:M}=re(F),A=_?.agentName??T.agent_name,G=_?.agentRole??T.agent_role,O=_?.agentId||A||G?{agent_id:_?.agentId,agent_name:A,agent_role:G,agent_framework:_?.agentFramework,parent_agent_id:_?.parentAgentId}:{},q=y?.tools?.length?ae(y.tools):void 0,R=null,$=h.enableOTel?ze.getTracer("zespan-sdk"):null;$&&(R=$.startSpan(`generateContent ${i}`,{kind:Ze.CLIENT,attributes:Ve(y,i)})),p&&h.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:N,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:h.environment,sdk_version:"1.0.0"});try{x=await Z(h,s,{text:JSON.stringify(y??""),phase:"pre",traceId:v,spanId:N,model:i,operation:r,agentName:_?.agentName});let L;if(h.configClient){let z=h.configClient.getTimeout(r);z&&(L=AbortSignal.timeout(z.timeoutMs))}let K=h.configClient?.getRetryPolicy(r)??null,H=h.configClient?.getConcurrencyLimit(r)??null,P=()=>ke(()=>W({traceId:v,spanId:N,userId:_?.userId,sessionId:_?.sessionId,tags:_?.tags,agentId:_?.agentId,agentName:A,agentRole:G,agentFramework:_?.agentFramework,parentAgentId:_?.parentAgentId,conversationTurn:S,availableTools:q},()=>f(y,L?{...k,signal:L}:k)),K),V=await(H!==null?Ie(`${h.projectId??"default"}:${r}`,H).run(P):P()),D=V.response,J=D.usageMetadata?.promptTokenCount??0,X=D.usageMetadata?.candidatesTokenCount??0,B=D.usageMetadata?.cachedContentTokenCount??0,se=D.usageMetadata?.thoughtsTokenCount??0,ue=pt(D),pe=(D.candidates?.[0]?.content?.parts??[]).filter(z=>z?.functionCall),ne=pe.map(z=>z.functionCall.name).filter(Boolean),de=(D.candidates?.[0]?.content?.parts??[]).filter(z=>z?.text&&!z?.functionCall).map(z=>z.text).join(""),le={};for(let z of pe)z.functionCall?.name&&(le[z.functionCall.name]=z.functionCall.args??{});I=await Z(h,s,{text:ue,phase:"post",traceId:v,spanId:N,model:i,operation:r,agentName:_?.agentName});let me=te(ue,I);return h.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:N,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-C,stream:!1,input_tokens:J,output_tokens:X,cached_tokens:B,reasoning_tokens:se,status:"success",finish_reason:D.candidates?.[0]?.finishReason??"",tools_used:ne,tool_call_args:Object.keys(le).length?le:void 0,prompt_hash:Ce(JSON.stringify(y)),...h.storePrompts?{prompt_text:h.redactForStorage(y),completion_text:h.redactForStorage(me)}:{},...ne.length>0&&de?{decision_rationale:de}:{},...O,span_kind:"llm",tool_definitions:q,conversation_turn:S,message_role:M,metadata:{usage:D.usageMetadata??null,guardrails:{pre:x?{matched:x.results.length,modified:!!x.modifiedText}:null,post:I?{matched:I.results.length,modified:!!I.modifiedText}:null},auto_detected:Object.keys(T).length?T:void 0,...m??{}},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:h.environment,sdk_version:"1.0.0"}),R&&(R.setAttribute("gen_ai.completion",me),R.setAttribute("gen_ai.usage.input_tokens",J),R.setAttribute("gen_ai.usage.output_tokens",X),R.setAttribute("gen_ai.usage.total_tokens",J+X),R.setAttribute("gen_ai.usage.cached_tokens",B),R.setAttribute("gen_ai.usage.reasoning_tokens",se),R.setAttribute("gen_ai.response.finish_reason",D.candidates?.[0]?.finishReason??""),R.setAttribute("gen_ai.response.latency",performance.now()-C),R.setStatus({code:ie.OK}),R.end()),V}catch(L){let K=L.__signal;if(L?.name==="TimeoutError"||L?.code==="ABORT_ERR")throw h.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:N,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-C,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:"timeout",error_message:String(L.message).slice(0,500),...O,span_kind:"llm",conversation_turn:S,message_role:M,metadata:{auto_detected:Object.keys(T).length?T:void 0},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:h.environment,sdk_version:"1.0.0"}),R&&(R.setStatus({code:ie.ERROR,message:L.message}),R.recordException(L),R.end()),L;let P=h.configClient?.getFallback(r)??null;if(P&&P.model){let J=String(L.message??"")+String(L.status??"")+String(L.code??"");if(P.conditions.length===0||P.conditions.some(B=>J.toLowerCase().includes(B.toLowerCase()))){let B=i,se={...o,model:P.model},ue=e(se,d),pe=ue.generateContent.bind(ue);h.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:N,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:P.model,operation:r,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:B,fallback_model:P.model,reason:String(L.message).slice(0,200)},environment:h.environment,sdk_version:"1.0.0"});try{let ne=await W({traceId:v,spanId:N,userId:_?.userId,sessionId:_?.sessionId,tags:_?.tags},()=>pe(y,k)),de=ne.response,le=de.usageMetadata?.promptTokenCount??0,me=de.usageMetadata?.candidatesTokenCount??0,z=de.usageMetadata?.cachedContentTokenCount??0;return h.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:N,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:P.model,operation:r,latency_ms:performance.now()-C,stream:!1,input_tokens:le,output_tokens:me,cached_tokens:z,status:"success",finish_reason:de.candidates?.[0]?.finishReason??"",prompt_hash:Ce(JSON.stringify(y)),...O,span_kind:"llm",conversation_turn:S,message_role:M,metadata:{usage:de.usageMetadata??null,fallback_used:!0,original_model:B,auto_detected:Object.keys(T).length?T:void 0},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:h.environment,sdk_version:"1.0.0"}),R&&(R.setStatus({code:ie.OK}),R.end()),ne}catch(ne){throw R&&(R.setStatus({code:ie.ERROR,message:ne.message}),R.recordException(ne),R.end()),ne}}}let V=L instanceof U,D=V?"error":L.status===429?"rate_limited":L.status===408?"timeout":"error";throw h.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:N,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-C,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:D,error_code:V?"guardrail_blocked":String(L.status??"unknown"),error_message:String(L.message).slice(0,500),...O,span_kind:"llm",conversation_turn:S,message_role:M,metadata:{auto_detected:Object.keys(T).length?T:void 0},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:h.environment,sdk_version:"1.0.0"}),R&&(R.setStatus({code:ie.ERROR,message:L.message}),R.recordException(L),R.end()),L}};let b=g.generateContentStream.bind(g);g.generateContentStream=async(y,k)=>{let h=j(),C=performance.now(),_=E(),N=u(),v=_?.traceId??u(),x=Ue(y),I=oe(x,i),{conversation_turn:F,message_role:T}=re(x),S=_?.agentName??I.agent_name,M=_?.agentRole??I.agent_role,A=_?.agentId||S||M?{agent_id:_?.agentId,agent_name:S,agent_role:M,agent_framework:_?.agentFramework,parent_agent_id:_?.parentAgentId}:{},G=y?.tools?.length?ae(y.tools):void 0,O=null,q=h.enableOTel?ze.getTracer("zespan-sdk"):null;q&&(O=q.startSpan(`generateContentStream ${i}`,{kind:Ze.CLIENT,attributes:Ve(y,i)}));try{let R=await W({traceId:v,spanId:N,userId:_?.userId,sessionId:_?.sessionId,tags:_?.tags,agentId:_?.agentId,agentName:S,agentRole:M,agentFramework:_?.agentFramework,parentAgentId:_?.parentAgentId,conversationTurn:F,availableTools:G},()=>b(y,k));return O&&O.setAttribute("gen_ai.request.stream",!0),R.stream=qt(R.stream,{lumiq:h,t0:C,traceId:v,spanId:N,parent:_,params:y,modelName:i,otelSpan:O,streamResult:R,agentCtx:A,tool_definitions:G,autoDetected:I,conversation_turn:F,message_role:T,operation:r}),R}catch(R){let $=R instanceof U,L=$?"error":R.status===429?"rate_limited":R.status===408?"timeout":"error";throw h.enqueue({event_id:u(),project_id:"",trace_id:v,span_id:N,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-C,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:L,error_code:$?"guardrail_blocked":String(R.status??"unknown"),error_message:String(R.message).slice(0,500),...A,span_kind:"llm",conversation_turn:F,message_role:T,metadata:{auto_detected:Object.keys(I).length?I:void 0},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:h.environment,sdk_version:"1.0.0"}),O&&(O.setStatus({code:ie.ERROR,message:R.message}),O.recordException(R),O.end()),R}};let w=g.startChat?.bind(g);return w&&(g.startChat=y=>{let k=w(y),h=k.sendMessage?.bind(k),C=k.sendMessageStream?.bind(k);return h&&(k.sendMessage=async(_,N)=>{let v=j(),x=performance.now(),I=E(),F=u(),T=I?.traceId??u(),S={...y||{},message:_},M=null,A=null,G=Ue(S),O=oe(G,i),{conversation_turn:q,message_role:R}=re(G),$=I?.agentName??O.agent_name,L=I?.agentRole??O.agent_role,K=I?.agentId||$||L?{agent_id:I?.agentId,agent_name:$,agent_role:L,agent_framework:I?.agentFramework,parent_agent_id:I?.parentAgentId}:{},H=y?.tools?.length?ae(y.tools):void 0,P=null,V=v.enableOTel?ze.getTracer("zespan-sdk"):null;V&&(P=V.startSpan(`sendMessage ${i}`,{kind:Ze.CLIENT,attributes:Ve(S,i)}));try{M=await Z(v,s,{text:JSON.stringify(S),phase:"pre",traceId:T,spanId:F,model:i,operation:r,agentName:I?.agentName});let D=await W({traceId:T,spanId:F,userId:I?.userId,sessionId:I?.sessionId,tags:I?.tags,agentId:I?.agentId,agentName:$,agentRole:L,agentFramework:I?.agentFramework,parentAgentId:I?.parentAgentId,conversationTurn:q,availableTools:H},()=>h(_,N)),J=D.response,X=J.usageMetadata?.promptTokenCount??0,B=J.usageMetadata?.candidatesTokenCount??0,se=J.usageMetadata?.cachedContentTokenCount??0,ue=J.usageMetadata?.thoughtsTokenCount??0,pe=pt(J),ne=(J.candidates?.[0]?.content?.parts??[]).filter(z=>z?.functionCall),de=ne.map(z=>z.functionCall.name).filter(Boolean),le={};for(let z of ne)z.functionCall?.name&&(le[z.functionCall.name]=z.functionCall.args??{});A=await Z(v,s,{text:pe,phase:"post",traceId:T,spanId:F,model:i,operation:r,agentName:I?.agentName});let me=te(pe,A);return v.enqueue({event_id:u(),project_id:"",trace_id:T,span_id:F,parent_span_id:I?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-x,stream:!1,input_tokens:X,output_tokens:B,cached_tokens:se,reasoning_tokens:ue,status:"success",finish_reason:J.candidates?.[0]?.finishReason??"",tools_used:de,tool_call_args:Object.keys(le).length?le:void 0,prompt_hash:Ce(JSON.stringify(S)),...v.storePrompts?{prompt_text:v.redactForStorage(S),completion_text:v.redactForStorage(me)}:{},...K,span_kind:"llm",tool_definitions:H,conversation_turn:q,message_role:R,metadata:{usage:J.usageMetadata??null,guardrails:{pre:M?{matched:M.results.length,modified:!!M.modifiedText}:null,post:A?{matched:A.results.length,modified:!!A.modifiedText}:null},auto_detected:Object.keys(O).length?O:void 0},user_id:I?.userId,session_id:I?.sessionId,tags:I?.tags,environment:v.environment,sdk_version:"1.0.0"}),P&&(P.setAttribute("gen_ai.completion",me),P.setAttribute("gen_ai.usage.input_tokens",X),P.setAttribute("gen_ai.usage.output_tokens",B),P.setAttribute("gen_ai.usage.total_tokens",X+B),P.setAttribute("gen_ai.usage.cached_tokens",se),P.setAttribute("gen_ai.usage.reasoning_tokens",ue),P.setAttribute("gen_ai.response.finish_reason",J.candidates?.[0]?.finishReason??""),P.setAttribute("gen_ai.response.latency",performance.now()-x),P.setStatus({code:ie.OK}),P.end()),D}catch(D){let J=D instanceof U,X=J?"error":D.status===429?"rate_limited":D.status===408?"timeout":"error";throw v.enqueue({event_id:u(),project_id:"",trace_id:T,span_id:F,parent_span_id:I?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-x,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:X,error_code:J?"guardrail_blocked":String(D.status??"unknown"),error_message:String(D.message).slice(0,500),...K,span_kind:"llm",conversation_turn:q,message_role:R,metadata:{auto_detected:Object.keys(O).length?O:void 0},user_id:I?.userId,session_id:I?.sessionId,tags:I?.tags,environment:v.environment,sdk_version:"1.0.0"}),P&&(P.setStatus({code:ie.ERROR,message:D.message}),P.recordException(D),P.end()),D}}),C&&(k.sendMessageStream=async(_,N)=>{let v=j(),x=performance.now(),I=E(),F=u(),T=I?.traceId??u(),S={...y||{},message:_},M=Ue(S),A=oe(M,i),{conversation_turn:G,message_role:O}=re(M),q=I?.agentName??A.agent_name,R=I?.agentRole??A.agent_role,$=I?.agentId||q||R?{agent_id:I?.agentId,agent_name:q,agent_role:R,agent_framework:I?.agentFramework,parent_agent_id:I?.parentAgentId}:{},L=y?.tools?.length?ae(y.tools):void 0,K=null,H=v.enableOTel?ze.getTracer("zespan-sdk"):null;H&&(K=H.startSpan(`sendMessageStream ${i}`,{kind:Ze.CLIENT,attributes:Ve(S,i)}));try{let P=await W({traceId:T,spanId:F,userId:I?.userId,sessionId:I?.sessionId,tags:I?.tags,agentId:I?.agentId,agentName:q,agentRole:R,agentFramework:I?.agentFramework,parentAgentId:I?.parentAgentId,conversationTurn:G,availableTools:L},()=>C(_,N));return K&&K.setAttribute("gen_ai.request.stream",!0),P.stream=qt(P.stream,{lumiq:v,t0:x,traceId:T,spanId:F,parent:I,params:S,modelName:i,otelSpan:K,streamResult:P,agentCtx:$,tool_definitions:L,autoDetected:A,conversation_turn:G,message_role:O,operation:r}),P}catch(P){let V=P instanceof U,D=V?"error":P.status===429?"rate_limited":P.status===408?"timeout":"error";throw v.enqueue({event_id:u(),project_id:"",trace_id:T,span_id:F,parent_span_id:I?.spanId,timestamp:new Date().toISOString(),provider:"google",model:i,operation:r,latency_ms:performance.now()-x,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:D,error_code:V?"guardrail_blocked":String(P.status??"unknown"),error_message:String(P.message).slice(0,500),...$,span_kind:"llm",conversation_turn:G,message_role:O,metadata:{auto_detected:Object.keys(A).length?A:void 0},user_id:I?.userId,session_id:I?.sessionId,tags:I?.tags,environment:v.environment,sdk_version:"1.0.0"}),K&&(K.setStatus({code:ie.ERROR,message:P.message}),K.recordException(P),K.end()),P}}),k}),g},a}import{createHash as Nn}from"crypto";import{SpanKind as Mn,SpanStatusCode as He,trace as qn}from"@opentelemetry/api";var mt=a=>Nn("sha256").update(a).digest("hex");function Pn(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 jn(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*Fn(a,t){let{lumiq:n,t0:e,traceId:s,spanId:r,parent:o,params:d,otelSpan:i,agentCtx:l,autoDetected:p,conversation_turn:c,message_role:m,tool_definitions:g}=t,f,b=0,w=0,y=0,k=0,h="",C=d.model,_=[],N="",v=[],{systemPrompt:x,conversationMessages:I}=Y(d.messages??[]);try{for await(let A of a){if(f||(f=performance.now()-e,n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:C,operation:"chat",latency_ms:f,ttft_ms:f,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:mt(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(I),...x?{system_prompt:n.redactForStorage(x)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.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)),A.choices?.[0]?.delta?.content&&(b++,N+=A.choices[0].delta.content),A.choices?.[0]?.finish_reason&&(h=A.choices[0].finish_reason),A.model&&(C=A.model),A.usage){w=Number(A.usage.prompt_tokens||w||0),b=Number(A.usage.completion_tokens||b||0),y=Number(A.usage.prompt_tokens_details?.cached_tokens||y||0);let G=A.usage.completion_tokens_details||{};k=Number(G.reasoning_tokens||k||0)}if(A.choices?.[0]?.delta?.tool_calls)for(let G of A.choices[0].delta.tool_calls)v.push(G),G.function?.name&&!_.includes(G.function.name)&&_.push(G.function.name);yield A}let{tools_used:F,tool_call_args:T}=$e(v),S=F.length>0?F:_,M=performance.now()-e;n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:C,operation:"chat",latency_ms:M,ttft_ms:f,stream:!0,input_tokens:w,output_tokens:b,cached_tokens:y,reasoning_tokens:k,status:"success",finish_reason:h||"stream_end",tools_used:S,tool_call_args:Object.keys(T).length?T:void 0,prompt_hash:mt(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(I),completion_text:n.redactForStorage(N),...x?{system_prompt:n.redactForStorage(x)}:{}}:{},...l,span_kind:"llm",tool_definitions:g,conversation_turn:c,message_role:m,metadata:{usage:{prompt_tokens:w,completion_tokens:b,cached_tokens:y,reasoning_tokens:k},auto_detected:Object.keys(p).length?p:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:"1.0.0"}),i&&(i.setAttribute("gen_ai.completion",N),i.setAttribute("gen_ai.usage.input_tokens",w),i.setAttribute("gen_ai.usage.output_tokens",b),i.setAttribute("gen_ai.usage.total_tokens",w+b),i.setAttribute("gen_ai.usage.reasoning_tokens",k),i.setAttribute("gen_ai.response.finish_reason",h||"stream_end"),i.setStatus({code:He.OK}),i.end())}catch(F){throw n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:C,operation:"chat",latency_ms:performance.now()-e,ttft_ms:f,stream:!0,input_tokens:0,output_tokens:b,cost_usd:0,status:"error",error_code:String(F.status??"unknown"),error_message:String(F.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:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:"1.0.0"}),i&&(i.setStatus({code:He.ERROR,message:F.message}),i.recordException(F),i.end()),F}}function jt(a,t={}){let n=a.chat.completions.create,e=Q(t.guardrails,"chat");return a.chat.completions.create=async(s,r)=>{let o=j(),d=performance.now(),i=E(),l=u(),p=i?.traceId??u(),c=s.model??"openrouter\u6A21\u578B",m=null,g=null;if(o.configClient){let _=o.configClient.getModel("chat",c);if(_!==c){let N=c;c=_,s={...s,model:_},o.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:N,override_model:_},environment:o.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}:{},b=oe(s.messages??[],s.model),{conversation_turn:w,message_role:y}=re(s.messages??[]),k=s.tools?.length?ae(s.tools):void 0,h=null,C=o.enableOTel?qn.getTracer("zespan-sdk"):null;C&&(h=C.startSpan(`chat.completions ${c}`,{kind:Mn.CLIENT,attributes:jn(s,c)}));try{m=await Z(o,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(a.chat.completions,s,r));if(s.stream===!0)return h&&h.setAttribute("gen_ai.request.stream",!0),Fn(_,{lumiq:o,t0:d,traceId:p,spanId:l,parent:i,params:s,otelSpan:h,agentCtx:f,autoDetected:b,conversation_turn:w,message_role:y,tool_definitions:k});let{prompt_tokens:v,completion_tokens:x,completion_tokens_details:I}=_.usage??{},F=I?.reasoning_tokens??0,T=_.model??s.model,{cost:S,source:M}=Pn(_,T,v??0,x??0);_.pricing||(_.pricing={cost_usd:S,source:M});let A=_.choices?.[0]?.message?.content||"";g=await Z(o,e,{text:A,phase:"post",traceId:p,spanId:l,model:T,operation:"chat"});let G=te(A,g);G!==A&&typeof _?.choices?.[0]?.message?.content=="string"&&(_.choices[0].message.content=G);let{tools_used:O,tool_call_args:q}=Ke(_.choices??[]);return o.enqueue({event_id:u(),project_id:"",trace_id:p,span_id:l,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:T,operation:"chat",latency_ms:performance.now()-d,stream:!1,input_tokens:v??0,output_tokens:x??0,cached_tokens:0,cost_usd:S,status:"success",finish_reason:_.choices?.[0]?.finish_reason??"",tools_used:O.length>0?O:_.choices?.[0]?.message?.tool_calls?.map(R=>R.function.name)??[],tool_call_args:Object.keys(q).length?q:void 0,prompt_hash:mt(JSON.stringify(s.messages)),...o.storePrompts?(()=>{let{systemPrompt:R,conversationMessages:$}=Y(s.messages??[]);return{prompt_text:o.redactForStorage($),completion_text:o.redactForStorage(G),...R?{system_prompt:o.redactForStorage(R)}:{}}})():{},...O.length>0&&A?{decision_rationale:A}:{},...f,span_kind:"llm",tool_definitions:k,conversation_turn:w,message_role:y,metadata:{usage:_.usage??null,pricing:_.pricing??{cost_usd:S,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(b).length?b:void 0},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:o.environment,sdk_version:"1.0.0"}),h&&(h.setAttribute("gen_ai.usage.input_tokens",v??0),h.setAttribute("gen_ai.usage.output_tokens",x??0),h.setAttribute("gen_ai.usage.total_tokens",(v??0)+(x??0)),h.setAttribute("gen_ai.usage.reasoning_tokens",F),h.setAttribute("gen_ai.completion",_.choices?.[0]?.message?.content||""),h.setAttribute("gen_ai.response.finish_reason",_.choices?.[0]?.finish_reason??""),h.setAttribute("gen_ai.response.latency",performance.now()-d),h.setStatus({code:He.OK}),h.end()),_}catch(_){let N=_ instanceof U,v=N?"error":_.status===429?"rate_limited":_.status===408?"timeout":"error";throw o.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:N?"guardrail_blocked":String(_.status??_.code??"unknown"),error_message:String(_.message).slice(0,500),...f,span_kind:"llm",conversation_turn:w,message_role:y,metadata:{auto_detected:Object.keys(b).length?b:void 0},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:o.environment,sdk_version:"1.0.0"}),h&&(h.setStatus({code:He.ERROR,message:_.message}),h.recordException(_),h.end()),_}},a}async function*Ft(a,t){let{lumiq:n,t0:e,traceId:s,spanId:r,parent:o,model:d,operation:i}=t,l=0,p=0,c=0,m="",g;try{for await(let f of a)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:r,parent_span_id:o?.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:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:"1.0.0"})}catch(f){throw n.enqueue({event_id:u(),project_id:"",trace_id:s,span_id:r,parent_span_id:o?.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:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:"1.0.0"}),f}}function En(a,t={}){let n=a,e=t.operation??"chat",s=Q(t.guardrails,e),r=n.converse?.bind(n);r&&(n.converse=async(i,...l)=>{let p=j(),c=performance.now(),m=E(),g=u(),f=m?.traceId??u(),b=i.modelId??"amazon.nova-lite-v1:0";p.configClient&&(b=p.configClient.getModel(e,b),b!==i.modelId&&(i={...i,modelId:b}));let w=JSON.stringify(i.messages??[]);await Z(p,s,{text:w,phase:"pre",traceId:f,spanId:g,model:b,operation:e,agentName:m?.agentName});try{let y=await W({traceId:f,spanId:g,userId:m?.userId,sessionId:m?.sessionId,tags:m?.tags},()=>r(i,...l)),k=y.usage??{},h=y.output?.message?.content??[],C=h.filter(x=>x.text).map(x=>x.text).join(""),_=h.filter(x=>x.toolUse),N=_.map(x=>x.toolUse?.name).filter(Boolean),v={};for(let x of _)x.toolUse?.name&&(v[x.toolUse.name]=x.toolUse.input??{});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:b,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:y.stopReason??"",span_kind:"llm",tools_used:N.length?N:void 0,tool_call_args:Object.keys(v).length?v:void 0,...p.storePrompts?{prompt_text:p.redactForStorage(i.messages??[]),completion_text:p.redactForStorage(C),...(()=>{let x=(i.system??[]).map(I=>I.text??"").filter(Boolean);return x.length?{system_prompt:p.redactForStorage(x.join(`
|
|
7
|
-
`))}:{}})()}:{},...N.length>0&&C?{decision_rationale:C}:{},user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:p.environment,sdk_version:"1.0.0"}),y}catch(y){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:b,operation:e,latency_ms:performance.now()-c,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:String(y.$metadata?.httpStatusCode??y.name??"unknown"),error_message:String(y.message).slice(0,500),user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:p.environment,sdk_version:"1.0.0"}),y}});let o=n.converseStream?.bind(n);o&&(n.converseStream=async(i,...l)=>{let p=j(),c=performance.now(),m=E(),g=u(),f=m?.traceId??u(),b=i.modelId??"amazon.nova-lite-v1:0",w=await W({traceId:f,spanId:g,userId:m?.userId,sessionId:m?.sessionId,tags:m?.tags},()=>o(i,...l));return{...w,stream:Ft(w.stream,{lumiq:p,t0:c,traceId:f,spanId:g,parent:m,model:b,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=j(),m=performance.now(),g=E(),f=u(),b=g?.traceId??u(),w=i?.input?.modelId??"amazon.nova-lite-v1:0",y=await W({traceId:b,spanId:f,userId:g?.userId,sessionId:g?.sessionId,tags:g?.tags},()=>d(i,...l));return{...y,stream:Ft(y.stream,{lumiq:c,t0:m,traceId:b,spanId:f,parent:g,model:w,operation:e})}}),a}function Gn(a,t={}){let n=a,e=t.operation??"chat",s=Q(t.guardrails,e),r=n.chat.complete.bind(n.chat);n.chat.complete=async(d,...i)=>{let l=j(),p=performance.now(),c=E(),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 b=JSON.stringify(d.messages??[]);await Z(l,s,{text:b,phase:"pre",traceId:g,spanId:m,model:f,operation:e});try{let w=await W({traceId:g,spanId:m,userId:c?.userId,sessionId:c?.sessionId,tags:c?.tags},()=>r(d,...i)),y=w.model??f,k=w.usage??{},h=Number(k.promptTokens??0),C=Number(k.completionTokens??0),_=w.choices?.[0]?.finishReason??"",N=w.choices?.[0]?.message?.content??"",v=w.choices?.[0]?.message?.tool_calls??[],x=v.map(S=>S.function?.name).filter(Boolean),I={};for(let S of v)if(S.function?.name)try{I[S.function.name]=JSON.parse(S.function.arguments??"{}")}catch{I[S.function.name]={}}await Z(l,s,{text:N,phase:"post",traceId:g,spanId:m,model:y,operation:e});let{systemPrompt:F,conversationMessages:T}=Y(d.messages??[]);return 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:y,operation:e,latency_ms:performance.now()-p,stream:!1,input_tokens:h,output_tokens:C,status:"success",finish_reason:_,span_kind:"llm",tools_used:x.length?x:void 0,tool_call_args:Object.keys(I).length?I:void 0,...l.storePrompts?{prompt_text:l.redactForStorage(T),completion_text:l.redactForStorage(N),...F?{system_prompt:l.redactForStorage(F)}:{}}:{},...x.length>0&&N?{decision_rationale:N}:{},user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:l.environment,sdk_version:"1.0.0"}),w}catch(w){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(w.message).slice(0,500),user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:l.environment,sdk_version:"1.0.0"}),w}};let o=n.chat.stream?.bind(n.chat);return o&&(n.chat.stream=async function*(d,...i){let l=j(),p=performance.now(),c=E(),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 b=0,w=0,y="",k;try{let h=await W({traceId:g,spanId:m,userId:c?.userId,sessionId:c?.sessionId,tags:c?.tags},()=>o(d,...i));for await(let C of h)k||(k=performance.now()-p),C.data?.usage&&(b=C.data.usage.promptTokens??b,w=C.data.usage.completionTokens??w),C.data?.choices?.[0]?.finishReason&&(y=C.data.choices[0].finishReason),C.data?.model&&(f=C.data.model),yield C;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:b,output_tokens:w,status:"success",finish_reason:y,user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:l.environment,sdk_version:"1.0.0"})}catch(h){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(h.message).slice(0,500),user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:l.environment,sdk_version:"1.0.0"}),h}}),a}function Dn(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 d=j(),i=performance.now(),l=E(),p=u(),c=l?.traceId??u(),m=r.model??"llama-3.3-70b-versatile";d.configClient&&(m=d.configClient.getModel(s,m),m!==r.model&&(r={...r,model:m}));try{let g=await W({traceId:c,spanId:p,userId:l?.userId,sessionId:l?.sessionId,tags:l?.tags},()=>e(r,...o)),f=g.usage??{},b=Number(f.prompt_tokens??0),w=Number(f.completion_tokens??0),y=g.choices?.[0]?.finish_reason??"",k=g.choices?.[0]?.message?.content??"",h=g.choices?.[0]?.message?.tool_calls??[],C=h.map(x=>x.function?.name).filter(Boolean),_={};for(let x of h)if(x.function?.name)try{_[x.function.name]=JSON.parse(x.function.arguments??"{}")}catch{_[x.function.name]={}}let{systemPrompt:N,conversationMessages:v}=Y(r.messages??[]);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:b,output_tokens:w,status:"success",finish_reason:y,span_kind:"llm",tools_used:C.length?C:void 0,tool_call_args:Object.keys(_).length?_:void 0,...d.storePrompts?{prompt_text:d.redactForStorage(v),completion_text:d.redactForStorage(k),...N?{system_prompt:d.redactForStorage(N)}:{}}:{},...C.length>0&&k?{decision_rationale:k}:{},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}},a}function Ln(a,t={}){let n=ot("openai").default??ot("openai"),e=new n({baseURL:a.baseURL,apiKey:a.apiKey??"litellm"});return We(e,{...t,providerOverride:"litellm"})}import{BaseCallbackHandler as Kn}from"@langchain/core/callbacks/base";import{createHash as $n}from"crypto";var Et=a=>$n("sha256").update(a).digest("hex"),Gt=["AgentExecutor","ReactAgent","ToolCallingAgent"],Dt=["DynamicStructuredTool","StructuredTool","DynamicTool","Tool"],Te=class extends Kn{name="ZespanCallbackHandler";guardrailSettings=Q(void 0,"chat");runs=new Map;constructor(t={}){super(),this.guardrailSettings=Q(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 r=e.additional_kwargs?.tool_calls;if(Array.isArray(r)&&r.length>0)return r.map(d=>({name:d.function?.name??"unknown",args:d.function?.arguments?JSON.parse(d.function.arguments):{},id:d.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?._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 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?._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 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,d=[],i;for(;(i=o.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 r=s?.message?.content;return typeof r=="string"?r:Array.isArray(r)?r.map(o=>typeof o?.text=="string"?o.text:"").filter(Boolean).join(`
|
|
8
|
-
`):""}async handleChatModelStart(t,n,e,s,r,o,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(`
|
|
9
|
-
`)),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,{...r,_lumiq_structured_messages:p,_lumiq_message_count:n[0]?.length??0})}catch{}}async handleLLMStart(t,n,e,s,r){try{let o=E(),d=j(),i=s?this.runs.get(s):void 0,l=u();ce(l);let p=i?.traceId??o?.traceId??u(),c=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,g=i?.spanKind==="agent"?i.agentId:void 0,f=i?.spanKind==="agent"?i.agentName:void 0;if(!f){let h=this.detectAgentFromPrompts(n,r);h&&(f=h,g=g??u())}let b=r?.invocation_params?.tools??r?.invocation_params?.functions??[],w=b.map(h=>h.function?.name??h.name??"unknown").filter(Boolean),y=this.detectConversationInfo(n,r),k=await Z(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??o?.spanId,model:c,agentId:g,agentName:f,spanKind:"llm",isStreaming:m,parent:o,prompts:n,structuredMessages:r?._lumiq_structured_messages,invocationParams:r?.invocation_params,preGuardrailResult:k,toolsAvailable:w,toolSchemas:d.storePrompts?b:void 0,conversationTurns:y.turns,hasMemory:y.hasMemory})}catch(o){if(o instanceof U)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=j();r.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)),...r.storePrompts?(()=>{let{systemPrompt:o,conversationMessages:d}=Y(s.structuredMessages??[]);return{prompt_text:r.redactForStorage(d.length?d: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:"1.0.0",finish_reason:"stream_start"})}catch{}}async handleLLMEnd(t,n){try{let e=this.runs.get(n);if(!e)return;ee(e.spanId),this.runs.delete(n);let s=j(),r=t.llmOutput?.tokenUsage?.promptTokens??t.llmOutput?.usageMetadata?.promptTokenCount??0,o=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 Z(s,this.guardrailSettings,{text:i,phase:"post",traceId:e.traceId,spanId:e.spanId,model:e.model,operation:"chat"}),b=te(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:r,output_tokens:o,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(w=>w.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?(()=>{let{systemPrompt:w,conversationMessages:y}=Y(e.structuredMessages??[]);return{prompt_text:s.redactForStorage(y.length?y:e.prompts),completion_text:s.redactForStorage(b),...w?{system_prompt:s.redactForStorage(w)}:{}}})():{},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;ee(e.spanId),this.runs.delete(n);let s=j();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 U?"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 r=E(),o=s?this.runs.get(s):void 0,d=u(),i=o?.traceId??r?.traceId??u(),l=Array.isArray(t?.id)?t.id:[],p=l.some(h=>Gt.includes(h)),c=!p&&l.some(h=>Dt.includes(h)),m=p?"agent":c?"tool":"general",g=p?u():void 0,f=p?l.find(h=>Gt.includes(h))??void 0:void 0,b=c?t.name??t.kwargs?.name??l.find(h=>Dt.includes(h)):void 0,w=c?JSON.stringify(n):void 0,y=o?.spanKind==="agent"?o.agentId:void 0,k=o?.spanKind==="agent"?o.agentName:void 0;this.runs.set(e,{t0:performance.now(),traceId:i,spanId:d,parentSpanId:o?.spanId??r?.spanId,model:"",agentId:g??o?.agentId,agentName:f??o?.agentName,spanKind:m,toolName:b,toolInput:w,prompts:[],parent:r,delegatedFromAgentId:y,delegatedFromAgentName:k})}catch{}}async handleChainEnd(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=j();if(e.spanKind==="tool"){let r=t?.output??t,o=typeof r=="string"?r:JSON.stringify(r);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: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:"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=j();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 r=E(),o=s?this.runs.get(s):void 0,d=u(),i=o?.traceId??r?.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:o?.spanId??r?.spanId,model:"",agentId:o?.agentId,agentName:o?.agentName,spanKind:"tool",toolName:p,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=j(),o=(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: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:"1.0.0"})}catch{}}async handleToolError(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=j();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=E(),r=j(),o=e?.traceId??s?.traceId??u(),d=u(),i=[t?.tool??"",typeof t?.toolInput=="string"?t.toolInput:JSON.stringify(t?.toolInput??""),t?.log??""].join(" | ");r.enqueue({event_id:u(),project_id:"",trace_id:o,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],...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:"1.0.0"})}catch{}}async handleRetrieverStart(t,n,e,s){try{let r=E(),o=s?this.runs.get(s):void 0,d=u(),i=o?.traceId??r?.traceId??u();this.runs.set(e,{t0:performance.now(),traceId:i,spanId:d,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=j();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=j();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{}}},Bn=Te;function Lt(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=E(),o=this.getParentAgentState(n,s),d=o?.traceId??r?.traceId??u(),i=u(),l=o?.spanId??r?.spanId,p=u(),c=Lt(t.invocationContext?.agent),m=t.userId||r?.userId,g=t.sessionId||r?.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: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,d=j();d.enqueue({event_id:u(),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:d.environment,sdk_version:"1.0.0"})}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??u(),d=Lt(t.invocationContext?.agent),i=n?.config?.systemInstruction,l;if(i){let m=typeof i=="string"?i:(i?.parts??[]).filter(g=>g?.text).map(g=>g.text).join("");m&&(l=m)}let p,c=n?.contents??[];if(c.length){let m=c.map(g=>{let f=g?.role??"user",b=(g?.parts??[]).filter(w=>typeof w?.text=="string"&&!w.thought).map(w=>w.text).join("");return b?`${f}: ${b}`:""}).filter(Boolean).join(`
|
|
10
|
-
`);m&&(p=m)}this.pendingModelCalls.set(this.agentKey(e,s),{t0:performance.now(),spanId:u(),traceId:o,parentSpanId:r?.spanId??"",model:d,agentId:r?.agentId,agentName:s,userId:r?.userId??t.userId,sessionId:r?.sessionId??t.sessionId,promptText:p,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:d,output:i,cached:l,reasoning:p}=Wn(n?.usageMetadata),c=n?.content?.parts??[],m=c.filter(b=>typeof b?.text=="string"&&!b.thought).map(b=>b.text).join(""),g=c.filter(b=>b?.functionCall?.name).map(b=>b.functionCall.name),f=j();f.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",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: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&&o.systemPrompt?{system_prompt:f.redactForStorage(o.systemPrompt)}:{},...f.storePrompts&&o.promptText?{prompt_text:f.redactForStorage(o.promptText)}:{},...f.storePrompts&&m?{completion_text:f.redactForStorage(m)}:{},user_id:o.userId,session_id:o.sessionId,environment:f.environment,sdk_version:"1.0.0"})}catch{}};beforeToolCallback=({tool:t,args:n,context:e})=>{try{let s=e.invocationId,r=e.agentName,o=e.functionCallId??u(),d=this.agentStates.get(this.agentKey(s,r));this.toolCallStates.set(o,{t0:performance.now(),spanId:u(),traceId:d?.traceId??u(),parentSpanId:d?.spanId??"",toolName:t?.name??"unknown_tool",args:n??{},agentId:d?.agentId,agentName:r,userId:d?.userId??e.userId,sessionId:d?.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 d=j();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:"",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: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 Jn}from"crypto";var gt=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 zn(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 Un(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 Bt(a){if(a instanceof U)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=Q(t.guardrails,"agent"),r=t.agentName??n.name??"ADKAgent",o=t.agentRole??(n.subAgents?.length?"coordinator":"specialist"),d=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 p=j(),c=performance.now(),m=E(),g=u(),f=u(),b=m?.traceId??u(),w=t.sessionId??m?.sessionId,y=typeof i=="string"?i:JSON.stringify(i??"");return W({traceId:b,spanId:f,userId:m?.userId,sessionId:w,tags:m?.tags,agentId:g,agentName:r,agentRole:o,agentFramework:"google-adk",parentAgentId:m?.agentId},async()=>{try{let k=await Z(p,s,{text:y,phase:"pre",traceId:b,spanId:f,model:d,operation:"agent"}),h=await e(i,l),{input:C,output:_,cached:N,reasoning:v}=$t(h),x=zn(h),I=Un(h),F=h?.candidates?.[0]?.finishReason??h?.finish_reason,T=await Z(p,s,{text:x,phase:"post",traceId:b,spanId:f,model:d,operation:"agent"}),S=te(x,T);return p.enqueue({event_id:u(),project_id:"",trace_id:b,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: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()-c,stream:!1,input_tokens:C,output_tokens:_,cached_tokens:N||void 0,reasoning_tokens:v||void 0,status:"success",finish_reason:F,tools_used:I.length?I.map(M=>M.name):void 0,tool_call_args:I.length?Object.fromEntries(I.map(M=>[M.name,M.args])):void 0,prompt_hash:gt(y),...p.storePrompts?{prompt_text:p.redactForStorage(y),completion_text:p.redactForStorage(S)}:{},...I.length>0&&x?{decision_rationale:x}:{},user_id:m?.userId,session_id:w,tags:m?.tags,environment:p.environment,sdk_version:"1.0.0",metadata:{guardrails:{pre:k?{matched:k.results.length,modified:!!k.modifiedText}:null,post:T?{matched:T.results.length,modified:!!T.modifiedText}:null}}}),h}catch(k){throw p.enqueue({event_id:u(),project_id:"",trace_id:b,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:r,agent_role:o,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:Bt(k),error_message:String(k?.message??k).slice(0,500),user_id:m?.userId,session_id:w,tags:m?.tags,environment:p.environment,sdk_version:"1.0.0"}),k}})},a}function _t(a,t={}){let n=a;if(typeof n?.runEphemeral=="function"){let e=n.runEphemeral.bind(n);n.runEphemeral=s=>Kt(e(s),s,t)}if(typeof n?.run=="function"){let e=n.run.bind(n);n.run=s=>Kt(e(s),s,t)}return a}async function*Kt(a,t,n){let e=j(),s=E(),r=s?.traceId??u(),o=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(""),b=new Map,w=new Map,y=new Set,k="",h=0,C=0,_=0,N=0;ce(o);try{yield*W({traceId:r,spanId:o,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 a){h++;let x=v.author??l;if(x!=="user"){x!==l&&!y.has(x)&&e.enqueue({event_id:u(),project_id:"",trace_id:r,span_id:u(),parent_span_id:o,timestamp:new Date().toISOString(),provider:"google-adk",model:c,operation:`handoff.${l}\u2192${x}`,span_kind:"handoff",agent_name:x,agent_role:"specialist",agent_framework:"google-adk",delegation_from_agent_id:d,delegation_from_agent_name:l,delegation_reason:`delegated to ${x}`,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"}),y.add(x),b.has(x)||(b.set(x,{spanId:u(),agentId:u(),t0:performance.now()}),w.set(x,{input:0,output:0,cached:0,reasoning:0}));let I=b.get(x),F=$t(v);if(F.input>0||F.output>0){let A=w.get(x);A.input+=F.input,A.output+=F.output,A.cached+=F.cached,A.reasoning+=F.reasoning,C+=F.input,_+=F.output,N+=F.cached}let T=v.content?.parts??[];for(let A of T){let G=A?.functionCall;if(!G?.name)continue;let O=u();e.enqueue({event_id:u(),project_id:"",trace_id:r,span_id:O,parent_span_id:I.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:c,operation:`tool.${G.name}`,span_kind:"tool",agent_id:I.agentId,agent_name:x,agent_framework:"google-adk",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",tools_used:[G.name],tool_call_id:O,tool_call_args:G.args??{},user_id:m,session_id:g,tags:s?.tags,environment:e.environment,sdk_version:"1.0.0"})}let S=T.filter(A=>typeof A?.text=="string"&&!A.thought).map(A=>A.text).join("");if(S&&(k+=S),typeof v.isFinalResponse=="function"?v.isFinalResponse():!!v.isFinalResponse||v.actions?.stateDelta!==void 0||v.turnComplete===!0){let A=T.map(q=>q?.functionCall?.name).filter(Boolean),G=T.filter(q=>typeof q?.text=="string"&&!q.thought).map(q=>q.text).join(""),O=w.get(x)??{input:0,output:0,cached:0,reasoning:0};e.enqueue({event_id:u(),project_id:"",trace_id:r,span_id:I.spanId,parent_span_id:o,timestamp:new Date().toISOString(),provider:"google-adk",model:c,operation:`agent.${x}`,span_kind:"agent",agent_id:I.agentId,agent_name:x,agent_role:x===l?p:"specialist",agent_framework:"google-adk",parent_agent_id:d,latency_ms:performance.now()-I.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:A.length?A:void 0,prompt_hash:f?gt(f):void 0,...e.storePrompts&&S?{completion_text:e.redactForStorage(S)}:{},...A.length>0&&G?{decision_rationale:G}:{},user_id:m,session_id:g,tags:s?.tags,environment:e.environment,sdk_version:"1.0.0"}),b.delete(x),w.delete(x)}}yield v}}),e.enqueue({event_id:u(),project_id:"",trace_id:r,span_id:o,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:C,output_tokens:_,cached_tokens:N||void 0,status:"success",prompt_hash:f?gt(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:h,agentsThatRan:[...y]}}),ee(o)}catch(v){throw ee(o),e.enqueue({event_id:u(),project_id:"",trace_id:r,span_id:o,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:Bt(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(a){let{coordinator:t,runner:n,agentName:e,agentRole:s,model:r,...o}=a,d=Xe(t,{...o,agentName:e??t?.name,agentRole:s??"coordinator",traceSubAgents:!0}),i=_t(n,{agentName:e??t?.name,agentRole:s??"coordinator",model:r});return{coordinator:d,runner:i}}var Jt=Xe;import{NodeTracerProvider as Vn}from"@opentelemetry/sdk-trace-node";import{ExportResultCode as Zt}from"@opentelemetry/core";function Hn(){try{return j()}catch{return null}}var ht=class{export(t,n){let e=Hn();if(!e){n({code:Zt.SUCCESS});return}try{for(let s of t){let{name:r}=s;if(!r.startsWith("ai."))continue;let o=s.attributes,d=E(),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(o["ai.model.provider"]??"unknown"),model:String(o["ai.model.id"]??"unknown"),latency_ms:p,stream:!1,status:m?"error":"success",environment:e.environment,sdk_version:"1.0.0"};r==="ai.generateText"||r==="ai.generateText.doGenerate"?(ee(i),e.enqueue({...g,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"?(ee(i),e.enqueue({...g,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"?(ee(i),e.enqueue({...g,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({...g,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({...g,operation:r==="ai.embed"?"embed":"embedMany",input_tokens:Number(o["ai.usage.tokens"]??0),output_tokens:0,span_kind:"embedding"})}}catch{}n({code:Zt.SUCCESS})}shutdown(){return Promise.resolve()}},yt=class{exporter;constructor(t){this.exporter=t}onStart(t,n){t.name?.startsWith("ai.")&&ce(t.spanContext().spanId)}onEnd(t){t.name.startsWith("ai.")&&ee(t.spanContext().spanId),this.exporter.export([t],()=>{})}forceFlush(){return Promise.resolve()}shutdown(){return Promise.resolve()}},Ye=null,ft=null;function kt(){return Ye||(Ye=new Vn,Ye.addSpanProcessor(new yt(new ht)),ft=Ye.getTracer("zespan-vercel-ai")),{tracer:ft,telemetry:{isEnabled:!0,tracer:ft}}}function zt(){return kt().telemetry}function Oe(){try{return j()}catch{return null}}function Qn(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=Q(t.guardrails,"chat")}_onLLMStart=async t=>{let n=t.id??u(),e=E(),s=u();ce(s);let r=t.detail?.llm?.metadata?.model_name??t.detail?.model??"unknown",o=e?.traceId??u(),d=JSON.stringify(t.detail?.messages??t.detail?.prompt??t.detail?.input??""),i;try{let l=Oe();l&&d&&(i=await Z(l,this.guardrailSettings,{text:d,phase:"pre",traceId:o,spanId:s,model:r,operation:"chat"}))}catch(l){if(l instanceof U)throw l}this.runs.set(n,{t0:performance.now(),traceId:o,spanId:s,parentSpanId:e?.spanId,model:r,preGuardrailResult:i,promptText:d})};_onLLMEnd=async t=>{let n=this.runs.get(t.id??"");if(!n)return;this.runs.delete(t.id),ee(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)??"",d=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"}),d=te(o,i)}catch(p){p instanceof U||(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:Qn(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:E()?.agentId,agent_name:E()?.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=Oe();if(!n)return;let e=E(),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=Oe();if(!n)return;let e=E(),s=t.detail?.toolCall?.name??t.detail?.name??"",r=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(r).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=E();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=Oe();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 Ut(a){let t=j(),n=E(),e=n?.traceId??u(),s=u(),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"}:{},d={};return{span:{setEvalScore(l,p){d[l]=p},async end(l){let p=performance.now()-r;try{t.enqueue({event_id:u(),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: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",...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=j(),e=E(),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=j(),r=E(),o=u(),d=r?.traceId??u(),i=performance.now(),l=this.tools.find(p=>p.name===t);try{let p=await W({traceId:d,spanId:o,userId:r?.userId,sessionId:r?.sessionId,tags:r?.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: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:"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:o,tool_call_args:n,tool_call_result:c.slice(0,10240),tool_definitions:l?[l]:void 0,user_id:r?.userId,session_id:r?.sessionId,tags:r?.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: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: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:o,tool_call_args:n,user_id:r?.userId,session_id:r?.sessionId,tags:r?.tags};throw s.enqueue(m),p}}delegateTo(t,n){let e=j(),s=E(),r={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(r)}};function Vt(a,t){let n=j(),e=E(),s=u(),r=e?.traceId??u(),o=u(),d={event_id:u(),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:"1.0.0",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(d),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 Xn}from"@opentelemetry/api";var It=class{onStart(t,n){let e=Xn.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 Ht,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=Ht.active();if(Object.keys(e).length>0){let r=Ae.getBaggage(s)??Ae.createBaggage();for(let[o,d]of Object.entries(e))r=r.setEntry(o,{value:d});s=Ae.setBaggage(s,r)}Ae.inject(s,a)}function st(a){return Ae.extract(Ht.active(),a)}var Yn={OTEL_EXPORTER_OTLP_ENDPOINT:"http://localhost:4318",OTEL_SERVICE_NAME:"crewai-agent"};function es(){return`
|
|
6
|
+
`)}return""},n=r?.contents??r,e=[],s=r?.systemInstruction??r?.system_instruction,a=t(s);if(a.length>0&&e.push({role:"system",content:a}),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 d=typeof o.role=="string"?o.role:"user",c=t(o);c.length>0&&e.push({role:d,content:c})}}return e}if(n&&typeof n=="object"){let o=typeof n.role=="string"?n.role:"user",d=t(n);d.length>0&&e.push({role:o,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 gt(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*Et(r,t){let{client:n,t0:e,traceId:s,spanId:a,parent:o,params:d,modelName:c,otelSpan:u,streamResult:l,agentCtx:i,tool_definitions:g,autoDetected:m,conversation_turn:h,message_role:w,operation:y}=t,f,I=0,k=0,A=0,_=0,R="",S=[],b="";try{for await(let v of r){f||(f=performance.now()-e,n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"google",model:c,operation:y,latency_ms:f,ttft_ms:f,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:Ae(JSON.stringify(d)),...n.storePrompts?{prompt_text:n.redactForStorage(d)}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x,finish_reason:"stream_start",...i,span_kind:"llm",tool_definitions:g,conversation_turn:h,message_role:w,metadata:{auto_detected:Object.keys(m).length?m:void 0}}),u&&u.setAttribute("gen_ai.response.time_to_first_token",f));let P=(typeof v?.text=="function"?v.text():v?.text)||v?.candidates?.[0]?.content?.parts?.map(T=>T?.text||"").join("")||"";P&&(I++,b+=P),v.candidates?.[0]?.finishReason&&(R=v.candidates[0].finishReason);let N=Array.isArray(v?.functionCalls)?v.functionCalls:v?.functionCalls?[v.functionCalls]:[],q=(v?.candidates?.[0]?.content?.parts??[]).map(T=>T?.functionCall).filter(Boolean);for(let T of[...N,...q])T?.name&&!S.includes(T.name)&&S.push(T.name);yield v}let C=await l?.response;C?.usageMetadata&&(k=Number(C.usageMetadata.promptTokenCount||k||0),I=Number(C.usageMetadata.candidatesTokenCount||I||0),A=Number(C.usageMetadata.cachedContentTokenCount||A||0),_=Number(C.usageMetadata.thoughtsTokenCount||_||0));let j=gt(C);j&&(b=j);let O=performance.now()-e;n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"google",model:c,operation:y,latency_ms:O,ttft_ms:f,stream:!0,input_tokens:k,output_tokens:I,cached_tokens:A,reasoning_tokens:_,status:"success",finish_reason:R||"stream_end",tools_used:S,prompt_hash:Ae(JSON.stringify(d)),...n.storePrompts?{prompt_text:n.redactForStorage(d),completion_text:n.redactForStorage(b)}:{},...i,span_kind:"llm",tool_definitions:g,conversation_turn:h,message_role:w,metadata:{usage:C?.usageMetadata??{promptTokenCount:k,candidatesTokenCount:I,cachedContentTokenCount:A,thoughtsTokenCount:_},auto_detected:Object.keys(m).length?m:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x}),u&&(u.setAttribute("gen_ai.completion",b),u.setAttribute("gen_ai.usage.input_tokens",k),u.setAttribute("gen_ai.usage.output_tokens",I),u.setAttribute("gen_ai.usage.total_tokens",k+I),u.setAttribute("gen_ai.usage.cached_tokens",A),u.setAttribute("gen_ai.usage.reasoning_tokens",_),u.setAttribute("gen_ai.response.finish_reason",R||"stream_end"),u.setStatus({code:ce.OK}),u.end())}catch(C){throw n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"google",model:c,operation:y,latency_ms:performance.now()-e,ttft_ms:f,stream:!0,input_tokens:0,output_tokens:I,cost_usd:0,status:"error",error_code:String(C.status??"unknown"),error_message:String(C.message).slice(0,500),...i,span_kind:"llm",conversation_turn:h,message_role:w,metadata:{auto_detected:Object.keys(m).length?m:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x}),u&&(u.setStatus({code:ce.ERROR,message:C.message}),u.recordException(C),u.end()),C}}function Gt(r,t={}){let n=r,e=n.getGenerativeModel.bind(n),s=ee(t.guardrails,"chat"),a=t.operation??"chat";return n.getGenerativeModel=(o,d)=>{let c=o.model,u=E(),l=!1,i=c,g;if(u.configClient){let f=u.configClient.getAbTest(a);if(f)Math.random()<f.splitPercentage/100?(i=c,c=f.candidateModel,o={...o,model:c},l=!0,g={ab_test:!0,ab_variant:"candidate"}):g={ab_test:!0,ab_variant:"control"};else{let I=u.configClient.getModel(a,c);I!==c&&(i=c,c=I,o={...o,model:I},l=!0)}}let m=e(o,d),h=m.generateContent.bind(m);m.generateContent=async(f,I)=>{let k=E(),A=performance.now(),_=F(),R=p(),S=_?.traceId??p(),b=null,C=null,j=Ze(f),O=oe(j,c),{conversation_turn:v,message_role:P}=ae(j),N=_?.agentName??O.agent_name,q=_?.agentRole??O.agent_role,T=_?.agentId||N||q?{agent_id:_?.agentId,agent_name:N,agent_role:q,agent_framework:_?.agentFramework,parent_agent_id:_?.parentAgentId}:{},D=f?.tools?.length?re(f.tools):void 0,M=null,B=k.enableOTel?Je.getTracer("zespan-sdk"):null;B&&(M=B.startSpan(`generateContent ${c}`,{kind:We.CLIENT,attributes:Ue(f,c)})),l&&k.enqueue({event_id:p(),project_id:"",trace_id:S,span_id:R,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:c,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:i,override_model:c},environment:k.environment,sdk_version:x});try{b=await Z(k,s,{text:JSON.stringify(f??""),phase:"pre",traceId:S,spanId:R,model:c,operation:a,agentName:_?.agentName});let K;if(k.configClient){let Q=k.configClient.getTimeout(a);Q&&(K=AbortSignal.timeout(Q.timeoutMs))}let z=k.configClient?.getRetryPolicy(a)??null,U=k.configClient?.getConcurrencyLimit(a)??null,G=()=>ge(()=>W({traceId:S,spanId:R,userId:_?.userId,sessionId:_?.sessionId,tags:_?.tags,agentId:_?.agentId,agentName:N,agentRole:q,agentFramework:_?.agentFramework,parentAgentId:_?.parentAgentId,conversationTurn:v,availableTools:D},()=>h(f,K?{...I,signal:K}:I)),z),H=await(U!==null?_e(`${k.projectId??"default"}:${a}`,U).run(G):G()),L=H.response,J=L.usageMetadata?.promptTokenCount??0,X=L.usageMetadata?.candidatesTokenCount??0,$=L.usageMetadata?.cachedContentTokenCount??0,Y=L.usageMetadata?.thoughtsTokenCount??0,pe=gt(L),fe=(L.candidates?.[0]?.content?.parts??[]).filter(Q=>Q?.functionCall),ie=fe.map(Q=>Q.functionCall.name).filter(Boolean),de=(L.candidates?.[0]?.content?.parts??[]).filter(Q=>Q?.text&&!Q?.functionCall).map(Q=>Q.text).join(""),ue={};for(let Q of fe)Q.functionCall?.name&&(ue[Q.functionCall.name]=Q.functionCall.args??{});C=await Z(k,s,{text:pe,phase:"post",traceId:S,spanId:R,model:c,operation:a,agentName:_?.agentName});let he=ne(pe,C);return k.enqueue({event_id:p(),project_id:"",trace_id:S,span_id:R,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:c,operation:a,latency_ms:performance.now()-A,stream:!1,input_tokens:J,output_tokens:X,cached_tokens:$,reasoning_tokens:Y,status:"success",finish_reason:L.candidates?.[0]?.finishReason??"",tools_used:ie,tool_call_args:Object.keys(ue).length?ue:void 0,prompt_hash:Ae(JSON.stringify(f)),...k.storePrompts?{prompt_text:k.redactForStorage(f),completion_text:k.redactForStorage(he)}:{},...ie.length>0&&de?{decision_rationale:de}:{},...T,span_kind:"llm",tool_definitions:D,conversation_turn:v,message_role:P,metadata:{usage:L.usageMetadata??null,guardrails:{pre:b?{matched:b.results.length,modified:!!b.modifiedText}:null,post:C?{matched:C.results.length,modified:!!C.modifiedText}:null},auto_detected:Object.keys(O).length?O:void 0,...g??{}},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:k.environment,sdk_version:x}),M&&(M.setAttribute("gen_ai.completion",he),M.setAttribute("gen_ai.usage.input_tokens",J),M.setAttribute("gen_ai.usage.output_tokens",X),M.setAttribute("gen_ai.usage.total_tokens",J+X),M.setAttribute("gen_ai.usage.cached_tokens",$),M.setAttribute("gen_ai.usage.reasoning_tokens",Y),M.setAttribute("gen_ai.response.finish_reason",L.candidates?.[0]?.finishReason??""),M.setAttribute("gen_ai.response.latency",performance.now()-A),M.setStatus({code:ce.OK}),M.end()),H}catch(K){let z=K.__signal;if(K?.name==="TimeoutError"||K?.code==="ABORT_ERR")throw k.enqueue({event_id:p(),project_id:"",trace_id:S,span_id:R,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:c,operation:a,latency_ms:performance.now()-A,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:"timeout",error_message:String(K.message).slice(0,500),...T,span_kind:"llm",conversation_turn:v,message_role:P,metadata:{auto_detected:Object.keys(O).length?O:void 0},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:k.environment,sdk_version:x}),M&&(M.setStatus({code:ce.ERROR,message:K.message}),M.recordException(K),M.end()),K;let G=k.configClient?.getFallback(a)??null;if(G&&G.model){let J=String(K.message??"")+String(K.status??"")+String(K.code??"");if(G.conditions.length===0||G.conditions.some($=>J.toLowerCase().includes($.toLowerCase()))){let $=c,Y={...o,model:G.model},pe=e(Y,d),fe=pe.generateContent.bind(pe);k.enqueue({event_id:p(),project_id:"",trace_id:S,span_id:R,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:G.model,operation:a,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:$,fallback_model:G.model,reason:String(K.message).slice(0,200)},environment:k.environment,sdk_version:x});try{let ie=await W({traceId:S,spanId:R,userId:_?.userId,sessionId:_?.sessionId,tags:_?.tags},()=>fe(f,I)),de=ie.response,ue=de.usageMetadata?.promptTokenCount??0,he=de.usageMetadata?.candidatesTokenCount??0,Q=de.usageMetadata?.cachedContentTokenCount??0;return k.enqueue({event_id:p(),project_id:"",trace_id:S,span_id:R,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:G.model,operation:a,latency_ms:performance.now()-A,stream:!1,input_tokens:ue,output_tokens:he,cached_tokens:Q,status:"success",finish_reason:de.candidates?.[0]?.finishReason??"",prompt_hash:Ae(JSON.stringify(f)),...T,span_kind:"llm",conversation_turn:v,message_role:P,metadata:{usage:de.usageMetadata??null,fallback_used:!0,original_model:$,auto_detected:Object.keys(O).length?O:void 0},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:k.environment,sdk_version:x}),M&&(M.setStatus({code:ce.OK}),M.end()),ie}catch(ie){throw M&&(M.setStatus({code:ce.ERROR,message:ie.message}),M.recordException(ie),M.end()),ie}}}let H=K instanceof V,L=H?"error":K.status===429?"rate_limited":K.status===408?"timeout":"error";throw k.enqueue({event_id:p(),project_id:"",trace_id:S,span_id:R,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:c,operation:a,latency_ms:performance.now()-A,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:L,error_code:H?"guardrail_blocked":String(K.status??"unknown"),error_message:String(K.message).slice(0,500),...T,span_kind:"llm",conversation_turn:v,message_role:P,metadata:{auto_detected:Object.keys(O).length?O:void 0},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:k.environment,sdk_version:x}),M&&(M.setStatus({code:ce.ERROR,message:K.message}),M.recordException(K),M.end()),K}};let w=m.generateContentStream.bind(m);m.generateContentStream=async(f,I)=>{let k=E(),A=performance.now(),_=F(),R=p(),S=_?.traceId??p(),b=Ze(f),C=oe(b,c),{conversation_turn:j,message_role:O}=ae(b),v=_?.agentName??C.agent_name,P=_?.agentRole??C.agent_role,N=_?.agentId||v||P?{agent_id:_?.agentId,agent_name:v,agent_role:P,agent_framework:_?.agentFramework,parent_agent_id:_?.parentAgentId}:{},q=f?.tools?.length?re(f.tools):void 0,T=null,D=k.enableOTel?Je.getTracer("zespan-sdk"):null;D&&(T=D.startSpan(`generateContentStream ${c}`,{kind:We.CLIENT,attributes:Ue(f,c)}));try{let M=await W({traceId:S,spanId:R,userId:_?.userId,sessionId:_?.sessionId,tags:_?.tags,agentId:_?.agentId,agentName:v,agentRole:P,agentFramework:_?.agentFramework,parentAgentId:_?.parentAgentId,conversationTurn:j,availableTools:q},()=>w(f,I));return T&&T.setAttribute("gen_ai.request.stream",!0),M.stream=Et(M.stream,{client:k,t0:A,traceId:S,spanId:R,parent:_,params:f,modelName:c,otelSpan:T,streamResult:M,agentCtx:N,tool_definitions:q,autoDetected:C,conversation_turn:j,message_role:O,operation:a}),M}catch(M){let B=M instanceof V,K=B?"error":M.status===429?"rate_limited":M.status===408?"timeout":"error";throw k.enqueue({event_id:p(),project_id:"",trace_id:S,span_id:R,parent_span_id:_?.spanId,timestamp:new Date().toISOString(),provider:"google",model:c,operation:a,latency_ms:performance.now()-A,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:K,error_code:B?"guardrail_blocked":String(M.status??"unknown"),error_message:String(M.message).slice(0,500),...N,span_kind:"llm",conversation_turn:j,message_role:O,metadata:{auto_detected:Object.keys(C).length?C:void 0},user_id:_?.userId,session_id:_?.sessionId,tags:_?.tags,environment:k.environment,sdk_version:x}),T&&(T.setStatus({code:ce.ERROR,message:M.message}),T.recordException(M),T.end()),M}};let y=m.startChat?.bind(m);return y&&(m.startChat=f=>{let I=y(f),k=I.sendMessage?.bind(I),A=I.sendMessageStream?.bind(I);return k&&(I.sendMessage=async(_,R)=>{let S=E(),b=performance.now(),C=F(),j=p(),O=C?.traceId??p(),v={...f||{},message:_},P=null,N=null,q=Ze(v),T=oe(q,c),{conversation_turn:D,message_role:M}=ae(q),B=C?.agentName??T.agent_name,K=C?.agentRole??T.agent_role,z=C?.agentId||B||K?{agent_id:C?.agentId,agent_name:B,agent_role:K,agent_framework:C?.agentFramework,parent_agent_id:C?.parentAgentId}:{},U=f?.tools?.length?re(f.tools):void 0,G=null,H=S.enableOTel?Je.getTracer("zespan-sdk"):null;H&&(G=H.startSpan(`sendMessage ${c}`,{kind:We.CLIENT,attributes:Ue(v,c)}));try{P=await Z(S,s,{text:JSON.stringify(v),phase:"pre",traceId:O,spanId:j,model:c,operation:a,agentName:C?.agentName});let L=await W({traceId:O,spanId:j,userId:C?.userId,sessionId:C?.sessionId,tags:C?.tags,agentId:C?.agentId,agentName:B,agentRole:K,agentFramework:C?.agentFramework,parentAgentId:C?.parentAgentId,conversationTurn:D,availableTools:U},()=>k(_,R)),J=L.response,X=J.usageMetadata?.promptTokenCount??0,$=J.usageMetadata?.candidatesTokenCount??0,Y=J.usageMetadata?.cachedContentTokenCount??0,pe=J.usageMetadata?.thoughtsTokenCount??0,fe=gt(J),ie=(J.candidates?.[0]?.content?.parts??[]).filter(Q=>Q?.functionCall),de=ie.map(Q=>Q.functionCall.name).filter(Boolean),ue={};for(let Q of ie)Q.functionCall?.name&&(ue[Q.functionCall.name]=Q.functionCall.args??{});N=await Z(S,s,{text:fe,phase:"post",traceId:O,spanId:j,model:c,operation:a,agentName:C?.agentName});let he=ne(fe,N);return S.enqueue({event_id:p(),project_id:"",trace_id:O,span_id:j,parent_span_id:C?.spanId,timestamp:new Date().toISOString(),provider:"google",model:c,operation:a,latency_ms:performance.now()-b,stream:!1,input_tokens:X,output_tokens:$,cached_tokens:Y,reasoning_tokens:pe,status:"success",finish_reason:J.candidates?.[0]?.finishReason??"",tools_used:de,tool_call_args:Object.keys(ue).length?ue:void 0,prompt_hash:Ae(JSON.stringify(v)),...S.storePrompts?{prompt_text:S.redactForStorage(v),completion_text:S.redactForStorage(he)}:{},...z,span_kind:"llm",tool_definitions:U,conversation_turn:D,message_role:M,metadata:{usage:J.usageMetadata??null,guardrails:{pre:P?{matched:P.results.length,modified:!!P.modifiedText}:null,post:N?{matched:N.results.length,modified:!!N.modifiedText}:null},auto_detected:Object.keys(T).length?T:void 0},user_id:C?.userId,session_id:C?.sessionId,tags:C?.tags,environment:S.environment,sdk_version:x}),G&&(G.setAttribute("gen_ai.completion",he),G.setAttribute("gen_ai.usage.input_tokens",X),G.setAttribute("gen_ai.usage.output_tokens",$),G.setAttribute("gen_ai.usage.total_tokens",X+$),G.setAttribute("gen_ai.usage.cached_tokens",Y),G.setAttribute("gen_ai.usage.reasoning_tokens",pe),G.setAttribute("gen_ai.response.finish_reason",J.candidates?.[0]?.finishReason??""),G.setAttribute("gen_ai.response.latency",performance.now()-b),G.setStatus({code:ce.OK}),G.end()),L}catch(L){let J=L instanceof V,X=J?"error":L.status===429?"rate_limited":L.status===408?"timeout":"error";throw S.enqueue({event_id:p(),project_id:"",trace_id:O,span_id:j,parent_span_id:C?.spanId,timestamp:new Date().toISOString(),provider:"google",model:c,operation:a,latency_ms:performance.now()-b,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:X,error_code:J?"guardrail_blocked":String(L.status??"unknown"),error_message:String(L.message).slice(0,500),...z,span_kind:"llm",conversation_turn:D,message_role:M,metadata:{auto_detected:Object.keys(T).length?T:void 0},user_id:C?.userId,session_id:C?.sessionId,tags:C?.tags,environment:S.environment,sdk_version:x}),G&&(G.setStatus({code:ce.ERROR,message:L.message}),G.recordException(L),G.end()),L}}),A&&(I.sendMessageStream=async(_,R)=>{let S=E(),b=performance.now(),C=F(),j=p(),O=C?.traceId??p(),v={...f||{},message:_},P=Ze(v),N=oe(P,c),{conversation_turn:q,message_role:T}=ae(P),D=C?.agentName??N.agent_name,M=C?.agentRole??N.agent_role,B=C?.agentId||D||M?{agent_id:C?.agentId,agent_name:D,agent_role:M,agent_framework:C?.agentFramework,parent_agent_id:C?.parentAgentId}:{},K=f?.tools?.length?re(f.tools):void 0,z=null,U=S.enableOTel?Je.getTracer("zespan-sdk"):null;U&&(z=U.startSpan(`sendMessageStream ${c}`,{kind:We.CLIENT,attributes:Ue(v,c)}));try{let G=await W({traceId:O,spanId:j,userId:C?.userId,sessionId:C?.sessionId,tags:C?.tags,agentId:C?.agentId,agentName:D,agentRole:M,agentFramework:C?.agentFramework,parentAgentId:C?.parentAgentId,conversationTurn:q,availableTools:K},()=>A(_,R));return z&&z.setAttribute("gen_ai.request.stream",!0),G.stream=Et(G.stream,{client:S,t0:b,traceId:O,spanId:j,parent:C,params:v,modelName:c,otelSpan:z,streamResult:G,agentCtx:B,tool_definitions:K,autoDetected:N,conversation_turn:q,message_role:T,operation:a}),G}catch(G){let H=G instanceof V,L=H?"error":G.status===429?"rate_limited":G.status===408?"timeout":"error";throw S.enqueue({event_id:p(),project_id:"",trace_id:O,span_id:j,parent_span_id:C?.spanId,timestamp:new Date().toISOString(),provider:"google",model:c,operation:a,latency_ms:performance.now()-b,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:L,error_code:H?"guardrail_blocked":String(G.status??"unknown"),error_message:String(G.message).slice(0,500),...B,span_kind:"llm",conversation_turn:q,message_role:T,metadata:{auto_detected:Object.keys(N).length?N:void 0},user_id:C?.userId,session_id:C?.sessionId,tags:C?.tags,environment:S.environment,sdk_version:x}),z&&(z.setStatus({code:ce.ERROR,message:G.message}),z.recordException(G),z.end()),G}}),I}),m},r}import{createHash as qn}from"crypto";import{SpanKind as Kn,SpanStatusCode as _t,trace as Ln}from"@opentelemetry/api";var qt=r=>qn("sha256").update(r).digest("hex");function Kt(r){let t=[];if(r?.systemInstruction){let e=typeof r.systemInstruction=="string"?r.systemInstruction:r.systemInstruction?.parts?.map(s=>s?.text||"").join("")||"";e&&t.push({role:"system",content:e})}let n=r?.contents;if(typeof n=="string")t.push({role:"user",content:n});else if(Array.isArray(n))for(let e of n){if(typeof e=="string"){t.push({role:"user",content:e});continue}let s=e?.role||"user",a=Array.isArray(e?.parts)?e.parts.map(o=>o?.text||"").filter(Boolean).join(""):e?.text||"";a&&t.push({role:s,content:a})}return t}function zn(r){try{if(typeof r?.text=="function")return r.text()||"";if(typeof r?.text=="string")return r.text}catch{}return(r?.candidates?.[0]?.content?.parts??[]).filter(t=>typeof t?.text=="string").map(t=>t.text).join("")}function Bn(r){return(r?.candidates?.[0]?.content?.parts??[]).filter(t=>t?.inlineData).length}function Lt(r){return{input:Number(r?.promptTokenCount||0),output:Number(r?.candidatesTokenCount||0),cached:Number(r?.cachedContentTokenCount||0),reasoning:Number(r?.thoughtsTokenCount||0)}}function zt(r,t={}){let e=r.models;if(!e||e.__zespanPatched)return r;let s=ee(t.guardrails,"chat"),a=t.operation??"chat",o=e.generateContent.bind(e),d=e.generateContentStream?.bind(e),c=e.embedContent?.bind(e),u=e.generateVideo?.bind(e);return e.generateContent=async l=>{let i=E(),g=performance.now(),m=F(),h=p(),w=m?.traceId??p(),y=l?.model??"gemini",f=!1,I=y,k;if(i.configClient){let T=i.configClient.getAbTest(a);if(T)Math.random()<T.splitPercentage/100?(I=y,y=T.candidateModel,l={...l,model:y},f=!0,k={ab_test:!0,ab_variant:"candidate"}):k={ab_test:!0,ab_variant:"control"};else{let D=i.configClient.getModel(a,y);D!==y&&(I=y,y=D,l={...l,model:D},f=!0)}}let A=Kt(l),_=oe(A,y),{conversation_turn:R,message_role:S}=ae(A),b=m?.agentName??_.agent_name,C=m?.agentRole??_.agent_role,j=m?.agentId||b||C?{agent_id:m?.agentId,agent_name:b,agent_role:C,agent_framework:m?.agentFramework,parent_agent_id:m?.parentAgentId}:{},O=l?.tools?.length?re(l.tools):void 0,v=null,P=i.enableOTel?Ln.getTracer("zespan-sdk"):null;P&&(v=P.startSpan(`generateContent ${y}`,{kind:Kn.CLIENT,attributes:{"gen_ai.system":"google","gen_ai.model":y,"gen_ai.operation.name":"chat"}})),f&&i.enqueue({event_id:p(),project_id:"",trace_id:w,span_id:h,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:y,operation:"model_override",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",metadata:{original_model:I,override_model:y},user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:i.environment,sdk_version:x});let N=null,q=null;try{N=await Z(i,s,{text:JSON.stringify(l??""),phase:"pre",traceId:w,spanId:h,model:y,operation:a,agentName:m?.agentName});let T;if(i.configClient){let Y=i.configClient.getTimeout(a);Y&&(T=AbortSignal.timeout(Y.timeoutMs))}let D=i.configClient?.getRetryPolicy(a)??null,M=i.configClient?.getConcurrencyLimit(a)??null,B=()=>ge(()=>W({traceId:w,spanId:h,userId:m?.userId,sessionId:m?.sessionId,tags:m?.tags,agentId:m?.agentId,agentName:b,agentRole:C,agentFramework:m?.agentFramework,parentAgentId:m?.parentAgentId,conversationTurn:R,availableTools:O},()=>o(T?{...l,fetchOptions:{signal:T}}:l)),D),K=await(M!==null?_e(`${i.projectId??"default"}:${a}`,M).run(B):B()),z=Lt(K?.usageMetadata),U=zn(K),G=Bn(K),H=G>0,L=(K?.candidates?.[0]?.content?.parts??[]).filter(Y=>Y?.functionCall),J=L.map(Y=>Y.functionCall.name).filter(Boolean),X={};for(let Y of L)Y.functionCall?.name&&(X[Y.functionCall.name]=Y.functionCall.args??{});q=await Z(i,s,{text:U,phase:"post",traceId:w,spanId:h,model:y,operation:a,agentName:m?.agentName});let $=ne(U,q);return i.enqueue({event_id:p(),project_id:"",trace_id:w,span_id:h,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:y,operation:a,latency_ms:performance.now()-g,stream:!1,input_tokens:z.input,output_tokens:z.output,cached_tokens:z.cached,reasoning_tokens:z.reasoning,status:"success",finish_reason:K?.candidates?.[0]?.finishReason??"",tools_used:J,tool_call_args:Object.keys(X).length?X:void 0,prompt_hash:qt(JSON.stringify(l)),...i.storePrompts?{prompt_text:i.redactForStorage(l),completion_text:i.redactForStorage($)}:{},...J.length>0&&U?{decision_rationale:U}:{},...j,span_kind:H?"image_gen":"llm",tool_definitions:O,conversation_turn:R,message_role:S,metadata:{usage:K?.usageMetadata??null,...H?{num_images:G}:{},guardrails:{pre:N?{matched:N.results.length,modified:!!N.modifiedText}:null,post:q?{matched:q.results.length,modified:!!q.modifiedText}:null},auto_detected:Object.keys(_).length?_:void 0,...k??{}},user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:i.environment,sdk_version:x}),v&&(v.setAttribute("gen_ai.completion",$),v.setAttribute("gen_ai.usage.input_tokens",z.input),v.setAttribute("gen_ai.usage.output_tokens",z.output),v.setAttribute("gen_ai.usage.total_tokens",z.input+z.output),v.setStatus({code:_t.OK}),v.end()),K}catch(T){throw T instanceof V?(v&&(v.setStatus({code:_t.ERROR,message:"Guardrail blocked"}),v.end()),T):(i.enqueue({event_id:p(),project_id:"",trace_id:w,span_id:h,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:y,operation:a,latency_ms:performance.now()-g,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_code:String(T.status??T.code??"unknown"),error_message:String(T.message).slice(0,500),...j,span_kind:"llm",user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:i.environment,sdk_version:x}),v&&(v.setStatus({code:_t.ERROR,message:T.message}),v.recordException(T),v.end()),T)}},d&&(e.generateContentStream=async l=>{let i=E(),g=performance.now(),m=F(),h=p(),w=m?.traceId??p(),y=l?.model??"gemini";if(i.configClient){let j=i.configClient.getModel(a,y);j!==y&&(y=j,l={...l,model:j})}let f=Kt(l),I=oe(f,y),{conversation_turn:k,message_role:A}=ae(f),_=m?.agentName??I.agent_name,R=m?.agentRole??I.agent_role,S=m?.agentId||_||R?{agent_id:m?.agentId,agent_name:_,agent_role:R,agent_framework:m?.agentFramework,parent_agent_id:m?.parentAgentId}:{},b=l?.tools?.length?re(l.tools):void 0,C=await d(l);return(async function*(){let j=0,O=0,v=0,P=0,N,q="",T="",D=[];try{for await(let M of C){N||(N=performance.now()-g);let B=M?.usageMetadata;B&&(j=Number(B.promptTokenCount||j),O=Number(B.candidatesTokenCount||O),v=Number(B.cachedContentTokenCount||v),P=Number(B.thoughtsTokenCount||P));let K=typeof M?.text=="function"?M.text():M?.text||"";K&&(T+=K),M?.candidates?.[0]?.finishReason&&(q=M.candidates[0].finishReason);let z=(M?.candidates?.[0]?.content?.parts??[]).map(U=>U?.functionCall).filter(Boolean);for(let U of z)U?.name&&!D.includes(U.name)&&D.push(U.name);yield M}i.enqueue({event_id:p(),project_id:"",trace_id:w,span_id:h,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:y,operation:a,latency_ms:performance.now()-g,ttft_ms:N,stream:!0,input_tokens:j,output_tokens:O,cached_tokens:v,reasoning_tokens:P,status:"success",finish_reason:q||"stream_end",tools_used:D,prompt_hash:qt(JSON.stringify(l)),...i.storePrompts?{prompt_text:i.redactForStorage(l),completion_text:i.redactForStorage(T)}:{},...S,span_kind:"llm",tool_definitions:b,conversation_turn:k,message_role:A,user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:i.environment,sdk_version:x})}catch(M){throw i.enqueue({event_id:p(),project_id:"",trace_id:w,span_id:h,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:y,operation:a,latency_ms:performance.now()-g,ttft_ms:N,stream:!0,input_tokens:j,output_tokens:O,status:"error",error_code:String(M.status??M.code??"unknown"),error_message:String(M.message).slice(0,500),...S,span_kind:"llm",user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:i.environment,sdk_version:x}),M}})()}),c&&(e.embedContent=async l=>{let i=E(),g=performance.now(),m=F(),h=p(),w=m?.traceId??p(),y=l?.model??"gemini-embedding";try{let f=await c(l),I=Lt(f?.usageMetadata);return i.enqueue({event_id:p(),project_id:"",trace_id:w,span_id:h,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:y,operation:"embedding",latency_ms:performance.now()-g,stream:!1,input_tokens:I.input,output_tokens:0,status:"success",span_kind:"embedding",user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:i.environment,sdk_version:x}),f}catch(f){throw i.enqueue({event_id:p(),project_id:"",trace_id:w,span_id:h,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:y,operation:"embedding",latency_ms:performance.now()-g,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_code:String(f.status??f.code??"unknown"),error_message:String(f.message).slice(0,500),span_kind:"embedding",environment:i.environment,sdk_version:x}),f}}),u&&(e.generateVideo=async l=>{let i=E(),g=performance.now(),m=F(),h=p(),w=m?.traceId??p(),y=l?.model??"veo";try{let f=await u(l);return i.enqueue({event_id:p(),project_id:"",trace_id:w,span_id:h,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:y,operation:"video_gen",latency_ms:performance.now()-g,stream:!1,input_tokens:0,output_tokens:0,status:"success",span_kind:"video_gen",metadata:{prompt:typeof l?.prompt=="string"?l.prompt.slice(0,500):void 0},user_id:m?.userId,session_id:m?.sessionId,tags:m?.tags,environment:i.environment,sdk_version:x}),f}catch(f){throw i.enqueue({event_id:p(),project_id:"",trace_id:w,span_id:h,parent_span_id:m?.spanId,timestamp:new Date().toISOString(),provider:"google",model:y,operation:"video_gen",latency_ms:performance.now()-g,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_code:String(f.status??f.code??"unknown"),error_message:String(f.message).slice(0,500),span_kind:"video_gen",environment:i.environment,sdk_version:x}),f}}),e.__zespanPatched=!0,r}import{createHash as $n}from"crypto";import{SpanKind as Wn,SpanStatusCode as Ve,trace as Jn}from"@opentelemetry/api";var ft=r=>$n("sha256").update(r).digest("hex");function Zn(r,t,n,e){let s=Number(r?.usage?.cost);if(Number.isFinite(s))return{cost:s,source:"usage.cost"};let a=r?.headers?.get?.("x-ratelimit-cost"),o=Number.parseFloat(String(a??""));return Number.isFinite(o)?{cost:o,source:"x-ratelimit-cost"}:{cost:0,source:"computed"}}function Un(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*Vn(r,t){let{client:n,t0:e,traceId:s,spanId:a,parent:o,params:d,otelSpan:c,agentCtx:u,autoDetected:l,conversation_turn:i,message_role:g,tool_definitions:m}=t,h,w=0,y=0,f=0,I=0,k="",A=d.model,_=[],R="",S=[],{systemPrompt:b,conversationMessages:C}=te(d.messages??[]);try{for await(let N of r){if(h||(h=performance.now()-e,n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:A,operation:"chat",latency_ms:h,ttft_ms:h,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:ft(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(C),...b?{system_prompt:n.redactForStorage(b)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x,finish_reason:"stream_start",...u,span_kind:"llm",tool_definitions:m,conversation_turn:i,message_role:g,metadata:{auto_detected:Object.keys(l).length?l:void 0}}),c&&c.setAttribute("gen_ai.response.time_to_first_token",h)),N.choices?.[0]?.delta?.content&&(w++,R+=N.choices[0].delta.content),N.choices?.[0]?.finish_reason&&(k=N.choices[0].finish_reason),N.model&&(A=N.model),N.usage){y=Number(N.usage.prompt_tokens||y||0),w=Number(N.usage.completion_tokens||w||0),f=Number(N.usage.prompt_tokens_details?.cached_tokens||f||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)S.push(q),q.function?.name&&!_.includes(q.function.name)&&_.push(q.function.name);yield N}let{tools_used:j,tool_call_args:O}=me(S),v=j.length>0?j:_,P=performance.now()-e;n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:A,operation:"chat",latency_ms:P,ttft_ms:h,stream:!0,input_tokens:y,output_tokens:w,cached_tokens:f,reasoning_tokens:I,status:"success",finish_reason:k||"stream_end",tools_used:v,tool_call_args:Object.keys(O).length?O:void 0,prompt_hash:ft(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(C),completion_text:n.redactForStorage(R),...b?{system_prompt:n.redactForStorage(b)}:{}}:{},...u,span_kind:"llm",tool_definitions:m,conversation_turn:i,message_role:g,metadata:{usage:{prompt_tokens:y,completion_tokens:w,cached_tokens:f,reasoning_tokens:I},auto_detected:Object.keys(l).length?l:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x}),c&&(c.setAttribute("gen_ai.completion",R),c.setAttribute("gen_ai.usage.input_tokens",y),c.setAttribute("gen_ai.usage.output_tokens",w),c.setAttribute("gen_ai.usage.total_tokens",y+w),c.setAttribute("gen_ai.usage.reasoning_tokens",I),c.setAttribute("gen_ai.response.finish_reason",k||"stream_end"),c.setStatus({code:Ve.OK}),c.end())}catch(j){throw n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:A,operation:"chat",latency_ms:performance.now()-e,ttft_ms:h,stream:!0,input_tokens:0,output_tokens:w,cost_usd:0,status:"error",error_code:String(j.status??"unknown"),error_message:String(j.message).slice(0,500),...u,span_kind:"llm",conversation_turn:i,message_role:g,metadata:{auto_detected:Object.keys(l).length?l:void 0},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x}),c&&(c.setStatus({code:Ve.ERROR,message:j.message}),c.recordException(j),c.end()),j}}function Bt(r,t={}){let n=r.chat.completions.create,e=ee(t.guardrails,"chat");return r.chat.completions.create=async(s,a)=>{let o=E(),d=performance.now(),c=F(),u=p(),l=c?.traceId??p(),i=s.model??"openrouter\u6A21\u578B",g=null,m=null;if(o.configClient){let _=o.configClient.getModel("chat",i);if(_!==i){let R=i;i=_,s={...s,model:_},o.enqueue({event_id:p(),project_id:"",trace_id:l,span_id:u,parent_span_id:c?.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:R,override_model:_},environment:o.environment,sdk_version:x})}}let h=c?.agentId?{agent_id:c.agentId,agent_name:c.agentName,agent_role:c.agentRole,agent_framework:c.agentFramework,parent_agent_id:c.parentAgentId}:{},w=oe(s.messages??[],s.model),{conversation_turn:y,message_role:f}=ae(s.messages??[]),I=s.tools?.length?re(s.tools):void 0,k=null,A=o.enableOTel?Jn.getTracer("zespan-sdk"):null;A&&(k=A.startSpan(`chat.completions ${i}`,{kind:Wn.CLIENT,attributes:Un(s,i)}));try{g=await Z(o,e,{text:JSON.stringify(s.messages??[]),phase:"pre",traceId:l,spanId:u,model:i,operation:"chat"});let _=await W({traceId:l,spanId:u,userId:c?.userId,sessionId:c?.sessionId,tags:c?.tags},()=>n.call(r.chat.completions,s,a));if(s.stream===!0)return k&&k.setAttribute("gen_ai.request.stream",!0),Vn(_,{client:o,t0:d,traceId:l,spanId:u,parent:c,params:s,otelSpan:k,agentCtx:h,autoDetected:w,conversation_turn:y,message_role:f,tool_definitions:I});let{prompt_tokens:S,completion_tokens:b,completion_tokens_details:C}=_.usage??{},j=C?.reasoning_tokens??0,O=_.model??s.model,{cost:v,source:P}=Zn(_,O,S??0,b??0);_.pricing||(_.pricing={cost_usd:v,source:P});let N=_.choices?.[0]?.message?.content||"";m=await Z(o,e,{text:N,phase:"post",traceId:l,spanId:u,model:O,operation:"chat"});let q=ne(N,m);q!==N&&typeof _?.choices?.[0]?.message?.content=="string"&&(_.choices[0].message.content=q);let{tools_used:T,tool_call_args:D}=Le(_.choices??[]);return o.enqueue({event_id:p(),project_id:"",trace_id:l,span_id:u,parent_span_id:c?.spanId,timestamp:new Date().toISOString(),provider:"openrouter",model:O,operation:"chat",latency_ms:performance.now()-d,stream:!1,input_tokens:S??0,output_tokens:b??0,cached_tokens:0,cost_usd:v,status:"success",finish_reason:_.choices?.[0]?.finish_reason??"",tools_used:T.length>0?T:_.choices?.[0]?.message?.tool_calls?.map(M=>M.function.name)??[],tool_call_args:Object.keys(D).length?D:void 0,prompt_hash:ft(JSON.stringify(s.messages)),...o.storePrompts?(()=>{let{systemPrompt:M,conversationMessages:B}=te(s.messages??[]);return{prompt_text:o.redactForStorage(B),completion_text:o.redactForStorage(q),...M?{system_prompt:o.redactForStorage(M)}:{}}})():{},...T.length>0&&N?{decision_rationale:N}:{},...h,span_kind:"llm",tool_definitions:I,conversation_turn:y,message_role:f,metadata:{usage:_.usage??null,pricing:_.pricing??{cost_usd:v,source:P},guardrails:{pre:g?{matched:g.results.length,modified:!!g.modifiedText}:null,post:m?{matched:m.results.length,modified:!!m.modifiedText}:null},auto_detected:Object.keys(w).length?w:void 0},user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:o.environment,sdk_version:x}),k&&(k.setAttribute("gen_ai.usage.input_tokens",S??0),k.setAttribute("gen_ai.usage.output_tokens",b??0),k.setAttribute("gen_ai.usage.total_tokens",(S??0)+(b??0)),k.setAttribute("gen_ai.usage.reasoning_tokens",j),k.setAttribute("gen_ai.completion",_.choices?.[0]?.message?.content||""),k.setAttribute("gen_ai.response.finish_reason",_.choices?.[0]?.finish_reason??""),k.setAttribute("gen_ai.response.latency",performance.now()-d),k.setStatus({code:Ve.OK}),k.end()),_}catch(_){let R=_ instanceof V,S=R?"error":_.status===429?"rate_limited":_.status===408?"timeout":"error";throw o.enqueue({event_id:p(),project_id:"",trace_id:l,span_id:u,parent_span_id:c?.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:S,error_code:R?"guardrail_blocked":String(_.status??_.code??"unknown"),error_message:String(_.message).slice(0,500),...h,span_kind:"llm",conversation_turn:y,message_role:f,metadata:{auto_detected:Object.keys(w).length?w:void 0},user_id:c?.userId,session_id:c?.sessionId,tags:c?.tags,environment:o.environment,sdk_version:x}),k&&(k.setStatus({code:Ve.ERROR,message:_.message}),k.recordException(_),k.end()),_}},r}async function*$t(r,t){let{client:n,t0:e,traceId:s,spanId:a,parent:o,model:d,operation:c,input:u}=t,l=0,i=0,g=0,m="",h,w="",y=[],f={},I="",k="",A=u?.messages??[],_=(u?.system??[]).map(S=>S.text??"").filter(Boolean),R=_.length?_.join(`
|
|
7
|
+
`):void 0;try{for await(let S of r){if(h||(h=performance.now()-e,n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:d,operation:c,latency_ms:h,ttft_ms:h,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(A),...R?{system_prompt:n.redactForStorage(R)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x})),"contentBlockStart"in S){let b=S.contentBlockStart?.start?.toolUse;b?.name&&(I=b.name,k="",y.includes(I)||y.push(I))}else if("contentBlockDelta"in S){let b=S.contentBlockDelta?.delta;b?.text&&(w+=b.text),b?.toolUse?.input&&(k+=b.toolUse.input)}else if("contentBlockStop"in S){if(I){try{f[I]=JSON.parse(k)}catch{f[I]=k}I="",k=""}}else"metadata"in S&&S.metadata?.usage?(l=S.metadata.usage.inputTokens??l,i=S.metadata.usage.outputTokens??i,g=S.metadata.usage.cacheReadInputTokens??g):"messageStop"in S&&S.messageStop?.stopReason&&(m=S.messageStop.stopReason);yield S}n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:d,operation:c,latency_ms:performance.now()-e,ttft_ms:h,stream:!0,input_tokens:l,output_tokens:i,cached_tokens:g,cost_usd:0,status:"success",finish_reason:m||"stream_end",span_kind:"llm",tools_used:y.length?y:void 0,tool_call_args:Object.keys(f).length?f:void 0,...y.length>0&&w?{decision_rationale:w}:{},...n.storePrompts?{prompt_text:n.redactForStorage(A),completion_text:n.redactForStorage(w),...R?{system_prompt:n.redactForStorage(R)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x})}catch(S){throw n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:d,operation:c,latency_ms:performance.now()-e,ttft_ms:h,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_code:String(S.$metadata?.httpStatusCode??S.name??"unknown"),error_message:String(S.message).slice(0,500),user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x}),S}}function Hn(r,t={}){let n=r,e=t.operation??"chat",s=ee(t.guardrails,e),a=n.converse?.bind(n);a&&(n.converse=async(c,...u)=>{let l=E(),i=performance.now(),g=F(),m=p(),h=g?.traceId??p(),w=c.modelId??"amazon.nova-lite-v1:0";l.configClient&&(w=l.configClient.getModel(e,w),w!==c.modelId&&(c={...c,modelId:w}));let y=JSON.stringify(c.messages??[]);await Z(l,s,{text:y,phase:"pre",traceId:h,spanId:m,model:w,operation:e,agentName:g?.agentName});try{let f=await W({traceId:h,spanId:m,userId:g?.userId,sessionId:g?.sessionId,tags:g?.tags},()=>a(c,...u)),I=f.usage??{},k=f.output?.message?.content??[],A=k.filter(b=>b.text).map(b=>b.text).join(""),_=k.filter(b=>b.toolUse),R=_.map(b=>b.toolUse?.name).filter(Boolean),S={};for(let b of _)b.toolUse?.name&&(S[b.toolUse.name]=b.toolUse.input??{});return l.enqueue({event_id:p(),project_id:"",trace_id:h,span_id:m,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:w,operation:e,latency_ms:performance.now()-i,stream:!1,input_tokens:Number(I.inputTokens??0),output_tokens:Number(I.outputTokens??0),cached_tokens:Number(I.cacheReadInputTokens??0),status:"success",finish_reason:f.stopReason??"",span_kind:"llm",tools_used:R.length?R:void 0,tool_call_args:Object.keys(S).length?S:void 0,...l.storePrompts?{prompt_text:l.redactForStorage(c.messages??[]),completion_text:l.redactForStorage(A),...(()=>{let b=(c.system??[]).map(C=>C.text??"").filter(Boolean);return b.length?{system_prompt:l.redactForStorage(b.join(`
|
|
8
|
+
`))}:{}})()}:{},...R.length>0&&A?{decision_rationale:A}:{},user_id:g?.userId,session_id:g?.sessionId,tags:g?.tags,environment:l.environment,sdk_version:x}),f}catch(f){throw l.enqueue({event_id:p(),project_id:"",trace_id:h,span_id:m,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"bedrock",model:w,operation:e,latency_ms:performance.now()-i,stream:!1,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:g?.userId,session_id:g?.sessionId,tags:g?.tags,environment:l.environment,sdk_version:x}),f}});let o=n.converseStream?.bind(n);o&&(n.converseStream=async(c,...u)=>{let l=E(),i=performance.now(),g=F(),m=p(),h=g?.traceId??p(),w=c.modelId??"amazon.nova-lite-v1:0",y=await W({traceId:h,spanId:m,userId:g?.userId,sessionId:g?.sessionId,tags:g?.tags},()=>o(c,...u));return{...y,stream:$t(y.stream,{client:l,t0:i,traceId:h,spanId:m,parent:g,model:w,operation:e,input:c})}});let d=n.send?.bind(n);return d&&(n.send=async(c,...u)=>{if(!(c?.constructor?.name==="ConverseStreamCommand"||c?.$name==="ConverseStream"))return d(c,...u);let i=E(),g=performance.now(),m=F(),h=p(),w=m?.traceId??p(),y=c?.input?.modelId??"amazon.nova-lite-v1:0",f=await W({traceId:w,spanId:h,userId:m?.userId,sessionId:m?.sessionId,tags:m?.tags},()=>d(c,...u));return{...f,stream:$t(f.stream,{client:i,t0:g,traceId:w,spanId:h,parent:m,model:y,operation:e,input:c.input})}}),r}function Qn(r,t={}){let n=r,e=t.operation??"chat",s=ee(t.guardrails,e),a=n.chat.complete.bind(n.chat);n.chat.complete=async(d,...c)=>{let u=E(),l=performance.now(),i=F(),g=p(),m=i?.traceId??p(),h=d.model??"mistral-small-latest";u.configClient&&(h=u.configClient.getModel(e,h),h!==d.model&&(d={...d,model:h}));let w=JSON.stringify(d.messages??[]);await Z(u,s,{text:w,phase:"pre",traceId:m,spanId:g,model:h,operation:e});try{let y=await W({traceId:m,spanId:g,userId:i?.userId,sessionId:i?.sessionId,tags:i?.tags},()=>a(d,...c)),f=y.model??h,I=y.usage??{},k=Number(I.promptTokens??0),A=Number(I.completionTokens??0),_=y.choices?.[0]?.finishReason??"",R=y.choices?.[0]?.message?.content??"",S=y.choices?.[0]?.message?.tool_calls??[],b=S.map(v=>v.function?.name).filter(Boolean),C={};for(let v of S)if(v.function?.name)try{C[v.function.name]=JSON.parse(v.function.arguments??"{}")}catch{C[v.function.name]={}}await Z(u,s,{text:R,phase:"post",traceId:m,spanId:g,model:f,operation:e});let{systemPrompt:j,conversationMessages:O}=te(d.messages??[]);return u.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:g,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:f,operation:e,latency_ms:performance.now()-l,stream:!1,input_tokens:k,output_tokens:A,status:"success",finish_reason:_,span_kind:"llm",tools_used:b.length?b:void 0,tool_call_args:Object.keys(C).length?C:void 0,...u.storePrompts?{prompt_text:u.redactForStorage(O),completion_text:u.redactForStorage(R),...j?{system_prompt:u.redactForStorage(j)}:{}}:{},...b.length>0&&R?{decision_rationale:R}:{},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:u.environment,sdk_version:x}),y}catch(y){throw u.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:g,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:h,operation:e,latency_ms:performance.now()-l,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_message:String(y.message).slice(0,500),user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:u.environment,sdk_version:x}),y}};let o=n.chat.stream?.bind(n.chat);return o&&(n.chat.stream=async function*(d,...c){let u=E(),l=performance.now(),i=F(),g=p(),m=i?.traceId??p(),h=d.model??"mistral-small-latest";u.configClient&&(h=u.configClient.getModel(e,h),h!==d.model&&(d={...d,model:h}));let w=0,y=0,f="",I,k="",A=[],_=[],{systemPrompt:R,conversationMessages:S}=te(d.messages??[]);try{let b=await W({traceId:m,spanId:g,userId:i?.userId,sessionId:i?.sessionId,tags:i?.tags},()=>o(d,...c));for await(let v of b){I||(I=performance.now()-l,u.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:g,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:h,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",...u.storePrompts?{prompt_text:u.redactForStorage(S),...R?{system_prompt:u.redactForStorage(R)}:{}}:{},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:u.environment,sdk_version:x}));let P=v.data?.choices?.[0]?.delta?.content;if(P&&(y++,k+=P),v.data?.usage&&(w=v.data.usage.promptTokens??w,y=v.data.usage.completionTokens??y),v.data?.choices?.[0]?.finishReason&&(f=v.data.choices[0].finishReason),v.data?.model&&(h=v.data.model),v.data?.choices?.[0]?.delta?.tool_calls)for(let N of v.data.choices[0].delta.tool_calls)_.push(N),N.function?.name&&!A.includes(N.function.name)&&A.push(N.function.name);yield v}let{tools_used:C,tool_call_args:j}=me(_),O=C.length>0?C:A;u.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:g,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:h,operation:e,latency_ms:performance.now()-l,ttft_ms:I,stream:!0,input_tokens:w,output_tokens:y,cost_usd:0,status:"success",finish_reason:f||"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&&k?{decision_rationale:k}:{},...u.storePrompts?{prompt_text:u.redactForStorage(S),completion_text:u.redactForStorage(k),...R?{system_prompt:u.redactForStorage(R)}:{}}:{},user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:u.environment,sdk_version:x})}catch(b){throw u.enqueue({event_id:p(),project_id:"",trace_id:m,span_id:g,parent_span_id:i?.spanId,timestamp:new Date().toISOString(),provider:"mistral",model:h,operation:e,latency_ms:performance.now()-l,ttft_ms:I,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_message:String(b.message).slice(0,500),user_id:i?.userId,session_id:i?.sessionId,tags:i?.tags,environment:u.environment,sdk_version:x}),b}}),r}import{createHash as Xn}from"crypto";var Wt=r=>Xn("sha256").update(r).digest("hex");async function*Yn(r,t){let{client:n,t0:e,traceId:s,spanId:a,parent:o,params:d,model:c,operation:u}=t,l,i=0,g=0,m=0,h="",w="",y=c,f=[],I=[],{systemPrompt:k,conversationMessages:A}=te(d.messages??[]);try{for await(let b of r){l||(l=performance.now()-e,n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"groq",model:y,operation:u,latency_ms:l,ttft_ms:l,stream:!0,input_tokens:0,output_tokens:0,cost_usd:0,status:"success",prompt_hash:Wt(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(A),...k?{system_prompt:n.redactForStorage(k)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x,finish_reason:"stream_start",span_kind:"llm"}));let C=b.choices?.[0]?.delta?.content;if(C&&(g++,w+=C),b.choices?.[0]?.finish_reason&&(h=b.choices[0].finish_reason),b.model&&(y=b.model),b.usage&&(i=Number(b.usage.prompt_tokens||i||0),g=Number(b.usage.completion_tokens||g||0),m=Number(b.usage.prompt_tokens_details?.cached_tokens||m||0)),b.choices?.[0]?.delta?.tool_calls)for(let j of b.choices[0].delta.tool_calls)I.push(j),j.function?.name&&!f.includes(j.function.name)&&f.push(j.function.name);yield b}let{tools_used:_,tool_call_args:R}=me(I),S=_.length>0?_:f;n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"groq",model:y,operation:u,latency_ms:performance.now()-e,ttft_ms:l,stream:!0,input_tokens:i,output_tokens:g,cached_tokens:m,status:"success",finish_reason:h||"stream_end",tools_used:S.length?S:void 0,tool_call_args:Object.keys(R).length?R:void 0,...S.length>0&&w?{decision_rationale:w}:{},prompt_hash:Wt(JSON.stringify(d.messages)),...n.storePrompts?{prompt_text:n.redactForStorage(A),completion_text:n.redactForStorage(w),...k?{system_prompt:n.redactForStorage(k)}:{}}:{},user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x,span_kind:"llm"})}catch(_){throw n.enqueue({event_id:p(),project_id:"",trace_id:s,span_id:a,parent_span_id:o?.spanId,timestamp:new Date().toISOString(),provider:"groq",model:y,operation:u,latency_ms:performance.now()-e,ttft_ms:l,stream:!0,input_tokens:0,output_tokens:g,cost_usd:0,status:"error",error_message:String(_.message).slice(0,500),user_id:o?.userId,session_id:o?.sessionId,tags:o?.tags,environment:n.environment,sdk_version:x}),_}}function es(r,t={}){let n=r,e=n.chat.completions.create.bind(n.chat.completions),s=t.operation??"chat";return n.chat.completions.create=async(a,...o)=>{let d=E(),c=performance.now(),u=F(),l=p(),i=u?.traceId??p(),g=a.model??"llama-3.3-70b-versatile";d.configClient&&(g=d.configClient.getModel(s,g),g!==a.model&&(a={...a,model:g}));try{let m=await W({traceId:i,spanId:l,userId:u?.userId,sessionId:u?.sessionId,tags:u?.tags},()=>e(a,...o));if(a.stream)return Yn(m,{client:d,t0:c,traceId:i,spanId:l,parent:u,params:a,model:g,operation:s});let h=m.usage??{},w=Number(h.prompt_tokens??0),y=Number(h.completion_tokens??0),f=m.choices?.[0]?.finish_reason??"",I=m.choices?.[0]?.message?.content??"",k=m.choices?.[0]?.message?.tool_calls??[],A=k.map(b=>b.function?.name).filter(Boolean),_={};for(let b of k)if(b.function?.name)try{_[b.function.name]=JSON.parse(b.function.arguments??"{}")}catch{_[b.function.name]={}}let{systemPrompt:R,conversationMessages:S}=te(a.messages??[]);return d.enqueue({event_id:p(),project_id:"",trace_id:i,span_id:l,parent_span_id:u?.spanId,timestamp:new Date().toISOString(),provider:"groq",model:g,operation:s,latency_ms:performance.now()-c,stream:!1,input_tokens:w,output_tokens:y,status:"success",finish_reason:f,span_kind:"llm",tools_used:A.length?A:void 0,tool_call_args:Object.keys(_).length?_:void 0,...d.storePrompts?{prompt_text:d.redactForStorage(S),completion_text:d.redactForStorage(I),...R?{system_prompt:d.redactForStorage(R)}:{}}:{},...A.length>0&&I?{decision_rationale:I}:{},user_id:u?.userId,session_id:u?.sessionId,tags:u?.tags,environment:d.environment,sdk_version:x}),m}catch(m){throw d.enqueue({event_id:p(),project_id:"",trace_id:i,span_id:l,parent_span_id:u?.spanId,timestamp:new Date().toISOString(),provider:"groq",model:g,operation:s,latency_ms:performance.now()-c,stream:!1,input_tokens:0,output_tokens:0,cost_usd:0,status:"error",error_message:String(m.message).slice(0,500),user_id:u?.userId,session_id:u?.sessionId,tags:u?.tags,environment:d.environment,sdk_version:x}),m}},r}function ts(r,t={}){let n=ot("openai").default??ot("openai"),e=new n({baseURL:r.baseURL,apiKey:r.apiKey??"litellm"});return Be(e,{...t,providerOverride:"litellm"})}import{BaseCallbackHandler as ns}from"@langchain/core/callbacks/base";import{createHash as ss}from"crypto";var Jt=r=>ss("sha256").update(r).digest("hex"),Zt=["AgentExecutor","ReactAgent","ToolCallingAgent"],Ut=["DynamicStructuredTool","StructuredTool","DynamicTool","Tool"],He=class extends ns{name="ZespanCallbackHandler";guardrailSettings=ee(void 0,"chat");runs=new Map;constructor(t={}){super(),this.guardrailSettings=ee(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 a=e.additional_kwargs?.tool_calls;if(Array.isArray(a)&&a.length>0)return a.map(d=>({name:d.function?.name??"unknown",args:d.function?.arguments?JSON.parse(d.function.arguments):{},id:d.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 a of s){let o=e.match(a);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(u=>u.test(e)))return{detected:!1};let o=/(?:step\s+\d+[:\s]|^\d+[\.\)]\s+)(.+)/gim,d=[],c;for(;(c=o.exec(t))!==null&&(c[1]&&d.push(c[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 a=s?.message?.content;return typeof a=="string"?a:Array.isArray(a)?a.map(o=>typeof o?.text=="string"?o.text:"").filter(Boolean).join(`
|
|
9
|
+
`):""}async handleChatModelStart(t,n,e,s,a,o,d,c){try{let u=n.map(i=>i.map(g=>{let m=g._getType?.()??g.role??"unknown",h=typeof g.content=="string"?g.content:JSON.stringify(g.content);return`${m}: ${h}`}).join(`
|
|
10
|
+
`)),l=n[0]?.map(i=>({role:i._getType?.()??i.role??"unknown",content:typeof i.content=="string"?i.content:JSON.stringify(i.content),tool_calls:i.tool_calls??i.additional_kwargs?.tool_calls}));await this.handleLLMStart(t,u,e,s,{...a,_zespan_structured_messages:l,_zespan_message_count:n[0]?.length??0})}catch{}}async handleLLMStart(t,n,e,s,a){try{let o=F(),d=E(),c=s?this.runs.get(s):void 0,u=p();le(u);let l=c?.traceId??o?.traceId??p(),i=a?.invocation_params?.model??a?.invocation_params?.model_name??t.kwargs?.model??t.kwargs?.modelName??t.id?.[t.id.length-1]??"unknown",g=a?.invocation_params?.stream===!0,m=c?.spanKind==="agent"?c.agentId:void 0,h=c?.spanKind==="agent"?c.agentName:void 0;if(!h){let k=this.detectAgentFromPrompts(n,a);k&&(h=k,m=m??p())}let w=a?.invocation_params?.tools??a?.invocation_params?.functions??[],y=w.map(k=>k.function?.name??k.name??"unknown").filter(Boolean),f=this.detectConversationInfo(n,a),I=await Z(d,this.guardrailSettings,{text:JSON.stringify(n??[]),phase:"pre",traceId:l,spanId:u,model:i,operation:"chat"});this.runs.set(e,{t0:performance.now(),traceId:l,spanId:u,parentSpanId:c?.spanId??o?.spanId,model:i,agentId:m,agentName:h,spanKind:"llm",isStreaming:g,parent:o,prompts:n,structuredMessages:a?._zespan_structured_messages,invocationParams:a?.invocation_params,preGuardrailResult:I,toolsAvailable:y,toolSchemas:d.storePrompts?w:void 0,conversationTurns:f.turns,hasMemory:f.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 a=E();a.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:Jt(JSON.stringify(s.prompts)),...a.storePrompts?(()=>{let{systemPrompt:o,conversationMessages:d}=te(s.structuredMessages??[]);return{prompt_text:a.redactForStorage(d.length?d:s.prompts),...o?{system_prompt:a.redactForStorage(o)}:{}}})():{},metadata:{invocation_params:s.invocationParams??null},user_id:s.parent?.userId,session_id:s.parent?.sessionId,tags:s.parent?.tags,environment:a.environment,sdk_version:x,finish_reason:"stream_start"})}catch{}}async handleLLMEnd(t,n){try{let e=this.runs.get(n);if(!e)return;se(e.spanId),this.runs.delete(n);let s=E(),a=t.llmOutput?.tokenUsage?.promptTokens??t.llmOutput?.usageMetadata?.promptTokenCount??0,o=t.llmOutput?.tokenUsage?.completionTokens??t.llmOutput?.usageMetadata?.candidatesTokenCount??0,d=performance.now()-e.t0,c=this.extractCompletion(t),u=this.extractToolCalls(t),l=t.llmOutput?.tokenUsage?.reasoningTokens??t.llmOutput?.usage?.completion_tokens_details?.reasoning_tokens??t.llmOutput?.reasoningTokens??t.llmOutput?.usageMetadata?.thoughtsTokenCount??0,i=t.llmOutput?.tokenUsage?.cachedTokens??t.llmOutput?.usage?.prompt_tokens_details?.cached_tokens??t.llmOutput?.usageMetadata?.cachedContentTokenCount??0,g=this.detectPlanning(c,e.prompts),m=t.generations?.[0]?.[0]?.generationInfo?.finish_reason??t.generations?.[0]?.[0]?.generationInfo?.finishReason??t.llmOutput?.finish_reason??(e.isStreaming?"stream_end":"stop"),h=await Z(s,this.guardrailSettings,{text:c,phase:"post",traceId:e.traceId,spanId:e.spanId,model:e.model,operation:"chat"}),w=ne(c,h);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:d,ttft_ms:e.ttft,stream:e.isStreaming??!1,input_tokens:a,output_tokens:o,status:"success",span_kind:"llm",agent_id:e.agentId,agent_name:e.agentName,finish_reason:m,prompt_hash:Jt(JSON.stringify(e.prompts)),tools_defined:e.toolsAvailable?.length?e.toolsAvailable:void 0,tools_used:u.length>0?u.map(y=>y.name):void 0,tool_calls:u.length>0?u:void 0,conversation_turns:e.conversationTurns,has_memory:e.hasMemory,reasoning_tokens:l>0?l:void 0,has_reasoning:l>0?!0:void 0,cached_tokens:i>0?i:void 0,has_planning:g.detected||void 0,planning_steps:g.steps,...s.storePrompts?(()=>{let{systemPrompt:y,conversationMessages:f}=te(e.structuredMessages??[]);return{prompt_text:s.redactForStorage(f.length?f:e.prompts),completion_text:s.redactForStorage(w),...y?{system_prompt:s.redactForStorage(y)}:{}}})():{},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:h?{matched:h.results.length,modified:!!h.modifiedText}:null}},user_id:e.parent?.userId,session_id:e.parent?.sessionId,tags:e.parent?.tags,environment:s.environment,sdk_version:x})}catch{}}async handleLLMError(t,n){try{let e=this.runs.get(n);if(!e)return;se(e.spanId),this.runs.delete(n);let s=E();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:x})}catch{}}async handleChainStart(t,n,e,s){try{let a=F(),o=s?this.runs.get(s):void 0,d=p(),c=o?.traceId??a?.traceId??p(),u=Array.isArray(t?.id)?t.id:[],l=u.some(k=>Zt.includes(k)),i=!l&&u.some(k=>Ut.includes(k)),g=l?"agent":i?"tool":"general",m=l?p():void 0,h=l?u.find(k=>Zt.includes(k))??void 0:void 0,w=i?t.name??t.kwargs?.name??u.find(k=>Ut.includes(k)):void 0,y=i?JSON.stringify(n):void 0,f=o?.spanKind==="agent"?o.agentId:void 0,I=o?.spanKind==="agent"?o.agentName:void 0;this.runs.set(e,{t0:performance.now(),traceId:c,spanId:d,parentSpanId:o?.spanId??a?.spanId,model:"",agentId:m??o?.agentId,agentName:h??o?.agentName,spanKind:g,toolName:w,toolInput:y,prompts:[],parent:a,delegatedFromAgentId:f,delegatedFromAgentName:I})}catch{}}async handleChainEnd(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=E();if(e.spanKind==="tool"){let a=t?.output??t,o=typeof a=="string"?a:JSON.stringify(a);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:x})}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:x})}catch{}}async handleChainError(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=E();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:x})}catch{}}async handleToolStart(t,n,e,s){try{let a=F(),o=s?this.runs.get(s):void 0,d=p(),c=o?.traceId??a?.traceId??p(),u=t.id?.[t.id.length-1]??"unknown_tool",l=t.name||t.kwargs?.name||u;this.runs.set(e,{t0:performance.now(),traceId:c,spanId:d,parentSpanId:o?.spanId??a?.spanId,model:"",agentId:o?.agentId,agentName:o?.agentName,spanKind:"tool",toolName:l,toolInput:n,prompts:[],parent:a})}catch{}}async handleToolEnd(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=E(),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:x})}catch{}}async handleToolError(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=E();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:x})}catch{}}async handleAgentAction(t,n){try{let e=this.runs.get(n),s=F(),a=E(),o=e?.traceId??s?.traceId??p(),d=p(),c=[t?.tool??"",typeof t?.toolInput=="string"?t.toolInput:JSON.stringify(t?.toolInput??""),t?.log??""].join(" | ");a.enqueue({event_id:p(),project_id:"",trace_id:o,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:[c],...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:a.environment,sdk_version:x})}catch{}}async handleRetrieverStart(t,n,e,s){try{let a=F(),o=s?this.runs.get(s):void 0,d=p(),c=o?.traceId??a?.traceId??p();this.runs.set(e,{t0:performance.now(),traceId:c,spanId:d,parentSpanId:o?.spanId??a?.spanId,model:"",agentId:o?.agentId,agentName:o?.agentName,spanKind:"retriever",query:n,prompts:[],parent:a})}catch{}}async handleRetrieverEnd(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=E();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:x})}catch{}}async handleRetrieverError(t,n){try{let e=this.runs.get(n);if(!e)return;this.runs.delete(n);let s=E();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:x})}catch{}}};function Vt(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 os(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 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,a=F(),o=this.getParentAgentState(n,s),d=o?.traceId??a?.traceId??p(),c=p(),u=o?.spanId??a?.spanId,l=p(),i=Vt(t.invocationContext?.agent),g=t.userId||a?.userId,m=t.sessionId||a?.sessionId;this.agentStates.set(this.agentKey(n,e),{t0:performance.now(),traceId:d,spanId:c,parentSpanId:u,agentId:l,agentName:e,model:i,userId:g,sessionId:m,parentAgentId:o?.agentId,parentAgentName:s})}catch{}};afterAgentCallback=t=>{try{let n=t.invocationId,e=t.agentName,s=this.agentKey(n,e),a=this.agentStates.get(s);if(!a)return;this.agentStates.delete(s);let o=!a.parentAgentName,d=E();d.enqueue({event_id:p(),project_id:"",trace_id:a.traceId,span_id:a.spanId,parent_span_id:a.parentSpanId,timestamp:new Date().toISOString(),provider:"google-adk",model:a.model,operation:`agent.${e}`,span_kind:"agent",agent_id:a.agentId,agent_name:e,agent_role:o?"coordinator":"specialist",agent_framework:"google-adk",parent_agent_id:a.parentAgentId,delegation_from_agent_id:a.parentAgentId,delegation_from_agent_name:a.parentAgentName,latency_ms:performance.now()-a.t0,stream:!1,input_tokens:0,output_tokens:0,status:"success",user_id:a.userId,session_id:a.sessionId,environment:d.environment,sdk_version:x})}catch{}};beforeModelCallback=({context:t,request:n})=>{try{let e=t.invocationId,s=t.agentName,a=this.agentStates.get(this.agentKey(e,s)),o=a?.traceId??p(),d=Vt(t.invocationContext?.agent),c=n?.config?.systemInstruction,u;if(c){let g=typeof c=="string"?c:(c?.parts??[]).filter(m=>m?.text).map(m=>m.text).join("");g&&(u=g)}let l,i=n?.contents??[];if(i.length){let g=i.map(m=>{let h=m?.role??"user",w=(m?.parts??[]).filter(y=>typeof y?.text=="string"&&!y.thought).map(y=>y.text).join("");return w?`${h}: ${w}`:""}).filter(Boolean).join(`
|
|
11
|
+
`);g&&(l=g)}this.pendingModelCalls.set(this.agentKey(e,s),{t0:performance.now(),spanId:p(),traceId:o,parentSpanId:a?.spanId??"",model:d,agentId:a?.agentId,agentName:s,userId:a?.userId??t.userId,sessionId:a?.sessionId??t.sessionId,promptText:l,systemPrompt:u})}catch{}};afterModelCallback=({context:t,response:n})=>{try{let e=t.invocationId,s=t.agentName,a=this.agentKey(e,s),o=this.pendingModelCalls.get(a);if(!o)return;this.pendingModelCalls.delete(a);let{input:d,output:c,cached:u,reasoning:l}=os(n?.usageMetadata),i=n?.content?.parts??[],g=i.filter(w=>typeof w?.text=="string"&&!w.thought).map(w=>w.text).join(""),m=i.filter(w=>w?.functionCall?.name).map(w=>w.functionCall.name),h=E();h.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:d,output_tokens:c,cached_tokens:u||void 0,reasoning_tokens:l||void 0,finish_reason:n?.finishReason??"stop",tools_used:m.length?m:void 0,status:n?.errorCode?"error":"success",error_message:n?.errorMessage,...h.storePrompts&&o.systemPrompt?{system_prompt:h.redactForStorage(o.systemPrompt)}:{},...h.storePrompts&&o.promptText?{prompt_text:h.redactForStorage(o.promptText)}:{},...h.storePrompts&&g?{completion_text:h.redactForStorage(g)}:{},user_id:o.userId,session_id:o.sessionId,environment:h.environment,sdk_version:x})}catch{}};beforeToolCallback=({tool:t,args:n,context:e})=>{try{let s=e.invocationId,a=e.agentName,o=e.functionCallId??p(),d=this.agentStates.get(this.agentKey(s,a));this.toolCallStates.set(o,{t0:performance.now(),spanId:p(),traceId:d?.traceId??p(),parentSpanId:d?.spanId??"",toolName:t?.name??"unknown_tool",args:n??{},agentId:d?.agentId,agentName:a,userId:d?.userId??e.userId,sessionId:d?.sessionId??e.sessionId})}catch{}};afterToolCallback=({tool:t,args:n,context:e,response:s})=>{try{let a=e.functionCallId??"",o=this.toolCallStates.get(a);if(!o)return;this.toolCallStates.delete(a);let d=E();d.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:a,tool_call_args:o.args,tool_call_result:JSON.stringify(s??{}).slice(0,10240),user_id:o.userId,session_id:o.sessionId,environment:d.environment,sdk_version:x})}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 as}from"crypto";var ht=r=>as("sha256").update(typeof r=="string"?r:JSON.stringify(r)).digest("hex");function rs(r){return r?.model??r?.config?.model??r?.llmConfig?.model??r?.generativeModel?.model??"gemini-2.5-flash"}function Qt(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 is(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 cs(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 Xt(r){if(r instanceof V)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 Xe(r,t={}){let n=r;if(typeof n?.run!="function")return r;let e=n.run.bind(n),s=ee(t.guardrails,"agent"),a=t.agentName??n.name??"ADKAgent",o=t.agentRole??(n.subAgents?.length?"coordinator":"specialist"),d=rs(n);return t.traceSubAgents!==!1&&Array.isArray(n.subAgents)&&(n.subAgents=n.subAgents.map(c=>Xe(c,{...t,agentRole:"specialist",traceSubAgents:!0}))),n.run=async(c,u)=>{let l=E(),i=performance.now(),g=F(),m=p(),h=p(),w=g?.traceId??p(),y=t.sessionId??g?.sessionId,f=typeof c=="string"?c:JSON.stringify(c??"");return W({traceId:w,spanId:h,userId:g?.userId,sessionId:y,tags:g?.tags,agentId:m,agentName:a,agentRole:o,agentFramework:"google-adk",parentAgentId:g?.agentId},async()=>{try{let I=await Z(l,s,{text:f,phase:"pre",traceId:w,spanId:h,model:d,operation:"agent"}),k=await e(c,u),{input:A,output:_,cached:R,reasoning:S}=Qt(k),b=is(k),C=cs(k),j=k?.candidates?.[0]?.finishReason??k?.finish_reason,O=await Z(l,s,{text:b,phase:"post",traceId:w,spanId:h,model:d,operation:"agent"}),v=ne(b,O);return l.enqueue({event_id:p(),project_id:"",trace_id:w,span_id:h,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:d,operation:"agent",span_kind:"agent",agent_id:m,agent_name:a,agent_role:o,agent_framework:"google-adk",parent_agent_id:g?.agentId,delegation_from_agent_id:g?.agentId,delegation_from_agent_name:g?.agentName,latency_ms:performance.now()-i,stream:!1,input_tokens:A,output_tokens:_,cached_tokens:R||void 0,reasoning_tokens:S||void 0,status:"success",finish_reason:j,tools_used:C.length?C.map(P=>P.name):void 0,tool_call_args:C.length?Object.fromEntries(C.map(P=>[P.name,P.args])):void 0,prompt_hash:ht(f),...l.storePrompts?{prompt_text:l.redactForStorage(f),completion_text:l.redactForStorage(v)}:{},...C.length>0&&b?{decision_rationale:b}:{},user_id:g?.userId,session_id:y,tags:g?.tags,environment:l.environment,sdk_version:x,metadata:{guardrails:{pre:I?{matched:I.results.length,modified:!!I.modifiedText}:null,post:O?{matched:O.results.length,modified:!!O.modifiedText}:null}}}),k}catch(I){throw l.enqueue({event_id:p(),project_id:"",trace_id:w,span_id:h,parent_span_id:g?.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:d,operation:"agent",span_kind:"agent",agent_id:m,agent_name:a,agent_role:o,agent_framework:"google-adk",parent_agent_id:g?.agentId,latency_ms:performance.now()-i,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_code:Xt(I),error_message:String(I?.message??I).slice(0,500),user_id:g?.userId,session_id:y,tags:g?.tags,environment:l.environment,sdk_version:x}),I}})},r}function yt(r,t={}){let n=r;if(typeof n?.runEphemeral=="function"){let e=n.runEphemeral.bind(n);n.runEphemeral=s=>Ht(e(s),s,t)}if(typeof n?.run=="function"){let e=n.run.bind(n);n.run=s=>Ht(e(s),s,t)}return r}async function*Ht(r,t,n){let e=E(),s=F(),a=s?.traceId??p(),o=p(),d=p(),c=performance.now(),u=n.agentName??"ADKAgent",l=n.agentRole??"coordinator",i=n.model??"gemini-2.5-flash",g=t?.userId??s?.userId,m=n.sessionId??t?.sessionId??s?.sessionId,h=(t?.newMessage?.parts??[]).map(S=>typeof S?.text=="string"?S.text:"").filter(Boolean).join(""),w=new Map,y=new Map,f=new Set,I="",k=0,A=0,_=0,R=0;le(o);try{yield*W({traceId:a,spanId:o,userId:g,sessionId:m,tags:s?.tags,agentId:d,agentName:u,agentRole:l,agentFramework:"google-adk",parentAgentId:s?.agentId},async function*(){for await(let S of r){k++;let b=S.author??u;if(b!=="user"){b!==u&&!f.has(b)&&e.enqueue({event_id:p(),project_id:"",trace_id:a,span_id:p(),parent_span_id:o,timestamp:new Date().toISOString(),provider:"google-adk",model:i,operation:`handoff.${u}\u2192${b}`,span_kind:"handoff",agent_name:b,agent_role:"specialist",agent_framework:"google-adk",delegation_from_agent_id:d,delegation_from_agent_name:u,delegation_reason:`delegated to ${b}`,latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",user_id:g,session_id:m,tags:s?.tags,environment:e.environment,sdk_version:x}),f.add(b),w.has(b)||(w.set(b,{spanId:p(),agentId:p(),t0:performance.now()}),y.set(b,{input:0,output:0,cached:0,reasoning:0}));let C=w.get(b),j=Qt(S);if(j.input>0||j.output>0){let N=y.get(b);N.input+=j.input,N.output+=j.output,N.cached+=j.cached,N.reasoning+=j.reasoning,A+=j.input,_+=j.output,R+=j.cached}let O=S.content?.parts??[];for(let N of O){let q=N?.functionCall;if(!q?.name)continue;let T=p();e.enqueue({event_id:p(),project_id:"",trace_id:a,span_id:T,parent_span_id:C.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:i,operation:`tool.${q.name}`,span_kind:"tool",agent_id:C.agentId,agent_name:b,agent_framework:"google-adk",latency_ms:0,stream:!1,input_tokens:0,output_tokens:0,status:"success",tools_used:[q.name],tool_call_id:T,tool_call_args:q.args??{},user_id:g,session_id:m,tags:s?.tags,environment:e.environment,sdk_version:x})}let v=O.filter(N=>typeof N?.text=="string"&&!N.thought).map(N=>N.text).join("");if(v&&(I+=v),typeof S.isFinalResponse=="function"?S.isFinalResponse():!!S.isFinalResponse||S.actions?.stateDelta!==void 0||S.turnComplete===!0){let N=O.map(D=>D?.functionCall?.name).filter(Boolean),q=O.filter(D=>typeof D?.text=="string"&&!D.thought).map(D=>D.text).join(""),T=y.get(b)??{input:0,output:0,cached:0,reasoning:0};e.enqueue({event_id:p(),project_id:"",trace_id:a,span_id:C.spanId,parent_span_id:o,timestamp:new Date().toISOString(),provider:"google-adk",model:i,operation:`agent.${b}`,span_kind:"agent",agent_id:C.agentId,agent_name:b,agent_role:b===u?l:"specialist",agent_framework:"google-adk",parent_agent_id:d,latency_ms:performance.now()-C.t0,stream:!1,input_tokens:T.input,output_tokens:T.output,cached_tokens:T.cached||void 0,reasoning_tokens:T.reasoning||void 0,status:"success",tools_used:N.length?N:void 0,prompt_hash:h?ht(h):void 0,...e.storePrompts&&v?{completion_text:e.redactForStorage(v)}:{},...N.length>0&&q?{decision_rationale:q}:{},user_id:g,session_id:m,tags:s?.tags,environment:e.environment,sdk_version:x}),w.delete(b),y.delete(b)}}yield S}}),e.enqueue({event_id:p(),project_id:"",trace_id:a,span_id:o,parent_span_id:s?.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:i,operation:"agent.run",span_kind:"agent",agent_id:d,agent_name:u,agent_role:l,agent_framework:"google-adk",parent_agent_id:s?.agentId,latency_ms:performance.now()-c,stream:!1,input_tokens:A,output_tokens:_,cached_tokens:R||void 0,status:"success",prompt_hash:h?ht(h):void 0,...e.storePrompts?{prompt_text:e.redactForStorage(h),completion_text:e.redactForStorage(I)}:{},user_id:g,session_id:m,tags:s?.tags,environment:e.environment,sdk_version:x,metadata:{eventCount:k,agentsThatRan:[...f]}}),se(o)}catch(S){throw se(o),e.enqueue({event_id:p(),project_id:"",trace_id:a,span_id:o,parent_span_id:s?.spanId,timestamp:new Date().toISOString(),provider:"google-adk",model:i,operation:"agent.run",span_kind:"agent",agent_id:d,agent_name:u,agent_role:l,agent_framework:"google-adk",parent_agent_id:s?.agentId,latency_ms:performance.now()-c,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_code:Xt(S),error_message:String(S?.message??S).slice(0,500),user_id:g,session_id:m,tags:s?.tags,environment:e.environment,sdk_version:x}),S}}function Yt(r){let{coordinator:t,runner:n,agentName:e,agentRole:s,model:a,...o}=r,d=Xe(t,{...o,agentName:e??t?.name,agentRole:s??"coordinator",traceSubAgents:!0}),c=yt(n,{agentName:e??t?.name,agentRole:s??"coordinator",model:a});return{coordinator:d,runner:c}}var en=Xe;import{NodeTracerProvider as ds}from"@opentelemetry/sdk-trace-node";import{ExportResultCode as tn}from"@opentelemetry/core";function ls(){try{return E()}catch{return null}}var It=class{export(t,n){let e=ls();if(!e){n({code:tn.SUCCESS});return}try{for(let s of t){let{name:a}=s;if(!a.startsWith("ai."))continue;let o=s.attributes,d=F(),c=s.spanContext().spanId,u=s.spanContext().traceId,l=s.duration[0]*1e3+s.duration[1]/1e6,i=new Date(s.startTime[0]*1e3+s.startTime[1]/1e6).toISOString(),g=s.status.code===2,m={event_id:p(),project_id:"",trace_id:d?.traceId??u,span_id:c,parent_span_id:d?.spanId,timestamp:i,provider:String(o["ai.model.provider"]??"unknown"),model:String(o["ai.model.id"]??"unknown"),latency_ms:l,stream:!1,status:g?"error":"success",environment:e.environment,sdk_version:x};a==="ai.generateText"||a==="ai.generateText.doGenerate"?(se(c),e.enqueue({...m,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"]))}:{}})):a==="ai.streamText"||a==="ai.streamText.doStream"?(se(c),e.enqueue({...m,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"})):a==="ai.generateObject"||a==="ai.generateObject.doGenerate"?(se(c),e.enqueue({...m,operation:"generateObject",input_tokens:Number(o["ai.usage.promptTokens"]??0),output_tokens:Number(o["ai.usage.completionTokens"]??0),span_kind:"general"})):a==="ai.toolCall"?e.enqueue({...m,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"])]:[]}):(a==="ai.embed"||a==="ai.embedMany")&&e.enqueue({...m,operation:a==="ai.embed"?"embed":"embedMany",input_tokens:Number(o["ai.usage.tokens"]??0),output_tokens:0,span_kind:"embedding"})}}catch{}n({code:tn.SUCCESS})}shutdown(){return Promise.resolve()}},St=class{exporter;constructor(t){this.exporter=t}onStart(t,n){t.name?.startsWith("ai.")&&le(t.spanContext().spanId)}onEnd(t){t.name.startsWith("ai.")&&se(t.spanContext().spanId),this.exporter.export([t],()=>{})}forceFlush(){return Promise.resolve()}shutdown(){return Promise.resolve()}},Ye=null,kt=null;function vt(){return Ye||(Ye=new ds,Ye.addSpanProcessor(new St(new It)),kt=Ye.getTracer("zespan-vercel-ai")),{tracer:kt,telemetry:{isEnabled:!0,tracer:kt}}}function nn(){return vt().telemetry}function Re(){try{return E()}catch{return null}}function us(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 et=class{runs=new Map;guardrailSettings;constructor(t={}){this.guardrailSettings=ee(t.guardrails,"chat")}_onLLMStart=async t=>{let n=t.id??p(),e=F(),s=p();le(s);let a=t.detail?.llm?.metadata?.model_name??t.detail?.model??"unknown",o=e?.traceId??p(),d=JSON.stringify(t.detail?.messages??t.detail?.prompt??t.detail?.input??""),c;try{let u=Re();u&&d&&(c=await Z(u,this.guardrailSettings,{text:d,phase:"pre",traceId:o,spanId:s,model:a,operation:"chat"}))}catch(u){if(u instanceof V)throw u}this.runs.set(n,{t0:performance.now(),traceId:o,spanId:s,parentSpanId:e?.spanId,model:a,preGuardrailResult:c,promptText:d})};_onLLMEnd=async t=>{let n=this.runs.get(t.id??"");if(!n)return;this.runs.delete(t.id),se(n.spanId);let e=Re();if(!e)return;let s=t.detail?.response??t.detail?.chunk??{},a=s.usage??s.raw?.usage??{},o=(typeof s.text=="string"?s.text:null)??(typeof s.message?.content=="string"?s.message.content:null)??"",d=o,c;if(o)try{c=await Z(e,this.guardrailSettings,{text:o,phase:"post",traceId:n.traceId,spanId:n.spanId,model:n.model,operation:"chat"}),d=ne(o,c)}catch(l){l instanceof V||(c=void 0)}let u=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:us(n.model),model:n.model,operation:"chat",latency_ms:performance.now()-n.t0,ttft_ms:n.ttft,stream:!1,input_tokens:a.promptTokens??a.prompt_tokens??0,output_tokens:a.completionTokens??a.completion_tokens??0,finish_reason:s.finishReason??s.finish_reason??"",status:"success",span_kind:"llm",agent_id:F()?.agentId,agent_name:F()?.agentName,...e.storePrompts?{prompt_text:e.redactForStorage(n.promptText??""),completion_text:e.redactForStorage(d)}:{},metadata:{guardrails:{pre:u?{matched:u.results.length,modified:!!u.modifiedText}:null,post:c?{matched:c.results.length,modified:!!c.modifiedText}:null}},environment:e.environment,sdk_version:x})}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=Re();if(!n)return;let e=F(),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:x})}catch{}};_onToolResult=t=>{let n=Re();if(!n)return;let e=F(),s=t.detail?.toolCall?.name??t.detail?.name??"",a=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(a).slice(0,10240),agent_id:e?.agentId,agent_name:e?.agentName,environment:n.environment,sdk_version:x})}catch{}};_onAgentStart=t=>{let n=t.id??p(),e=F();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=Re();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:x})}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 sn(r){let t=E(),n=F(),e=n?.traceId??p(),s=p(),a=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:r.span_kind??"general"}:{},d={};return{span:{setEvalScore(u,l){d[u]=l},async end(u){let l=performance.now()-a;try{t.enqueue({event_id:p(),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:l,stream:!1,input_tokens:u.input_tokens??0,output_tokens:u.output_tokens??0,cost_usd:u.cost_usd,status:u.status,error_message:u.error_message?.slice(0,500),environment:t.environment,eval_scores:Object.keys(d).length?d:void 0,sdk_version:x,...o})}catch{}}},run:u=>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},u)}}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=E(),e=F(),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:x,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=E(),a=F(),o=p(),d=a?.traceId??p(),c=performance.now(),u=this.tools.find(l=>l.name===t);try{let l=await W({traceId:d,spanId:o,userId:a?.userId,sessionId:a?.sessionId,tags:a?.tags,agentId:this.agentId,agentName:this.name,agentRole:this.role,agentFramework:this.framework},()=>e()),i=typeof l=="string"?l:JSON.stringify(l),g={event_id:p(),project_id:"",trace_id:d,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"tool",model:"",operation:`tool.${t}`,latency_ms:performance.now()-c,stream:!1,input_tokens:0,output_tokens:0,status:"success",environment:s.environment,sdk_version:x,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:i.slice(0,10240),tool_definitions:u?[u]:void 0,user_id:a?.userId,session_id:a?.sessionId,tags:a?.tags};return s.enqueue(g),l}catch(l){let i=l instanceof Error?l.message:String(l),g={event_id:p(),project_id:"",trace_id:d,span_id:o,parent_span_id:a?.spanId,timestamp:new Date().toISOString(),provider:"tool",model:"",operation:`tool.${t}`,latency_ms:performance.now()-c,stream:!1,input_tokens:0,output_tokens:0,status:"error",error_message:i.slice(0,500),environment:s.environment,sdk_version:x,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:a?.userId,session_id:a?.sessionId,tags:a?.tags};throw s.enqueue(g),l}}delegateTo(t,n){let e=E(),s=F(),a={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:x,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(a)}};function on(r,t){let n=E(),e=F(),s=p(),a=e?.traceId??p(),o=p(),d={event_id:p(),project_id:"",trace_id:a,span_id:o,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:x,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:a,spanId:o,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 tt(s,r)))}import{propagation as ps}from"@opentelemetry/api";var wt=class{onStart(t,n){let e=ps.getBaggage(n);e&&e.getAllEntries().forEach(([s,a])=>{t.setAttribute(`baggage.${s}`,a.value)})}onEnd(){}shutdown(){return Promise.resolve()}forceFlush(){return Promise.resolve()}};import{context as an,propagation as Ne}from"@opentelemetry/api";function nt(r,t,n){let e={};t&&(e["agent.delegation.reason"]=t),n&&(e["agent.task"]=n);let s=an.active();if(Object.keys(e).length>0){let a=Ne.getBaggage(s)??Ne.createBaggage();for(let[o,d]of Object.entries(e))a=a.setEntry(o,{value:d});s=Ne.setBaggage(s,a)}Ne.inject(s,r)}function st(r){return Ne.extract(an.active(),r)}var ms={OTEL_EXPORTER_OTLP_ENDPOINT:"http://localhost:4318",OTEL_SERVICE_NAME:"crewai-agent"};function gs(){return`
|
|
11
12
|
pip install openinference-instrumentation-crewai opentelemetry-exporter-otlp-proto-grpc
|
|
12
13
|
|
|
13
14
|
from openinference.instrumentation.crewai import CrewAIInstrumentor
|
|
@@ -20,4 +21,4 @@ provider = TracerProvider()
|
|
|
20
21
|
provider.add_span_processor(BatchSpanProcessor(OTLPSpanExporter(endpoint="http://localhost:4318")))
|
|
21
22
|
trace.set_tracer_provider(provider)
|
|
22
23
|
CrewAIInstrumentor().instrument()
|
|
23
|
-
`.trim()}function
|
|
24
|
+
`.trim()}function _s(r,t){let n={};return nt(n,t),{...r,_zespan_trace:n}}function fs(r){let t=r._zespan_trace??{};return st(t)}function hs(r){return{OTEL_EXPORTER_OTLP_ENDPOINT:r,OTEL_EXPORTER_OTLP_PROTOCOL:"http/protobuf",OTEL_SERVICE_NAME:"pydantic-ai-agent",LOGFIRE_SEND_TO_LOGFIRE:"false"}}var Dr={init:pt,getClient:E,getZespanClient:E,wrapOpenAI:Be,wrapAnthropic:Ft,wrapGoogle:Gt,wrapGoogleGenAI:zt,wrapOpenRouter:Bt,wrapADKAgent:en,wrapADKRunner:yt,instrumentADK:Yt,startSpan:sn,withContext:rt,withTrace:it,withAgent:on,initOTel:je,getTracer:ye,createSpan:Ct,withSpan:xt,PromptClient:we,CallbackHandler:He,ADKCallbackHandler:Qe,autopatch:qe,markFrameworkActive:le,markFrameworkInactive:se,instrumentVercelAI:vt,getZespanVercelTelemetry:nn,ZespanLlamaIndexHandler:et};export{tt as AgentContext,wt as BaggageSpanProcessor,ms as CREWAI_OTEL_ENV,V as GuardrailBlockedError,we as PromptClient,ve as SpanKind,Se as SpanStatusCode,Qe as ZespanADKCallbackHandler,He as ZespanCallbackHandler,Ke as ZespanClient,et as ZespanLlamaIndexHandler,_s as attachTraceToAutoGenMessage,qe as autopatch,Ct as createSpan,st as extractAgentContext,st as extractAutoGenContext,fs as extractTraceFromAutoGenMessage,gs as getCrewAIInstrumentationGuide,hs as getPydanticAIConfig,ye as getTracer,E as getZespanClient,nn as getZespanVercelTelemetry,bn as init,je as initOTel,pt as initZespan,nt as injectAgentContext,nt as injectAutoGenContext,Yt as instrumentADK,vt as instrumentVercelAI,le as markFrameworkActive,se as markFrameworkInactive,sn as startSpan,on as withAgent,xt as withSpan,kn as withZespan,rt as withZespanContext,it as withZespanTrace,Xe as wrapADKAgent,yt as wrapADKRunner,Ft as wrapAnthropic,Hn as wrapBedrock,Gt as wrapGoogle,zt as wrapGoogleGenAI,es as wrapGroq,ts as wrapLiteLLM,Qn as wrapMistral,Be as wrapOpenAI,Bt as wrapOpenRouter,en as wrapZespanADKAgent,Dr as zespan};
|