cartaid/shopping-assistant-api
1
0
Fork 0
mirror of https://github.com/Shchoholiev/shopping-assistant-api.git synced 2025-04-12 01:48:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

SA-29 working GetChatCompletion

Browse Source
This commit is contained in:
Mykyta Dubovyi 2023-10-19 12:56:40 +03:00
parent 534fc3293d
commit 2b9453b09f
2 changed files with 73 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
ShoppingAssistantApi.Infrastructure/Services/OpenAiService.cs
View File

@ -1,6 +1,7 @@
using System; using System;
using System.Net.Http.Headers; using System.Net.Http.Headers;
using System.Text; using System.Text;
using MongoDB.Bson;
using Newtonsoft.Json; using Newtonsoft.Json;
using Newtonsoft.Json.Serialization; using Newtonsoft.Json.Serialization;
using ShoppingAssistantApi.Application.IServices; using ShoppingAssistantApi.Application.IServices;
@ -25,32 +26,24 @@ public class OpenAiService : IOpenAiService
public OpenAiService(HttpClient client) public OpenAiService(HttpClient client)
{ {
_httpClient = client; _httpClient = client;
_httpClient.DefaultRequestHeaders.Authorization =
new AuthenticationHeaderValue("Bearer", "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");
_httpClient.DefaultRequestHeaders.Accept.Add(new MediaTypeWithQualityHeaderValue("application/json"));
} }
public async Task<OpenAiMessage> GetChatCompletion(ChatCompletionRequest chat, CancellationToken cancellationToken) public async Task<OpenAiMessage> GetChatCompletion(ChatCompletionRequest chat, CancellationToken cancellationToken)
{ {
chat.Stream = false; chat.Stream = false;
var jsonBody = JsonConvert.SerializeObject(chat, _jsonSettings); var jsonBody = JsonConvert.SerializeObject(chat, _jsonSettings);
var body = new StringContent(jsonBody, Encoding.UTF8, "application/json"); var body = new StringContent(jsonBody, Encoding.UTF8, /*change file appsettings.Develop.json*/"application/json");
using var httpResponse = await _httpClient.PostAsync(/*api url*/"https://api.openai.com/v1/completions", body, cancellationToken);
using var httpResponse = await _httpClient.PostAsync("chat/completions", body, cancellationToken);
var responseBody = await httpResponse.Content.ReadAsStringAsync(cancellationToken); var responseBody = await httpResponse.Content.ReadAsStringAsync(cancellationToken);
var responses = new List<OpenAiMessage>(); var data = JsonConvert.DeserializeObject<OpenAiResponse>(responseBody);
foreach (var line in responseBody.Split(new[] {"\n\n"}, StringSplitOptions.RemoveEmptyEntries))
{
if (line.Trim() == "[DONE]") break;
var json = line.Substring(6); return data.Choices[0].Message;
var OpenAiMessage = JsonConvert.DeserializeObject<OpenAiMessage>(json, _jsonSettings);
responses.Add(OpenAiMessage);
}
return responses.Count > 0 ? responses.Last() : null;
} }
public async IAsyncEnumerable<string> GetChatCompletionStream(ChatCompletionRequest chat, CancellationToken cancellationToken) public async IAsyncEnumerable<string> GetChatCompletionStream(ChatCompletionRequest chat, CancellationToken cancellationToken)
@ -59,7 +52,7 @@ public class OpenAiService : IOpenAiService
var jsonBody = JsonConvert.SerializeObject(chat, _jsonSettings); var jsonBody = JsonConvert.SerializeObject(chat, _jsonSettings);
var body = new StringContent(jsonBody, Encoding.UTF8, "application/json"); var body = new StringContent(jsonBody, Encoding.UTF8, "application/json");
var request = new HttpRequestMessage(HttpMethod.Post, "chat/completions") var request = new HttpRequestMessage(HttpMethod.Post, "https://api.openai.com/v1/completions")
{ {
Content = body Content = body
}; };

70
ShoppingAssistantApi.UnitTests/OpenAiServiceTests.cs
View File

@ -24,8 +24,68 @@ public class OpenAiServiceTests
_openAiService = new OpenAiService(_httpClient); _openAiService = new OpenAiService(_httpClient);
} }
//[Fact]
//public async Task GetChatCompletion_ValidChat_ReturnsNewMessage()
//{
// // Arrange
// _mockHttpMessageHandler
// .Protected()
// .Setup<Task<HttpResponseMessage>>(
// "SendAsync",
// ItExpr.IsAny<HttpRequestMessage>(),
// ItExpr.IsAny<CancellationToken>()
// )
// .ReturnsAsync(new HttpResponseMessage
// {
// StatusCode = HttpStatusCode.OK,
// Content = new StringContent(@"
// {
// ""id"": ""chatcmpl-89OMdgTZXOLAXv7bPUJ4SwrPpS5Md"",
// ""object"": ""chat.completion"",
// ""created"": 1697249299,
// ""model"": ""gpt-3.5-turbo-0613"",
// ""choices"": [
// {
// ""index"": 0,
// ""message"": {
// ""role"": ""assistant"",
// ""content"": ""Hello World!""
// },
// ""finish_reason"": ""stop""
// }
// ],
// ""usage"": {
// ""prompt_tokens"": 10,
// ""completion_tokens"": 3,
// ""total_tokens"": 13
// }
// }"),
// });
// var chat = new ChatCompletionRequest
// {
// Messages = new List<OpenAiMessage>
// {
// new OpenAiMessage
// {
// Role = OpenAiRole.User,
// Content = "Return Hello World!"
// }
// }
// };
// // Act
// var newMessage = await _openAiService.GetChatCompletion(chat, CancellationToken.None);
// // Assert
// Assert.NotNull(newMessage);
// Assert.Equal("Hello World!", newMessage.Content);
//}
// TODO: Add more tests
[Fact] [Fact]
public async Task GetChatCompletion_ValidChat_ReturnsNewMessage() public async Task GetChatCompletionStream_ValidChat_ReturnsNewMessage()
{ {
// Arrange // Arrange
_mockHttpMessageHandler _mockHttpMessageHandler
@ -61,7 +121,7 @@ public class OpenAiServiceTests
} }
}"), }"),
}); });
var chat = new ChatCompletionRequest var chat = new ChatCompletionRequest
{ {
Messages = new List<OpenAiMessage> Messages = new List<OpenAiMessage>
@ -75,12 +135,10 @@ public class OpenAiServiceTests
}; };
// Act // Act
var newMessage = await _openAiService.GetChatCompletion(chat, CancellationToken.None); var newMessage = _openAiService.GetChatCompletionStream(chat, CancellationToken.None);
// Assert // Assert
Assert.NotNull(newMessage); Assert.NotNull(newMessage);
Assert.Equal("Hello, World!", newMessage.Content); Assert.Equal("Hello World!", newMessage.ToString());
} }
// TODO: Add more tests
} }