Merge pull request #1136 from iceljc/master
refine web search model select
This commit is contained in:
commit
944869ed79
|
|
@ -86,6 +86,7 @@ public class ReasoningSetting
|
|||
|
||||
public class WebSearchSetting
|
||||
{
|
||||
public bool IsDefault { get; set; }
|
||||
public string? SearchContextSize { get; set; }
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -53,8 +53,16 @@ public class WebIntelligentSearchFn : IFunctionCallback
|
|||
{
|
||||
try
|
||||
{
|
||||
var provider = "openai";
|
||||
var defaultModel = "gpt-4o-mini-search-preview";
|
||||
|
||||
var llmProviderService = _services.GetRequiredService<ILlmProviderService>();
|
||||
var completion = CompletionProvider.GetChatCompletion(_services, provider: "openai", model: "gpt-4o-search-preview");
|
||||
var models = llmProviderService.GetProviderModels(provider);
|
||||
var webSearchModel = models.FirstOrDefault(x => x.WebSearch?.IsDefault == true)?.Name
|
||||
?? models.FirstOrDefault(x => x.WebSearch != null)?.Name
|
||||
?? defaultModel;
|
||||
|
||||
var completion = CompletionProvider.GetChatCompletion(_services, provider: provider, model: webSearchModel);
|
||||
var response = await completion.GetChatCompletions(agent, dialogs);
|
||||
return response.Content;
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue