Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -5,8 +5,8 @@ from dotenv import load_dotenv
|
|
5 |
|
6 |
from langchain.chains import ConversationalRetrievalChain
|
7 |
from langchain_community.vectorstores import FAISS
|
8 |
-
from langchain_community.chat_models import AzureChatOpenAI
|
9 |
from langchain_community.embeddings import AzureOpenAIEmbeddings
|
|
|
10 |
|
11 |
# Patch Gradio bug (schema parsing issue)
|
12 |
import gradio_client.utils
|
@@ -17,25 +17,20 @@ load_dotenv()
|
|
17 |
AZURE_OPENAI_API_KEY = os.getenv("AZURE_OPENAI_API_KEY")
|
18 |
AZURE_OPENAI_ENDPOINT = os.getenv("AZURE_OPENAI_ENDPOINT")
|
19 |
AZURE_OPENAI_LLM_DEPLOYMENT = os.getenv("AZURE_OPENAI_LLM_DEPLOYMENT")
|
20 |
-
AZURE_OPENAI_EMBEDDING_DEPLOYMENT = os.getenv("AZURE_OPENAI_EMBEDDING_DEPLOYMENT")
|
21 |
-
|
22 |
-
if not all([AZURE_OPENAI_API_KEY, AZURE_OPENAI_ENDPOINT, AZURE_OPENAI_LLM_DEPLOYMENT, AZURE_OPENAI_EMBEDDING_DEPLOYMENT]):
|
23 |
-
raise ValueError("Azure OpenAI environment variables missing.")
|
24 |
|
25 |
-
|
26 |
-
|
27 |
|
28 |
# Suppress warnings
|
29 |
warnings.filterwarnings("ignore")
|
30 |
|
31 |
-
# Initialize embedding model
|
32 |
embeddings = AzureOpenAIEmbeddings(
|
33 |
-
azure_deployment=
|
34 |
azure_endpoint=AZURE_OPENAI_ENDPOINT,
|
35 |
openai_api_key=AZURE_OPENAI_API_KEY,
|
36 |
openai_api_version="2024-08-01-preview",
|
37 |
chunk_size=1000
|
38 |
-
#validate_base_url=False
|
39 |
)
|
40 |
|
41 |
# Load FAISS vector store
|
|
|
5 |
|
6 |
from langchain.chains import ConversationalRetrievalChain
|
7 |
from langchain_community.vectorstores import FAISS
|
|
|
8 |
from langchain_community.embeddings import AzureOpenAIEmbeddings
|
9 |
+
from langchain_community.chat_models import AzureChatOpenAI
|
10 |
|
11 |
# Patch Gradio bug (schema parsing issue)
|
12 |
import gradio_client.utils
|
|
|
17 |
AZURE_OPENAI_API_KEY = os.getenv("AZURE_OPENAI_API_KEY")
|
18 |
AZURE_OPENAI_ENDPOINT = os.getenv("AZURE_OPENAI_ENDPOINT")
|
19 |
AZURE_OPENAI_LLM_DEPLOYMENT = os.getenv("AZURE_OPENAI_LLM_DEPLOYMENT")
|
|
|
|
|
|
|
|
|
20 |
|
21 |
+
if not all([AZURE_OPENAI_API_KEY, AZURE_OPENAI_ENDPOINT, AZURE_OPENAI_LLM_DEPLOYMENT]):
|
22 |
+
raise ValueError("Azure OpenAI environment variables are missing.")
|
23 |
|
24 |
# Suppress warnings
|
25 |
warnings.filterwarnings("ignore")
|
26 |
|
27 |
+
# Initialize embedding model (must be deployed separately)
|
28 |
embeddings = AzureOpenAIEmbeddings(
|
29 |
+
azure_deployment="text-embedding-ada-002", # make sure this deployment exists
|
30 |
azure_endpoint=AZURE_OPENAI_ENDPOINT,
|
31 |
openai_api_key=AZURE_OPENAI_API_KEY,
|
32 |
openai_api_version="2024-08-01-preview",
|
33 |
chunk_size=1000
|
|
|
34 |
)
|
35 |
|
36 |
# Load FAISS vector store
|