|
39 | 39 |
|
40 | 40 | from anthropic.resources import AsyncMessages, Messages |
41 | 41 |
|
42 | | - try: |
43 | | - # http://github.com/anthropics/anthropic-sdk-python/commit/bc9d11cd2addec6976c46db10b7c89a8c276101a |
44 | | - from anthropic.types import ( |
45 | | - RawMessageStartEvent, |
46 | | - RawMessageDeltaEvent, |
47 | | - RawMessageStopEvent, |
48 | | - RawContentBlockStartEvent, |
49 | | - RawContentBlockDeltaEvent, |
50 | | - RawContentBlockStopEvent, |
51 | | - ) |
52 | | - except ImportError: |
53 | | - RawMessageStartEvent = None |
54 | | - RawMessageDeltaEvent = None |
55 | | - RawMessageStopEvent = None |
56 | | - RawContentBlockStartEvent = None |
57 | | - RawContentBlockDeltaEvent = None |
58 | | - RawContentBlockStopEvent = None |
59 | | - |
60 | 42 | from anthropic.types import ( |
61 | 43 | MessageStartEvent, |
62 | 44 | MessageDeltaEvent, |
@@ -435,20 +417,7 @@ def new_iterator() -> "Iterator[MessageStreamEvent]": |
435 | 417 | content_blocks: "list[str]" = [] |
436 | 418 |
|
437 | 419 | for event in old_iterator: |
438 | | - if ( |
439 | | - RawMessageStartEvent is not None |
440 | | - and not isinstance( |
441 | | - event, |
442 | | - ( |
443 | | - RawMessageStartEvent, |
444 | | - RawMessageDeltaEvent, |
445 | | - RawMessageStopEvent, |
446 | | - RawContentBlockStartEvent, |
447 | | - RawContentBlockDeltaEvent, |
448 | | - RawContentBlockStopEvent, |
449 | | - ), |
450 | | - ) |
451 | | - ) or not isinstance( |
| 420 | + if not isinstance( |
452 | 421 | event, |
453 | 422 | ( |
454 | 423 | MessageStartEvent, |
@@ -500,20 +469,7 @@ async def new_iterator_async() -> "AsyncIterator[MessageStreamEvent]": |
500 | 469 | content_blocks: "list[str]" = [] |
501 | 470 |
|
502 | 471 | async for event in old_iterator: |
503 | | - if ( |
504 | | - RawMessageStartEvent is not None |
505 | | - and not isinstance( |
506 | | - event, |
507 | | - ( |
508 | | - RawMessageStartEvent, |
509 | | - RawMessageDeltaEvent, |
510 | | - RawMessageStopEvent, |
511 | | - RawContentBlockStartEvent, |
512 | | - RawContentBlockDeltaEvent, |
513 | | - RawContentBlockStopEvent, |
514 | | - ), |
515 | | - ) |
516 | | - ) or not isinstance( |
| 472 | + if not isinstance( |
517 | 473 | event, |
518 | 474 | ( |
519 | 475 | MessageStartEvent, |
|
0 commit comments