fix: import type hints from beartype

Signed-off-by: thiswillbeyourgithub
<[email protected]>

fix: import typing from beartype

Signed-off-by: thiswillbeyourgithub
<[email protected]>

Author: thiswillbeyourgithub

wdoc/utils/batch_file_loader.py

@@ -15,7 +15,7 @@
 from tqdm import tqdm
 from functools import cache as memoizer
 import time
-from typing import List, Tuple, Union, Optional
+from beartype.typing import List, Tuple, Union, Optional
 import random
 from multiprocessing.context import TimeoutError as MultiprocessTimeoutError
 

wdoc/utils/customs/compressed_embeddings_cache.py

@@ -4,7 +4,7 @@
 This is basically the exact same code but with added compression
 """
 from pathlib import Path
-from typing import Iterator, List, Optional, Sequence, Tuple, Union
+from beartype.typing import Iterator, List, Optional, Sequence, Tuple, Union
 
 from langchain_core.stores import ByteStore
 

wdoc/utils/customs/fix_llm_caching.py

@@ -9,7 +9,7 @@
 
 import json
 from pathlib import Path
-from typing import Union, Any, Optional, Generator
+from beartype.typing import Union, Any, Optional, Generator
 
 from langchain_core.caches import BaseCache
 

wdoc/utils/embeddings.py

@@ -3,7 +3,7 @@
 * Loads and store embeddings for each document.
 """
 
-from typing import List, Union, Optional, Any, Tuple, Callable
+from beartype.typing import List, Union, Optional, Any, Tuple, Callable
 # import math
 import hashlib
 import os

wdoc/utils/env.py

@@ -6,7 +6,7 @@
 
 import sys
 import os
-from typing import Optional, Union, Literal
+from beartype.typing import Optional, Union, Literal
 from beartype import beartype, BeartypeConf
 from beartype.door import is_bearable
 

wdoc/utils/interact.py

@@ -2,7 +2,7 @@
 Code related to the prompt (in the sense of "directly ask the user a question")
 """
 
-from typing import Optional, Tuple, Any
+from beartype.typing import Optional, Tuple, Any
 import time
 from pathlib import Path
 import json

wdoc/utils/llm.py

@@ -3,7 +3,7 @@
 counting callback.
 """
 
-from typing import Union, List, Any, Optional, Dict
+from beartype.typing import Union, List, Any, Optional, Dict
 import os
 
 from langchain_core.callbacks import BaseCallbackHandler

wdoc/utils/loaders.py

@@ -7,7 +7,7 @@
 import sys
 import os
 import time
-from typing import List, Union, Any, Optional, Callable, Dict, Tuple
+from beartype.typing import List, Union, Any, Optional, Callable, Dict, Tuple
 import signal
 from contextlib import contextmanager
 import traceback

wdoc/utils/logger.py

@@ -8,7 +8,7 @@
 from tqdm import tqdm
 from loguru import logger
 from pathlib import Path
-from typing import Type, Callable, Optional, Union, List, Dict
+from beartype.typing import Type, Callable, Optional, Union, List, Dict
 from rich.markdown import Markdown
 from rich.console import Console
 from platformdirs import user_cache_dir, user_log_dir

wdoc/utils/misc.py

@@ -3,7 +3,7 @@
 """
 
 import sys
-from typing import List, Union, Callable, get_type_hints, Literal
+from beartype.typing import List, Union, Callable, get_type_hints, Literal
 from joblib import Memory
 from joblib import hash as jhash
 import uuid