BotSharp/src/Infrastructure/BotSharp.OpenAPI/Controllers/LlmProviderController.cs

31 lines
885 B
C#
Raw Normal View History

using BotSharp.Abstraction.MLTasks;
using BotSharp.Abstraction.MLTasks.Settings;
namespace BotSharp.OpenAPI.Controllers;
[Authorize]
[ApiController]
public class LlmProviderController : ControllerBase
{
private readonly IServiceProvider _services;
private readonly ILlmProviderService _llmProvider;
public LlmProviderController(IServiceProvider services, ILlmProviderService llmProvider)
{
_services = services;
_llmProvider = llmProvider;
}
[HttpGet("/llm-providers")]
public IEnumerable<string> GetLlmProviders()
{
return _llmProvider.GetProviders();
}
[HttpGet("/llm-provider/{provider}/models")]
public IEnumerable<LlmModelSetting> GetLlmProviderModels([FromRoute] string provider)
{
2024-06-24 19:32:52 +00:00
var list = _llmProvider.GetProviderModels(provider);
return list.Where(x => !x.ImageGeneration);
}
}