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

sagemaker + llm creator class

Browse Source
This commit is contained in:
Alex
2023-09-29 01:09:01 +01:00
parent c1c54f4848
commit e4be38b9f7
5 changed files with 57 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
application/llm/openai.py
View File

@@ -1,4 +1,5 @@
from application.llm.base import BaseLLM
from application.core.settings import settings
class OpenAILLM(BaseLLM):
@@ -44,9 +45,9 @@ class AzureOpenAILLM(OpenAILLM):
def __init__(self, openai_api_key, openai_api_base, openai_api_version, deployment_name):
super().__init__(openai_api_key)
self.api_base = openai_api_base
self.api_version = openai_api_version
self.deployment_name = deployment_name
self.api_base = settings.OPENAI_API_BASE,
self.api_version = settings.OPENAI_API_VERSION,
self.deployment_name = settings.AZURE_DEPLOYMENT_NAME,
def _get_openai(self):
openai = super()._get_openai()