From d760022ebdae946c738aa01037a53ce5abaf9744 Mon Sep 17 00:00:00 2001
From: guidooswald <guidooswald>
Date: Tue, 2 Sep 2025 09:33:35 +0000
Subject: [PATCH 1/3] Claude-3-7 model is not available in free edition

...and deprecated
---
 product_demos/Data-Science/ai-agent/config.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/product_demos/Data-Science/ai-agent/config.py b/product_demos/Data-Science/ai-agent/config.py
index b0922c39..dc24c228 100644
--- a/product_demos/Data-Science/ai-agent/config.py
+++ b/product_demos/Data-Science/ai-agent/config.py
@@ -22,4 +22,4 @@
 ENDPOINT_NAME = f'{MODEL_NAME}_{catalog}_{db}'[:60]
 
 # This must be a tool-enabled model
-LLM_ENDPOINT_NAME = 'databricks-claude-3-7-sonnet'
+LLM_ENDPOINT_NAME = 'databricks-llama-4-maverick'

From 2d89226ffd813b50828b5925b1d7794e2edb3809 Mon Sep 17 00:00:00 2001
From: guidooswald <guidooswald>
Date: Tue, 2 Sep 2025 14:05:40 +0000
Subject: [PATCH 2/3] Eval dataset creation also uses claude

not available in free edition and deprecated...
Removing the display command saves some execution time
---
 .../ai-agent/_resources/04-eval-dataset-generation.py        | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/product_demos/Data-Science/ai-agent/_resources/04-eval-dataset-generation.py b/product_demos/Data-Science/ai-agent/_resources/04-eval-dataset-generation.py
index 5d826b6f..19722260 100644
--- a/product_demos/Data-Science/ai-agent/_resources/04-eval-dataset-generation.py
+++ b/product_demos/Data-Science/ai-agent/_resources/04-eval-dataset-generation.py
@@ -86,7 +86,7 @@ def generate_question(email: pd.Series, row_id: pd.Series) -> pd.Series:
 final_df_raw = spark.sql("""
 SELECT 
   question,
-  AI_QUERY("databricks-claude-3-7-sonnet", prompt) AS expected_facts_json
+  AI_QUERY("databricks-llama-4-maverick", prompt) AS expected_facts_json
 FROM customer_test_questions
 """)
 
@@ -107,4 +107,5 @@ def generate_question(email: pd.Series, row_id: pd.Series) -> pd.Series:
 
 # COMMAND ----------
 
-display(eval_df)
+# The display command can be skipped, since it isn't shown on the calling noterbook and just costs time...
+# display(eval_df)

From 2ebfb17a4d65c3f684f0cd2c5769270455d5cf58 Mon Sep 17 00:00:00 2001
From: guidooswald <guidooswald>
Date: Tue, 2 Sep 2025 15:34:14 +0000
Subject: [PATCH 3/3] deactivate folder deletion

all files got downloaded every time, since the volume got wiped before....
---
 product_demos/Data-Science/ai-agent/_resources/01-setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/product_demos/Data-Science/ai-agent/_resources/01-setup.py b/product_demos/Data-Science/ai-agent/_resources/01-setup.py
index 5e45e27d..b6bef690 100644
--- a/product_demos/Data-Science/ai-agent/_resources/01-setup.py
+++ b/product_demos/Data-Science/ai-agent/_resources/01-setup.py
@@ -18,8 +18,8 @@
 
 # COMMAND ----------
 
-for f in dbutils.fs.ls(volume_folder):
-  dbutils.fs.rm(f.path, True)  
+# for f in dbutils.fs.ls(volume_folder):
+#   dbutils.fs.rm(f.path, True)  
 #dbutils.fs.rm(volume_folder, True)
 
 # COMMAND ----------