Merge pull request #2 from oxylabs/remove-cached-file

remove cached file
oxylabs · Mar 13, 2024 · 3bf623f · 3bf623f
2 parents 861c38e + 4ca092c
commit 3bf623f
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/_oxyparser_cache_oxylabs b/_oxyparser_cache_oxylabs
diff --git a/oxyparser/ai/query.py b/oxyparser/ai/query.py
@@ -4,9 +4,9 @@
 import time
 
 import structlog
+from litellm import acompletion
 from openai import BadRequestError, RateLimitError
 from tenacity import retry, stop_after_attempt, wait_exponential
-from litellm import acompletion
 
 from oxyparser.settings import settings
 
@@ -20,7 +20,7 @@
 def extract_json_from_ai_response(response: str) -> dict[str, str]:
     pattern = r"{(.*?)}"
     matched_items = re.findall(pattern, response, re.DOTALL)
-    item = f"".join(matched_items)
+    item = "".join(matched_items)
     return json.loads(f"{{{item}}}") if item else {}
 
 

diff --git a/oxyparser/cache/utils.py b/oxyparser/cache/utils.py
@@ -1,10 +1,11 @@
 from typing import Literal
 
+from oxyparser.cache.base import BaseCache
 from oxyparser.cache.file import FileCache
 from oxyparser.cache.redis import RedisCache
 
 
-def get_cache(cache_type: Literal["file", "redis"] | None) -> FileCache | RedisCache | None:
+def get_cache(cache_type: Literal["file", "redis"] | None) -> BaseCache:
     mapping = {"file": FileCache, "redis": RedisCache}
 
     cache_cls = mapping.get(str(cache_type))

diff --git a/oxyparser/oxyparser.py b/oxyparser/oxyparser.py
@@ -1,4 +1,4 @@
-from typing import Any, Type, Literal
+from typing import Any, Literal, Type
 
 import structlog
 from pydantic import BaseModel, ValidationError