clean up

kyegomez · Jul 29, 2023 · 22fc173 · 22fc173
1 parent 04caaa2
commit 22fc173
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 2 deletions.
diff --git a/example.py b/example.py
@@ -1,11 +1,12 @@
+import os
 from tree_of_thoughts.openai_models import OpenAILanguageModel
 from tree_of_thoughts.treeofthoughts import MonteCarloTreeofThoughts
 
 
 api_model= "gpt-3.5-turbo"
 
-
-model = OpenAILanguageModel(api_key='', api_model=api_model)
+api_key = os.getenv("OPENAI_API_KEY")
+model = OpenAILanguageModel(api_key=api_key, api_model=api_model)
 
 
 # Initialize the MonteCarloTreeofThoughts class with the model

diff --git a/example_tot2.py → examples/example_tot2.py b/example_tot2.py → examples/example_tot2.py
diff --git a/example_totA.py → examples/example_totA.py b/example_totA.py → examples/example_totA.py
diff --git a/example_totdfs.py → examples/example_totdfs.py b/example_totdfs.py → examples/example_totdfs.py