Avoid setting the env variables

code_examples/1_basics/2_first_programs.py

@@ -247,10 +247,10 @@ def _(mo):
 
 
 @app.cell(hide_code=True)
-def _(mo, openai_api_key):
-    import os
+def _(mo, openai_api_key, litellm):
+    import litellm
     mo.stop(not openai_api_key.value)
-    os.environ["OPENAI_API_KEY"] = openai_api_key.value
+    litellm.openai_key = openai_api_key.value
     return
 
 

code_examples/1_basics/4_conversational_applications.py

@@ -99,9 +99,9 @@ def _(mo):
 
 @app.cell(hide_code=True)
 def _(mo, openai_api_key):
-    import os
+    import litellm
     mo.stop(not openai_api_key.value)
-    os.environ["OPENAI_API_KEY"] = openai_api_key.value
+    litellm.openai_key = openai_api_key.value
     return
 
 

code_examples/1_basics/6_training_programs.py

@@ -200,9 +200,9 @@ def _(mo):
 
 @app.cell(hide_code=True)
 def _(mo, openai_api_key):
-    import os
+    import litellm
     mo.stop(not openai_api_key.value)
-    os.environ["OPENAI_API_KEY"] = openai_api_key.value
+    litellm.openai_key = openai_api_key.value
     return