Skip to content

Commit 0a31dda

Browse files
(feat:storage) use get storage
1 parent 38476cf commit 0a31dda

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

application/llm/google_ai.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55

66
from application.llm.base import BaseLLM
77
from application.storage.storage_creator import StorageCreator
8-
from application.core.settings import settings
98

109

1110
class GoogleLLM(BaseLLM):
@@ -14,7 +13,7 @@ def __init__(self, api_key=None, user_api_key=None, *args, **kwargs):
1413
self.api_key = api_key
1514
self.user_api_key = user_api_key
1615
self.client = genai.Client(api_key=self.api_key)
17-
self.storage = StorageCreator.create_storage(getattr(settings, "STORAGE_TYPE", "local"))
16+
self.storage = StorageCreator.get_storage()
1817

1918
def get_supported_attachment_types(self):
2019
"""

application/llm/openai.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def __init__(self, api_key=None, user_api_key=None, *args, **kwargs):
1919
self.client = OpenAI(api_key=api_key)
2020
self.api_key = api_key
2121
self.user_api_key = user_api_key
22-
self.storage = StorageCreator.create_storage(getattr(settings, "STORAGE_TYPE", "local"))
22+
self.storage = StorageCreator.get_storage()
2323

2424
def _clean_messages_openai(self, messages):
2525
cleaned_messages = []

0 commit comments

Comments
 (0)