cleanup

feiran-jia · Sep 16, 2023 · 45ddfa2 · 45ddfa2
1 parent 1517105
commit 45ddfa2
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 1 deletion.
diff --git a/autogen/oai/openai_utils.py b/autogen/oai/openai_utils.py
@@ -75,6 +75,7 @@ def config_list_openai_aoai(
     """
     if "OPENAI_API_KEY" not in os.environ and exclude != "openai":
         try:
+            print(f"{key_file_path}/{openai_api_key_file}")
             with open(f"{key_file_path}/{openai_api_key_file}") as key_file:
                 os.environ["OPENAI_API_KEY"] = key_file.read().strip()
         except FileNotFoundError:

diff --git a/test/__init__.py b/test/__init__.py
diff --git a/test/test_notebook.py b/test/test_notebook.py
@@ -19,7 +19,7 @@ def run_notebook(input_nb, output_nb="executed_openai_notebook.ipynb", save=Fals
     from nbconvert.preprocessors import CellExecutionError
 
     try:
-        nb_loc = os.path.join(here, os.pardir, os.pardir, "notebook")
+        nb_loc = os.path.join(here, os.pardir, "notebook")
         file_path = os.path.join(nb_loc, input_nb)
         with open(file_path) as nb_file:
             nb = nbformat.read(nb_file, as_version=4)