Skip to content

Commit 8fbfb39

Browse files
committed
update based on PR comments
1 parent acff84e commit 8fbfb39

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

quick_start_demo/5-min-Mosaic-AI-Agents-Demo-with-Synthetic-Eval.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -53,18 +53,13 @@
5353

5454
# Get current user's name & email to ensure each user doesn't over-write other user's outputs
5555
w = WorkspaceClient()
56-
user_email = w.current_user.me().user_name
57-
user_name = user_email.split("@")[0].replace(".", "_")
58-
59-
print(f"User name: {user_name}")
6056

6157
# COMMAND ----------
6258

6359
########################
6460
# #### ✅✏️ CONFIGURE THE UC MODEL NAME HERE
6561
########################
66-
#UC_MODEL_NAME = f"<ENTER_YOUR_CATALOG_NAME>.<ENTER_YOUR_SCHEMA_NAME>.db_docs__{user_name}"
67-
UC_MODEL_NAME = f"yz_agent_demo.playground.db_docs__{user_name}"
62+
UC_MODEL_NAME = "catalog.schema.my_demo_agent"
6863

6964
# COMMAND ----------
7065

@@ -190,7 +185,7 @@
190185
# MAGIC import json
191186
# MAGIC
192187
# MAGIC DEFAULT_CONFIG = {
193-
# MAGIC 'endpoint_name': "agents-demo-gpt4o",
188+
# MAGIC 'endpoint_name': "databricks-meta-llama-3-1-70b-instruct",
194189
# MAGIC 'temperature': 0.01,
195190
# MAGIC 'max_tokens': 1000,
196191
# MAGIC 'system_prompt': """You are a helpful assistant that answers questions about Databricks. Questions unrelated to Databricks are irrelevant.
@@ -458,7 +453,7 @@
458453
fc_agent = FunctionCallingAgent()
459454

460455
dev_config = {
461-
'endpoint_name': "agents-demo-gpt4o",
456+
'endpoint_name': "databricks-meta-llama-3-1-70b-instruct",
462457
'temperature': 0.01,
463458
'max_tokens': 1000,
464459
'system_prompt': """You are a helpful assistant that answers questions about Databricks. Questions unrelated to Databricks are irrelevant.

0 commit comments

Comments
 (0)