Skip to content

Commit 221856a

Browse files
committed
update example now that StreamEnded is handled gracefully
1 parent 55e9f1a commit 221856a

File tree

3 files changed

+4
-21
lines changed
  • examples

3 files changed

+4
-21
lines changed

examples/responses-function-call/src/main.rs

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -271,11 +271,6 @@ async fn run_streaming() -> Result<(), Box<dyn Error>> {
271271
write!(stdout_lock, "{}: {}\n", delta.event_type(), delta.delta)?;
272272
stdout().flush()?;
273273
}
274-
ResponseStreamEvent::ResponseCompleted(completed) => {
275-
// todo remove once StreamEnded is fixed
276-
writeln!(stdout_lock, "{}\n", completed.event_type(),)?;
277-
break;
278-
}
279274
_ => {
280275
writeln!(stdout_lock, "{}: skipping\n", event.event_type())?;
281276
}
@@ -346,11 +341,6 @@ async fn run_streaming() -> Result<(), Box<dyn Error>> {
346341
write!(stdout_lock, "{}: {}\n", delta.event_type(), delta.delta)?;
347342
stdout().flush()?;
348343
}
349-
ResponseStreamEvent::ResponseCompleted(completed) => {
350-
// todo remove once StreamEnded is fixed
351-
writeln!(stdout_lock, "{}\n", completed.event_type(),)?;
352-
break;
353-
}
354344
_ => {
355345
writeln!(stdout_lock, "{}: skipping\n", event.event_type())?;
356346
}

examples/responses-stream/src/main.rs

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
use async_openai::{
22
Client,
3+
traits::EventType,
34
types::responses::{CreateResponseArgs, ResponseStreamEvent},
45
};
56
use futures::StreamExt;
@@ -25,13 +26,8 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
2526
ResponseStreamEvent::ResponseOutputTextDelta(delta) => {
2627
write!(lock, "{}", delta.delta)?;
2728
}
28-
ResponseStreamEvent::ResponseCompleted(_)
29-
| ResponseStreamEvent::ResponseIncomplete(_)
30-
| ResponseStreamEvent::ResponseFailed(_) => {
31-
break;
32-
}
3329
_ => {
34-
writeln!(lock, "\n{response_event:#?}")?;
30+
writeln!(lock, "\n{}: skipping\n", response_event.event_type())?;
3531
}
3632
},
3733
Err(e) => {

examples/responses-structured-outputs/src/main.rs

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ use std::error::Error;
22

33
use async_openai::{
44
config::OpenAIConfig,
5+
traits::EventType,
56
types::{
67
chat::ResponseFormatJsonSchema,
78
responses::{
@@ -370,12 +371,8 @@ async fn streaming_structured_output(client: &Client<OpenAIConfig>) -> Result<()
370371
final_response = Some(completed.response);
371372
break;
372373
}
373-
ResponseStreamEvent::ResponseFailed(_)
374-
| ResponseStreamEvent::ResponseIncomplete(_) => {
375-
break;
376-
}
377374
_ => {
378-
// Ignore other events
375+
writeln!(lock, "\n{}: skipping\n", event.event_type())?;
379376
}
380377
},
381378
Err(e) => {

0 commit comments

Comments
 (0)