yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-09-26 00:24:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Notebook/hierarchy flow (#482)

Browse Source 
* Notebook showing how to use select speaker to control conversation flow.

* pytest associated with notebook.

* Added llm_config to assistant and user proxy agent, and clarified why we set use_cache to false, as requested in the review.

* Added a @pytest.mark.skipif decorator like other tests to run it only in one py version, 3.10

* Fixed config warning.

* Removd llm_config to UserProxyAgent

* Fixed minor typos.

* Reran outputs

* Remopved llm_config from user_proxy_agent

* Colab Badge link updated.

* pre-commit formatting changes.

* Fixed base_url

---------

Co-authored-by: Chi Wang <wang.chi@microsoft.com>
This commit is contained in:
Joshua Kim 2023-11-08 09:06:45 +11:00 committed by GitHub
parent 65ec0b19de
commit 2a96e4d9d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 492 additions and 92 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

392
notebook/agentchat_hierarchy_flow_using_select_speaker.ipynb Normal file
View File

File diff suppressed because one or more lines are too long

192
test/test_notebook.py
View File

@ -1,92 +1,100 @@
import sys import sys
import os import os
import pytest import pytest
try: try:
import openai import openai
skip = False skip = False
except ImportError: except ImportError:
skip = True skip = True
here = os.path.abspath(os.path.dirname(__file__)) here = os.path.abspath(os.path.dirname(__file__))
def run_notebook(input_nb, output_nb="executed_openai_notebook.ipynb", save=False): def run_notebook(input_nb, output_nb="executed_openai_notebook.ipynb", save=False):
import nbformat import nbformat
from nbconvert.preprocessors import ExecutePreprocessor from nbconvert.preprocessors import ExecutePreprocessor
from nbconvert.preprocessors import CellExecutionError from nbconvert.preprocessors import CellExecutionError
try: try:
nb_loc = os.path.join(here, os.pardir, "notebook") nb_loc = os.path.join(here, os.pardir, "notebook")
file_path = os.path.join(nb_loc, input_nb) file_path = os.path.join(nb_loc, input_nb)
with open(file_path) as nb_file: with open(file_path) as nb_file:
nb = nbformat.read(nb_file, as_version=4) nb = nbformat.read(nb_file, as_version=4)
preprocessor = ExecutePreprocessor(timeout=4800, kernel_name="python3") preprocessor = ExecutePreprocessor(timeout=4800, kernel_name="python3")
preprocessor.preprocess(nb, {"metadata": {"path": nb_loc}}) preprocessor.preprocess(nb, {"metadata": {"path": nb_loc}})
output_file_name = "executed_openai_notebook_output.txt" output_file_name = "executed_openai_notebook_output.txt"
output_file = os.path.join(here, output_file_name) output_file = os.path.join(here, output_file_name)
with open(output_file, "a") as nb_output_file: with open(output_file, "a") as nb_output_file:
for cell in nb.cells: for cell in nb.cells:
if cell.cell_type == "code" and "outputs" in cell: if cell.cell_type == "code" and "outputs" in cell:
for output in cell.outputs: for output in cell.outputs:
if "text" in output: if "text" in output:
nb_output_file.write(output["text"].strip() + "\n") nb_output_file.write(output["text"].strip() + "\n")
elif "data" in output and "text/plain" in output["data"]: elif "data" in output and "text/plain" in output["data"]:
nb_output_file.write(output["data"]["text/plain"].strip() + "\n") nb_output_file.write(output["data"]["text/plain"].strip() + "\n")
except CellExecutionError: except CellExecutionError:
raise raise
finally: finally:
if save: if save:
with open(os.path.join(here, output_nb), "w", encoding="utf-8") as nb_executed_file: with open(os.path.join(here, output_nb), "w", encoding="utf-8") as nb_executed_file:
nbformat.write(nb, nb_executed_file) nbformat.write(nb, nb_executed_file)
@pytest.mark.skipif( @pytest.mark.skipif(
skip or not sys.version.startswith("3.11"), skip or not sys.version.startswith("3.11"),
reason="do not run if openai is not installed or py!=3.11", reason="do not run if openai is not installed or py!=3.11",
) )
def test_agentchat_auto_feedback_from_code(save=False): def test_agentchat_auto_feedback_from_code(save=False):
run_notebook("agentchat_auto_feedback_from_code_execution.ipynb", save=save) run_notebook("agentchat_auto_feedback_from_code_execution.ipynb", save=save)
@pytest.mark.skipif( @pytest.mark.skipif(
skip or not sys.version.startswith("3.10"), skip or not sys.version.startswith("3.10"),
reason="do not run if openai is not installed or py!=3.10", reason="do not run if openai is not installed or py!=3.10",
) )
def _test_oai_completion(save=False): def _test_oai_completion(save=False):
run_notebook("oai_completion.ipynb", save=save) run_notebook("oai_completion.ipynb", save=save)
@pytest.mark.skipif( @pytest.mark.skipif(
skip or not sys.version.startswith("3.10"), skip or not sys.version.startswith("3.10"),
reason="do not run if openai is not installed or py!=3.10", reason="do not run if openai is not installed or py!=3.10",
) )
def test_agentchat_function_call(save=False): def test_agentchat_function_call(save=False):
run_notebook("agentchat_function_call.ipynb", save=save) run_notebook("agentchat_function_call.ipynb", save=save)
@pytest.mark.skipif( @pytest.mark.skipif(
skip or not sys.version.startswith("3.10"), skip or not sys.version.startswith("3.10"),
reason="do not run if openai is not installed or py!=3.10", reason="do not run if openai is not installed or py!=3.10",
) )
def _test_agentchat_MathChat(save=False): def _test_agentchat_MathChat(save=False):
run_notebook("agentchat_MathChat.ipynb", save=save) run_notebook("agentchat_MathChat.ipynb", save=save)
@pytest.mark.skipif( @pytest.mark.skipif(
skip or not sys.version.startswith("3.11"), skip or not sys.version.startswith("3.11"),
reason="do not run if openai is not installed or py!=3.11", reason="do not run if openai is not installed or py!=3.11",
) )
def _test_oai_chatgpt_gpt4(save=False): def _test_oai_chatgpt_gpt4(save=False):
run_notebook("oai_chatgpt_gpt4.ipynb", save=save) run_notebook("oai_chatgpt_gpt4.ipynb", save=save)
if __name__ == "__main__": @pytest.mark.skipif(
test_agentchat_auto_feedback_from_code(save=True) skip or not sys.version.startswith("3.10"),
# test_oai_chatgpt_gpt4(save=True) reason="do not run if openai is not installed or py!=3.10",
# test_oai_completion(save=True) )
# test_agentchat_MathChat(save=True) def test_hierarchy_flow_using_select_speaker(save=False):
# test_agentchat_function_call(save=True) run_notebook("agentchat_hierarchy_flow_using_select_speaker.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)