Skip to content

Commit

Permalink
fix: 修复"openai_api_type=azure"时,知识库索引构建失败的问题 (#1158)
Browse files Browse the repository at this point in the history
  • Loading branch information
ZhengQinyu authored Sep 11, 2024
1 parent 1fe23b7 commit ce53e71
Showing 1 changed file with 3 additions and 9 deletions.
12 changes: 3 additions & 9 deletions modules/index_func.py
Original file line number Diff line number Diff line change
@@ -1,16 +1,10 @@
import hashlib
import logging
import os

import PyPDF2
from langchain_community.chat_models import ChatOpenAI
from langchain_community.embeddings.huggingface import HuggingFaceEmbeddings
from langchain_community.vectorstores import FAISS
from langchain_openai import OpenAIEmbeddings
from langchain_openai import OpenAIEmbeddings, AzureOpenAIEmbeddings
from tqdm import tqdm

from modules.config import local_embedding
from modules.presets import *
from modules.utils import *


Expand Down Expand Up @@ -144,11 +138,11 @@ def construct_index(
model="text-embedding-3-large",
)
else:
embeddings = OpenAIEmbeddings(
embeddings = AzureOpenAIEmbeddings(
deployment=os.environ["AZURE_EMBEDDING_DEPLOYMENT_NAME"],
openai_api_key=os.environ["AZURE_OPENAI_API_KEY"],
model=os.environ["AZURE_EMBEDDING_MODEL_NAME"],
openai_api_base=os.environ["AZURE_OPENAI_API_BASE_URL"],
azure_endpoint=os.environ["AZURE_OPENAI_API_BASE_URL"],
openai_api_type="azure",
)
if os.path.exists(index_path) and load_from_cache_if_possible:
Expand Down

0 comments on commit ce53e71

Please sign in to comment.