15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

chore: fix typo.
Some checks are pending
sync2gitee / repo-sync (push) Waiting to run
Details
Typos Check / Spell Check with Typos (push) Waiting to run
Details

This commit is contained in:
CaptainB 2024-12-20 12:33:08 +08:00
parent 77aa83d829
commit 6c7ae3f79f
2 changed files with 2 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/setting/models_provider/impl/azure_model_provider/model/embedding.py
View File

@ -16,14 +16,10 @@ from setting.models_provider.base_model_provider import MaxKBBaseModel
class AzureOpenAIEmbeddingModel(MaxKBBaseModel, AzureOpenAIEmbeddings):
@staticmethod
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
return AzureOpenAIEmbeddings(
return AzureOpenAIEmbeddingModel(
model=model_name,
openai_api_key=model_credential.get('api_key'),
azure_endpoint=model_credential.get('api_base'),
openai_api_version=model_credential.get('api_version'),
openai_api_type="azure",
)
def is_cache_model(self):
return True

5
apps/setting/models_provider/impl/gemini_model_provider/model/embedding.py
View File

@ -16,10 +16,7 @@ from setting.models_provider.base_model_provider import MaxKBBaseModel
class GeminiEmbeddingModel(MaxKBBaseModel, GoogleGenerativeAIEmbeddings):
@staticmethod
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
return GoogleGenerativeAIEmbeddings(
return GeminiEmbeddingModel(
google_api_key=model_credential.get('api_key'),
model=model_name,
)
def is_cache_model(self):
return False