(feat:storage) use get storage

This commit is contained in:
ManishMadan2882 committed 2025-04-22 01:41:53 +05:30
1 parent 38476cfeb8
commit 0a31ddaae6
2 files changed
+2 -3

No files matched your search

+1 -2
View File
@@ -5,7 +5,6 @@ import json
from application.llm.base import BaseLLM
from application.storage.storage_creator import StorageCreator
from application.core.settings import settings
class GoogleLLM(BaseLLM):
@@ -14,7 +13,7 @@ class GoogleLLM(BaseLLM):
self.api_key = api_key
self.user_api_key = user_api_key
self.client = genai.Client(api_key=self.api_key)
self.storage = StorageCreator.create_storage(getattr(settings, "STORAGE_TYPE", "local"))
self.storage = StorageCreator.get_storage()
def get_supported_attachment_types(self):
"""
+1 -1
View File
@@ -19,7 +19,7 @@ class OpenAILLM(BaseLLM):
self.client = OpenAI(api_key=api_key)
self.api_key = api_key
self.user_api_key = user_api_key
self.storage = StorageCreator.create_storage(getattr(settings, "STORAGE_TYPE", "local"))
self.storage = StorageCreator.get_storage()
def _clean_messages_openai(self, messages):
cleaned_messages = []