Skip to content

Commit c08814e

Browse files
committed
Optimize user initial test model use experience
1 parent ebd5414 commit c08814e

2 files changed

Lines changed: 6 additions & 4 deletions

File tree

src/BE/DB/Init/InitService.cs

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
using Chats.BE.DB.Enums;
22
using Chats.BE.Services;
33
using Chats.BE.Services.Configs;
4+
using Chats.BE.Services.Models.ChatServices.Test;
45
using System.Text.Json;
56

67
namespace Chats.BE.DB.Init;
@@ -32,10 +33,10 @@ private static async Task InsertInitialData(IServiceScope scope, ChatsDB db, Can
3233
Name = "Hello-World Model",
3334
UpdatedAt = DateTime.UtcNow,
3435
CreatedAt = DateTime.UtcNow,
35-
DeploymentName = "hello-world",
36-
ReasoningEffortOptions = "",
37-
SupportedImageSizes = "",
36+
DeploymentName = Test2ChatService.ModelName,
3837
AllowStreaming = true,
38+
ContextWindow = 64000,
39+
MaxResponseTokens = 16000,
3940
};
4041
ModelKey modelKey = new()
4142
{

src/BE/Services/Models/ChatServices/Test/Test2ChatService.cs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,9 +131,10 @@ async IAsyncEnumerable<ChatSegment> TextOnly([EnumeratorCancellation] Cancellati
131131
}
132132
}
133133

134+
internal const string ModelName = "test-model";
134135
public override Task<string[]> ListModels(ModelKey modelKey, CancellationToken cancellationToken)
135136
{
136-
return Task.FromResult(new[] { "test-model" });
137+
return Task.FromResult(new[] { ModelName });
137138
}
138139

139140
internal static IEnumerable<string> UnicodeCharacterSplit(string input)

0 commit comments

Comments
 (0)