public void TestClassifyDocument()
{
string remoteFileName = "TestClassifyDocument.docx";
this.UploadFileToStorage(
remoteDataFolder + "/" + remoteFileName,
null,
null,
File.ReadAllBytes(LocalTestDataFolder + localFile)
);
var request = new ClassifyDocumentRequest(
name: remoteFileName,
folder: remoteDataFolder,
bestClassesCount: "3"
);
var actual = this.WordsApi.ClassifyDocument(request);
Assert.AreEqual("Hobbies_&_Interests", actual.BestClassName);
Assert.NotNull(actual.BestResults);
Assert.AreEqual(3, actual.BestResults.Count);
}
/// <summary>
/// A simple function that takes a string and does a ToUpper
/// </summary>
/// <param name="input"></param>
/// <param name="context"></param>
/// <returns></returns>
public string FunctionHandler(JObject input, ILambdaContext context)
{
AmazonComprehendClient comprehendClient = new AmazonComprehendClient(Amazon.RegionEndpoint.USEast1);
var detectEntitiesRequest = new ClassifyDocumentRequest()
{
Text = input["Text"].ToString(),
EndpointArn = Environment.GetEnvironmentVariable("DOCUMENT_CLASSIFICATION_ENDPOINT")
};
var detectEntitiesResponse = comprehendClient.ClassifyDocumentAsync(detectEntitiesRequest).Result;
var bestEntity = new DocumentClass();
foreach (var e in detectEntitiesResponse.Classes)
{
if (e.Score > bestEntity.Score)
{
bestEntity = e;
}
}
return(JsonConvert.SerializeObject(bestEntity));
}
