Skip to content

Commit 19a3515

Browse files
some changes
1 parent bff155c commit 19a3515

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

griptape/drivers/assistant/openai_assistant_driver.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@
1414
if TYPE_CHECKING:
1515
import openai
1616
from openai import AssistantEventHandler
17+
from openai.types.beta.threads import Text, TextDelta
18+
from openai.types.beta.threads.runs import ToolCall, ToolCallDelta
1719

1820

1921
@define
@@ -24,15 +26,13 @@ def _create_event_handler_class() -> type[AssistantEventHandler]: # pyright: ig
2426
AssistantEventHandler = import_optional_dependency("openai").AssistantEventHandler # noqa: N806
2527

2628
class EventHandler(AssistantEventHandler):
27-
# Pyright can't verify override since base class is in TYPE_CHECKING
2829
@override
29-
def on_text_delta(self, delta, snapshot) -> None: # pyright: ignore[reportGeneralTypeIssues,reportUndefinedVariable] # noqa: ANN001
30+
def on_text_delta(self, delta: TextDelta, snapshot: Text) -> None: # pyright: ignore[reportGeneralTypeIssues,reportUndefinedVariable] Pyright can't verify override since base class is in TYPE_CHECKING
3031
if delta.value is not None:
3132
EventBus.publish_event(TextChunkEvent(token=delta.value))
3233

33-
# Pyright can't verify override since base class is in TYPE_CHECKING
3434
@override
35-
def on_tool_call_delta(self, delta, snapshot) -> None: # pyright: ignore[reportGeneralTypeIssues,reportUndefinedVariable] # noqa: ANN001
35+
def on_tool_call_delta(self, delta: ToolCallDelta, snapshot: ToolCall) -> None: # pyright: ignore[reportGeneralTypeIssues,reportUndefinedVariable] Pyright can't verify override since base class is in TYPE_CHECKING
3636
if delta.type == "code_interpreter" and delta.code_interpreter is not None:
3737
if delta.code_interpreter.input:
3838
EventBus.publish_event(TextChunkEvent(token=delta.code_interpreter.input))

0 commit comments

Comments
 (0)