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

Files changed (1) hide show
  1. rag/llm/embedding_model.py +1 -1
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