Remove AzureOpenAI references and update imports to use only llm_openai

basic_llama_agent.py

@@ -1,7 +1,7 @@
 
 from llama_index.core.agent.workflow import FunctionAgent
 
-from llms import llm_azure_openai, llm_openai
+from llms import llm_openai
 from tools import (generate_implications, get_lead_up_events, get_news,
                    get_social_media_opinions)
 

llms.py

@@ -1,18 +1,7 @@
 import os
 
 from llama_index.llms.openai import OpenAI
-from llama_index.llms.azure_openai import AzureOpenAI
 
-# llm_azure_openai = AzureOpenAI(
-#     model="gpt-4o-mini",
-#     engine=os.environ.get("AZURE_OPENAI_MODEL"),
-#     temperature=0.0,
-#     azure_endpoint=os.environ.get("AZURE_OPENAI_ENDPOINT"),
-#     api_key=os.environ.get("AZURE_OPENAI_API_KEY"),
-#     api_version=os.environ.get("AZURE_OPENAI_API_VERSION"),
-# )
-
-llm_azure_openai = None
 
 llm_openai = OpenAI(
     model="gpt-4o-mini",

tools.py

@@ -5,7 +5,7 @@ from typing import Dict, List
 import requests
 from llama_index.core.llms import ChatMessage
 
-from llms import llm_openai as llm_azure_openai
+from llms import llm_openai
 
 
 last_news = []
@@ -76,7 +76,7 @@ def analyze_sentiment(text: str) -> str:
     question: Tell me if the sentiment of this news is positive, negative, or neutral? 
     context: {text}"""
     try:
-        result = llm_azure_openai.chat(
+        result = llm_openai.chat(
             messages=[ChatMessage(role="user", content=prompt)]
         )
     except Exception as e:
@@ -105,7 +105,7 @@ def recognize_entities(text: str) -> List[str]:
     question: Tell me entities mentioned in this news? 
     context: {text}"""
     try:
-        result = llm_azure_openai.chat(
+        result = llm_openai.chat(
             messages=[ChatMessage(role="user", content=prompt)]
         )
     except Exception as e:
@@ -132,7 +132,7 @@ def generate_implications(article_index: int) -> str:
     summary = article["summary"]
     prompt = f"question: What are the possible implications of this news? context: {summary}"
     try:
-        result = llm_azure_openai.chat(
+        result = llm_openai.chat(
             messages=[ChatMessage(role="user", content=prompt)]
         )
     except Exception as e:
@@ -206,7 +206,7 @@ def get_lead_up_events(article_index: int) -> str:
     </historical_context>
     """
     try:
-        result = llm_azure_openai.chat(
+        result = llm_openai.chat(
             messages=[ChatMessage(role="user", content=prompt)]
         )
     except Exception as e: