MindWorkAI/AI-Studio
MindWorkAI/AI-Studio
1
0
Fork 0
mirror of https://github.com/MindWorkAI/AI-Studio.git synced 2026-08-23 11:22:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix SSE parsing without space after data prefix

Browse Source
This commit is contained in:
Thorsten Sommer 2026-08-12 14:47:14 +02:00
parent aa11e1c352
commit 10a37ddac9
Signed by untrusted user who does not match committer: tsommer
GPG Key ID: 371BBA77A02C0108
1 changed files with 28 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
app/MindWork AI Studio/Provider/BaseProvider.cs
View File

@ -269,10 +269,10 @@ public abstract class BaseProvider : IProvider, ISecretId
{ {
exception = new(); exception = new();
if (!line.StartsWith("data: ", StringComparison.InvariantCulture)) if (!TryGetServerSentEventData(line, out var jsonData))
return false; return false;
var jsonData = line[6..].Trim(); jsonData = jsonData.Trim();
if (string.IsNullOrWhiteSpace(jsonData) || jsonData is "[DONE]") if (string.IsNullOrWhiteSpace(jsonData) || jsonData is "[DONE]")
return false; return false;
@ -304,6 +304,21 @@ public abstract class BaseProvider : IProvider, ISecretId
} }
} }
private static bool TryGetServerSentEventData(string line, out string data)
{
const string DATA_PREFIX = "data:";
data = string.Empty;
if (!line.StartsWith(DATA_PREFIX, StringComparison.InvariantCulture))
return false;
data = line[DATA_PREFIX.Length..];
if (data.StartsWith(' '))
data = data[1..];
return true;
}
private static bool IsProviderStreamFailure(JsonElement root) private static bool IsProviderStreamFailure(JsonElement root)
{ {
var eventType = TryGetString(root, "type"); var eventType = TryGetString(root, "type");
@ -661,13 +676,13 @@ public abstract class BaseProvider : IProvider, ISecretId
if (this.TryCreateProviderRequestExceptionFromStreamLine(providerName, line, out var providerRequestException)) if (this.TryCreateProviderRequestExceptionFromStreamLine(providerName, line, out var providerRequestException))
throw providerRequestException; throw providerRequestException;
// Skip lines that do not start with "data: ". Regard // Skip lines that do not start with "data:". According
// to the specification, we only want to read the data lines: // to the specification, we only want to read the data lines:
if (!line.StartsWith("data: ", StringComparison.InvariantCulture)) if (!TryGetServerSentEventData(line, out var jsonData))
continue; continue;
// Check if the line is the end of the stream: // Check if the line is the end of the stream:
if (line.StartsWith("data: [DONE]", StringComparison.InvariantCulture)) if (jsonData is "[DONE]")
yield break; yield break;
// //
@ -681,10 +696,6 @@ public abstract class BaseProvider : IProvider, ISecretId
try try
{ {
// We know that the line starts with "data: ". Hence, we can
// skip the first 6 characters to get the JSON data after that.
var jsonData = line[6..];
// Deserialize the JSON data: // Deserialize the JSON data:
providerResponse = JsonSerializer.Deserialize<TAnnotation>(jsonData, JSON_SERIALIZER_OPTIONS); providerResponse = JsonSerializer.Deserialize<TAnnotation>(jsonData, JSON_SERIALIZER_OPTIONS);
@ -713,10 +724,6 @@ public abstract class BaseProvider : IProvider, ISecretId
TDelta? providerResponse; TDelta? providerResponse;
try try
{ {
// We know that the line starts with "data: ". Hence, we can
// skip the first 6 characters to get the JSON data after that.
var jsonData = line[6..];
// Deserialize the JSON data: // Deserialize the JSON data:
providerResponse = JsonSerializer.Deserialize<TDelta>(jsonData, JSON_SERIALIZER_OPTIONS); providerResponse = JsonSerializer.Deserialize<TDelta>(jsonData, JSON_SERIALIZER_OPTIONS);
@ -866,20 +873,19 @@ public abstract class BaseProvider : IProvider, ISecretId
if (line.StartsWith("event: response.completed", StringComparison.InvariantCulture)) if (line.StartsWith("event: response.completed", StringComparison.InvariantCulture))
yield break; yield break;
if (!TryGetServerSentEventData(line, out var jsonData))
continue;
// //
// Find delta lines: // Find delta lines:
// //
if (line.StartsWith(""" if (jsonData.StartsWith("""
data: {"type":"response.output_text.delta" {"type":"response.output_text.delta"
""", StringComparison.InvariantCulture)) """, StringComparison.InvariantCulture))
{ {
TDelta? providerResponse; TDelta? providerResponse;
try try
{ {
// We know that the line starts with "data: ". Hence, we can
// skip the first 6 characters to get the JSON data after that.
var jsonData = line[6..];
// Deserialize the JSON data: // Deserialize the JSON data:
providerResponse = JsonSerializer.Deserialize<TDelta>(jsonData, JSON_SERIALIZER_OPTIONS); providerResponse = JsonSerializer.Deserialize<TDelta>(jsonData, JSON_SERIALIZER_OPTIONS);
@ -903,18 +909,14 @@ public abstract class BaseProvider : IProvider, ISecretId
// //
// Find annotation added lines: // Find annotation added lines:
// //
else if (annotationSupported && line.StartsWith( else if (annotationSupported && jsonData.StartsWith(
""" """
data: {"type":"response.output_text.annotation.added" {"type":"response.output_text.annotation.added"
""", StringComparison.InvariantCulture)) """, StringComparison.InvariantCulture))
{ {
TAnnotation? providerResponse; TAnnotation? providerResponse;
try try
{ {
// We know that the line starts with "data: ". Hence, we can
// skip the first 6 characters to get the JSON data after that.
var jsonData = line[6..];
// Deserialize the JSON data: // Deserialize the JSON data:
providerResponse = JsonSerializer.Deserialize<TAnnotation>(jsonData, JSON_SERIALIZER_OPTIONS); providerResponse = JsonSerializer.Deserialize<TAnnotation>(jsonData, JSON_SERIALIZER_OPTIONS);