Pārlūkot izejas kodu

move import positions (#2753)

### What problem does this PR solve?

### Type of change

- [x] Refactoring
tags/v0.13.0
Kevin Hu pirms 1 gada
vecāks
revīzija
7f44cf543a
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. 1
    1
      rag/llm/embedding_model.py

+ 1
- 1
rag/llm/embedding_model.py Parādīt failu

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


Notiek ielāde…
Atcelt
Saglabāt