Skip to content

Commit b25b38a

Browse files
Add extra headers
1 parent 0f269de commit b25b38a

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

packages/dd-trace/src/aiguard/sdk.js

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ const {
1515
} = require('./tags')
1616
const log = require('../log')
1717
const telemetryMetrics = require('../telemetry/metrics')
18+
const tracerVersion = require('../../../../package.json').version
1819

1920
const appsecMetrics = telemetryMetrics.manager.namespace('appsec')
2021

@@ -63,6 +64,9 @@ class AIGuard extends NoopAIGuard {
6364
this.#headers = {
6465
'DD-API-KEY': config.apiKey,
6566
'DD-APPLICATION-KEY': config.appKey,
67+
'DD-AI-GUARD-VERSION': tracerVersion,
68+
'DD-AI-GUARD-SOURCE': 'SDK',
69+
'DD-AI-GUARD-LANGUAGE': 'nodejs'
6670
}
6771
const endpoint = config.experimental.aiguard.endpoint || `https://app.${config.site}/api/v2/ai-guard`
6872
this.#evaluateUrl = `${endpoint}/evaluate`

packages/dd-trace/test/aiguard/index.spec.js

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ const sinon = require('sinon')
88
const agent = require('../plugins/agent')
99
const NoopAIGuard = require('../../src/aiguard/noop')
1010
const AIGuard = require('../../src/aiguard/sdk')
11+
const tracerVersion = require('../../../../package.json').version
1112
const telemetryMetrics = require('../../src/telemetry/metrics')
1213
const appsecNamespace = telemetryMetrics.manager.namespace('appsec')
1314

@@ -110,7 +111,10 @@ describe('AIGuard SDK', () => {
110111
'Content-Type': 'application/json',
111112
'Content-Length': Buffer.byteLength(postData),
112113
'DD-API-KEY': config.apiKey,
113-
'DD-APPLICATION-KEY': config.appKey
114+
'DD-APPLICATION-KEY': config.appKey,
115+
'DD-AI-GUARD-VERSION': tracerVersion,
116+
'DD-AI-GUARD-SOURCE': 'SDK',
117+
'DD-AI-GUARD-LANGUAGE': 'nodejs'
114118
},
115119
body: postData,
116120
signal: sinon.match.instanceOf(AbortSignal)

0 commit comments

Comments
 (0)