Update app.py

app.py

@@ -4,12 +4,20 @@ import gradio as gr
 from langchain_community.utilities.sql_database import SQLDatabase
 from langchain_community.agent_toolkits import create_sql_agent
 
-from langchain_openai import ChatOpenAI
+from langchain_openai import ChatOpenaI, AzureChatOpenAI
 
 ccms_db_loc = 'ccms.db'
 
 ccms_db = SQLDatabase.from_uri(f"sqlite:///{ccms_db_loc}")
 
+gpt4o_azure = AzureChatOpenAI(
+    model_name='gpt-4o',
+    api_key=os.environ["AZURE_OPENAI_KEY"],
+    azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
+    api_version="2024-02-01",
+    temperature=0
+)
+
 gpt4o = ChatOpenAI(
     model_name='gpt-4o',
     api_key=os.environ["OPENAI_API_KEY"],
@@ -23,7 +31,7 @@ gpt35 = ChatOpenAI(
 )
 
 sqlite_agent = create_sql_agent(
-    gpt4o,
+    gpt4o_azure,
     db=ccms_db,
     agent_type="openai-tools",
     agent_executor_kwargs={'handle_parsing_errors':True},