BotSharp/src/Infrastructure/BotSharp.OpenAPI/Controllers/LlmProviderController.cs
2024-01-16 11:38:45 -06:00

30 lines
829 B
C#

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)
{
return _llmProvider.GetProviderModels(provider);
}
}