Skip to content

Commit

Permalink
fix(email_organizer): typing
Browse files Browse the repository at this point in the history
  • Loading branch information
idiotWu committed Dec 31, 2024
1 parent f40547f commit 0956d74
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions npiai/tools/email_organizer/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,7 @@ async def summarize_stream(
output_columns: List[Column],
with_pdf_attachments: bool = False,
concurrency: int = 1,
):
) -> AsyncGenerator[EmailSummary, None]:
"""
Summarize emails into a table.
Expand All @@ -181,7 +181,7 @@ async def summarize_stream(
lock = asyncio.Lock()
index = 0

async def summarize_email(results_queue: asyncio.Queue[FilterResult]):
async def summarize_email(results_queue: asyncio.Queue[EmailSummary]):
nonlocal index
async with lock:
if index >= len(email_or_id_list):
Expand Down

0 comments on commit 0956d74

Please sign in to comment.