Skip to content

Commit 3a7fd63

Browse files
committed
Expand tests.
1 parent b3a200d commit 3a7fd63

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

packages/sdk/server-ai/__tests__/LDAIClientImpl.test.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ it('returns model config with interpolated prompts', async () => {
3939
};
4040

4141
const mockVariation = {
42-
model: { modelId: 'example-provider', name: 'imagination' },
42+
model: { modelId: 'example-provider', name: 'imagination', temperature: 0.7, maxTokens: 4096 },
4343
prompt: [
4444
{ role: 'system', content: 'Hello {{name}}' },
4545
{ role: 'user', content: 'Score: {{score}}' },
@@ -56,7 +56,7 @@ it('returns model config with interpolated prompts', async () => {
5656
const result = await client.modelConfig(key, testContext, defaultValue, variables);
5757

5858
expect(result).toEqual({
59-
model: { modelId: 'example-provider', name: 'imagination' },
59+
model: { modelId: 'example-provider', name: 'imagination', temperature: 0.7, maxTokens: 4096 },
6060
prompt: [
6161
{ role: 'system', content: 'Hello John' },
6262
{ role: 'user', content: 'Score: 42' },

0 commit comments

Comments
 (0)