openlit 1.13.0 → 1.14.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/README.md +34 -1
- package/dist/__tests__/manual-trace.test.js +118 -0
- package/dist/__tests__/manual-trace.test.js.map +1 -0
- package/dist/__tests__/register.test.js +90 -0
- package/dist/__tests__/register.test.js.map +1 -0
- package/dist/config.d.ts +7 -1
- package/dist/config.js +3 -1
- package/dist/config.js.map +1 -1
- package/dist/evals/__tests__/offline.test.js +418 -0
- package/dist/evals/__tests__/offline.test.js.map +1 -0
- package/dist/evals/index.d.ts +2 -5
- package/dist/evals/index.js +12 -26
- package/dist/evals/index.js.map +1 -1
- package/dist/evals/offline.d.ts +6 -0
- package/dist/evals/offline.js +347 -0
- package/dist/evals/offline.js.map +1 -0
- package/dist/evals/types.d.ts +69 -18
- package/dist/evals/types.js +20 -0
- package/dist/evals/types.js.map +1 -1
- package/dist/guard/__tests__/integration.test.js +178 -0
- package/dist/guard/__tests__/integration.test.js.map +1 -0
- package/dist/guard/__tests__/pii.test.js +109 -0
- package/dist/guard/__tests__/pii.test.js.map +1 -0
- package/dist/guard/__tests__/pipeline.test.js +162 -0
- package/dist/guard/__tests__/pipeline.test.js.map +1 -0
- package/dist/guard/__tests__/prompt-injection.test.js +92 -0
- package/dist/guard/__tests__/prompt-injection.test.js.map +1 -0
- package/dist/guard/base.d.ts +75 -14
- package/dist/guard/base.js +130 -53
- package/dist/guard/base.js.map +1 -1
- package/dist/guard/custom.d.ts +19 -0
- package/dist/guard/custom.js +46 -0
- package/dist/guard/custom.js.map +1 -0
- package/dist/guard/index.d.ts +24 -5
- package/dist/guard/index.js +39 -21
- package/dist/guard/index.js.map +1 -1
- package/dist/guard/integration.d.ts +32 -0
- package/dist/guard/integration.js +328 -0
- package/dist/guard/integration.js.map +1 -0
- package/dist/guard/moderation.d.ts +18 -0
- package/dist/guard/moderation.js +61 -0
- package/dist/guard/moderation.js.map +1 -0
- package/dist/guard/pii.d.ts +17 -0
- package/dist/guard/pii.js +94 -0
- package/dist/guard/pii.js.map +1 -0
- package/dist/guard/pipeline.d.ts +18 -0
- package/dist/guard/pipeline.js +96 -0
- package/dist/guard/pipeline.js.map +1 -0
- package/dist/guard/prompt-injection.d.ts +20 -6
- package/dist/guard/prompt-injection.js +59 -68
- package/dist/guard/prompt-injection.js.map +1 -1
- package/dist/guard/schema.d.ts +20 -0
- package/dist/guard/schema.js +99 -0
- package/dist/guard/schema.js.map +1 -0
- package/dist/guard/sensitive-topic.d.ts +22 -6
- package/dist/guard/sensitive-topic.js +78 -64
- package/dist/guard/sensitive-topic.js.map +1 -1
- package/dist/guard/topic-restriction.d.ts +21 -6
- package/dist/guard/topic-restriction.js +43 -64
- package/dist/guard/topic-restriction.js.map +1 -1
- package/dist/helpers.d.ts +133 -2
- package/dist/helpers.js +415 -5
- package/dist/helpers.js.map +1 -1
- package/dist/index.d.ts +54 -30
- package/dist/index.js +104 -22
- package/dist/index.js.map +1 -1
- package/dist/instrumentation/__tests__/ai21-rag-trace-comparison.test.d.ts +13 -0
- package/dist/instrumentation/__tests__/ai21-rag-trace-comparison.test.js +176 -0
- package/dist/instrumentation/__tests__/ai21-rag-trace-comparison.test.js.map +1 -0
- package/dist/instrumentation/__tests__/ai21-trace-comparison.test.d.ts +10 -0
- package/dist/instrumentation/__tests__/ai21-trace-comparison.test.js +203 -0
- package/dist/instrumentation/__tests__/ai21-trace-comparison.test.js.map +1 -0
- package/dist/instrumentation/__tests__/assemblyai-wrapper.test.js +262 -0
- package/dist/instrumentation/__tests__/assemblyai-wrapper.test.js.map +1 -0
- package/dist/instrumentation/__tests__/astra-trace-comparison.test.d.ts +7 -0
- package/dist/instrumentation/__tests__/astra-trace-comparison.test.js +160 -0
- package/dist/instrumentation/__tests__/astra-trace-comparison.test.js.map +1 -0
- package/dist/instrumentation/__tests__/browser-use-wrapper.test.js +284 -0
- package/dist/instrumentation/__tests__/browser-use-wrapper.test.js.map +1 -0
- package/dist/instrumentation/__tests__/cache-pricing.test.d.ts +12 -0
- package/dist/instrumentation/__tests__/cache-pricing.test.js +95 -0
- package/dist/instrumentation/__tests__/cache-pricing.test.js.map +1 -0
- package/dist/instrumentation/__tests__/elevenlabs-wrapper.test.d.ts +1 -0
- package/dist/instrumentation/__tests__/elevenlabs-wrapper.test.js +282 -0
- package/dist/instrumentation/__tests__/elevenlabs-wrapper.test.js.map +1 -0
- package/dist/instrumentation/__tests__/firecrawl-wrapper.test.d.ts +1 -0
- package/dist/instrumentation/__tests__/firecrawl-wrapper.test.js +238 -0
- package/dist/instrumentation/__tests__/firecrawl-wrapper.test.js.map +1 -0
- package/dist/instrumentation/__tests__/gpu-collector.test.d.ts +1 -0
- package/dist/instrumentation/__tests__/gpu-collector.test.js +247 -0
- package/dist/instrumentation/__tests__/gpu-collector.test.js.map +1 -0
- package/dist/instrumentation/__tests__/gradient-trace-comparison.test.d.ts +11 -0
- package/dist/instrumentation/__tests__/gradient-trace-comparison.test.js +391 -0
- package/dist/instrumentation/__tests__/gradient-trace-comparison.test.js.map +1 -0
- package/dist/instrumentation/__tests__/groq-trace-comparison.test.js +150 -0
- package/dist/instrumentation/__tests__/groq-trace-comparison.test.js.map +1 -1
- package/dist/instrumentation/__tests__/helpers-genai.test.d.ts +14 -0
- package/dist/instrumentation/__tests__/helpers-genai.test.js +347 -0
- package/dist/instrumentation/__tests__/helpers-genai.test.js.map +1 -0
- package/dist/instrumentation/__tests__/langchain-role-map.test.d.ts +1 -0
- package/dist/instrumentation/__tests__/langchain-role-map.test.js +19 -0
- package/dist/instrumentation/__tests__/langchain-role-map.test.js.map +1 -0
- package/dist/instrumentation/__tests__/mem0-trace-comparison.test.d.ts +13 -0
- package/dist/instrumentation/__tests__/mem0-trace-comparison.test.js +208 -0
- package/dist/instrumentation/__tests__/mem0-trace-comparison.test.js.map +1 -0
- package/dist/instrumentation/__tests__/pg-trace-comparison.test.d.ts +21 -0
- package/dist/instrumentation/__tests__/pg-trace-comparison.test.js +455 -0
- package/dist/instrumentation/__tests__/pg-trace-comparison.test.js.map +1 -0
- package/dist/instrumentation/__tests__/transformers-trace-comparison.test.d.ts +11 -0
- package/dist/instrumentation/__tests__/transformers-trace-comparison.test.js +214 -0
- package/dist/instrumentation/__tests__/transformers-trace-comparison.test.js.map +1 -0
- package/dist/instrumentation/__tests__/vertexai-trace-comparison.test.d.ts +6 -0
- package/dist/instrumentation/__tests__/vertexai-trace-comparison.test.js +409 -0
- package/dist/instrumentation/__tests__/vertexai-trace-comparison.test.js.map +1 -0
- package/dist/instrumentation/__tests__/vllm-trace-comparison.test.d.ts +1 -0
- package/dist/instrumentation/__tests__/vllm-trace-comparison.test.js +519 -0
- package/dist/instrumentation/__tests__/vllm-trace-comparison.test.js.map +1 -0
- package/dist/instrumentation/ai21/index.d.ts +11 -0
- package/dist/instrumentation/ai21/index.js +67 -0
- package/dist/instrumentation/ai21/index.js.map +1 -0
- package/dist/instrumentation/ai21/wrapper.d.ts +54 -0
- package/dist/instrumentation/ai21/wrapper.js +511 -0
- package/dist/instrumentation/ai21/wrapper.js.map +1 -0
- package/dist/instrumentation/anthropic/wrapper.js +56 -2
- package/dist/instrumentation/anthropic/wrapper.js.map +1 -1
- package/dist/instrumentation/assemblyai/index.d.ts +12 -0
- package/dist/instrumentation/assemblyai/index.js +91 -0
- package/dist/instrumentation/assemblyai/index.js.map +1 -0
- package/dist/instrumentation/assemblyai/wrapper.d.ts +35 -0
- package/dist/instrumentation/assemblyai/wrapper.js +233 -0
- package/dist/instrumentation/assemblyai/wrapper.js.map +1 -0
- package/dist/instrumentation/astra/index.d.ts +11 -0
- package/dist/instrumentation/astra/index.js +87 -0
- package/dist/instrumentation/astra/index.js.map +1 -0
- package/dist/instrumentation/astra/wrapper.d.ts +39 -0
- package/dist/instrumentation/astra/wrapper.js +349 -0
- package/dist/instrumentation/astra/wrapper.js.map +1 -0
- package/dist/instrumentation/azure-ai-inference/wrapper.d.ts +13 -0
- package/dist/instrumentation/azure-ai-inference/wrapper.js +60 -1
- package/dist/instrumentation/azure-ai-inference/wrapper.js.map +1 -1
- package/dist/instrumentation/bedrock/wrapper.d.ts +13 -0
- package/dist/instrumentation/bedrock/wrapper.js +78 -2
- package/dist/instrumentation/bedrock/wrapper.js.map +1 -1
- package/dist/instrumentation/browser-use/index.d.ts +11 -0
- package/dist/instrumentation/browser-use/index.js +98 -0
- package/dist/instrumentation/browser-use/index.js.map +1 -0
- package/dist/instrumentation/browser-use/wrapper.d.ts +12 -0
- package/dist/instrumentation/browser-use/wrapper.js +706 -0
- package/dist/instrumentation/browser-use/wrapper.js.map +1 -0
- package/dist/instrumentation/claude-agent-sdk/wrapper.js +90 -4
- package/dist/instrumentation/claude-agent-sdk/wrapper.js.map +1 -1
- package/dist/instrumentation/cohere/wrapper.d.ts +13 -0
- package/dist/instrumentation/cohere/wrapper.js +59 -0
- package/dist/instrumentation/cohere/wrapper.js.map +1 -1
- package/dist/instrumentation/cursor-sdk/wrapper.js +111 -15
- package/dist/instrumentation/cursor-sdk/wrapper.js.map +1 -1
- package/dist/instrumentation/elevenlabs/index.d.ts +12 -0
- package/dist/instrumentation/elevenlabs/index.js +92 -0
- package/dist/instrumentation/elevenlabs/index.js.map +1 -0
- package/dist/instrumentation/elevenlabs/wrapper.d.ts +34 -0
- package/dist/instrumentation/elevenlabs/wrapper.js +280 -0
- package/dist/instrumentation/elevenlabs/wrapper.js.map +1 -0
- package/dist/instrumentation/firecrawl/index.d.ts +13 -0
- package/dist/instrumentation/firecrawl/index.js +115 -0
- package/dist/instrumentation/firecrawl/index.js.map +1 -0
- package/dist/instrumentation/firecrawl/wrapper.d.ts +22 -0
- package/dist/instrumentation/firecrawl/wrapper.js +378 -0
- package/dist/instrumentation/firecrawl/wrapper.js.map +1 -0
- package/dist/instrumentation/google-ai/wrapper.d.ts +13 -0
- package/dist/instrumentation/google-ai/wrapper.js +85 -1
- package/dist/instrumentation/google-ai/wrapper.js.map +1 -1
- package/dist/instrumentation/gpu/index.d.ts +38 -0
- package/dist/instrumentation/gpu/index.js +233 -0
- package/dist/instrumentation/gpu/index.js.map +1 -0
- package/dist/instrumentation/gradient/index.d.ts +12 -0
- package/dist/instrumentation/gradient/index.js +74 -0
- package/dist/instrumentation/gradient/index.js.map +1 -0
- package/dist/instrumentation/gradient/utils.d.ts +14 -0
- package/dist/instrumentation/gradient/utils.js +119 -0
- package/dist/instrumentation/gradient/utils.js.map +1 -0
- package/dist/instrumentation/gradient/wrapper.d.ts +76 -0
- package/dist/instrumentation/gradient/wrapper.js +584 -0
- package/dist/instrumentation/gradient/wrapper.js.map +1 -0
- package/dist/instrumentation/groq/wrapper.js +42 -0
- package/dist/instrumentation/groq/wrapper.js.map +1 -1
- package/dist/instrumentation/huggingface/wrapper.d.ts +13 -0
- package/dist/instrumentation/huggingface/wrapper.js +66 -0
- package/dist/instrumentation/huggingface/wrapper.js.map +1 -1
- package/dist/instrumentation/index.js +26 -0
- package/dist/instrumentation/index.js.map +1 -1
- package/dist/instrumentation/langchain/wrapper.js +5 -8
- package/dist/instrumentation/langchain/wrapper.js.map +1 -1
- package/dist/instrumentation/langgraph/wrapper.js +12 -11
- package/dist/instrumentation/langgraph/wrapper.js.map +1 -1
- package/dist/instrumentation/llamaindex/wrapper.d.ts +14 -1
- package/dist/instrumentation/llamaindex/wrapper.js +68 -0
- package/dist/instrumentation/llamaindex/wrapper.js.map +1 -1
- package/dist/instrumentation/mcp/index.d.ts +20 -0
- package/dist/instrumentation/mcp/index.js +282 -0
- package/dist/instrumentation/mcp/index.js.map +1 -0
- package/dist/instrumentation/mcp/wrapper.d.ts +72 -0
- package/dist/instrumentation/mcp/wrapper.js +571 -0
- package/dist/instrumentation/mcp/wrapper.js.map +1 -0
- package/dist/instrumentation/mem0/index.d.ts +17 -0
- package/dist/instrumentation/mem0/index.js +113 -0
- package/dist/instrumentation/mem0/index.js.map +1 -0
- package/dist/instrumentation/mem0/wrapper.d.ts +37 -0
- package/dist/instrumentation/mem0/wrapper.js +298 -0
- package/dist/instrumentation/mem0/wrapper.js.map +1 -0
- package/dist/instrumentation/mistral/wrapper.d.ts +13 -0
- package/dist/instrumentation/mistral/wrapper.js +60 -1
- package/dist/instrumentation/mistral/wrapper.js.map +1 -1
- package/dist/instrumentation/ollama/wrapper.d.ts +13 -0
- package/dist/instrumentation/ollama/wrapper.js +67 -1
- package/dist/instrumentation/ollama/wrapper.js.map +1 -1
- package/dist/instrumentation/openai/wrapper.d.ts +13 -0
- package/dist/instrumentation/openai/wrapper.js +89 -3
- package/dist/instrumentation/openai/wrapper.js.map +1 -1
- package/dist/instrumentation/pg/index.d.ts +20 -0
- package/dist/instrumentation/pg/index.js +74 -0
- package/dist/instrumentation/pg/index.js.map +1 -0
- package/dist/instrumentation/pg/wrapper.d.ts +66 -0
- package/dist/instrumentation/pg/wrapper.js +441 -0
- package/dist/instrumentation/pg/wrapper.js.map +1 -0
- package/dist/instrumentation/replicate/wrapper.d.ts +13 -0
- package/dist/instrumentation/replicate/wrapper.js +61 -0
- package/dist/instrumentation/replicate/wrapper.js.map +1 -1
- package/dist/instrumentation/strands/processor.d.ts +6 -0
- package/dist/instrumentation/strands/processor.js +73 -3
- package/dist/instrumentation/strands/processor.js.map +1 -1
- package/dist/instrumentation/strands/utils.d.ts +2 -2
- package/dist/instrumentation/strands/utils.js +15 -2
- package/dist/instrumentation/strands/utils.js.map +1 -1
- package/dist/instrumentation/together/wrapper.d.ts +13 -0
- package/dist/instrumentation/together/wrapper.js +59 -1
- package/dist/instrumentation/together/wrapper.js.map +1 -1
- package/dist/instrumentation/transformers/index.d.ts +35 -0
- package/dist/instrumentation/transformers/index.js +119 -0
- package/dist/instrumentation/transformers/index.js.map +1 -0
- package/dist/instrumentation/transformers/wrapper.d.ts +49 -0
- package/dist/instrumentation/transformers/wrapper.js +472 -0
- package/dist/instrumentation/transformers/wrapper.js.map +1 -0
- package/dist/instrumentation/vercel-ai/wrapper.d.ts +13 -0
- package/dist/instrumentation/vercel-ai/wrapper.js +106 -1
- package/dist/instrumentation/vercel-ai/wrapper.js.map +1 -1
- package/dist/instrumentation/vertexai/index.d.ts +11 -0
- package/dist/instrumentation/vertexai/index.js +70 -0
- package/dist/instrumentation/vertexai/index.js.map +1 -0
- package/dist/instrumentation/vertexai/wrapper.d.ts +67 -0
- package/dist/instrumentation/vertexai/wrapper.js +529 -0
- package/dist/instrumentation/vertexai/wrapper.js.map +1 -0
- package/dist/instrumentation/vllm/index.d.ts +13 -0
- package/dist/instrumentation/vllm/index.js +67 -0
- package/dist/instrumentation/vllm/index.js.map +1 -0
- package/dist/instrumentation/vllm/wrapper.d.ts +60 -0
- package/dist/instrumentation/vllm/wrapper.js +482 -0
- package/dist/instrumentation/vllm/wrapper.js.map +1 -0
- package/dist/llm/index.d.ts +9 -2
- package/dist/llm/index.js.map +1 -1
- package/dist/manual-trace.d.ts +40 -0
- package/dist/manual-trace.js +103 -0
- package/dist/manual-trace.js.map +1 -0
- package/dist/otel/metrics.d.ts +11 -0
- package/dist/otel/metrics.js +48 -0
- package/dist/otel/metrics.js.map +1 -1
- package/dist/register.d.ts +4 -0
- package/dist/register.js +52 -0
- package/dist/register.js.map +1 -0
- package/dist/score/__tests__/score.test.d.ts +1 -0
- package/dist/score/__tests__/score.test.js +143 -0
- package/dist/score/__tests__/score.test.js.map +1 -0
- package/dist/score/index.d.ts +2 -0
- package/dist/score/index.js +6 -0
- package/dist/score/index.js.map +1 -0
- package/dist/score/score.d.ts +15 -0
- package/dist/score/score.js +135 -0
- package/dist/score/score.js.map +1 -0
- package/dist/semantic-convention.d.ts +273 -0
- package/dist/semantic-convention.js +317 -0
- package/dist/semantic-convention.js.map +1 -1
- package/dist/types.d.ts +15 -1
- package/package.json +18 -1
- package/dist/evals/__tests__/anthropic.test.js +0 -9
- package/dist/evals/__tests__/anthropic.test.js.map +0 -1
- package/dist/evals/__tests__/base.test.js +0 -37
- package/dist/evals/__tests__/base.test.js.map +0 -1
- package/dist/evals/__tests__/core.test.js +0 -33
- package/dist/evals/__tests__/core.test.js.map +0 -1
- package/dist/evals/__tests__/metrics.test.js +0 -59
- package/dist/evals/__tests__/metrics.test.js.map +0 -1
- package/dist/evals/__tests__/openai.test.js +0 -9
- package/dist/evals/__tests__/openai.test.js.map +0 -1
- package/dist/evals/__tests__/providers.test.js +0 -14
- package/dist/evals/__tests__/providers.test.js.map +0 -1
- package/dist/evals/__tests__/utils.test.js +0 -46
- package/dist/evals/__tests__/utils.test.js.map +0 -1
- package/dist/evals/all.d.ts +0 -4
- package/dist/evals/all.js +0 -35
- package/dist/evals/all.js.map +0 -1
- package/dist/evals/base.d.ts +0 -15
- package/dist/evals/base.js +0 -51
- package/dist/evals/base.js.map +0 -1
- package/dist/evals/bias.d.ts +0 -4
- package/dist/evals/bias.js +0 -35
- package/dist/evals/bias.js.map +0 -1
- package/dist/evals/hallucination.d.ts +0 -4
- package/dist/evals/hallucination.js +0 -32
- package/dist/evals/hallucination.js.map +0 -1
- package/dist/evals/llm/anthropic.d.ts +0 -5
- package/dist/evals/llm/anthropic.js +0 -42
- package/dist/evals/llm/anthropic.js.map +0 -1
- package/dist/evals/llm/openai.d.ts +0 -6
- package/dist/evals/llm/openai.js +0 -28
- package/dist/evals/llm/openai.js.map +0 -1
- package/dist/evals/llm/providers.d.ts +0 -7
- package/dist/evals/llm/providers.js +0 -10
- package/dist/evals/llm/providers.js.map +0 -1
- package/dist/evals/metrics.d.ts +0 -9
- package/dist/evals/metrics.js +0 -38
- package/dist/evals/metrics.js.map +0 -1
- package/dist/evals/toxicity.d.ts +0 -4
- package/dist/evals/toxicity.js +0 -33
- package/dist/evals/toxicity.js.map +0 -1
- package/dist/evals/utils.d.ts +0 -4
- package/dist/evals/utils.js +0 -39
- package/dist/evals/utils.js.map +0 -1
- package/dist/guard/__tests__/gaurd.test.js +0 -136
- package/dist/guard/__tests__/gaurd.test.js.map +0 -1
- package/dist/guard/__tests__/utils.test.js +0 -64
- package/dist/guard/__tests__/utils.test.js.map +0 -1
- package/dist/guard/all.d.ts +0 -8
- package/dist/guard/all.js +0 -28
- package/dist/guard/all.js.map +0 -1
- package/dist/guard/types.d.ts +0 -26
- package/dist/guard/types.js +0 -4
- package/dist/guard/types.js.map +0 -1
- package/dist/guard/utils.d.ts +0 -13
- package/dist/guard/utils.js +0 -79
- package/dist/guard/utils.js.map +0 -1
- /package/dist/{evals/__tests__/anthropic.test.d.ts → __tests__/manual-trace.test.d.ts} +0 -0
- /package/dist/{evals/__tests__/metrics.test.d.ts → __tests__/register.test.d.ts} +0 -0
- /package/dist/evals/__tests__/{base.test.d.ts → offline.test.d.ts} +0 -0
- /package/dist/{evals/__tests__/core.test.d.ts → guard/__tests__/integration.test.d.ts} +0 -0
- /package/dist/{evals/__tests__/openai.test.d.ts → guard/__tests__/pii.test.d.ts} +0 -0
- /package/dist/{evals/__tests__/providers.test.d.ts → guard/__tests__/pipeline.test.d.ts} +0 -0
- /package/dist/{evals/__tests__/utils.test.d.ts → guard/__tests__/prompt-injection.test.d.ts} +0 -0
- /package/dist/{guard/__tests__/gaurd.test.d.ts → instrumentation/__tests__/assemblyai-wrapper.test.d.ts} +0 -0
- /package/dist/{guard/__tests__/utils.test.d.ts → instrumentation/__tests__/browser-use-wrapper.test.d.ts} +0 -0
|
@@ -0,0 +1,441 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
14
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
15
|
+
}) : function(o, v) {
|
|
16
|
+
o["default"] = v;
|
|
17
|
+
});
|
|
18
|
+
var __importStar = (this && this.__importStar) || (function () {
|
|
19
|
+
var ownKeys = function(o) {
|
|
20
|
+
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
21
|
+
var ar = [];
|
|
22
|
+
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
23
|
+
return ar;
|
|
24
|
+
};
|
|
25
|
+
return ownKeys(o);
|
|
26
|
+
};
|
|
27
|
+
return function (mod) {
|
|
28
|
+
if (mod && mod.__esModule) return mod;
|
|
29
|
+
var result = {};
|
|
30
|
+
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
31
|
+
__setModuleDefault(result, mod);
|
|
32
|
+
return result;
|
|
33
|
+
};
|
|
34
|
+
})();
|
|
35
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
36
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
37
|
+
};
|
|
38
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
39
|
+
const api_1 = require("@opentelemetry/api");
|
|
40
|
+
const core_1 = require("@opentelemetry/core");
|
|
41
|
+
const semantic_conventions_1 = require("@opentelemetry/semantic-conventions");
|
|
42
|
+
const config_1 = __importDefault(require("../../config"));
|
|
43
|
+
const constant_1 = require("../../constant");
|
|
44
|
+
const helpers_1 = __importStar(require("../../helpers"));
|
|
45
|
+
const metrics_1 = __importDefault(require("../../otel/metrics"));
|
|
46
|
+
const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
|
|
47
|
+
const base_wrapper_1 = __importDefault(require("../base-wrapper"));
|
|
48
|
+
/**
|
|
49
|
+
* PostgreSQL (`pg`) instrumentation wrapper.
|
|
50
|
+
*
|
|
51
|
+
* Mirrors the Python SDK psycopg instrumentation
|
|
52
|
+
* (sdk/python/src/openlit/instrumentation/psycopg) for the Node `pg` driver:
|
|
53
|
+
* - parses the SQL operation (SELECT/INSERT/...) and target table for span naming,
|
|
54
|
+
* - detects pgvector similarity operators (`<=>`, `<->`, `<#>`),
|
|
55
|
+
* - honors a `captureDbParameters` toggle to record query parameters
|
|
56
|
+
* (off by default — parameters can contain PII/secrets),
|
|
57
|
+
* - sets the same `db.*` semantic-convention attributes the Python SDK sets.
|
|
58
|
+
*
|
|
59
|
+
* The `pg` `query()` method has three call shapes which must all be handled:
|
|
60
|
+
* 1. callback: client.query(text, values?, (err, res) => {}) → no return value
|
|
61
|
+
* 2. promise: const res = await client.query(text, values?) → returns a thenable
|
|
62
|
+
* 3. submittable: client.query(new Query(...)) / cursors → returns the submittable
|
|
63
|
+
*/
|
|
64
|
+
// Maximum number of query parameters to capture (matches Python MAX_PARAMS_COUNT).
|
|
65
|
+
const MAX_PARAMS_COUNT = 50;
|
|
66
|
+
// Maximum captured parameter value length (matches Python MAX_PARAM_LENGTH).
|
|
67
|
+
const MAX_PARAM_LENGTH = 256;
|
|
68
|
+
// Map of leading SQL keyword → db.operation.name value (mirrors Python SQL_OPERATION_MAP).
|
|
69
|
+
const SQL_OPERATION_MAP = {
|
|
70
|
+
SELECT: semantic_convention_1.default.DB_OPERATION_SELECT,
|
|
71
|
+
INSERT: semantic_convention_1.default.DB_OPERATION_INSERT,
|
|
72
|
+
UPDATE: semantic_convention_1.default.DB_OPERATION_UPDATE,
|
|
73
|
+
DELETE: semantic_convention_1.default.DB_OPERATION_DELETE,
|
|
74
|
+
COPY: semantic_convention_1.default.DB_OPERATION_COPY,
|
|
75
|
+
CREATE: semantic_convention_1.default.DB_OPERATION_CREATE,
|
|
76
|
+
ALTER: semantic_convention_1.default.DB_OPERATION_ALTER,
|
|
77
|
+
DROP: semantic_convention_1.default.DB_OPERATION_DROP,
|
|
78
|
+
TRUNCATE: semantic_convention_1.default.DB_OPERATION_TRUNCATE,
|
|
79
|
+
COMMIT: semantic_convention_1.default.DB_OPERATION_COMMIT,
|
|
80
|
+
ROLLBACK: semantic_convention_1.default.DB_OPERATION_ROLLBACK,
|
|
81
|
+
};
|
|
82
|
+
// Table-name extraction patterns (mirrors Python TABLE_PATTERNS).
|
|
83
|
+
const TABLE_PATTERNS = [
|
|
84
|
+
/INSERT\s+INTO\s+["']?(\w+)["']?/i,
|
|
85
|
+
/UPDATE\s+["']?(\w+)["']?/i,
|
|
86
|
+
/DELETE\s+FROM\s+["']?(\w+)["']?/i,
|
|
87
|
+
/FROM\s+["']?(\w+)["']?/i,
|
|
88
|
+
/TRUNCATE\s+(?:TABLE\s+)?["']?(\w+)["']?/i,
|
|
89
|
+
/CREATE\s+(?:TABLE|INDEX)\s+(?:IF\s+NOT\s+EXISTS\s+)?["']?(\w+)["']?/i,
|
|
90
|
+
/ALTER\s+TABLE\s+["']?(\w+)["']?/i,
|
|
91
|
+
/DROP\s+(?:TABLE|INDEX)\s+(?:IF\s+EXISTS\s+)?["']?(\w+)["']?/i,
|
|
92
|
+
/COPY\s+["']?(\w+)["']?/i,
|
|
93
|
+
];
|
|
94
|
+
function recordDbMetrics(dbOperation, connInfo, durationSeconds) {
|
|
95
|
+
if (config_1.default.disableMetrics) {
|
|
96
|
+
return;
|
|
97
|
+
}
|
|
98
|
+
const applicationName = config_1.default.applicationName || '';
|
|
99
|
+
const environment = config_1.default.environment || '';
|
|
100
|
+
const attributes = {
|
|
101
|
+
[semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME]: constant_1.SDK_NAME,
|
|
102
|
+
[semantic_conventions_1.ATTR_SERVICE_NAME]: applicationName,
|
|
103
|
+
[semantic_conventions_1.SEMRESATTRS_DEPLOYMENT_ENVIRONMENT]: environment,
|
|
104
|
+
[semantic_convention_1.default.DB_SYSTEM_NAME]: PgWrapper.dbSystem,
|
|
105
|
+
[semantic_convention_1.default.DB_OPERATION_NAME]: dbOperation,
|
|
106
|
+
[semantic_convention_1.default.SERVER_ADDRESS]: connInfo.host,
|
|
107
|
+
[semantic_convention_1.default.SERVER_PORT]: connInfo.port,
|
|
108
|
+
};
|
|
109
|
+
metrics_1.default.dbRequests?.add(1, attributes);
|
|
110
|
+
metrics_1.default.dbClientOperationDuration?.record(durationSeconds, attributes);
|
|
111
|
+
}
|
|
112
|
+
class PgWrapper extends base_wrapper_1.default {
|
|
113
|
+
/** Detect the SQL operation verb from a query string (mirrors Python parse_sql_operation). */
|
|
114
|
+
static parseSqlOperation(query) {
|
|
115
|
+
if (query === null || query === undefined) {
|
|
116
|
+
return semantic_convention_1.default.DB_OPERATION_QUERY;
|
|
117
|
+
}
|
|
118
|
+
const queryStr = String(query).trim();
|
|
119
|
+
if (!queryStr) {
|
|
120
|
+
return semantic_convention_1.default.DB_OPERATION_QUERY;
|
|
121
|
+
}
|
|
122
|
+
const queryUpper = queryStr.toUpperCase();
|
|
123
|
+
for (const [keyword, operation] of Object.entries(SQL_OPERATION_MAP)) {
|
|
124
|
+
if (queryUpper.startsWith(keyword)) {
|
|
125
|
+
return operation;
|
|
126
|
+
}
|
|
127
|
+
}
|
|
128
|
+
// Handle WITH ... SELECT/INSERT/UPDATE/DELETE (CTEs).
|
|
129
|
+
if (queryUpper.startsWith('WITH')) {
|
|
130
|
+
for (const [keyword, operation] of Object.entries(SQL_OPERATION_MAP)) {
|
|
131
|
+
if (queryUpper.includes(keyword)) {
|
|
132
|
+
return operation;
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
}
|
|
136
|
+
return semantic_convention_1.default.DB_OPERATION_QUERY;
|
|
137
|
+
}
|
|
138
|
+
/** Extract the target table name for span naming (mirrors Python extract_table_name). */
|
|
139
|
+
static extractTableName(query) {
|
|
140
|
+
if (query === null || query === undefined) {
|
|
141
|
+
return 'unknown';
|
|
142
|
+
}
|
|
143
|
+
const queryStr = String(query);
|
|
144
|
+
for (const pattern of TABLE_PATTERNS) {
|
|
145
|
+
const match = pattern.exec(queryStr);
|
|
146
|
+
if (match) {
|
|
147
|
+
return match[1];
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
return 'unknown';
|
|
151
|
+
}
|
|
152
|
+
/** Detect pgvector similarity operator → metric name (mirrors Python detect_special_features). */
|
|
153
|
+
static detectSimilarityMetric(query) {
|
|
154
|
+
if (query === null || query === undefined) {
|
|
155
|
+
return undefined;
|
|
156
|
+
}
|
|
157
|
+
const queryStr = String(query);
|
|
158
|
+
if (queryStr.includes('<=>'))
|
|
159
|
+
return 'cosine';
|
|
160
|
+
if (queryStr.includes('<->'))
|
|
161
|
+
return 'l2';
|
|
162
|
+
if (queryStr.includes('<#>'))
|
|
163
|
+
return 'inner_product';
|
|
164
|
+
return undefined;
|
|
165
|
+
}
|
|
166
|
+
/** Detect full-text search keywords (mirrors Python detect_special_features). */
|
|
167
|
+
static detectFullTextSearch(query) {
|
|
168
|
+
if (query === null || query === undefined) {
|
|
169
|
+
return false;
|
|
170
|
+
}
|
|
171
|
+
const queryUpper = String(query).toUpperCase();
|
|
172
|
+
return (queryUpper.includes('TSVECTOR') ||
|
|
173
|
+
queryUpper.includes('TSQUERY') ||
|
|
174
|
+
queryUpper.includes('TS_RANK') ||
|
|
175
|
+
queryUpper.includes('WEBSEARCH_TO_TSQUERY'));
|
|
176
|
+
}
|
|
177
|
+
/** Resolve batch size when values is an array of parameter arrays (executemany-style). */
|
|
178
|
+
static resolveBatchSize(values) {
|
|
179
|
+
if (!Array.isArray(values) || values.length === 0) {
|
|
180
|
+
return undefined;
|
|
181
|
+
}
|
|
182
|
+
if (Array.isArray(values[0])) {
|
|
183
|
+
return values.length;
|
|
184
|
+
}
|
|
185
|
+
return undefined;
|
|
186
|
+
}
|
|
187
|
+
/** Build the human-readable query summary (mirrors Python get_query_summary). */
|
|
188
|
+
static getQuerySummary(dbOperation, tableName, query) {
|
|
189
|
+
let summary = `${dbOperation} ${tableName}`;
|
|
190
|
+
const metric = PgWrapper.detectSimilarityMetric(query);
|
|
191
|
+
if (metric) {
|
|
192
|
+
summary += ` (vector ${metric})`;
|
|
193
|
+
}
|
|
194
|
+
if (PgWrapper.detectFullTextSearch(query)) {
|
|
195
|
+
summary += ' (full-text)';
|
|
196
|
+
}
|
|
197
|
+
return summary;
|
|
198
|
+
}
|
|
199
|
+
/** Sanitize a parameter value for safe inclusion in a span (mirrors Python sanitize_parameter). */
|
|
200
|
+
static sanitizeParameter(value) {
|
|
201
|
+
if (value === null || value === undefined) {
|
|
202
|
+
return 'NULL';
|
|
203
|
+
}
|
|
204
|
+
if (value instanceof Uint8Array || Buffer.isBuffer(value)) {
|
|
205
|
+
return `<bytes:${value.length}>`;
|
|
206
|
+
}
|
|
207
|
+
let strValue;
|
|
208
|
+
try {
|
|
209
|
+
strValue = typeof value === 'object' ? JSON.stringify(value) : String(value);
|
|
210
|
+
}
|
|
211
|
+
catch {
|
|
212
|
+
return '<unrepresentable>';
|
|
213
|
+
}
|
|
214
|
+
if (strValue.length > MAX_PARAM_LENGTH) {
|
|
215
|
+
return `${strValue.slice(0, MAX_PARAM_LENGTH)}...<truncated:${strValue.length}>`;
|
|
216
|
+
}
|
|
217
|
+
return strValue;
|
|
218
|
+
}
|
|
219
|
+
/**
|
|
220
|
+
* Extract { host, port, database } from a `pg` Client / Pool instance.
|
|
221
|
+
* `pg` stores connection params on the instance (and Pool on `options`).
|
|
222
|
+
* Defaults mirror the Python psycopg instrumentation (localhost:5432).
|
|
223
|
+
*/
|
|
224
|
+
static extractConnectionInfo(instance) {
|
|
225
|
+
const defaults = { host: 'localhost', port: 5432 };
|
|
226
|
+
try {
|
|
227
|
+
const params = instance?.connectionParameters || instance?.options || instance || {};
|
|
228
|
+
const host = params.host || defaults.host;
|
|
229
|
+
const port = params.port ? Number(params.port) : defaults.port;
|
|
230
|
+
const database = params.database;
|
|
231
|
+
return { host, port: Number.isFinite(port) ? port : defaults.port, database };
|
|
232
|
+
}
|
|
233
|
+
catch {
|
|
234
|
+
return defaults;
|
|
235
|
+
}
|
|
236
|
+
}
|
|
237
|
+
/**
|
|
238
|
+
* Resolve the SQL text and the parameter values from the `pg` query() arguments.
|
|
239
|
+
* Supports: (text, values?, cb?), ({ text, values, name }, cb?), and submittables.
|
|
240
|
+
*/
|
|
241
|
+
static resolveQuery(args) {
|
|
242
|
+
const first = args[0];
|
|
243
|
+
if (typeof first === 'string') {
|
|
244
|
+
return { queryText: first, values: Array.isArray(args[1]) ? args[1] : undefined };
|
|
245
|
+
}
|
|
246
|
+
if (first && typeof first === 'object') {
|
|
247
|
+
// Query config object or a Submittable (Query/Cursor) — both may carry `text`/`values`.
|
|
248
|
+
const queryText = 'text' in first ? first.text : undefined;
|
|
249
|
+
const values = 'values' in first ? first.values : undefined;
|
|
250
|
+
return { queryText, values };
|
|
251
|
+
}
|
|
252
|
+
return { queryText: undefined, values: undefined };
|
|
253
|
+
}
|
|
254
|
+
/** Pull the last argument if it is a node-style callback. */
|
|
255
|
+
static extractCallback(args) {
|
|
256
|
+
const last = args[args.length - 1];
|
|
257
|
+
return typeof last === 'function' ? last : undefined;
|
|
258
|
+
}
|
|
259
|
+
/** Stamp all span attributes (mirrors Python common_psycopg_logic). */
|
|
260
|
+
static setSpanAttributes(span, { dbOperation, tableName, queryText, values, connInfo, durationSeconds, rowCount, batchSize, captureDbParameters, sdkVersion, }) {
|
|
261
|
+
const applicationName = config_1.default.applicationName || '';
|
|
262
|
+
const environment = config_1.default.environment || '';
|
|
263
|
+
span.setAttribute(semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME, constant_1.SDK_NAME);
|
|
264
|
+
span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_VECTORDB);
|
|
265
|
+
span.setAttribute(semantic_convention_1.default.DB_SYSTEM_NAME, PgWrapper.dbSystem);
|
|
266
|
+
span.setAttribute(semantic_convention_1.default.DB_OPERATION_NAME, dbOperation);
|
|
267
|
+
span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, connInfo.host);
|
|
268
|
+
span.setAttribute(semantic_convention_1.default.SERVER_PORT, connInfo.port);
|
|
269
|
+
span.setAttribute(semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT, environment);
|
|
270
|
+
span.setAttribute(semantic_conventions_1.ATTR_SERVICE_NAME, applicationName);
|
|
271
|
+
span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, durationSeconds);
|
|
272
|
+
if (sdkVersion) {
|
|
273
|
+
span.setAttribute(semantic_convention_1.default.DB_SDK_VERSION, sdkVersion);
|
|
274
|
+
}
|
|
275
|
+
if (connInfo.database && connInfo.database !== 'unknown') {
|
|
276
|
+
span.setAttribute(semantic_convention_1.default.DB_NAMESPACE_POSTGRESQL, connInfo.database);
|
|
277
|
+
}
|
|
278
|
+
if (tableName && tableName !== 'unknown') {
|
|
279
|
+
span.setAttribute(semantic_convention_1.default.DB_COLLECTION_NAME, tableName);
|
|
280
|
+
}
|
|
281
|
+
if (typeof batchSize === 'number' && batchSize > 0) {
|
|
282
|
+
span.setAttribute(semantic_convention_1.default.DB_BATCH_SIZE, batchSize);
|
|
283
|
+
}
|
|
284
|
+
// Capture query text when message-content capture is enabled (parity w/ Python).
|
|
285
|
+
if (config_1.default.captureMessageContent && queryText) {
|
|
286
|
+
let queryStr = String(queryText);
|
|
287
|
+
const maxLen = config_1.default.maxContentLength;
|
|
288
|
+
if (typeof maxLen === 'number' && maxLen > 0 && queryStr.length > maxLen) {
|
|
289
|
+
queryStr = queryStr.slice(0, maxLen);
|
|
290
|
+
}
|
|
291
|
+
span.setAttribute(semantic_convention_1.default.DB_QUERY_TEXT, queryStr);
|
|
292
|
+
}
|
|
293
|
+
// Capture query parameters (OTel per-index convention) only when explicitly enabled.
|
|
294
|
+
const paramsForCapture = typeof batchSize === 'number' && batchSize > 0 && Array.isArray(values)
|
|
295
|
+
? values[0]
|
|
296
|
+
: values;
|
|
297
|
+
if (captureDbParameters && paramsForCapture !== null && paramsForCapture !== undefined) {
|
|
298
|
+
if (Array.isArray(paramsForCapture)) {
|
|
299
|
+
paramsForCapture.slice(0, MAX_PARAMS_COUNT).forEach((value, idx) => {
|
|
300
|
+
span.setAttribute(`${semantic_convention_1.default.DB_QUERY_PARAMETER}.${idx}`, PgWrapper.sanitizeParameter(value));
|
|
301
|
+
});
|
|
302
|
+
}
|
|
303
|
+
else if (typeof paramsForCapture === 'object') {
|
|
304
|
+
Object.entries(paramsForCapture)
|
|
305
|
+
.slice(0, MAX_PARAMS_COUNT)
|
|
306
|
+
.forEach(([key, value]) => {
|
|
307
|
+
span.setAttribute(`${semantic_convention_1.default.DB_QUERY_PARAMETER}.${key}`, PgWrapper.sanitizeParameter(value));
|
|
308
|
+
});
|
|
309
|
+
}
|
|
310
|
+
}
|
|
311
|
+
span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, PgWrapper.getQuerySummary(dbOperation, tableName, queryText));
|
|
312
|
+
if (typeof rowCount === 'number' && rowCount >= 0) {
|
|
313
|
+
span.setAttribute(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, rowCount);
|
|
314
|
+
}
|
|
315
|
+
// pgvector similarity metric.
|
|
316
|
+
const metric = PgWrapper.detectSimilarityMetric(queryText);
|
|
317
|
+
if (metric) {
|
|
318
|
+
span.setAttribute(semantic_convention_1.default.DB_SEARCH_SIMILARITY_METRIC, metric);
|
|
319
|
+
}
|
|
320
|
+
(0, helpers_1.applyCustomSpanAttributes)(span);
|
|
321
|
+
span.setStatus({ code: api_1.SpanStatusCode.OK });
|
|
322
|
+
recordDbMetrics(dbOperation, connInfo, durationSeconds);
|
|
323
|
+
}
|
|
324
|
+
/**
|
|
325
|
+
* Build the patched `query()` replacement. Handles callback, promise, and
|
|
326
|
+
* submittable forms while always producing a single CLIENT span.
|
|
327
|
+
*/
|
|
328
|
+
static _patchQuery(tracer, config = {}) {
|
|
329
|
+
const captureDbParameters = Boolean(config.captureDbParameters);
|
|
330
|
+
const sdkVersion = config.sdkVersion;
|
|
331
|
+
return (originalQuery) => {
|
|
332
|
+
return function (...args) {
|
|
333
|
+
if ((0, core_1.isTracingSuppressed)(api_1.context.active())) {
|
|
334
|
+
return originalQuery.apply(this, args);
|
|
335
|
+
}
|
|
336
|
+
const { queryText, values } = PgWrapper.resolveQuery(args);
|
|
337
|
+
const dbOperation = PgWrapper.parseSqlOperation(queryText);
|
|
338
|
+
const tableName = PgWrapper.extractTableName(queryText);
|
|
339
|
+
const batchSize = PgWrapper.resolveBatchSize(values);
|
|
340
|
+
const connInfo = PgWrapper.extractConnectionInfo(this);
|
|
341
|
+
const spanName = typeof batchSize === 'number' && batchSize > 0
|
|
342
|
+
? `${dbOperation} ${tableName} (batch)`
|
|
343
|
+
: `${dbOperation} ${tableName}`;
|
|
344
|
+
const span = tracer.startSpan(spanName, { kind: api_1.SpanKind.CLIENT });
|
|
345
|
+
const startTime = Date.now();
|
|
346
|
+
const finalize = (result) => {
|
|
347
|
+
try {
|
|
348
|
+
const durationSeconds = (Date.now() - startTime) / 1000;
|
|
349
|
+
const rowCount = result && typeof result.rowCount === 'number' ? result.rowCount : undefined;
|
|
350
|
+
PgWrapper.setSpanAttributes(span, {
|
|
351
|
+
dbOperation,
|
|
352
|
+
tableName,
|
|
353
|
+
queryText,
|
|
354
|
+
values,
|
|
355
|
+
connInfo,
|
|
356
|
+
durationSeconds,
|
|
357
|
+
rowCount,
|
|
358
|
+
batchSize,
|
|
359
|
+
captureDbParameters,
|
|
360
|
+
sdkVersion,
|
|
361
|
+
});
|
|
362
|
+
}
|
|
363
|
+
catch (attrErr) {
|
|
364
|
+
helpers_1.default.handleException(span, attrErr);
|
|
365
|
+
}
|
|
366
|
+
finally {
|
|
367
|
+
span.end();
|
|
368
|
+
}
|
|
369
|
+
};
|
|
370
|
+
const fail = (error) => {
|
|
371
|
+
helpers_1.default.handleException(span, error);
|
|
372
|
+
span.end();
|
|
373
|
+
};
|
|
374
|
+
return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), () => {
|
|
375
|
+
const callback = PgWrapper.extractCallback(args);
|
|
376
|
+
// ── Callback form ────────────────────────────────────────────────
|
|
377
|
+
if (callback) {
|
|
378
|
+
const wrappedCallback = function (err, res) {
|
|
379
|
+
if (err) {
|
|
380
|
+
fail(err);
|
|
381
|
+
}
|
|
382
|
+
else {
|
|
383
|
+
finalize(res);
|
|
384
|
+
}
|
|
385
|
+
return callback.apply(this, [err, res]);
|
|
386
|
+
};
|
|
387
|
+
const newArgs = args.slice(0, -1).concat(wrappedCallback);
|
|
388
|
+
try {
|
|
389
|
+
return originalQuery.apply(this, newArgs);
|
|
390
|
+
}
|
|
391
|
+
catch (syncErr) {
|
|
392
|
+
fail(syncErr);
|
|
393
|
+
throw syncErr;
|
|
394
|
+
}
|
|
395
|
+
}
|
|
396
|
+
// ── Promise / submittable form ───────────────────────────────────
|
|
397
|
+
let result;
|
|
398
|
+
try {
|
|
399
|
+
result = originalQuery.apply(this, args);
|
|
400
|
+
}
|
|
401
|
+
catch (syncErr) {
|
|
402
|
+
fail(syncErr);
|
|
403
|
+
throw syncErr;
|
|
404
|
+
}
|
|
405
|
+
if (result && typeof result.then === 'function') {
|
|
406
|
+
return result.then((res) => {
|
|
407
|
+
finalize(res);
|
|
408
|
+
return res;
|
|
409
|
+
}, (err) => {
|
|
410
|
+
fail(err);
|
|
411
|
+
throw err;
|
|
412
|
+
});
|
|
413
|
+
}
|
|
414
|
+
// Submittable (Query/Cursor) — no promise, no callback. Listen for completion.
|
|
415
|
+
if (result && typeof result.on === 'function') {
|
|
416
|
+
let settled = false;
|
|
417
|
+
result.on('end', () => {
|
|
418
|
+
if (settled)
|
|
419
|
+
return;
|
|
420
|
+
settled = true;
|
|
421
|
+
finalize(result);
|
|
422
|
+
});
|
|
423
|
+
result.on('error', (err) => {
|
|
424
|
+
if (settled)
|
|
425
|
+
return;
|
|
426
|
+
settled = true;
|
|
427
|
+
fail(err);
|
|
428
|
+
});
|
|
429
|
+
return result;
|
|
430
|
+
}
|
|
431
|
+
// Unknown shape — close the span without a row count.
|
|
432
|
+
finalize(result);
|
|
433
|
+
return result;
|
|
434
|
+
});
|
|
435
|
+
};
|
|
436
|
+
};
|
|
437
|
+
}
|
|
438
|
+
}
|
|
439
|
+
PgWrapper.dbSystem = semantic_convention_1.default.DB_SYSTEM_POSTGRESQL;
|
|
440
|
+
exports.default = PgWrapper;
|
|
441
|
+
//# sourceMappingURL=wrapper.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/pg/wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAQ4B;AAC5B,8CAA0D;AAC1D,8EAI6C;AAC7C,0DAAyC;AACzC,6CAA0C;AAC1C,yDAAyE;AACzE,iEAAyC;AACzC,oFAA2D;AAC3D,mEAA0C;AAE1C;;;;;;;;;;;;;;;GAeG;AAEH,mFAAmF;AACnF,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAC5B,6EAA6E;AAC7E,MAAM,gBAAgB,GAAG,GAAG,CAAC;AAE7B,2FAA2F;AAC3F,MAAM,iBAAiB,GAA2B;IAChD,MAAM,EAAE,6BAAkB,CAAC,mBAAmB;IAC9C,MAAM,EAAE,6BAAkB,CAAC,mBAAmB;IAC9C,MAAM,EAAE,6BAAkB,CAAC,mBAAmB;IAC9C,MAAM,EAAE,6BAAkB,CAAC,mBAAmB;IAC9C,IAAI,EAAE,6BAAkB,CAAC,iBAAiB;IAC1C,MAAM,EAAE,6BAAkB,CAAC,mBAAmB;IAC9C,KAAK,EAAE,6BAAkB,CAAC,kBAAkB;IAC5C,IAAI,EAAE,6BAAkB,CAAC,iBAAiB;IAC1C,QAAQ,EAAE,6BAAkB,CAAC,qBAAqB;IAClD,MAAM,EAAE,6BAAkB,CAAC,mBAAmB;IAC9C,QAAQ,EAAE,6BAAkB,CAAC,qBAAqB;CACnD,CAAC;AAEF,kEAAkE;AAClE,MAAM,cAAc,GAAa;IAC/B,kCAAkC;IAClC,2BAA2B;IAC3B,kCAAkC;IAClC,yBAAyB;IACzB,0CAA0C;IAC1C,sEAAsE;IACtE,kCAAkC;IAClC,8DAA8D;IAC9D,yBAAyB;CAC1B,CAAC;AAOF,SAAS,eAAe,CACtB,WAAmB,EACnB,QAAwC,EACxC,eAAuB;IAEvB,IAAI,gBAAa,CAAC,cAAc,EAAE,CAAC;QACjC,OAAO;IACT,CAAC;IAED,MAAM,eAAe,GAAG,gBAAa,CAAC,eAAe,IAAI,EAAE,CAAC;IAC5D,MAAM,WAAW,GAAG,gBAAa,CAAC,WAAW,IAAI,EAAE,CAAC;IAEpD,MAAM,UAAU,GAAe;QAC7B,CAAC,8CAAuB,CAAC,EAAE,mBAAQ;QACnC,CAAC,wCAAiB,CAAC,EAAE,eAAe;QACpC,CAAC,yDAAkC,CAAC,EAAE,WAAW;QACjD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,SAAS,CAAC,QAAQ;QACvD,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,EAAE,WAAW;QACnD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,QAAQ,CAAC,IAAI;QAClD,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,QAAQ,CAAC,IAAI;KAChD,CAAC;IAEF,iBAAO,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IACvC,iBAAO,CAAC,yBAAyB,EAAE,MAAM,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;AACzE,CAAC;AAED,MAAM,SAAU,SAAQ,sBAAW;IAGjC,8FAA8F;IAC9F,MAAM,CAAC,iBAAiB,CAAC,KAAc;QACrC,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,6BAAkB,CAAC,kBAAkB,CAAC;QAC/C,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,6BAAkB,CAAC,kBAAkB,CAAC;QAC/C,CAAC;QACD,MAAM,UAAU,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;QAC1C,KAAK,MAAM,CAAC,OAAO,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACrE,IAAI,UAAU,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;gBACnC,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;QACD,sDAAsD;QACtD,IAAI,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAClC,KAAK,MAAM,CAAC,OAAO,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBACrE,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;oBACjC,OAAO,SAAS,CAAC;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,6BAAkB,CAAC,kBAAkB,CAAC;IAC/C,CAAC;IAED,yFAAyF;IACzF,MAAM,CAAC,gBAAgB,CAAC,KAAc;QACpC,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACrC,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;YAClB,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,kGAAkG;IAClG,MAAM,CAAC,sBAAsB,CAAC,KAAc;QAC1C,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,QAAQ,CAAC;QAC9C,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAC1C,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,eAAe,CAAC;QACrD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,iFAAiF;IACjF,MAAM,CAAC,oBAAoB,CAAC,KAAc;QACxC,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/C,OAAO,CACL,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC;YAC/B,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC9B,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC9B,UAAU,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAC5C,CAAC;IACJ,CAAC;IAED,0FAA0F;IAC1F,MAAM,CAAC,gBAAgB,CAAC,MAAe;QACrC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7B,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,iFAAiF;IACjF,MAAM,CAAC,eAAe,CAAC,WAAmB,EAAE,SAAiB,EAAE,KAAc;QAC3E,IAAI,OAAO,GAAG,GAAG,WAAW,IAAI,SAAS,EAAE,CAAC;QAC5C,MAAM,MAAM,GAAG,SAAS,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;QACvD,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,IAAI,YAAY,MAAM,GAAG,CAAC;QACnC,CAAC;QACD,IAAI,SAAS,CAAC,oBAAoB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,IAAI,cAAc,CAAC;QAC5B,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,mGAAmG;IACnG,MAAM,CAAC,iBAAiB,CAAC,KAAc;QACrC,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,IAAI,KAAK,YAAY,UAAU,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1D,OAAO,UAAW,KAAoB,CAAC,MAAM,GAAG,CAAC;QACnD,CAAC;QACD,IAAI,QAAgB,CAAC;QACrB,IAAI,CAAC;YACH,QAAQ,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/E,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,mBAAmB,CAAC;QAC7B,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,GAAG,gBAAgB,EAAE,CAAC;YACvC,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC,iBAAiB,QAAQ,CAAC,MAAM,GAAG,CAAC;QACnF,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;OAIG;IACH,MAAM,CAAC,qBAAqB,CAAC,QAAa;QACxC,MAAM,QAAQ,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QACnD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,QAAQ,EAAE,oBAAoB,IAAI,QAAQ,EAAE,OAAO,IAAI,QAAQ,IAAI,EAAE,CAAC;YACrF,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI,CAAC;YAC1C,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;YAC/D,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;YACjC,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC;QAChF,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,QAAQ,CAAC;QAClB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,YAAY,CAAC,IAAW;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACtB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;QACpF,CAAC;QACD,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACvC,wFAAwF;YACxF,MAAM,SAAS,GAAG,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAC3D,MAAM,MAAM,GAAG,QAAQ,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;YAC5D,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;QAC/B,CAAC;QACD,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IACrD,CAAC;IAED,6DAA6D;IAC7D,MAAM,CAAC,eAAe,CAAC,IAAW;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACnC,OAAO,OAAO,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IACvD,CAAC;IAED,uEAAuE;IACvE,MAAM,CAAC,iBAAiB,CACtB,IAAU,EACV,EACE,WAAW,EACX,SAAS,EACT,SAAS,EACT,MAAM,EACN,QAAQ,EACR,eAAe,EACf,QAAQ,EACR,SAAS,EACT,mBAAmB,EACnB,UAAU,GAYX;QAED,MAAM,eAAe,GAAG,gBAAa,CAAC,eAAe,IAAI,EAAE,CAAC;QAC5D,MAAM,WAAW,GAAG,gBAAa,CAAC,WAAW,IAAI,EAAE,CAAC;QAEpD,IAAI,CAAC,YAAY,CAAC,8CAAuB,EAAE,mBAAQ,CAAC,CAAC;QACrD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,8BAA8B,CAAC,CAAC;QAC1G,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,SAAS,CAAC,QAAQ,CAAC,CAAC;QACzE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,WAAW,CAAC,CAAC;QACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QACpE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QACjE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,2BAA2B,EAAE,WAAW,CAAC,CAAC;QAC/E,IAAI,CAAC,YAAY,CAAC,wCAAiB,EAAE,eAAe,CAAC,CAAC;QACtD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,eAAe,CAAC,CAAC;QACpF,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,QAAQ,CAAC,QAAQ,IAAI,QAAQ,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YACzD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,uBAAuB,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACnF,CAAC;QACD,IAAI,SAAS,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YACzC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YACnD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;QACjE,CAAC;QAED,iFAAiF;QACjF,IAAI,gBAAa,CAAC,qBAAqB,IAAI,SAAS,EAAE,CAAC;YACrD,IAAI,QAAQ,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;YACjC,MAAM,MAAM,GAAG,gBAAa,CAAC,gBAAgB,CAAC;YAC9C,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,MAAM,EAAE,CAAC;gBACzE,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YACvC,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;QAChE,CAAC;QAED,qFAAqF;QACrF,MAAM,gBAAgB,GACpB,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,GAAG,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YACrE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YACX,CAAC,CAAC,MAAM,CAAC;QAEb,IAAI,mBAAmB,IAAI,gBAAgB,KAAK,IAAI,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACvF,IAAI,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBACpC,gBAAgB,CAAC,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;oBACjE,IAAI,CAAC,YAAY,CACf,GAAG,6BAAkB,CAAC,kBAAkB,IAAI,GAAG,EAAE,EACjD,SAAS,CAAC,iBAAiB,CAAC,KAAK,CAAC,CACnC,CAAC;gBACJ,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,OAAO,gBAAgB,KAAK,QAAQ,EAAE,CAAC;gBAChD,MAAM,CAAC,OAAO,CAAC,gBAA2C,CAAC;qBACxD,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC;qBAC1B,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;oBACxB,IAAI,CAAC,YAAY,CACf,GAAG,6BAAkB,CAAC,kBAAkB,IAAI,GAAG,EAAE,EACjD,SAAS,CAAC,iBAAiB,CAAC,KAAK,CAAC,CACnC,CAAC;gBACJ,CAAC,CAAC,CAAC;YACP,CAAC;QACH,CAAC;QAED,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,gBAAgB,EACnC,SAAS,CAAC,eAAe,CAAC,WAAW,EAAE,SAAS,EAAE,SAAS,CAAC,CAC7D,CAAC;QAEF,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,QAAQ,IAAI,CAAC,EAAE,CAAC;YAClD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,QAAQ,CAAC,CAAC;QAC5E,CAAC;QAED,8BAA8B;QAC9B,MAAM,MAAM,GAAG,SAAS,CAAC,sBAAsB,CAAC,SAAS,CAAC,CAAC;QAC3D,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,2BAA2B,EAAE,MAAM,CAAC,CAAC;QAC5E,CAAC;QAED,IAAA,mCAAyB,EAAC,IAAI,CAAC,CAAC;QAChC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5C,eAAe,CAAC,WAAW,EAAE,QAAQ,EAAE,eAAe,CAAC,CAAC;IAC1D,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,WAAW,CAAC,MAAc,EAAE,SAA0B,EAAE;QAC7D,MAAM,mBAAmB,GAAG,OAAO,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAChE,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;QAErC,OAAO,CAAC,aAAsC,EAAE,EAAE;YAChD,OAAO,UAAqB,GAAG,IAAW;gBACxC,IAAI,IAAA,0BAAmB,EAAC,aAAO,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;oBAC1C,OAAO,aAAa,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACzC,CAAC;gBAED,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;gBAC3D,MAAM,WAAW,GAAG,SAAS,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;gBAC3D,MAAM,SAAS,GAAG,SAAS,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;gBACxD,MAAM,SAAS,GAAG,SAAS,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;gBACrD,MAAM,QAAQ,GAAG,SAAS,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;gBACvD,MAAM,QAAQ,GACZ,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,GAAG,CAAC;oBAC5C,CAAC,CAAC,GAAG,WAAW,IAAI,SAAS,UAAU;oBACvC,CAAC,CAAC,GAAG,WAAW,IAAI,SAAS,EAAE,CAAC;gBACpC,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBACnE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAE7B,MAAM,QAAQ,GAAG,CAAC,MAAW,EAAE,EAAE;oBAC/B,IAAI,CAAC;wBACH,MAAM,eAAe,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBACxD,MAAM,QAAQ,GACZ,MAAM,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;wBAC9E,SAAS,CAAC,iBAAiB,CAAC,IAAI,EAAE;4BAChC,WAAW;4BACX,SAAS;4BACT,SAAS;4BACT,MAAM;4BACN,QAAQ;4BACR,eAAe;4BACf,QAAQ;4BACR,SAAS;4BACT,mBAAmB;4BACnB,UAAU;yBACX,CAAC,CAAC;oBACL,CAAC;oBAAC,OAAO,OAAY,EAAE,CAAC;wBACtB,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;oBAC/C,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,GAAG,EAAE,CAAC;oBACb,CAAC;gBACH,CAAC,CAAC;gBAEF,MAAM,IAAI,GAAG,CAAC,KAAU,EAAE,EAAE;oBAC1B,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;oBAC3C,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC,CAAC;gBAEF,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,GAAG,EAAE;oBAC9D,MAAM,QAAQ,GAAG,SAAS,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;oBAEjD,oEAAoE;oBACpE,IAAI,QAAQ,EAAE,CAAC;wBACb,MAAM,eAAe,GAAG,UAAqB,GAAQ,EAAE,GAAQ;4BAC7D,IAAI,GAAG,EAAE,CAAC;gCACR,IAAI,CAAC,GAAG,CAAC,CAAC;4BACZ,CAAC;iCAAM,CAAC;gCACN,QAAQ,CAAC,GAAG,CAAC,CAAC;4BAChB,CAAC;4BACD,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;wBAC1C,CAAC,CAAC;wBACF,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;wBAC1D,IAAI,CAAC;4BACH,OAAO,aAAa,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;wBAC5C,CAAC;wBAAC,OAAO,OAAY,EAAE,CAAC;4BACtB,IAAI,CAAC,OAAO,CAAC,CAAC;4BACd,MAAM,OAAO,CAAC;wBAChB,CAAC;oBACH,CAAC;oBAED,oEAAoE;oBACpE,IAAI,MAAW,CAAC;oBAChB,IAAI,CAAC;wBACH,MAAM,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;oBAC3C,CAAC;oBAAC,OAAO,OAAY,EAAE,CAAC;wBACtB,IAAI,CAAC,OAAO,CAAC,CAAC;wBACd,MAAM,OAAO,CAAC;oBAChB,CAAC;oBAED,IAAI,MAAM,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBAChD,OAAO,MAAM,CAAC,IAAI,CAChB,CAAC,GAAQ,EAAE,EAAE;4BACX,QAAQ,CAAC,GAAG,CAAC,CAAC;4BACd,OAAO,GAAG,CAAC;wBACb,CAAC,EACD,CAAC,GAAQ,EAAE,EAAE;4BACX,IAAI,CAAC,GAAG,CAAC,CAAC;4BACV,MAAM,GAAG,CAAC;wBACZ,CAAC,CACF,CAAC;oBACJ,CAAC;oBAED,+EAA+E;oBAC/E,IAAI,MAAM,IAAI,OAAO,MAAM,CAAC,EAAE,KAAK,UAAU,EAAE,CAAC;wBAC9C,IAAI,OAAO,GAAG,KAAK,CAAC;wBACpB,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;4BACpB,IAAI,OAAO;gCAAE,OAAO;4BACpB,OAAO,GAAG,IAAI,CAAC;4BACf,QAAQ,CAAC,MAAM,CAAC,CAAC;wBACnB,CAAC,CAAC,CAAC;wBACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAQ,EAAE,EAAE;4BAC9B,IAAI,OAAO;gCAAE,OAAO;4BACpB,OAAO,GAAG,IAAI,CAAC;4BACf,IAAI,CAAC,GAAG,CAAC,CAAC;wBACZ,CAAC,CAAC,CAAC;wBACH,OAAO,MAAM,CAAC;oBAChB,CAAC;oBAED,sDAAsD;oBACtD,QAAQ,CAAC,MAAM,CAAC,CAAC;oBACjB,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;;AAjYM,kBAAQ,GAAG,6BAAkB,CAAC,oBAAoB,CAAC;AAoY5D,kBAAe,SAAS,CAAC"}
|
|
@@ -4,6 +4,19 @@ declare class ReplicateWrapper extends BaseWrapper {
|
|
|
4
4
|
static aiSystem: string;
|
|
5
5
|
static serverAddress: string;
|
|
6
6
|
static serverPort: number;
|
|
7
|
+
/**
|
|
8
|
+
* Stamp `openlit.agent.version_hash` (auto) and `gen_ai.agent.version`
|
|
9
|
+
* (user override, if set) on the span and return the same attributes so
|
|
10
|
+
* the caller can merge them into the inference event extras.
|
|
11
|
+
*/
|
|
12
|
+
static _stampAgentVersion(span: Span, args: {
|
|
13
|
+
systemInstructionsJson?: string;
|
|
14
|
+
toolDefinitionsJson?: string;
|
|
15
|
+
primaryModel?: string;
|
|
16
|
+
temperature?: number | null;
|
|
17
|
+
top_p?: number | null;
|
|
18
|
+
max_tokens?: number | null;
|
|
19
|
+
}): Record<string, string>;
|
|
7
20
|
static _patchRun(tracer: Tracer): any;
|
|
8
21
|
static _run({ args, genAIEndpoint, response, span, }: {
|
|
9
22
|
args: any[];
|
|
@@ -51,6 +51,41 @@ function spanCreationAttrs(operationName, requestModel) {
|
|
|
51
51
|
};
|
|
52
52
|
}
|
|
53
53
|
class ReplicateWrapper extends base_wrapper_1.default {
|
|
54
|
+
/**
|
|
55
|
+
* Stamp `openlit.agent.version_hash` (auto) and `gen_ai.agent.version`
|
|
56
|
+
* (user override, if set) on the span and return the same attributes so
|
|
57
|
+
* the caller can merge them into the inference event extras.
|
|
58
|
+
*/
|
|
59
|
+
static _stampAgentVersion(span, args) {
|
|
60
|
+
const out = {};
|
|
61
|
+
try {
|
|
62
|
+
const versionHash = helpers_1.default.computeAgentVersionHash({
|
|
63
|
+
systemInstructions: args.systemInstructionsJson ?? null,
|
|
64
|
+
toolDefinitions: args.toolDefinitionsJson ?? null,
|
|
65
|
+
primaryModel: args.primaryModel ?? null,
|
|
66
|
+
runtimeConfig: {
|
|
67
|
+
temperature: args.temperature ?? null,
|
|
68
|
+
top_p: args.top_p ?? null,
|
|
69
|
+
max_tokens: args.max_tokens ?? null,
|
|
70
|
+
provider: semantic_convention_1.default.GEN_AI_SYSTEM_REPLICATE,
|
|
71
|
+
},
|
|
72
|
+
providers: [semantic_convention_1.default.GEN_AI_SYSTEM_REPLICATE],
|
|
73
|
+
});
|
|
74
|
+
if (versionHash) {
|
|
75
|
+
out[semantic_convention_1.default.OPENLIT_AGENT_VERSION_HASH] = versionHash;
|
|
76
|
+
span.setAttribute(semantic_convention_1.default.OPENLIT_AGENT_VERSION_HASH, versionHash);
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
catch {
|
|
80
|
+
// Hash computation must never fail the wrapped call.
|
|
81
|
+
}
|
|
82
|
+
const versionLabel = (0, helpers_1.getCurrentAgentVersion)();
|
|
83
|
+
if (versionLabel) {
|
|
84
|
+
out[semantic_convention_1.default.GEN_AI_AGENT_VERSION] = versionLabel;
|
|
85
|
+
span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_VERSION, versionLabel);
|
|
86
|
+
}
|
|
87
|
+
return out;
|
|
88
|
+
}
|
|
54
89
|
static _patchRun(tracer) {
|
|
55
90
|
const genAIEndpoint = 'replicate.run';
|
|
56
91
|
return (originalMethod) => {
|
|
@@ -127,13 +162,36 @@ class ReplicateWrapper extends base_wrapper_1.default {
|
|
|
127
162
|
span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['stop']);
|
|
128
163
|
let inputMessagesJson;
|
|
129
164
|
let outputMessagesJson;
|
|
165
|
+
// Replicate language models commonly accept a `system_prompt` input.
|
|
166
|
+
const systemPrompt = typeof input.system_prompt === 'string' ? input.system_prompt :
|
|
167
|
+
typeof input.system === 'string' ? input.system : '';
|
|
168
|
+
// Compute system_instructions JSON regardless of captureContent so the
|
|
169
|
+
// version hash stays consistent across runs even when content capture
|
|
170
|
+
// is disabled.
|
|
171
|
+
const systemInstructionsJson = systemPrompt
|
|
172
|
+
? JSON.stringify([{ type: 'text', content: systemPrompt }])
|
|
173
|
+
: undefined;
|
|
130
174
|
if (captureContent) {
|
|
131
175
|
const messages = prompt ? [{ role: 'user', content: prompt }] : [];
|
|
132
176
|
inputMessagesJson = helpers_1.default.buildInputMessages(messages);
|
|
133
177
|
span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, inputMessagesJson);
|
|
134
178
|
outputMessagesJson = helpers_1.default.buildOutputMessages(outputText, 'stop');
|
|
135
179
|
span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, outputMessagesJson);
|
|
180
|
+
if (systemInstructionsJson) {
|
|
181
|
+
span.setAttribute(semantic_convention_1.default.GEN_AI_SYSTEM_INSTRUCTIONS, systemInstructionsJson);
|
|
182
|
+
}
|
|
136
183
|
}
|
|
184
|
+
const versionExtras = ReplicateWrapper._stampAgentVersion(span, {
|
|
185
|
+
systemInstructionsJson,
|
|
186
|
+
primaryModel: requestModel,
|
|
187
|
+
temperature: typeof input.temperature === 'number' ? input.temperature : null,
|
|
188
|
+
top_p: typeof input.top_p === 'number' ? input.top_p : null,
|
|
189
|
+
max_tokens: typeof input.max_tokens === 'number'
|
|
190
|
+
? input.max_tokens
|
|
191
|
+
: typeof input.max_new_tokens === 'number'
|
|
192
|
+
? input.max_new_tokens
|
|
193
|
+
: null,
|
|
194
|
+
});
|
|
137
195
|
if (!config_1.default.disableEvents) {
|
|
138
196
|
const eventAttrs = {
|
|
139
197
|
[semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TEXT_COMPLETION,
|
|
@@ -145,12 +203,15 @@ class ReplicateWrapper extends base_wrapper_1.default {
|
|
|
145
203
|
[semantic_convention_1.default.GEN_AI_OUTPUT_TYPE]: outputType,
|
|
146
204
|
[semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: promptTokens,
|
|
147
205
|
[semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: completionTokens,
|
|
206
|
+
...versionExtras,
|
|
148
207
|
};
|
|
149
208
|
if (captureContent) {
|
|
150
209
|
if (inputMessagesJson)
|
|
151
210
|
eventAttrs[semantic_convention_1.default.GEN_AI_INPUT_MESSAGES] = inputMessagesJson;
|
|
152
211
|
if (outputMessagesJson)
|
|
153
212
|
eventAttrs[semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES] = outputMessagesJson;
|
|
213
|
+
if (systemInstructionsJson)
|
|
214
|
+
eventAttrs[semantic_convention_1.default.GEN_AI_SYSTEM_INSTRUCTIONS] = systemInstructionsJson;
|
|
154
215
|
}
|
|
155
216
|
helpers_1.default.emitInferenceEvent(span, eventAttrs);
|
|
156
217
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/replicate/wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAAwF;AACxF,0DAAyC;AACzC,
|
|
1
|
+
{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/replicate/wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAAwF;AACxF,0DAAyC;AACzC,yDAIuB;AACvB,oFAA2D;AAC3D,mEAAkE;AAElE,SAAS,iBAAiB,CACxB,aAAqB,EACrB,YAAoB;IAEpB,OAAO;QACL,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,aAAa;QACpD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,gBAAgB,CAAC,QAAQ;QACzE,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,YAAY;QACvD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,gBAAgB,CAAC,aAAa;QACnE,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,gBAAgB,CAAC,UAAU;KAC9D,CAAC;AACJ,CAAC;AAED,MAAM,gBAAiB,SAAQ,sBAAW;IAKxC;;;;OAIG;IACH,MAAM,CAAC,kBAAkB,CACvB,IAAU,EACV,IAOC;QAED,MAAM,GAAG,GAA2B,EAAE,CAAC;QACvC,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,iBAAa,CAAC,uBAAuB,CAAC;gBACxD,kBAAkB,EAAE,IAAI,CAAC,sBAAsB,IAAI,IAAI;gBACvD,eAAe,EAAE,IAAI,CAAC,mBAAmB,IAAI,IAAI;gBACjD,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,IAAI;gBACvC,aAAa,EAAE;oBACb,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI;oBACrC,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,IAAI;oBACzB,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI;oBACnC,QAAQ,EAAE,6BAAkB,CAAC,uBAAuB;iBACrD;gBACD,SAAS,EAAE,CAAC,6BAAkB,CAAC,uBAAuB,CAAC;aACxD,CAAC,CAAC;YACH,IAAI,WAAW,EAAE,CAAC;gBAChB,GAAG,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,GAAG,WAAW,CAAC;gBACjE,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,0BAA0B,EAC7C,WAAW,CACZ,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,qDAAqD;QACvD,CAAC;QACD,MAAM,YAAY,GAAG,IAAA,gCAAsB,GAAE,CAAC;QAC9C,IAAI,YAAY,EAAE,CAAC;YACjB,GAAG,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,GAAG,YAAY,CAAC;YAC5D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,YAAY,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,MAAM,CAAC,SAAS,CAAC,MAAc;QAC7B,MAAM,aAAa,GAAG,eAAe,CAAC;QACtC,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,IAAI,IAAA,8BAAoB,GAAE;oBAAE,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACpE,MAAM,UAAU,GAAG,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC9D,MAAM,YAAY,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,UAAU,CAAC;gBAC5D,MAAM,QAAQ,GAAG,GAAG,6BAAkB,CAAC,qCAAqC,IAAI,YAAY,EAAE,CAAC;gBAC/F,MAAM,YAAY,GAAG,IAAA,mCAAyB,GAAE,IAAI,aAAO,CAAC,MAAM,EAAE,CAAC;gBACrE,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE;oBACtC,IAAI,EAAE,cAAQ,CAAC,MAAM;oBACrB,UAAU,EAAE,iBAAiB,CAAC,6BAAkB,CAAC,qCAAqC,EAAE,YAAY,CAAC;iBACtG,EAAE,YAAY,CAAC,CAAC;gBACjB,OAAO,aAAO;qBACX,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,YAAY,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBAClD,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC1C,CAAC,CAAC;qBACD,IAAI,CAAC,CAAC,QAAa,EAAE,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;qBACvF,KAAK,CAAC,CAAC,CAAM,EAAE,EAAE;oBAChB,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBACvC,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE;wBAC9B,aAAa;wBACb,KAAK,EAAE,YAAY;wBACnB,QAAQ,EAAE,gBAAgB,CAAC,QAAQ;wBACnC,aAAa,EAAE,gBAAgB,CAAC,aAAa;wBAC7C,UAAU,EAAE,gBAAgB,CAAC,UAAU;wBACvC,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,IAAI,QAAQ;qBAC5C,CAAC,CAAC;oBACH,IAAI,CAAC,GAAG,EAAE,CAAC;oBACX,MAAM,CAAC,CAAC;gBACV,CAAC,CAAC,CAAC;YACP,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAChB,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,IAAI,GAML;QACC,IAAI,YAA4C,CAAC;QACjD,IAAI,CAAC;YACH,MAAM,cAAc,GAAG,gBAAa,CAAC,qBAAqB,CAAC;YAE3D,MAAM,UAAU,GAAG,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAC9B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;YAClC,MAAM,MAAM,GAAW,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC;YAC1C,MAAM,YAAY,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,UAAU,CAAC;YAE5D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;YAEtE,IAAI,UAAU,GAAG,EAAE,CAAC;YACpB,IAAI,UAAU,GAAG,6BAAkB,CAAC,uBAAuB,CAAC;YAE5D,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACjC,UAAU,GAAG,QAAQ,CAAC;YACxB,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACnC,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACjC,CAAC;iBAAM,IAAI,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACpD,UAAU,GAAG,6BAAkB,CAAC,uBAAuB,CAAC;gBACxD,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YACxC,CAAC;YAED,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;YAErE,MAAM,YAAY,GAAG,iBAAa,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC9D,MAAM,gBAAgB,GAAG,iBAAa,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAEtE,MAAM,WAAW,GAAG,gBAAa,CAAC,WAAW,IAAI,EAAE,CAAC;YACpD,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CAAC,YAAY,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,CAAC,CAAC;YAEvG,gBAAgB,CAAC,qBAAqB,CAAC,IAAI,EAAE;gBAC3C,aAAa;gBACb,KAAK,EAAE,YAAY;gBACnB,IAAI;gBACJ,QAAQ,EAAE,gBAAgB,CAAC,QAAQ;gBACnC,aAAa,EAAE,gBAAgB,CAAC,aAAa;gBAC7C,UAAU,EAAE,gBAAgB,CAAC,UAAU;aACxC,CAAC,CAAC;YAEH,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;YAC1E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,YAAY,CAAC,CAAC;YAC9E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,gBAAgB,CAAC,CAAC;YACnF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;YAE9E,IAAI,iBAAqC,CAAC;YAC1C,IAAI,kBAAsC,CAAC;YAC3C,qEAAqE;YACrE,MAAM,YAAY,GAChB,OAAO,KAAK,CAAC,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;gBAC/D,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;YACvD,uEAAuE;YACvE,sEAAsE;YACtE,eAAe;YACf,MAAM,sBAAsB,GAAuB,YAAY;gBAC7D,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;gBAC3D,CAAC,CAAC,SAAS,CAAC;YACd,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnE,iBAAiB,GAAG,iBAAa,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;gBAC/D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,iBAAiB,CAAC,CAAC;gBAC/E,kBAAkB,GAAG,iBAAa,CAAC,mBAAmB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;gBAC3E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,sBAAsB,EAAE,kBAAkB,CAAC,CAAC;gBACjF,IAAI,sBAAsB,EAAE,CAAC;oBAC3B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,sBAAsB,CAAC,CAAC;gBAC3F,CAAC;YACH,CAAC;YAED,MAAM,aAAa,GAAG,gBAAgB,CAAC,kBAAkB,CAAC,IAAI,EAAE;gBAC9D,sBAAsB;gBACtB,YAAY,EAAE,YAAY;gBAC1B,WAAW,EAAE,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI;gBAC7E,KAAK,EAAE,OAAO,KAAK,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI;gBAC3D,UAAU,EACR,OAAO,KAAK,CAAC,UAAU,KAAK,QAAQ;oBAClC,CAAC,CAAC,KAAK,CAAC,UAAU;oBAClB,CAAC,CAAC,OAAO,KAAK,CAAC,cAAc,KAAK,QAAQ;wBACxC,CAAC,CAAC,KAAK,CAAC,cAAc;wBACtB,CAAC,CAAC,IAAI;aACb,CAAC,CAAC;YAEH,IAAI,CAAC,gBAAa,CAAC,aAAa,EAAE,CAAC;gBACjC,MAAM,UAAU,GAAe;oBAC7B,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,qCAAqC;oBAC/F,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,YAAY;oBACvD,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,EAAE,YAAY;oBACxD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,gBAAgB,CAAC,aAAa;oBACnE,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,gBAAgB,CAAC,UAAU;oBAC7D,CAAC,6BAAkB,CAAC,6BAA6B,CAAC,EAAE,CAAC,MAAM,CAAC;oBAC5D,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,EAAE,UAAU;oBACnD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,YAAY;oBAC5D,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,gBAAgB;oBACjE,GAAG,aAAa;iBACjB,CAAC;gBACF,IAAI,cAAc,EAAE,CAAC;oBACnB,IAAI,iBAAiB;wBAAE,UAAU,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,GAAG,iBAAiB,CAAC;oBAChG,IAAI,kBAAkB;wBAAE,UAAU,CAAC,6BAAkB,CAAC,sBAAsB,CAAC,GAAG,kBAAkB,CAAC;oBACnG,IAAI,sBAAsB;wBAAE,UAAU,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,GAAG,sBAAsB,CAAC;gBACjH,CAAC;gBACD,iBAAa,CAAC,kBAAkB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;YACrD,CAAC;YAED,YAAY,GAAG;gBACb,aAAa;gBACb,KAAK,EAAE,YAAY;gBACnB,IAAI;gBACJ,QAAQ,EAAE,gBAAgB,CAAC,QAAQ;aACpC,CAAC;YACF,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YACvC,MAAM,CAAC,CAAC;QACV,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,YAAY,EAAE,CAAC;gBACjB,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;IACH,CAAC;;AAzNM,yBAAQ,GAAG,6BAAkB,CAAC,uBAAuB,CAAC;AACtD,8BAAa,GAAG,mBAAmB,CAAC;AACpC,2BAAU,GAAG,GAAG,CAAC;AA0N1B,kBAAe,gBAAgB,CAAC"}
|
|
@@ -42,4 +42,10 @@ export declare class StrandsSpanProcessor implements SpanProcessor {
|
|
|
42
42
|
private static _extractToolCallIdFromSpanEvents;
|
|
43
43
|
private _extractAndSetContent;
|
|
44
44
|
private _emitChatInferenceEvent;
|
|
45
|
+
/**
|
|
46
|
+
* Compute and write `openlit.agent.version_hash` (auto) and
|
|
47
|
+
* `gen_ai.agent.version` (user override) onto a Strands chat span.
|
|
48
|
+
* Returns the same attributes for inclusion in the inference event.
|
|
49
|
+
*/
|
|
50
|
+
private _stampChatAgentVersion;
|
|
45
51
|
}
|