Skip to content

Commit 44a95b3

Browse files
authored
chore(streaming): make event method static (#42)
1 parent 534fe82 commit 44a95b3

File tree

1 file changed

+16
-14
lines changed

1 file changed

+16
-14
lines changed

src/openai_responses/streaming.py

Lines changed: 16 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -67,12 +67,13 @@ def to_sse_event(self) -> Tuple[Optional[bytes], Optional[bytes]]:
6767

6868

6969
class BaseEventStream:
70+
@staticmethod
7071
@overload
71-
def event(self, event: Literal["thread.created"], data: Thread) -> Event: ...
72+
def event(event: Literal["thread.created"], data: Thread) -> Event: ...
7273

74+
@staticmethod
7375
@overload
7476
def event(
75-
self,
7677
event: Literal[
7778
"thread.run.created",
7879
"thread.run.queued",
@@ -88,9 +89,9 @@ def event(
8889
data: Run,
8990
) -> Event: ...
9091

92+
@staticmethod
9193
@overload
9294
def event(
93-
self,
9495
event: Literal[
9596
"thread.run.step.created",
9697
"thread.run.step.in_progress",
@@ -102,14 +103,13 @@ def event(
102103
data: RunStep,
103104
) -> Event: ...
104105

106+
@staticmethod
105107
@overload
106-
def event(
107-
self, event: Literal["thread.run.step.delta"], data: RunStepDelta
108-
) -> Event: ...
108+
def event(event: Literal["thread.run.step.delta"], data: RunStepDelta) -> Event: ...
109109

110+
@staticmethod
110111
@overload
111112
def event(
112-
self,
113113
event: Literal[
114114
"thread.message.created",
115115
"thread.message.in_progress",
@@ -119,20 +119,22 @@ def event(
119119
data: Message,
120120
) -> Event: ...
121121

122+
@staticmethod
122123
@overload
123-
def event(
124-
self, event: Literal["thread.message.delta"], data: MessageDelta
125-
) -> Event: ...
124+
def event(event: Literal["thread.message.delta"], data: MessageDelta) -> Event: ...
126125

126+
@staticmethod
127127
@overload
128-
def event(self, event: Literal["error"], data: ErrorObject) -> Event: ...
128+
def event(event: Literal["error"], data: ErrorObject) -> Event: ...
129129

130+
@staticmethod
130131
@overload
131-
def event(self, event: None, data: ChatCompletionChunk) -> Event: ...
132+
def event(event: None, data: ChatCompletionChunk) -> Event: ...
132133

133-
def event(self, event: Optional[EventType], data: EventData) -> Event:
134+
@staticmethod
135+
def event(event: Optional[EventType], data: EventData) -> Event:
134136
"""
135-
Create a server sent event payload with optional event.type and event.data payloads
137+
Create a server sent event payload with event.type (optional) and event.data payloads
136138
"""
137139
return Event(event, data)
138140

0 commit comments

Comments
 (0)