53 lines
1.6 KiB
C#
53 lines
1.6 KiB
C#
|
using BotSharp.Core.Engines;
|
|||
|
|
using BotSharp.Platform.Models;
|
|||
|
|
using Microsoft.AspNetCore.Mvc;
|
|||
|
|
using Microsoft.Extensions.Configuration;
|
|||
|
|
using Newtonsoft.Json;
|
|||
|
|
using Newtonsoft.Json.Linq;
|
|||
|
|
using Newtonsoft.Json.Serialization;
|
|||
|
|
using BotSharp.Platform.Dialogflow.Models;
|
|||
|
|
using System;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using System.IO;
|
|||
|
|
using System.Linq;
|
|||
|
|
using System.Text;
|
|||
|
|
using System.Text.RegularExpressions;
|
|||
|
|
using System.Threading.Tasks;
|
|||
|
|
using BotSharp.Platform.Models.MachineLearning;
|
|||
|
|
using BotSharp.Platform.Abstraction;
|
|||
|
|
|
|||
|
|
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;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|