Make RequiredActionUpdate a RunUpdate (#128)

openai · Jul 17, 2024 · 3e12a62 · 3e12a62
1 parent 39941ca
commit 3e12a62
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/Custom/Assistants/Streaming/RequiredActionUpdate.cs b/src/Custom/Assistants/Streaming/RequiredActionUpdate.cs
@@ -11,7 +11,7 @@ namespace OpenAI.Assistants;
 /// Distinct <see cref="RequiredActionUpdate"/> 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.
 /// </remarks>
-public class RequiredActionUpdate : StreamingUpdate
+public class RequiredActionUpdate : RunUpdate
 {
     /// <inheritdoc cref="InternalRequiredFunctionToolCall.InternalName"/>
     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.
     /// </summary>
     /// <returns></returns>
-    public ThreadRun GetThreadRun() => _run;
+    public ThreadRun GetThreadRun() => Value;
 
     internal static IEnumerable<RequiredActionUpdate> DeserializeRequiredActionUpdates(JsonElement element)
     {