Shifted from openai to azureopenai

src/llm.py

@@ -7,7 +7,7 @@ from typing import Literal
 
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.pydantic_v1 import BaseModel, Field
-from langchain_openai import ChatOpenAI
+from langchain_openai import AzureChatOpenAI
 
 #%%
 # Data model
@@ -21,7 +21,7 @@ class RouteQuery(BaseModel):
 
 
 # LLM with function call
-llm = ChatOpenAI(model="gpt-4o-mini", temperature=0.3)
+llm = AzureChatOpenAI(model="gpt-4o-mini", temperature=0.3)
 structured_llm_router = llm.with_structured_output(RouteQuery)
 
 #%%
@@ -60,7 +60,7 @@ class GradeDocuments(BaseModel):
 
 #%%
 # LLM with function call
-llm = ChatOpenAI(model="gpt-4o-mini", temperature=0.3)
+llm = AzureChatOpenAI(model="gpt-4o-mini", temperature=0.3)
 structured_llm_grader = llm.with_structured_output(GradeDocuments)
 
 # Prompt
@@ -90,7 +90,7 @@ from langchain_core.output_parsers import StrOutputParser
 prompt = hub.pull("rlm/rag-prompt")
 
 # LLM
-llm = ChatOpenAI(model_name="gpt-4o-mini", temperature=0.3)
+llm = AzureChatOpenAI(model_name="gpt-4o-mini", temperature=0.3)
 
 
 # Post-processing
@@ -120,7 +120,7 @@ class GradeHallucinations(BaseModel):
 
 
 # LLM with function call
-llm = ChatOpenAI(model="gpt-4o-mini", temperature=0.3)
+llm = AzureChatOpenAI(model="gpt-4o-mini", temperature=0.3)
 structured_llm_grader = llm.with_structured_output(GradeHallucinations)
 
 # Prompt
@@ -150,7 +150,7 @@ class GradeAnswer(BaseModel):
 
 
 # LLM with function call
-llm = ChatOpenAI(model="gpt-4o-mini", temperature=0.3)
+llm = AzureChatOpenAI(model="gpt-4o-mini", temperature=0.3)
 structured_llm_grader = llm.with_structured_output(GradeAnswer)
 
 # Prompt
@@ -170,7 +170,7 @@ answer_grader.invoke({"question": question, "generation": generation})
 ### Question Re-writer
 
 # LLM
-llm = ChatOpenAI(model="gpt-4o-mini", temperature=0.3)
+llm = AzureChatOpenAI(model="gpt-4o-mini", temperature=0.3)
 
 # Prompt
 system = """You a question re-writer that converts an input question to a better version that is optimized \n 

vectordb/08d73b15-e800-45c5-a450-5b9d696166f3/length.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:71bde5af77b7e5e53efccdfbabf3da1e125a2ebfdc944d9b9ed18b04ecf527fe
+oid sha256:36145df1efa1a3bfba46c79e2b76437f2fd83cbfea85af22e6d7bebd2958e995
 size 4000