diff --git a/src/Custom/Assistants/Streaming/RequiredActionUpdate.cs b/src/Custom/Assistants/Streaming/RequiredActionUpdate.cs
index cd6db5a98..2c1899236 100644
--- a/src/Custom/Assistants/Streaming/RequiredActionUpdate.cs
+++ b/src/Custom/Assistants/Streaming/RequiredActionUpdate.cs
@@ -11,7 +11,7 @@ namespace OpenAI.Assistants;
/// Distinct instances will generated for each required action, meaning that
/// parallel function calling will present multiple updates even if the tool calls arrive at the same time.
///
-public class RequiredActionUpdate : StreamingUpdate
+public class RequiredActionUpdate : RunUpdate
{
///
public string FunctionName => AsFunctionCall?.FunctionName;
@@ -24,13 +24,11 @@ public class RequiredActionUpdate : StreamingUpdate
private InternalRequiredFunctionToolCall AsFunctionCall => _requiredAction as InternalRequiredFunctionToolCall;
- private readonly ThreadRun _run;
private readonly RequiredAction _requiredAction;
internal RequiredActionUpdate(ThreadRun run, RequiredAction action)
- : base(StreamingUpdateReason.RunRequiresAction)
+ : base(run, StreamingUpdateReason.RunRequiresAction)
{
- _run = run;
_requiredAction = action;
}
@@ -39,7 +37,7 @@ internal RequiredActionUpdate(ThreadRun run, RequiredAction action)
/// update.
///
///
- public ThreadRun GetThreadRun() => _run;
+ public ThreadRun GetThreadRun() => Value;
internal static IEnumerable DeserializeRequiredActionUpdates(JsonElement element)
{