Procházet zdrojové kódy

move import positions (#2753)

### What problem does this PR solve?

### Type of change

- [x] Refactoring
tags/v0.13.0
Kevin Hu před 1 rokem
rodič
revize
7f44cf543a
Žádný účet není propojen s e-mailovou adresou tvůrce revize
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1
    1
      rag/llm/embedding_model.py

+ 1
- 1
rag/llm/embedding_model.py Zobrazit soubor

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


Načítá se…
Zrušit
Uložit