Kevin Hu
commited on
Commit
·
f711e2a
1
Parent(s):
fe03167
move import positions (#2753)
Browse files### What problem does this PR solve?
### Type of change
- [x] Refactoring
rag/llm/embedding_model.py
CHANGED
@@ -18,7 +18,6 @@ from typing import Optional
|
|
18 |
import threading
|
19 |
import requests
|
20 |
from huggingface_hub import snapshot_download
|
21 |
-
from openai.lib.azure import AzureOpenAI
|
22 |
from zhipuai import ZhipuAI
|
23 |
import os
|
24 |
from abc import ABC
|
@@ -137,6 +136,7 @@ class LocalAIEmbed(Base):
|
|
137 |
|
138 |
class AzureEmbed(OpenAIEmbed):
|
139 |
def __init__(self, key, model_name, **kwargs):
|
|
|
140 |
self.client = AzureOpenAI(api_key=key, azure_endpoint=kwargs["base_url"], api_version="2024-02-01")
|
141 |
self.model_name = model_name
|
142 |
|
|
|
18 |
import threading
|
19 |
import requests
|
20 |
from huggingface_hub import snapshot_download
|
|
|
21 |
from zhipuai import ZhipuAI
|
22 |
import os
|
23 |
from abc import ABC
|
|
|
136 |
|
137 |
class AzureEmbed(OpenAIEmbed):
|
138 |
def __init__(self, key, model_name, **kwargs):
|
139 |
+
from openai.lib.azure import AzureOpenAI
|
140 |
self.client = AzureOpenAI(api_key=key, azure_endpoint=kwargs["base_url"], api_version="2024-02-01")
|
141 |
self.model_name = model_name
|
142 |
|