Skip to content

Commit 53c5433

Browse files
import cache dir from logger instead of misc
1 parent 89aee7f commit 53c5433

2 files changed

Lines changed: 3 additions & 7 deletions

File tree

DocToolsLLM/DocToolsLLM.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"""
44

55
# import this first because it sets the logging level
6-
from .utils.logger import whi, yel, red, md_printer, log, set_docstring
6+
from .utils.logger import whi, yel, red, md_printer, log, set_docstring, config_dir, cache_dir
77

88
import sys
99
import faulthandler
@@ -27,7 +27,7 @@
2727
# cannot be lazy loaded because some are not callable but objects directly
2828
from .utils.misc import (
2929
ankiconnect, debug_chain, model_name_matcher,
30-
cache_dir, average_word_length, wpm, get_splitter,
30+
average_word_length, wpm, get_splitter,
3131
check_docs_tkn_length, get_tkn_length,
3232
extra_args_keys, disable_internet)
3333
from .utils.prompts import PR_CONDENSE_QUESTION, PR_EVALUATE_DOC, PR_ANSWER_ONE_DOC, PR_COMBINE_INTERMEDIATE_ANSWERS

DocToolsLLM/utils/misc.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111
import json
1212
import re
1313
from pathlib import Path
14-
from platformdirs import user_cache_dir
1514
from difflib import get_close_matches
1615
from bs4 import BeautifulSoup
1716
import hashlib
@@ -23,7 +22,7 @@
2322
from langchain_core.runnables import RunnableLambda
2423
from langchain_core.runnables import chain
2524

26-
from .logger import red, yel
25+
from .logger import red, yel, cache_dir
2726
from .typechecker import optional_typecheck
2827
from .flags import is_verbose
2928

@@ -55,9 +54,6 @@ def language_detector(text: str) -> float:
5554
def language_detector(text: str) -> None:
5655
return None
5756

58-
assert Path(user_cache_dir()).exists(), f"User cache dir not found: '{user_cache_dir()}'"
59-
cache_dir = (Path(user_cache_dir()) / "DocToolsLLM").resolve()
60-
cache_dir.mkdir(exist_ok=True)
6157
loaddoc_cache_dir = (cache_dir / "loaddoc_cache")
6258
loaddoc_cache_dir.mkdir(exist_ok=True)
6359
loaddoc_cache = Memory(loaddoc_cache_dir, verbose=0)

0 commit comments

Comments
 (0)