Skip to content

Commit 944869e

Browse files
authored
Merge pull request #1136 from iceljc/master
refine web search model select
2 parents aca1145 + 6466092 commit 944869e

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

src/Infrastructure/BotSharp.Abstraction/MLTasks/Settings/LlmModelSetting.cs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,7 @@ public class ReasoningSetting
8686

8787
public class WebSearchSetting
8888
{
89+
public bool IsDefault { get; set; }
8990
public string? SearchContextSize { get; set; }
9091
}
9192

src/Infrastructure/BotSharp.Core/WebSearch/Functions/WebIntelligentSearchFn.cs

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,16 @@ private async Task<string> GetChatCompletion(Agent agent, List<RoleDialogModel>
5353
{
5454
try
5555
{
56+
var provider = "openai";
57+
var defaultModel = "gpt-4o-mini-search-preview";
58+
5659
var llmProviderService = _services.GetRequiredService<ILlmProviderService>();
57-
var completion = CompletionProvider.GetChatCompletion(_services, provider: "openai", model: "gpt-4o-search-preview");
60+
var models = llmProviderService.GetProviderModels(provider);
61+
var webSearchModel = models.FirstOrDefault(x => x.WebSearch?.IsDefault == true)?.Name
62+
?? models.FirstOrDefault(x => x.WebSearch != null)?.Name
63+
?? defaultModel;
64+
65+
var completion = CompletionProvider.GetChatCompletion(_services, provider: provider, model: webSearchModel);
5866
var response = await completion.GetChatCompletions(agent, dialogs);
5967
return response.Content;
6068
}

0 commit comments

Comments
 (0)