46 lines
1.3 KiB
C#
46 lines
1.3 KiB
C#
using BotSharp.Platform.Models;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using BotSharp.Platform.Dialogflow.Models;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using BotSharp.Platform.Models.MachineLearning;
|
|
|
|
namespace BotSharp.Platform.Dialogflow.Controllers
|
|
{
|
|
/// <summary>
|
|
/// You can post your training data to this endpoint to train a new model for a project.
|
|
/// This request will wait for the server answer: either the model was trained successfully or the training exited with an error.
|
|
/// </summary>
|
|
[Route("v1/[controller]")]
|
|
public class TrainController : ControllerBase
|
|
{
|
|
private DialogflowAi<AgentModel> builder;
|
|
|
|
public TrainController(DialogflowAi<AgentModel> platform)
|
|
{
|
|
builder = platform;
|
|
}
|
|
|
|
[HttpPost]
|
|
public async Task<ActionResult<ModelMetaData>> Train([FromQuery] string agentId)
|
|
{
|
|
var agent = await builder.GetAgentById(agentId);
|
|
|
|
if(agent == null)
|
|
{
|
|
agent = await builder.GetAgentByName(agentId);
|
|
}
|
|
|
|
var corpus = await builder.ExtractorCorpus(agent);
|
|
|
|
var meta = await builder.Train(agent, corpus, new BotTrainOptions { });
|
|
|
|
return meta;
|
|
}
|
|
}
|
|
}
|