diff --git a/rag/llm/embedding_model.py b/rag/llm/embedding_model.py index 39704590a..ba73cdfba 100644 --- a/rag/llm/embedding_model.py +++ b/rag/llm/embedding_model.py @@ -18,7 +18,6 @@ from typing import Optional import threading import requests from huggingface_hub import snapshot_download -from openai.lib.azure import AzureOpenAI from zhipuai import ZhipuAI import os from abc import ABC @@ -137,6 +136,7 @@ class LocalAIEmbed(Base): class AzureEmbed(OpenAIEmbed): def __init__(self, key, model_name, **kwargs): + from openai.lib.azure import AzureOpenAI self.client = AzureOpenAI(api_key=key, azure_endpoint=kwargs["base_url"], api_version="2024-02-01") self.model_name = model_name