SysModeler commited on
Commit
7e38020
·
verified ·
1 Parent(s): 60ffbab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -10
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
- print(f"Using embedding deployment: {AZURE_OPENAI_EMBEDDING_DEPLOYMENT}")
26
- print(f"Using LLM deployment: {AZURE_OPENAI_LLM_DEPLOYMENT}")
27
 
28
  # Suppress warnings
29
  warnings.filterwarnings("ignore")
30
 
31
- # Initialize embedding model
32
  embeddings = AzureOpenAIEmbeddings(
33
- azure_deployment=AZURE_OPENAI_EMBEDDING_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