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

SA-243 Save message at the end of product search

Browse Source 
- Add message saving on "[DONE]" text in SearchProductAsync()
This commit is contained in:
shchoholiev 2023-12-17 17:10:41 +00:00
parent 07186c3052
commit e13bb4bbed
2 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -73,7 +73,11 @@ public class OpenAiService : IOpenAiService
if (string.IsNullOrEmpty(line)) continue;
var json = line?.Substring(6, line.Length - 6);
if (json == "[DONE]") yield break;
if (json == "[DONE]")
{
yield return json;
yield break;
}
var data = JsonConvert.DeserializeObject<OpenAiResponse>(json);

17
ShoppingAssistantApi.Infrastructure/Services/ProductService.cs
View File

@ -95,7 +95,20 @@ public class ProductService : IProductService
await foreach (var data in _openAiService.GetChatCompletionStream(chatRequest, cancellationToken))
{
if (data.Contains('['))
if (data == "[DONE]")
{
if (!string.IsNullOrEmpty(messageBuffer.Text))
{
_ = await _wishlistsService.AddMessageToPersonalWishlistAsync(wishlistId, new MessageDto()
{
Text = messageBuffer.Text,
Role = MessageRoles.Application.ToString(),
}, cancellationToken);
}
yield break;
}
else if (data.Contains('['))
{
dataTypeHolder = data;
}
@ -110,6 +123,8 @@ public class ProductService : IProductService
Text = messageBuffer.Text,
Role = MessageRoles.Application.ToString(),
}, cancellationToken);
messageBuffer.Text = string.Empty;
}
dataTypeHolder += data;