diff --git a/agents-workshop/agent_setup/agent.py b/agents-workshop/agent_setup/agent.py
index 03574f6..f0f8c36 100644
--- a/agents-workshop/agent_setup/agent.py
+++ b/agents-workshop/agent_setup/agent.py
@@ -33,7 +33,8 @@
 # Define your LLM endpoint and system prompt
 ############################################
 #LLM_ENDPOINT_NAME = "databricks-meta-llama-3-3-70b-instruct"
-LLM_ENDPOINT_NAME = "databricks-claude-3-7-sonnet"
+#LLM_ENDPOINT_NAME = "databricks-claude-sonnet-4-6"
+LLM_ENDPOINT_NAME = "databricks-gpt-oss-120b"
 llm = ChatDatabricks(endpoint=LLM_ENDPOINT_NAME)
 
 system_prompt = "You are a customer success specialist that helps users with product questions as part of a Databricks lab. Use tools to retrieve all information needed and help customers fully understand the products they're asking about. Aim to provide value in every interaction."