Streaming BAML Functions

1
import asyncio
2
from baml_client import b, partial_types, types
3
4
# Using both async iteration and get_final_response() from a stream
5
async def example1(receipt: str):
6
    stream = b.stream.ExtractReceiptInfo(receipt)
7
8
    async for partial in stream:
9
        print(f"partial: parsed {len(partial.items)} items (object: {partial})")
10
11
    final = await stream.get_final_response()
12
    print(f"final: {len(final.items)} items (object: {final})")
13
14
# Using only async iteration of a stream
15
async def example2(receipt: str):
16
    async for partial in b.stream.ExtractReceiptInfo(receipt):
17
        print(f"partial: parsed {len(partial.items)} items (object: {partial})")
18
19
# Using only get_final_response() of a stream
20
#
21
# In this case, you should just use b.ExtractReceiptInfo(receipt) instead,
22
# which is faster and more efficient.
23
async def example3(receipt: str):
24
    final = await b.stream.ExtractReceiptInfo(receipt).get_final_response()
25
    print(f"final: {len(final.items)} items (object: {final})")
26
27
receipt = """
28
04/14/2024 1:05 pm
29
30
Ticket: 220000082489
31
Register: Shop Counter
32
Employee: Connor
33
Customer: Sam
34
Item	#	Price
35
Guide leash (1 Pair) uni UNI
36
1	$34.95
37
The Index Town Walls
38
1	$35.00
39
Boot Punch
40
3	$60.00
41
Subtotal	$129.95
42
Tax ($129.95 @ 9%)	$11.70
43
Total Tax	$11.70
44
Total	$141.65
45
"""
46
47
if __name__ == '__main__':
48
    asyncio.run(example1(receipt))
49
    asyncio.run(example2(receipt))
50
    asyncio.run(example3(receipt))