From 5ecbe46082cc4e19e89c035b72659ec2da0d8235 Mon Sep 17 00:00:00 2001
From: Nick Bobrowski <39348559+bonk1t@users.noreply.github.com>
Date: Mon, 9 Dec 2024 01:44:30 +0000
Subject: [PATCH] Import fix

---
 agency_swarm/util/streaming/gradio_event_handler.py | 3 ++-
 agency_swarm/util/streaming/term_event_handler.py   | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/agency_swarm/util/streaming/gradio_event_handler.py b/agency_swarm/util/streaming/gradio_event_handler.py
index fe4ee3ce..2206a3d6 100644
--- a/agency_swarm/util/streaming/gradio_event_handler.py
+++ b/agency_swarm/util/streaming/gradio_event_handler.py
@@ -1,5 +1,5 @@
 import queue
-from typing import Type, override
+from typing import Type
 
 from openai.types.beta.threads import Message
 from openai.types.beta.threads.runs.run_step import RunStep
@@ -9,6 +9,7 @@
     FunctionToolCall,
     ToolCall,
 )
+from typing_extensions import override
 
 from agency_swarm.messages import MessageOutput
 from agency_swarm.util.streaming import (
diff --git a/agency_swarm/util/streaming/term_event_handler.py b/agency_swarm/util/streaming/term_event_handler.py
index 9b229a9e..8d1e7454 100644
--- a/agency_swarm/util/streaming/term_event_handler.py
+++ b/agency_swarm/util/streaming/term_event_handler.py
@@ -1,4 +1,4 @@
-from typing import Type, override
+from typing import Type
 
 from openai.types.beta.threads import Message
 from openai.types.beta.threads.runs import RunStep