From 93b818dbff126047dc8975c649f867c346258a08 Mon Sep 17 00:00:00 2001
From: Daniel OBrien <dobrien@launchdarkly.com>
Date: Mon, 11 Nov 2024 08:55:46 -0500
Subject: [PATCH] feat: update model signature

---
 packages/sdk/server-ai/src/api/config/LDAIConfig.ts | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/packages/sdk/server-ai/src/api/config/LDAIConfig.ts b/packages/sdk/server-ai/src/api/config/LDAIConfig.ts
index 0d47f6eac..5bd8748bc 100644
--- a/packages/sdk/server-ai/src/api/config/LDAIConfig.ts
+++ b/packages/sdk/server-ai/src/api/config/LDAIConfig.ts
@@ -9,6 +9,16 @@ export interface LDModelConfig {
    */
   modelId?: string;
 
+  /**
+   * The temperature of the model.
+   */
+  temperature?: number;
+
+  /**
+   * The maximum number of tokens to generate.
+   */
+  maxTokens?: number;
+
   /**
    * And additional model specific information.
    */