@forge/llm 0.2.0-next.2-experimental-f76634b → 0.2.0-next.3

package/out/__test__/error-handling.test.js

@@ -102,4 +102,24 @@ describe('Error Handling', () => {
             traceId
         }, { code, message, context, ...extraFields }));
     });
+    it('should include atl-traceid when x-trace-id and x-b3-traceid is not present', async () => {
+        const code = 'UNSUPPORTED_AI_MODEL';
+        const message = 'Requested claude-3-5-sonnet-v2@20241022 not available for vendor: Claude';
+        const traceId = 'atl-trace-456';
+        const apiResponse = new Response(JSON.stringify({
+            code: code,
+            message: message
+        }), {
+            status: 400,
+            statusText: 'Bad Request',
+            headers: {
+                'atl-traceid': traceId
+            }
+        });
+        await expect(async () => (0, error_handling_1.checkResponseError)(apiResponse)).rejects.toMatchError(new errors_1.ForgeLlmAPIError({
+            status: 400,
+            statusText: 'Bad Request',
+            traceId
+        }, { code, message }));
+    });
 });

package/out/__test__/llm-api.test.js

@@ -10,8 +10,34 @@ function setupEnvironment(response) {
     const mockedFetch = jest.fn().mockResolvedValue(response || new Response(null, { status: 200 }));
     global.__forge_fetch__ = jest.fn((_ctx, path, options) => mockedFetch(path, options));
 }
+jest.mock('@forge/api', () => {
+    const actual = jest.requireActual('@forge/api');
+    return {
+        ...actual,
+        __getRuntime: jest.fn(() => ({
+            tracing: { traceId: 'trace-test', spanId: 'span-test' }
+        }))
+    };
+});
 describe('llm api', () => {
-    it('should make ForgeLlm request', async () => {
+    it('should make ForgeLlm request with correct headers', async () => {
+        const llmResponse = (0, test_helpers_1.constructLlmResponse)();
+        const apiResponse = new Response(JSON.stringify(llmResponse), {
+            status: 200
+        });
+        const mockedFetch = jest.fn().mockResolvedValue(apiResponse);
+        global.__forge_fetch__ = jest.fn((_ctx, path, options) => mockedFetch(path, options));
+        const llmAPI = new llm_api_1.LlmApiImpl((0, fetch_wrapper_1.getFetchWrapper)());
+        const prompt = (0, test_helpers_1.constructPrompt)();
+        await llmAPI.chat(prompt);
+        const [[, options]] = mockedFetch.mock.calls;
+        expect(options.headers).toEqual({
+            'Content-Type': 'application/json',
+            'x-b3-traceid': 'trace-test',
+            'x-b3-spanid': 'span-test'
+        });
+    });
+    it('should make ForgeLlm request with correct body', async () => {
         const llmResponse = (0, test_helpers_1.constructLlmResponse)();
         const apiResponse = new Response(JSON.stringify(llmResponse), {
             status: 200
@@ -34,7 +60,6 @@ describe('llm api', () => {
         const [[path, options]] = mockedFetch.mock.calls;
         expect(path).toBe('https://llm/claude-sonnet-4-20250514');
         expect(options.method).toBe('POST');
-        expect(options.headers).toEqual({ 'Content-Type': 'application/json' });
         expect(JSON.parse(options.body)).toEqual({
             messages: [
                 {

package/out/fetch-wrapper.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"fetch-wrapper.d.ts","sourceRoot":"","sources":["../src/fetch-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,WAAW,EAAe,MAAM,YAAY,CAAC;AAEnE,wBAAgB,eAAe,IAAI,WAAW,CAW7C"}
+{"version":3,"file":"fetch-wrapper.d.ts","sourceRoot":"","sources":["../src/fetch-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,EAA6B,WAAW,EAAe,MAAM,YAAY,CAAC;AAEjF,wBAAgB,eAAe,IAAI,WAAW,CAc7C"}

package/out/fetch-wrapper.js

@@ -1,14 +1,18 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getFetchWrapper = void 0;
+const api_1 = require("@forge/api");
 function getFetchWrapper() {
+    const { tracing } = (0, api_1.__getRuntime)();
     return async function (path, options) {
         const model = path?.split('/').pop();
         return await global.__forge_fetch__({ type: 'llm', model: model }, path, {
             ...options,
             headers: {
                 ...options?.headers,
-                'Content-Type': 'application/json'
+                'Content-Type': 'application/json',
+                'x-b3-traceid': tracing.traceId,
+                'x-b3-spanid': tracing.spanId
             }
         });
     };

package/out/utils/error-handling.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"error-handling.d.ts","sourceRoot":"","sources":["../../src/utils/error-handling.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuC,UAAU,EAAoB,MAAM,WAAW,CAAC;AAE9F,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAMzC,wBAAgB,YAAY,CAAC,IAAI,EAAE,OAAO,GAAG,IAAI,IAAI,UAAU,CAE9D;AAED,wBAAsB,kBAAkB,CAAC,QAAQ,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAwB7E;AAED,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,GAAG,SAAS,CAMnE"}
+{"version":3,"file":"error-handling.d.ts","sourceRoot":"","sources":["../../src/utils/error-handling.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuC,UAAU,EAAoB,MAAM,WAAW,CAAC;AAE9F,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAQzC,wBAAgB,YAAY,CAAC,IAAI,EAAE,OAAO,GAAG,IAAI,IAAI,UAAU,CAE9D;AAED,wBAAsB,kBAAkB,CAAC,QAAQ,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAwB7E;AAED,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,GAAG,SAAS,CAMnE"}

package/out/utils/error-handling.js

@@ -4,7 +4,7 @@ exports.safeGetParsedBody = exports.checkResponseError = exports.isForgeError =
 const errors_1 = require("../errors");
 const text_1 = require("../text");
 function extractTraceId(response) {
-    return response.headers.get('x-b3-traceid') || response.headers.get('x-trace-id');
+    return (response.headers.get('x-b3-traceid') || response.headers.get('x-trace-id') || response.headers.get('atl-traceid'));
 }
 function isForgeError(body) {
     return typeof body === 'object' && body !== null && 'code' in body && 'message' in body;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@forge/llm",
-  "version": "0.2.0-next.2-experimental-f76634b",
+  "version": "0.2.0-next.3",
   "description": "Forge LLM SDK",
   "main": "out/index.js",
   "types": "out/index.d.ts",
@@ -10,7 +10,7 @@
   "author": "Atlassian",
   "license": "SEE LICENSE IN LICENSE.txt",
   "dependencies": {
-    "@forge/api": "^6.3.0-next.2-experimental-f76634b"
+    "@forge/api": "^6.3.0-next.2"
   },
   "devDependencies": {
     "@types/node": "20.19.1",