Update jupyter_ai_config.py

deutsche-nationalbibliothek · Jun 19, 2024 · 7e203c2 · 7e203c2
1 parent b4f5073
commit 7e203c2
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/jupyter_ai_config.py b/jupyter_ai_config.py
@@ -13,4 +13,10 @@
     "openai": api_key
 }
 
-config.default_llm = "openai:gpt-3.5-turbo"
+# Define default language model and embedding model after activation
+def set_defaults():
+    config.default_llm = "openai:gpt-3.5-turbo"
+    config.default_embedding_model = "openai:embed-gpt-3.5-turbo"
+
+# Call set_defaults() to set defaults after extension activation
+set_defaults()