|
|
|
|
|
|
|
|
return np.array(res.data[0].embedding), res.usage.total_tokens |
|
|
return np.array(res.data[0].embedding), res.usage.total_tokens |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class AzureEmbed(Base): |
|
|
|
|
|
|
|
|
class AzureEmbed(OpenAIEmbed): |
|
|
def __init__(self, key, model_name, **kwargs): |
|
|
def __init__(self, key, model_name, **kwargs): |
|
|
self.client = AzureOpenAI(api_key=key, azure_endpoint=kwargs["base_url"], api_version="2024-02-01") |
|
|
self.client = AzureOpenAI(api_key=key, azure_endpoint=kwargs["base_url"], api_version="2024-02-01") |
|
|
self.model_name = model_name |
|
|
self.model_name = model_name |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BaiChuanEmbed(OpenAIEmbed): |
|
|
class BaiChuanEmbed(OpenAIEmbed): |
|
|
def __init__(self, key, |
|
|
def __init__(self, key, |
|
|
model_name='Baichuan-Text-Embedding', |
|
|
model_name='Baichuan-Text-Embedding', |