mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-31 21:02:33 +00:00
c4f8b1c761
* api_base -> base_url (#383) * InvalidRequestError -> BadRequestError (#389) * remove api_key_path; close #388 * close #402 (#403) * openai client (#419) * openai client * client test * _client -> client * _client -> client * extra kwargs * Completion -> client (#426) * Completion -> client * Completion -> client * Completion -> client * Completion -> client * support aoai * fix test error * remove commented code * support aoai * annotations * import * reduce test * skip test * skip test * skip test * debug test * rename test * update workflow * update workflow * env * py version * doc improvement * docstr update * openai<1 * add tiktoken to dependency * filter_func * async test * dependency * migration guide (#477) * migration guide * change in kwargs * simplify header * update optigude description * deal with azure gpt-3.5 * add back test_eval_math_responses * timeout * Add back tests for RetrieveChat (#480) * Add back tests for RetrieveChat * Fix format * Update dependencies order * Fix path * Fix path * Fix path * Fix tests * Add not run openai on MacOS or Win * Update skip openai tests * Remove unnecessary dependencies, improve format * Add py3.8 for testing qdrant * Fix multiline error of windows * Add openai tests * Add dependency mathchat, remove unused envs * retrieve chat is tested * bump version to 0.2.0b1 --------- Co-authored-by: Li Jiang <bnujli@gmail.com>
93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
import sys
|
|
import os
|
|
import pytest
|
|
|
|
try:
|
|
import openai
|
|
|
|
skip = False
|
|
except ImportError:
|
|
skip = True
|
|
|
|
|
|
here = os.path.abspath(os.path.dirname(__file__))
|
|
|
|
|
|
def run_notebook(input_nb, output_nb="executed_openai_notebook.ipynb", save=False):
|
|
import nbformat
|
|
from nbconvert.preprocessors import ExecutePreprocessor
|
|
from nbconvert.preprocessors import CellExecutionError
|
|
|
|
try:
|
|
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)
|
|
preprocessor = ExecutePreprocessor(timeout=4800, kernel_name="python3")
|
|
preprocessor.preprocess(nb, {"metadata": {"path": nb_loc}})
|
|
|
|
output_file_name = "executed_openai_notebook_output.txt"
|
|
output_file = os.path.join(here, output_file_name)
|
|
with open(output_file, "a") as nb_output_file:
|
|
for cell in nb.cells:
|
|
if cell.cell_type == "code" and "outputs" in cell:
|
|
for output in cell.outputs:
|
|
if "text" in output:
|
|
nb_output_file.write(output["text"].strip() + "\n")
|
|
elif "data" in output and "text/plain" in output["data"]:
|
|
nb_output_file.write(output["data"]["text/plain"].strip() + "\n")
|
|
except CellExecutionError:
|
|
raise
|
|
finally:
|
|
if save:
|
|
with open(os.path.join(here, output_nb), "w", encoding="utf-8") as nb_executed_file:
|
|
nbformat.write(nb, nb_executed_file)
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
skip or not sys.version.startswith("3.11"),
|
|
reason="do not run if openai is not installed or py!=3.11",
|
|
)
|
|
def test_agentchat_auto_feedback_from_code(save=False):
|
|
run_notebook("agentchat_auto_feedback_from_code_execution.ipynb", save=save)
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
skip or not sys.version.startswith("3.10"),
|
|
reason="do not run if openai is not installed or py!=3.10",
|
|
)
|
|
def _test_oai_completion(save=False):
|
|
run_notebook("oai_completion.ipynb", save=save)
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
skip or not sys.version.startswith("3.10"),
|
|
reason="do not run if openai is not installed or py!=3.10",
|
|
)
|
|
def test_agentchat_function_call(save=False):
|
|
run_notebook("agentchat_function_call.ipynb", save=save)
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
skip or not sys.version.startswith("3.10"),
|
|
reason="do not run if openai is not installed or py!=3.10",
|
|
)
|
|
def _test_agentchat_MathChat(save=False):
|
|
run_notebook("agentchat_MathChat.ipynb", save=save)
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
skip or not sys.version.startswith("3.11"),
|
|
reason="do not run if openai is not installed or py!=3.11",
|
|
)
|
|
def _test_oai_chatgpt_gpt4(save=False):
|
|
run_notebook("oai_chatgpt_gpt4.ipynb", save=save)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_agentchat_auto_feedback_from_code(save=True)
|
|
# test_oai_chatgpt_gpt4(save=True)
|
|
# test_oai_completion(save=True)
|
|
# test_agentchat_MathChat(save=True)
|
|
# test_agentchat_function_call(save=True)
|