variable for slackbot model

PrefectHQ · Sep 13, 2024 · 0af998b · 0af998b
1 parent d06031b
commit 0af998b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cookbook/slackbot/start.py b/cookbook/slackbot/start.py
@@ -3,6 +3,7 @@
 import re
 import warnings
 from contextlib import contextmanager
+from typing import cast
 
 import marvin
 import uvicorn
@@ -37,7 +38,7 @@
 @contextmanager
 def engage_marvin_bot(instructions: str):
     with Assistant(
-        model="gpt-4o",
+        model=cast(str, Variable.get("marvin_bot_model", "gpt-4o")),
         name="Marvin (from Hitchhiker's Guide to the Galaxy)",
         tools=[
             search_prefect_2x_docs,