Default azure credential auth is made as the default

Author: Prajwal-Microsoft

AzureFunctions/km-charts-function/function_app.py

@@ -21,17 +21,6 @@ def get_db_connection():
     password = os.environ.get("SQLDB_PASSWORD")
 
     # Attempt connection using Username & Password
-    try:
-        conn = pyodbc.connect(
-            f"DRIVER={driver};SERVER={server};DATABASE={database};UID={username};PWD={password}",
-            timeout=5
-        )
-        logging.info("Connected using Username & Password")
-        return conn
-    except pyodbc.Error as e:
-        print(f"Failed with Username & Password: {str(e)}")
-
-    # If first attempt fails, try Azure Default Credential
     try:
         credential = DefaultAzureCredential()
 
@@ -50,10 +39,16 @@ def get_db_connection():
         )
 
         logging.info("Connected using Default Azure Credential")
+
         return conn
-    except Exception as e:
+    except pyodbc.Error as e:
         logging.error(f"Failed with Default Credential: {str(e)}")
-        return None  # Return None if both attempts fail
+        conn = pyodbc.connect(
+            f"DRIVER={driver};SERVER={server};DATABASE={database};UID={username};PWD={password}",
+            timeout=5
+        )
+        logging.info("Connected using Username & Password")
+        return conn
 
 # add post methods - filters will come in the body (request.body), if body is not empty, update the where clause in the query
 @app.route(route="get_metrics", methods=["GET","POST"], auth_level=func.AuthLevel.ANONYMOUS)

AzureFunctions/km-rag-function/function_app.py

@@ -49,17 +49,6 @@ def get_db_connection():
     password = os.environ.get("SQLDB_PASSWORD")
 
     # Attempt connection using Username & Password
-    try:
-        conn = pyodbc.connect(
-            f"DRIVER={driver};SERVER={server};DATABASE={database};UID={username};PWD={password}",
-            timeout=5
-        )
-        logging.info("Connected using Username & Password")
-        return conn
-    except pyodbc.Error as e:
-        print(f"Failed with Username & Password: {str(e)}")
-
-    # If first attempt fails, try Azure Default Credential
     try:
         credential = DefaultAzureCredential()
 
@@ -78,10 +67,17 @@ def get_db_connection():
         )
 
         logging.info("Connected using Default Azure Credential")
+
         return conn
-    except Exception as e:
+    except pyodbc.Error as e:
         logging.error(f"Failed with Default Credential: {str(e)}")
-        return None  # Return None if both attempts fail
+        conn = pyodbc.connect(
+            f"DRIVER={driver};SERVER={server};DATABASE={database};UID={username};PWD={password}",
+            timeout=5
+        )
+        
+        logging.info("Connected using Username & Password")
+        return conn
 
 class ChatWithDataPlugin:
     @kernel_function(name="Greeting", description="Respond to any greeting or general questions")