Skip to content

Commit a801718

Browse files
fix: type hint for doc_total_cost
Signed-off-by: thiswillbeyourgithub <[email protected]>
1 parent 7f08200 commit a801718

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

wdoc/utils/tasks/summarize.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ class wdocSummary:
7171
Total tokens used across all operations.
7272
doc_total_tokens_str : str
7373
Human-readable string representation of token usage.
74-
doc_total_cost : float
74+
doc_total_cost : Union[float, int]
7575
Total cost in dollars for LLM usage.
7676
author : Optional[str]
7777
Document author if available in metadata.
@@ -88,7 +88,7 @@ class wdocSummary:
8888
doc_total_tokens: Dict[str, int]
8989
doc_total_tokens_sum: int
9090
doc_total_tokens_str: str
91-
doc_total_cost: float
91+
doc_total_cost: Union[float, int]
9292
author: Optional[str]
9393
n_chunk: int
9494

0 commit comments

Comments
 (0)