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 |
|