51 lines
1.6 KiB
C#
51 lines
1.6 KiB
C#
using Bot.Rasa.Agents;
|
|
using Bot.Rasa.Models;
|
|
using CustomEntityFoundation;
|
|
using Newtonsoft.Json;
|
|
using Newtonsoft.Json.Serialization;
|
|
using RestSharp;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Text;
|
|
|
|
namespace Bot.Rasa.Console
|
|
{
|
|
public static class RequestExtension
|
|
{
|
|
public static AgentResponse TextRequest(this RasaConsole console, String agentId, String text)
|
|
{
|
|
var client = new RestClient($"{console.options.HostUrl}");
|
|
|
|
var request = new RestRequest("parse?project={project}&q={text}", Method.GET);
|
|
request.AddUrlSegment("project", agentId);
|
|
request.AddUrlSegment("text", text);
|
|
|
|
var response = client.Execute<AgentResponse>(request);
|
|
|
|
return response.Data;
|
|
}
|
|
|
|
public static bool Train(this RasaConsole console, EntityDbContext dc, String agentId)
|
|
{
|
|
var agent = dc.Agent().Find(agentId);
|
|
var corpus = agent.GrabCorpus(dc);
|
|
|
|
string json = JsonConvert.SerializeObject(new { rasa_nlu_data = corpus },
|
|
new JsonSerializerSettings
|
|
{
|
|
ContractResolver = new CamelCasePropertyNamesContractResolver()
|
|
});
|
|
|
|
var client = new RestClient($"{console.options.HostUrl}");
|
|
var request = new RestRequest("train", Method.POST);
|
|
request.AddQueryParameter("project", agentId);
|
|
request.AddParameter("application/json", json, ParameterType.RequestBody);
|
|
|
|
var response = client.Execute(request);
|
|
|
|
return true;
|
|
}
|
|
}
|
|
}
|