@forge/llm 0.3.2-next.1 → 0.4.0

package/CHANGELOG.md

@@ -1,5 +1,23 @@
 # @forge/llm
 
+## 0.4.0
+
+### Minor Changes
+
+- 7c05ef4: Adding support for new list method to retrieve the supported AI models
+
+### Patch Changes
+
+- Updated dependencies [69e97f1]
+- Updated dependencies [8f24ce1]
+  - @forge/api@7.0.0
+
+## 0.4.0-next.2
+
+### Minor Changes
+
+- 7c05ef4: Adding support for new list method to retrieve the supported AI models
+
 ## 0.3.2-next.1
 
 ### Patch Changes

package/README.md

@@ -5,11 +5,37 @@ Library for Forge LLM
 
 ## Usage Example
 
-### Supported Models
+The SDK supports a `chat` and a `stream` interface available to use for all supported models.
 
-For the list of supported models visit [LLM Models](https://go.atlassian.com/forge-llms-api-reference/#supported-models)
+### List - Supported Models
 
-The SDK supports a `chat` and a `stream` interface.
+For the list of supported models, visit [LLM Models](https://go.atlassian.com/forge-llms-api-reference/#supported-models)
+
+You can also dynamically get the list of available models via the `list` method:
+
+```typescript
+import { list } from '@forge/llm';
+
+const response = await list();
+```
+
+#### Response Structure
+The response is a `ModelListResponse` object:
+
+```json
+{
+  "models": [
+    {
+      "model": "claude-sonnet-4-20250514",
+      "status": "active"
+    },
+    {
+      "model": "claude-opus-4-20250514",
+      "status": "deprecated"
+    }
+  ]
+}
+```
 
 ### Chat
 

package/out/index.d.ts

@@ -1,5 +1,6 @@
 export * from './interfaces/types';
 declare const chat: (prompt: import("./interfaces/types").Prompt) => Promise<import("./interfaces/internal").LlmResponse>;
 declare const stream: (prompt: import("./interfaces/types").Prompt) => Promise<import("./streaming/stream-response-wrapper").StreamResponse>;
-export { chat, stream };
+declare const list: () => Promise<import("./interfaces/internal").ModelListResponse>;
+export { chat, stream, list };
 //# sourceMappingURL=index.d.ts.map

package/out/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,cAAc,oBAAoB,CAAC;AAInC,QAAA,MAAM,IAAI,uGAAoE,CAAC;AAE/E,QAAA,MAAM,MAAM,wHAAwE,CAAC;AAErF,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,cAAc,oBAAoB,CAAC;AAInC,QAAA,MAAM,IAAI,uGAAoE,CAAC;AAE/E,QAAA,MAAM,MAAM,wHAAwE,CAAC;AAErF,QAAA,MAAM,IAAI,kEAAsB,CAAC;AAEjC,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC"}

package/out/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.stream = exports.chat = void 0;
+exports.list = exports.stream = exports.chat = void 0;
 const tslib_1 = require("tslib");
 const llm_api_1 = require("./llm-api");
 tslib_1.__exportStar(require("./interfaces/types"), exports);
@@ -9,3 +9,5 @@ const chat = (...args) => llmApi.chat(...args);
 exports.chat = chat;
 const stream = (...args) => llmApi.stream(...args);
 exports.stream = stream;
+const list = () => llmApi.list();
+exports.list = list;

package/out/interfaces/internal.d.ts

@@ -2,6 +2,7 @@ import { ChatResponse, Prompt, StreamResponse } from './types';
 export interface LlmApi {
     chat(prompt: Prompt): Promise<ChatResponse>;
     stream(prompt: Prompt): Promise<StreamResponse>;
+    list(): Promise<ModelListResponse>;
 }
 export interface ClientArgs {
     model: string;
@@ -9,6 +10,7 @@ export interface ClientArgs {
 }
 export interface LlmClient {
     post(args: ClientArgs): Promise<Response>;
+    get(): Promise<Response>;
 }
 export interface LlmRequest {
     messages: Message[];
@@ -81,4 +83,11 @@ export interface TextPart {
     type: 'text';
     text: string;
 }
+export interface ModelListResponse {
+    models: ModelDetails[];
+}
+export interface ModelDetails {
+    model: string;
+    status: 'active' | 'deprecated';
+}
 //# sourceMappingURL=internal.d.ts.map

package/out/interfaces/internal.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"internal.d.ts","sourceRoot":"","sources":["../../src/interfaces/internal.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAE/D,MAAM,WAAW,MAAM;IACrB,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IAC5C,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;CACjD;AAED,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;CAC3C;AAED,MAAM,WAAW,UAAU;IACzB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IACf,WAAW,CAAC,EAAE,UAAU,CAAC;CAC1B;AAED,oBAAY,OAAO,GAAG,aAAa,GAAG,WAAW,GAAG,gBAAgB,GAAG,WAAW,CAAC;AAEnF,MAAM,WAAW,IAAI;IACnB,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACrC,CAAC;CACH;AAED,oBAAY,UAAU,GAClB,MAAM,GACN,MAAM,GACN,UAAU,GACV;IACE,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;CACH,CAAC;AAEN,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,QAAQ,CAAC;CAChB;AAED,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;CACd;AACD,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,WAAW,CAAC;IAClB,UAAU,CAAC,EAAE,QAAQ,EAAE,CAAC;CACzB;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,UAAU,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AAED,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,KAAK,CAAC,EAAE,KAAK,CAAC;CACf;AAED,MAAM,WAAW,MAAM;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,gBAAgB,CAAC;CAC3B;AAED,MAAM,WAAW,KAAK;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,oBAAY,OAAO,GAAG,MAAM,GAAG,WAAW,EAAE,CAAC;AAE7C,oBAAY,WAAW,GAAG,QAAQ,CAAC;AAEnC,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd"}
+{"version":3,"file":"internal.d.ts","sourceRoot":"","sources":["../../src/interfaces/internal.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAE/D,MAAM,WAAW,MAAM;IACrB,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IAC5C,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;IAChD,IAAI,IAAI,OAAO,CAAC,iBAAiB,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC1C,GAAG,IAAI,OAAO,CAAC,QAAQ,CAAC,CAAC;CAC1B;AAED,MAAM,WAAW,UAAU;IACzB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IACf,WAAW,CAAC,EAAE,UAAU,CAAC;CAC1B;AAED,oBAAY,OAAO,GAAG,aAAa,GAAG,WAAW,GAAG,gBAAgB,GAAG,WAAW,CAAC;AAEnF,MAAM,WAAW,IAAI;IACnB,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACrC,CAAC;CACH;AAED,oBAAY,UAAU,GAClB,MAAM,GACN,MAAM,GACN,UAAU,GACV;IACE,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;CACH,CAAC;AAEN,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,QAAQ,CAAC;CAChB;AAED,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;CACd;AACD,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,WAAW,CAAC;IAClB,UAAU,CAAC,EAAE,QAAQ,EAAE,CAAC;CACzB;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,UAAU,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AAED,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,KAAK,CAAC,EAAE,KAAK,CAAC;CACf;AAED,MAAM,WAAW,MAAM;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,gBAAgB,CAAC;CAC3B;AAED,MAAM,WAAW,KAAK;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,oBAAY,OAAO,GAAG,MAAM,GAAG,WAAW,EAAE,CAAC;AAE7C,oBAAY,WAAW,GAAG,QAAQ,CAAC;AAEnC,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,iBAAiB;IAChC,MAAM,EAAE,YAAY,EAAE,CAAC;CACxB;AAED,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,QAAQ,GAAG,YAAY,CAAC;CACjC"}

package/out/interfaces/types.d.ts

@@ -1,12 +1,12 @@
 import { ForgeLlmError, PromptValidationError } from '../errors';
 import { StreamResponse } from '../streaming/stream-response-wrapper';
-import { LlmRequest, LlmResponse, Message, Tool, ToolCall, Usage } from './internal';
+import { LlmRequest, LlmResponse, Message, Tool, ToolCall, Usage, ModelListResponse, ModelDetails } from './internal';
 export declare type Prompt = LlmRequest & {
     model: string;
 };
 export declare type ChatResponse = LlmResponse;
 export type { StreamResponse };
 export declare type ToolSchema = Tool;
-export type { Message, ToolCall, Usage };
+export type { Message, ToolCall, Usage, ModelListResponse, ModelDetails };
 export { ForgeLlmError, PromptValidationError };
 //# sourceMappingURL=types.d.ts.map

package/out/interfaces/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/interfaces/types.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,aAAa,EAAE,qBAAqB,EAAE,MAAM,WAAW,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,sCAAsC,CAAC;AACtE,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAErF,oBAAY,MAAM,GAAG,UAAU,GAAG;IAChC,KAAK,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,oBAAY,YAAY,GAAG,WAAW,CAAC;AAEvC,YAAY,EAAE,cAAc,EAAE,CAAC;AAE/B,oBAAY,UAAU,GAAG,IAAI,CAAC;AAE9B,YAAY,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;AAEzC,OAAO,EAAE,aAAa,EAAE,qBAAqB,EAAE,CAAC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/interfaces/types.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,aAAa,EAAE,qBAAqB,EAAE,MAAM,WAAW,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,sCAAsC,CAAC;AACtE,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,iBAAiB,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAEtH,oBAAY,MAAM,GAAG,UAAU,GAAG;IAChC,KAAK,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,oBAAY,YAAY,GAAG,WAAW,CAAC;AAEvC,YAAY,EAAE,cAAc,EAAE,CAAC;AAE/B,oBAAY,UAAU,GAAG,IAAI,CAAC;AAE9B,YAAY,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,iBAAiB,EAAE,YAAY,EAAE,CAAC;AAE1E,OAAO,EAAE,aAAa,EAAE,qBAAqB,EAAE,CAAC"}

package/out/llm-api.d.ts

@@ -1,7 +1,8 @@
-import { ChatResponse, Prompt, StreamResponse } from './interfaces/types';
+import { ChatResponse, ModelListResponse, Prompt, StreamResponse } from './interfaces/types';
 import { LlmApi, LlmClient } from './interfaces/internal';
 export declare class LlmApiImpl implements LlmApi {
     llmClient: LlmClient;
+    list: () => Promise<ModelListResponse>;
     chat: (prompt: Prompt) => Promise<ChatResponse>;
     stream: (prompt: Prompt) => Promise<StreamResponse>;
 }

package/out/llm-api.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"llm-api.d.ts","sourceRoot":"","sources":["../src/llm-api.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAE1E,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAK1D,qBAAa,UAAW,YAAW,MAAM;IACvC,SAAS,EAAE,SAAS,CAAuB;IAE3C,IAAI,WAAkB,MAAM,KAAG,QAAQ,YAAY,CAAC,CAgBlD;IAEF,MAAM,WAAkB,MAAM,KAAG,QAAQ,cAAc,CAAC,CAgBtD;CACH"}
+{"version":3,"file":"llm-api.d.ts","sourceRoot":"","sources":["../src/llm-api.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAE7F,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAK1D,qBAAa,UAAW,YAAW,MAAM;IACvC,SAAS,EAAE,SAAS,CAAuB;IAE3C,IAAI,QAAa,QAAQ,iBAAiB,CAAC,CAMzC;IAEF,IAAI,WAAkB,MAAM,KAAG,QAAQ,YAAY,CAAC,CAgBlD;IAEF,MAAM,WAAkB,MAAM,KAAG,QAAQ,cAAc,CAAC,CAgBtD;CACH"}

package/out/llm-api.js

@@ -7,6 +7,11 @@ const error_handling_1 = require("./utils/error-handling");
 const llm_client_1 = require("./llm-client");
 class LlmApiImpl {
     llmClient = new llm_client_1.LlmClientImpl();
+    list = async () => {
+        const response = await this.llmClient.get();
+        await (0, error_handling_1.checkResponseError)(response);
+        return await (0, response_mapper_1.mapForgeLLMBatchResponse)(response);
+    };
     chat = async (prompt) => {
         (0, validators_1.validatePrompt)(prompt);
         const { model, ...request } = prompt;

package/out/llm-client.d.ts

@@ -4,6 +4,8 @@ export declare class LlmClientImpl implements LlmClient {
         traceId: string;
         spanId: string;
     };
+    private static LLM_BASE_URL;
+    get(): Promise<Response>;
     post(args: ClientArgs): Promise<Response>;
     private buildForgeLlmUrl;
 }

package/out/llm-client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"llm-client.d.ts","sourceRoot":"","sources":["../src/llm-client.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAE9D,qBAAa,aAAc,YAAW,SAAS;IAC7C,OAAO;;;MAA2B;IAE5B,IAAI,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,QAAQ,CAAC;IAkB/C,OAAO,CAAC,gBAAgB;CAIzB"}
+{"version":3,"file":"llm-client.d.ts","sourceRoot":"","sources":["../src/llm-client.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAE9D,qBAAa,aAAc,YAAW,SAAS;IAC7C,OAAO;;;MAA2B;IAElC,OAAO,CAAC,MAAM,CAAC,YAAY,CAAiB;IAEtC,GAAG,IAAI,OAAO,CAAC,QAAQ,CAAC;IAexB,IAAI,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,QAAQ,CAAC;IAkB/C,OAAO,CAAC,gBAAgB;CAGzB"}

package/out/llm-client.js

@@ -4,6 +4,21 @@ exports.LlmClientImpl = void 0;
 const api_1 = require("@forge/api");
 class LlmClientImpl {
     tracing = (0, api_1.__getRuntime)()?.tracing;
+    static LLM_BASE_URL = 'https://llm';
+    async get() {
+        const { traceId, spanId } = this.tracing;
+        const options = {
+            method: 'GET',
+            headers: {
+                'Content-Type': 'application/json',
+                'x-b3-traceid': traceId,
+                'x-b3-spanid': spanId
+            }
+        };
+        return await global.__forge_fetch__({ type: 'llm' }, LlmClientImpl.LLM_BASE_URL, {
+            ...options
+        });
+    }
     async post(args) {
         const { model, body } = args;
         const { traceId, spanId } = this.tracing;
@@ -22,8 +37,7 @@ class LlmClientImpl {
         });
     }
     buildForgeLlmUrl(model) {
-        const baseUrl = 'https://llm';
-        return `${baseUrl}/${encodeURIComponent(model)}`;
+        return `${LlmClientImpl.LLM_BASE_URL}/${encodeURIComponent(model)}`;
     }
 }
 exports.LlmClientImpl = LlmClientImpl;

package/out/response-mapper.d.ts

@@ -1,6 +1,5 @@
 import { APIResponse } from '@forge/api';
-import { ChatResponse } from './interfaces/types';
 import { StreamResponse } from './streaming/stream-response-wrapper';
-export declare const mapForgeLLMBatchResponse: (response: APIResponse) => Promise<ChatResponse>;
+export declare function mapForgeLLMBatchResponse<T>(response: APIResponse): Promise<T>;
 export declare const mapForgeLLMStreamResponse: (response: Response) => StreamResponse;
 //# sourceMappingURL=response-mapper.d.ts.map

package/out/response-mapper.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"response-mapper.d.ts","sourceRoot":"","sources":["../src/response-mapper.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,OAAO,EAAyB,cAAc,EAAE,MAAM,qCAAqC,CAAC;AAE5F,eAAO,MAAM,wBAAwB,aAAoB,WAAW,KAAG,QAAQ,YAAY,CAG1F,CAAC;AAEF,eAAO,MAAM,yBAAyB,aAAc,QAAQ,KAAG,cAQ9D,CAAC"}
+{"version":3,"file":"response-mapper.d.ts","sourceRoot":"","sources":["../src/response-mapper.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAGzC,OAAO,EAAyB,cAAc,EAAE,MAAM,qCAAqC,CAAC;AAE5F,wBAAsB,wBAAwB,CAAC,CAAC,EAAE,QAAQ,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,CAGnF;AAED,eAAO,MAAM,yBAAyB,aAAc,QAAQ,KAAG,cAQ9D,CAAC"}

package/out/response-mapper.js

@@ -2,10 +2,10 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.mapForgeLLMStreamResponse = exports.mapForgeLLMBatchResponse = void 0;
 const stream_response_wrapper_1 = require("./streaming/stream-response-wrapper");
-const mapForgeLLMBatchResponse = async (response) => {
+async function mapForgeLLMBatchResponse(response) {
     const chatResponse = await response.json();
     return chatResponse;
-};
+}
 exports.mapForgeLLMBatchResponse = mapForgeLLMBatchResponse;
 const mapForgeLLMStreamResponse = (response) => {
     if (response.body) {

package/package.json

@@ -1,13 +1,13 @@
 {
   "name": "@forge/llm",
-  "version": "0.3.2-next.1",
+  "version": "0.4.0",
   "description": "Forge LLM SDK",
   "main": "out/index.js",
   "types": "out/index.d.ts",
   "author": "Atlassian",
   "license": "SEE LICENSE IN LICENSE.txt",
   "dependencies": {
-    "@forge/api": "^7.0.0-next.3"
+    "@forge/api": "^7.0.0"
   },
   "devDependencies": {
     "@types/node": "20.19.1",