zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: fix typo.

Browse Source
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): class AzureOpenAIEmbeddingModel(MaxKBBaseModel, AzureOpenAIEmbeddings):
@staticmethod @staticmethod
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs): def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
return AzureOpenAIEmbeddings( return AzureOpenAIEmbeddingModel(
model=model_name, model=model_name,
openai_api_key=model_credential.get('api_key'), openai_api_key=model_credential.get('api_key'),
azure_endpoint=model_credential.get('api_base'), azure_endpoint=model_credential.get('api_base'),
openai_api_version=model_credential.get('api_version'), openai_api_version=model_credential.get('api_version'),
openai_api_type="azure", 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): class GeminiEmbeddingModel(MaxKBBaseModel, GoogleGenerativeAIEmbeddings):
@staticmethod @staticmethod
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs): 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'), google_api_key=model_credential.get('api_key'),
model=model_name, model=model_name,
) )
def is_cache_model(self):
return False