LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-29 08:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

(feat:storage) use get storage

Browse Source
This commit is contained in:
ManishMadan2882
2025-04-22 01:41:53 +05:30
parent 38476cfeb8
commit 0a31ddaae6
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
application/llm/google_ai.py
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):
"""

2
application/llm/openai.py
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 = []