2024-01-16 17:38:45 +00:00
|
|
|
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);
|
2024-01-16 17:38:45 +00:00
|
|
|
}
|
|
|
|
|
}
|