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
|