hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-21 00:36:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: azure customize model name duplicate (#2073)

Browse Source
This commit is contained in:
takatost
2024-01-17 21:17:59 +08:00
committed by GitHub
parent 8799c888e3
commit 1d91535ba6
2 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/core/model_runtime/model_providers/azure_openai/llm/llm.py
View File

@@ -1,3 +1,4 @@
import copy
import logging import logging
from typing import Generator, List, Optional, Union, cast from typing import Generator, List, Optional, Union, cast
@@ -625,9 +626,10 @@ class AzureOpenAILargeLanguageModel(_CommonAzureOpenAI, LargeLanguageModel):
def _get_ai_model_entity(base_model_name: str, model: str) -> AzureBaseModel: def _get_ai_model_entity(base_model_name: str, model: str) -> AzureBaseModel:
for ai_model_entity in LLM_BASE_MODELS: for ai_model_entity in LLM_BASE_MODELS:
if ai_model_entity.base_model_name == base_model_name: if ai_model_entity.base_model_name == base_model_name:
ai_model_entity.entity.model = model ai_model_entity_copy = copy.deepcopy(ai_model_entity)
ai_model_entity.entity.label.en_US = model ai_model_entity_copy.entity.model = model
ai_model_entity.entity.label.zh_Hans = model ai_model_entity_copy.entity.label.en_US = model
return ai_model_entity ai_model_entity_copy.entity.label.zh_Hans = model
return ai_model_entity_copy
return None return None

10
api/core/model_runtime/model_providers/azure_openai/text_embedding/text_embedding.py
View File

@@ -1,4 +1,5 @@
import base64 import base64
import copy
import time import time
from typing import Optional, Tuple from typing import Optional, Tuple
@@ -186,9 +187,10 @@ class AzureOpenAITextEmbeddingModel(_CommonAzureOpenAI, TextEmbeddingModel):
def _get_ai_model_entity(base_model_name: str, model: str) -> AzureBaseModel: def _get_ai_model_entity(base_model_name: str, model: str) -> AzureBaseModel:
for ai_model_entity in EMBEDDING_BASE_MODELS: for ai_model_entity in EMBEDDING_BASE_MODELS:
if ai_model_entity.base_model_name == base_model_name: if ai_model_entity.base_model_name == base_model_name:
ai_model_entity.entity.model = model ai_model_entity_copy = copy.deepcopy(ai_model_entity)
ai_model_entity.entity.label.en_US = model ai_model_entity_copy.entity.model = model
ai_model_entity.entity.label.zh_Hans = model ai_model_entity_copy.entity.label.en_US = model
return ai_model_entity ai_model_entity_copy.entity.label.zh_Hans = model
return ai_model_entity_copy
return None return None