@@ -1,3 +1,4 @@ | |||
import copy | |||
import logging | |||
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: | |||
for ai_model_entity in LLM_BASE_MODELS: | |||
if ai_model_entity.base_model_name == base_model_name: | |||
ai_model_entity.entity.model = model | |||
ai_model_entity.entity.label.en_US = model | |||
ai_model_entity.entity.label.zh_Hans = model | |||
return ai_model_entity | |||
ai_model_entity_copy = copy.deepcopy(ai_model_entity) | |||
ai_model_entity_copy.entity.model = model | |||
ai_model_entity_copy.entity.label.en_US = model | |||
ai_model_entity_copy.entity.label.zh_Hans = model | |||
return ai_model_entity_copy | |||
return None |
@@ -1,4 +1,5 @@ | |||
import base64 | |||
import copy | |||
import time | |||
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: | |||
for ai_model_entity in EMBEDDING_BASE_MODELS: | |||
if ai_model_entity.base_model_name == base_model_name: | |||
ai_model_entity.entity.model = model | |||
ai_model_entity.entity.label.en_US = model | |||
ai_model_entity.entity.label.zh_Hans = model | |||
return ai_model_entity | |||
ai_model_entity_copy = copy.deepcopy(ai_model_entity) | |||
ai_model_entity_copy.entity.model = model | |||
ai_model_entity_copy.entity.label.en_US = model | |||
ai_model_entity_copy.entity.label.zh_Hans = model | |||
return ai_model_entity_copy | |||
return None |