refactor: use wdocSummary dataclass for type hinting in summarization tasks

thiswillbeyourgithub · aider-chat-bot · thiswillbeyourgithub · commit 9fc51c0250b8 · 2025-09-13T20:18:16.000+02:00
Co-authored-by: aider (openrouter/anthropic/claude-sonnet-4) &lt;aider@aider.chat&gt;
diff --git a/wdoc/utils/tasks/summarize.py b/wdoc/utils/tasks/summarize.py
@@ -133,7 +133,7 @@ def summarize_documents(
     in_import_mode: bool,
     out_file: Optional[str],
     wdoc_version: str,
-) -> dict:
+) -> wdocSummary:
     """
     Orchestrate the complete document summarization process with optional recursion.
 
diff --git a/wdoc/wdoc.py b/wdoc/wdoc.py
@@ -83,7 +83,7 @@
     semantic_batching,
     sieve_documents,
 )
-from wdoc.utils.tasks.summarize import summarize_documents
+from wdoc.utils.tasks.summarize import summarize_documents, wdocSummary
 
 logger.info("Starting wdoc")
 
@@ -668,7 +668,7 @@ def print_exception(exc_type, exc_value, exc_traceback):
                         self.interaction_settings
                     )
 
-    def summary_task(self) -> dict:
+    def summary_task(self) -> wdocSummary:
         docs_tkn_cost = {}
         for doc in self.loaded_docs:
             meta = doc.metadata["path"]

Original file line number	Diff line number	Diff line change
`@@ -83,7 +83,7 @@`
`83`	`83`	`semantic_batching,`
`84`	`84`	`sieve_documents,`
`85`	`85`	`)`
`86`		`-from wdoc.utils.tasks.summarize import summarize_documents`
	`86`	`+from wdoc.utils.tasks.summarize import summarize_documents, wdocSummary`
`87`	`87`
`88`	`88`	`logger.info("Starting wdoc")`
`89`	`89`
`@@ -668,7 +668,7 @@ def print_exception(exc_type, exc_value, exc_traceback):`
`668`	`668`	`self.interaction_settings`
`669`	`669`	`)`
`670`	`670`
`671`		`- def summary_task(self) -> dict:`
	`671`	`+ def summary_task(self) -> wdocSummary:`
`672`	`672`	`docs_tkn_cost = {}`
`673`	`673`	`for doc in self.loaded_docs:`
`674`	`674`	`meta = doc.metadata["path"]`