Skip to content

Commit 6759c4a

Browse files
committed
Add type safety for ZenML secrets in Hugging Face deployment
Modify Hugging Face space deployment to ensure ZenML store secrets are converted to strings before adding, preventing potential type-related errors during deployment.
1 parent da289e2 commit 6759c4a

File tree

1 file changed

+15
-10
lines changed

1 file changed

+15
-10
lines changed

llm-complete-guide/steps/rag_deployment.py

Lines changed: 15 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -89,16 +89,21 @@ def gradio_rag_deployment() -> None:
8989
exist_ok=True,
9090
token=get_hf_token(),
9191
)
92-
api.add_space_secret(
93-
repo_id=hf_repo_id,
94-
key="ZENML_STORE_API_KEY",
95-
value=ZENML_API_TOKEN,
96-
)
97-
api.add_space_secret(
98-
repo_id=hf_repo_id,
99-
key="ZENML_STORE_URL",
100-
value=ZENML_STORE_URL,
101-
)
92+
93+
# Ensure values are strings
94+
if ZENML_API_TOKEN is not None:
95+
api.add_space_secret(
96+
repo_id=hf_repo_id,
97+
key="ZENML_STORE_API_KEY",
98+
value=str(ZENML_API_TOKEN),
99+
)
100+
101+
if ZENML_STORE_URL is not None:
102+
api.add_space_secret(
103+
repo_id=hf_repo_id,
104+
key="ZENML_STORE_URL",
105+
value=str(ZENML_STORE_URL),
106+
)
102107

103108
files_to_upload = {
104109
"deployment_hf.py": "app.py",

0 commit comments

Comments
 (0)