fix forward reference type hints

Signed-off-by: thiswillbeyourgithub
<26625900+thiswillbeyourgithub@users.noreply.github.com>
Signed-off-by: thiswillbeyourgithub <26625900+thiswillbeyourgithub@users.noreply.github.com>

Author: thiswillbeyourgithub

wdoc/utils/tasks/query.py

@@ -556,15 +556,21 @@ def semantic_batching(
 
 
 def pbar_chain(
-    llm: Union["langchain_litellm.ChatLiteLLM", "FakeListChatModel"],
+    llm: Union[
+        "langchain_litellm.ChatLiteLLM",
+        "langchain_community.chat_models.fake.FakeListChatModel",
+    ],
     len_func: str,
     **tqdm_kwargs,
 ) -> RunnableLambda:
     "create a chain that just sets a tqdm progress bar"
 
     def actual_pbar_chain(
         inputs: Union[dict, List],
-        llm: Union["langchain_litellm.ChatLiteLLM", "FakeListChatModel"] = llm,
+        llm: Union[
+            "langchain_litellm.ChatLiteLLM",
+            "langchain_community.chat_models.fake.FakeListChatModel",
+        ] = llm,
     ) -> Union[dict, List]:
 
         llm.callbacks[0].pbar.append(
@@ -584,13 +590,19 @@ def actual_pbar_chain(
 
 
 def pbar_closer(
-    llm: Union["langchain_litellm.ChatLiteLLM", "FakeListChatModel"],
+    llm: Union[
+        "langchain_litellm.ChatLiteLLM",
+        "langchain_community.chat_models.fake.FakeListChatModel",
+    ],
 ) -> RunnableLambda:
     "close a pbar created by pbar_chain"
 
     def actual_pbar_closer(
         inputs: Union[dict, List],
-        llm: Union["langchain_litellm.ChatLiteLLM", "FakeListChatModel"] = llm,
+        llm: Union[
+            "langchain_litellm.ChatLiteLLM",
+            "langchain_community.chat_models.fake.FakeListChatModel",
+        ] = llm,
     ) -> Union[dict, List]:
         pbar = llm.callbacks[0].pbar[-1]
         pbar.update(pbar.total - pbar.n)

wdoc/wdoc.py

@@ -647,7 +647,7 @@ def print_exception(exc_type, exc_value, exc_traceback):
                         self.interaction_settings
                     )
 
-    def summary_task(self) -> "wdocSummary":
+    def summary_task(self) -> "wdoc.utils.tasks.summarize.wdocSummary":
         from wdoc.utils.tasks.summarize import summarize_documents
 
         docs_tkn_cost = {}