ag2 0.8.0b1__tar.gz → 0.8.1__tar.gz

{ag2-0.8.0b1/ag2.egg-info → ag2-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ag2
-Version: 0.8.0b1
+Version: 0.8.1
 Summary: Alias package for pyautogen
 Home-page: https://github.com/ag2ai/ag2
 Author: Chi Wang & Qingyun Wu
@@ -39,6 +39,7 @@ Provides-Extra: teachable
 Provides-Extra: lmm
 Provides-Extra: graph
 Provides-Extra: gemini
+Provides-Extra: gemini-realtime
 Provides-Extra: together
 Provides-Extra: websurfer
 Provides-Extra: redis
@@ -52,6 +53,7 @@ Provides-Extra: groq
 Provides-Extra: cohere
 Provides-Extra: ollama
 Provides-Extra: bedrock
+Provides-Extra: deepseek
 Provides-Extra: commsagent-discord
 Provides-Extra: commsagent-slack
 Provides-Extra: commsagent-telegram

{ag2-0.8.0b1 → ag2-0.8.1/ag2.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ag2
-Version: 0.8.0b1
+Version: 0.8.1
 Summary: Alias package for pyautogen
 Home-page: https://github.com/ag2ai/ag2
 Author: Chi Wang & Qingyun Wu
@@ -39,6 +39,7 @@ Provides-Extra: teachable
 Provides-Extra: lmm
 Provides-Extra: graph
 Provides-Extra: gemini
+Provides-Extra: gemini-realtime
 Provides-Extra: together
 Provides-Extra: websurfer
 Provides-Extra: redis
@@ -52,6 +53,7 @@ Provides-Extra: groq
 Provides-Extra: cohere
 Provides-Extra: ollama
 Provides-Extra: bedrock
+Provides-Extra: deepseek
 Provides-Extra: commsagent-discord
 Provides-Extra: commsagent-slack
 Provides-Extra: commsagent-telegram

ag2-0.8.1/ag2.egg-info/requires.txt

@@ -0,0 +1,151 @@
+pyautogen==0.8.1
+
+[anthropic]
+pyautogen[anthropic]==0.8.1
+
+[autobuild]
+pyautogen[autobuild]==0.8.1
+
+[bedrock]
+pyautogen[bedrock]==0.8.1
+
+[blendsearch]
+pyautogen[blendsearch]==0.8.1
+
+[browser-use]
+pyautogen[browser-use]==0.8.1
+
+[captainagent]
+pyautogen[captainagent]==0.8.1
+
+[cerebras]
+pyautogen[cerebras]==0.8.1
+
+[cohere]
+pyautogen[cohere]==0.8.1
+
+[commsagent-discord]
+pyautogen[commsagent-discord]==0.8.1
+
+[commsagent-slack]
+pyautogen[commsagent-slack]==0.8.1
+
+[commsagent-telegram]
+pyautogen[commsagent-telegram]==0.8.1
+
+[cosmosdb]
+pyautogen[cosmosdb]==0.8.1
+
+[crawl4ai]
+pyautogen[crawl4ai]==0.8.1
+
+[deepseek]
+pyautogen[deepseek]==0.8.1
+
+[dev]
+pyautogen[dev]==0.8.1
+
+[docs]
+pyautogen[docs]==0.8.1
+
+[flaml]
+pyautogen[flaml]==0.8.1
+
+[gemini]
+pyautogen[gemini]==0.8.1
+
+[gemini-realtime]
+pyautogen[gemini-realtime]==0.8.1
+
+[graph]
+pyautogen[graph]==0.8.1
+
+[graph-rag-falkor-db]
+pyautogen[graph-rag-falkor-db]==0.8.1
+
+[groq]
+pyautogen[groq]==0.8.1
+
+[interop]
+pyautogen[interop]==0.8.1
+
+[interop-crewai]
+pyautogen[interop-crewai]==0.8.1
+
+[interop-langchain]
+pyautogen[interop-langchain]==0.8.1
+
+[interop-pydantic-ai]
+pyautogen[interop-pydantic-ai]==0.8.1
+
+[jupyter-executor]
+pyautogen[jupyter-executor]==0.8.1
+
+[lint]
+pyautogen[lint]==0.8.1
+
+[lmm]
+pyautogen[lmm]==0.8.1
+
+[long-context]
+pyautogen[long-context]==0.8.1
+
+[mathchat]
+pyautogen[mathchat]==0.8.1
+
+[mistral]
+pyautogen[mistral]==0.8.1
+
+[neo4j]
+pyautogen[neo4j]==0.8.1
+
+[ollama]
+pyautogen[ollama]==0.8.1
+
+[openai]
+pyautogen[openai]==0.8.1
+
+[openai-realtime]
+pyautogen[openai-realtime]==0.8.1
+
+[rag]
+pyautogen[rag]==0.8.1
+
+[redis]
+pyautogen[redis]==0.8.1
+
+[retrievechat]
+pyautogen[retrievechat]==0.8.1
+
+[retrievechat-couchbase]
+pyautogen[retrievechat-couchbase]==0.8.1
+
+[retrievechat-mongodb]
+pyautogen[retrievechat-mongodb]==0.8.1
+
+[retrievechat-pgvector]
+pyautogen[retrievechat-pgvector]==0.8.1
+
+[retrievechat-qdrant]
+pyautogen[retrievechat-qdrant]==0.8.1
+
+[teachable]
+pyautogen[teachable]==0.8.1
+
+[test]
+pyautogen[test]==0.8.1
+
+[together]
+pyautogen[together]==0.8.1
+
+[twilio]
+pyautogen[twilio]==0.8.1
+
+[types]
+pyautogen[types]==0.8.1
+
+[websockets]
+pyautogen[websockets]==0.8.1
+
+[websurfer]
+pyautogen[websurfer]==0.8.1

{ag2-0.8.0b1 → ag2-0.8.1}/pyproject.toml

@@ -80,7 +80,7 @@ openai = [
 ]
 
 openai-realtime = [
-    "autogen[openai]",
+    "pyautogen[openai]",
     "openai[realtime]",
 ]
 
@@ -139,6 +139,10 @@ rag = [
     "chromadb>=0.5,<1",
     "llama-index>=0.12,<1",
     "llama-index-vector-stores-chroma==0.4.1",
+    "llama-index-vector-stores-mongodb==0.6.0",
+    "llama-index-embeddings-huggingface==0.5.2",
+    "llama-index-llms-langchain==0.6.0",
+    "requests>=2.32.3,<3",
 ]
 
 
@@ -154,10 +158,10 @@ browser-use = [
 
 neo4j = [
     "docx2txt==0.8",
-    "llama-index==0.12.19",
+    "llama-index==0.12.22",
     "llama-index-graph-stores-neo4j==0.4.6",
-    "llama-index-core==0.12.19",
-    "llama-index-readers-web==0.3.5",
+    "llama-index-core==0.12.22",
+    "llama-index-readers-web==0.3.7",
 ]
 
 # used for agentchat_realtime_swarm notebook and realtime agent twilio demo
@@ -172,7 +176,7 @@ interop-crewai = [
     "weaviate-client>=4,<5; python_version>='3.10' and python_version<'3.13'",
 ]
 interop-langchain = ["langchain-community>=0.3.12,<1"]
-interop-pydantic-ai = ["pydantic-ai==0.0.24"]
+interop-pydantic-ai = ["pydantic-ai==0.0.31"]
 interop =[
     "pyautogen[interop-crewai, interop-langchain, interop-pydantic-ai]",
 ]
@@ -188,7 +192,7 @@ lmm = ["replicate", "pillow"]
 graph = ["networkx", "matplotlib"]
 gemini = [
     "google-api-core",
-    "google-genai>=1.2.0,<2.0",
+    "google-genai>=1.2.0",
     "google-cloud-aiplatform",
     "google-auth",
     "pillow",
@@ -196,11 +200,16 @@ gemini = [
     "jsonref>=1,<2",
 ]
 
+gemini-realtime = [
+    "pyautogen[gemini]",
+    "pyautogen[websockets]",
+]
+
 together = ["together>=1.2"]
 websurfer = ["beautifulsoup4", "markdownify", "pdfminer.six", "pathvalidate"]
 redis = ["redis"]
 cosmosdb = ["azure-cosmos>=4.2.0"]
-websockets = ["websockets>=14.0,<15"]
+websockets = ["websockets>=14.0,<16"]
 long-context = ["llmlingua<0.3"]
 anthropic = ["anthropic[vertex]>=0.23.1"]
 cerebras = ["cerebras_cloud_sdk>=1.0.0"]
@@ -209,8 +218,9 @@ groq = ["groq>=0.9.0"]
 cohere = ["cohere>=5.13.5"]
 ollama = ["ollama>=0.4.5", "fix_busted_json>=0.0.18"]
 bedrock = ["boto3>=1.34.149"]
+deepseek = ["pyautogen[openai]"]
 
-commsagent-discord = ["discord.py>=2.4.0,<2.5"]
+commsagent-discord = ["discord.py>=2.4.0,<2.6"]
 commsagent-slack = ["slack_sdk>=3.33.0,<3.40"]
 commsagent-telegram = ["telethon>=1.38.1, <2"]
 
@@ -223,23 +233,24 @@ test = [
     "nbformat==5.10.4",
     "pytest-cov==6.0.0",
     "pytest-asyncio==0.25.3",
-    "pytest==8.3.4",
+    "pytest==8.3.5",
     "mock==5.1.0",
     "pandas==2.2.3",
-    "fastapi==0.115.8",
+    "fastapi==0.115.11",
 ]
 
 docs = [
-    "mkdocs-material==9.6.4",
-    "mkdocstrings[python]==0.28.1",
+    "mkdocs-material==9.6.7",
+    "mkdocstrings[python]==0.28.2",
     "mkdocs-literate-nav==0.6.1",
     "mdx-include==1.4.2",
     "mkdocs-git-revision-date-localized-plugin==1.3.0",
     "mike==2.1.3",
-    "typer==0.15.1",
+    "typer==0.15.2",
     "mkdocs-minify-plugin==0.8.0",
     "mkdocs-macros-plugin==1.3.7",  # includes with variables
     "mkdocs-glightbox==0.4.0",  # img zoom
+    "mkdocs-ezlinks-plugin==0.1.14",  # converts abs links to rel links
     "pillow",  # required for mkdocs-glightbo
     "cairosvg",  # required for mkdocs-glightbo
     "pdoc3==0.11.5",
@@ -255,7 +266,7 @@ types = [
 ]
 
 lint = [
-    "ruff==0.9.7",
+    "ruff==0.9.9",
     "codespell==2.4.1",
     "pyupgrade-directories==0.3.0",
 ]
@@ -265,7 +276,7 @@ dev = [
     "pyautogen[lint,test,types,docs]",
     "pre-commit==4.1.0",
     "detect-secrets==1.5.0",
-    "uv==0.6.2",
+    "uv==0.6.4",
 ]
 
 
@@ -294,23 +305,31 @@ exclude = ["test", "notebook"]
 "autogen" = "autogen"
 "autogen/agentchat/contrib/captainagent/tools" = "autogen/agentchat/contrib/captainagent/tools"
 
-
 [tool.pytest.ini_options]
 addopts = '--cov=autogen --cov-append --cov-branch --cov-report=xml -m "not conda"'
 testpaths = [
     "test",
 ]
+
 markers = [
     "conda: test related to conda forge distribution",
     "all",
     "openai",
+    "openai_realtime",
     "gemini",
+    "gemini_realtime",
     "anthropic",
     "deepseek",
+    "cosmosdb",
+    "ollama",
+    "bedrock",
+    "cerebras",
+    "aux_neg_flag",
+    # optional deps
     "redis",
     "docker",
     "docs",
-    # optional deps
+    "rag",
     "jupyter_executor",
     "retrievechat",
     "retrievechat_pgvector",
@@ -323,6 +342,10 @@ markers = [
     "browser_use",
     "crawl4ai",
     "websockets",
+    "commsagent_discord",
+    "commsagent_slack",
+    "commsagent-telegram",
+    "lmm",
 ]
 
 [tool.black]
@@ -379,6 +402,8 @@ select = [
 #    "UP",     # pyupgrade              https://docs.astral.sh/ruff/rules/#pydocstyle-d
 ]
 
+extend-select = ["D417"]
+
 ignore = ["E501", "F403", "C901",
     "E402",
     "E721",
@@ -404,9 +429,12 @@ convention = "google"
 
 [tool.mypy]
 files = [
+    "autogen/agentchat/agent.py",
     "autogen/agentchat/contrib/rag",
     "autogen/agentchat/contrib/graph_rag",
+    "autogen/agentchat/contrib/swarm_agent.py",
     "autogen/agentchat/realtime_agent",
+    "autogen/agentchat/utils.py",
     "autogen/agents",
     "autogen/coding",
     "autogen/exception_utils.py",
@@ -418,9 +446,11 @@ files = [
     "autogen/oai/oai_models",
     "autogen/oai/openai_utils.py",
     "autogen/tools",
+    "autogen/_website",
     "website/*.py",
+    "test/agentchat/contrib/rag",
     "test/agentchat/contrib/graph_rag",
-#    "test/agentchat/contrib/rag",
+    "test/agentchat/contrib/test_swarm.py",
     "test/agentchat/realtime_agent",
     "test/agents",
     "test/conftest.py",

{ag2-0.8.0b1 → ag2-0.8.1}/setup_ag2.py

@@ -55,6 +55,7 @@ setuptools.setup(
         "lmm": ["pyautogen[lmm]==" + __version__],
         "graph": ["pyautogen[graph]==" + __version__],
         "gemini": ["pyautogen[gemini]==" + __version__],
+        "gemini-realtime": ["pyautogen[gemini-realtime]==" + __version__],
         "together": ["pyautogen[together]==" + __version__],
         "websurfer": ["pyautogen[websurfer]==" + __version__],
         "redis": ["pyautogen[redis]==" + __version__],
@@ -68,6 +69,7 @@ setuptools.setup(
         "cohere": ["pyautogen[cohere]==" + __version__],
         "ollama": ["pyautogen[ollama]==" + __version__],
         "bedrock": ["pyautogen[bedrock]==" + __version__],
+        "deepseek": ["pyautogen[deepseek]==" + __version__],
         "commsagent-discord": ["pyautogen[commsagent-discord]==" + __version__],
         "commsagent-slack": ["pyautogen[commsagent-slack]==" + __version__],
         "commsagent-telegram": ["pyautogen[commsagent-telegram]==" + __version__],

{ag2-0.8.0b1 → ag2-0.8.1}/test/test_code_utils.py

@@ -8,7 +8,6 @@
 
 import os
 import tempfile
-import unittest
 from io import StringIO
 from types import SimpleNamespace
 from unittest.mock import patch
@@ -24,15 +23,10 @@ from autogen.code_utils import (
     execute_code,
     extract_code,
     get_powershell_command,
-    improve_code,
-    improve_function,
     in_docker_container,
     infer_lang,
     is_docker_running,
 )
-from autogen.import_utils import skip_on_missing_imports
-
-from .conftest import Credentials
 
 here = os.path.abspath(os.path.dirname(__file__))
 
@@ -387,69 +381,40 @@ def test_create_virtual_env_with_extra_args():
         assert venv_context.env_name == os.path.split(temp_dir)[1]
 
 
-@skip_on_missing_imports(["openai"])
-def _test_improve(credentials_all: Credentials):
-    config_list = credentials_all.config_list
-    improved, _ = improve_function(
-        "autogen/math_utils.py",
-        "solve_problem",
-        "Solve math problems accurately, by avoiding calculation errors and reduce reasoning errors.",
-        config_list=config_list,
-    )
-    with open(f"{here}/math_utils.py.improved", "w") as f:
-        f.write(improved)
-    suggestion, _ = improve_code(
-        ["autogen/code_utils.py", "autogen/math_utils.py"],
-        "leverage generative AI smartly and cost-effectively",
-        config_list=config_list,
-    )
-    print(suggestion)
-    improvement, cost = improve_code(
-        ["autogen/code_utils.py", "autogen/math_utils.py"],
-        "leverage generative AI smartly and cost-effectively",
-        suggest_only=False,
-        config_list=config_list,
-    )
-    print(cost)
-    with open(f"{here}/suggested_improvement.txt", "w") as f:
-        f.write(improvement)
-
-
-class TestContentStr(unittest.TestCase):
+class TestContentStr:
     def test_string_content(self):
-        self.assertEqual(content_str("simple string"), "simple string")
+        assert content_str("simple string") == "simple string"
 
     def test_list_of_text_content(self):
         content = [{"type": "text", "text": "hello"}, {"type": "text", "text": " world"}]
-        self.assertEqual(content_str(content), "hello world")
+        assert content_str(content) == "hello world"
 
     def test_mixed_content(self):
         content = [{"type": "text", "text": "hello"}, {"type": "image_url", "url": "http://example.com/image.png"}]
-        self.assertEqual(content_str(content), "hello<image>")
+        assert content_str(content) == "hello<image>"
 
     def test_invalid_content(self):
         content = [{"type": "text", "text": "hello"}, {"type": "wrong_type", "url": "http://example.com/image.png"}]
-        with self.assertRaises(ValueError) as context:
+        with pytest.raises(ValueError):
             content_str(content)
-        self.assertIn("Wrong content format", str(context.exception))
 
     def test_empty_list(self):
-        self.assertEqual(content_str([]), "")
+        assert content_str([]) == ""
 
     def test_non_dict_in_list(self):
         content = ["string", {"type": "text", "text": "text"}]
-        with self.assertRaises(TypeError):
+        with pytest.raises(TypeError):
             content_str(content)
 
 
-class TestGetPowerShellCommand(unittest.TestCase):
+class TestGetPowerShellCommand:
     @patch("subprocess.run")
     def test_get_powershell_command_powershell(self, mock_subprocess_run):
         # Set up the mock to return a successful result for 'powershell'
         mock_subprocess_run.return_value.returncode = 0
         mock_subprocess_run.return_value.stdout = StringIO("5")
 
-        self.assertEqual(get_powershell_command(), "powershell")
+        assert get_powershell_command() == "powershell"
 
     @patch("subprocess.run")
     def test_get_powershell_command_pwsh(self, mock_subprocess_run):
@@ -458,18 +423,18 @@ class TestGetPowerShellCommand(unittest.TestCase):
         mock_subprocess_run.return_value.returncode = 0
         mock_subprocess_run.return_value.stdout = StringIO("7")
 
-        self.assertEqual(get_powershell_command(), "pwsh")
+        assert get_powershell_command() == "pwsh"
 
     @patch("subprocess.run")
     def test_get_powershell_command_not_found(self, mock_subprocess_run):
         mock_subprocess_run.side_effect = [FileNotFoundError, FileNotFoundError]
-        with self.assertRaises(FileNotFoundError):
+        with pytest.raises(FileNotFoundError):
             get_powershell_command()
 
     @patch("subprocess.run")
     def test_get_powershell_command_no_permission(self, mock_subprocess_run):
         mock_subprocess_run.side_effect = [PermissionError, FileNotFoundError]
-        with self.assertRaises(PermissionError):
+        with pytest.raises(PermissionError):
             get_powershell_command()
 
 

{ag2-0.8.0b1 → ag2-0.8.1}/test/test_logging.py

@@ -14,7 +14,7 @@ from unittest.mock import Mock, patch
 
 import pytest
 
-from autogen.import_utils import optional_import_block, skip_on_missing_imports
+from autogen.import_utils import optional_import_block, run_for_optional_imports
 
 with optional_import_block() as result:
     import openai  # noqa: F401
@@ -160,7 +160,7 @@ def test_log_function_use(db_connection):
         assert row["returns"] == json.dumps(returns)
 
 
-@skip_on_missing_imports(["openai"], "openai")
+@run_for_optional_imports(["openai"], "openai")
 def test_log_new_agent(db_connection):
     from autogen import AssistantAgent
 
@@ -186,7 +186,7 @@ def test_log_new_agent(db_connection):
         assert row["init_args"] == json.dumps(init_args)
 
 
-@skip_on_missing_imports(["openai"], "openai")
+@run_for_optional_imports(["openai"], "openai")
 def test_log_oai_wrapper(db_connection):
     from autogen import OpenAIWrapper
 
@@ -213,7 +213,7 @@ def test_log_oai_wrapper(db_connection):
         assert "base_config" in saved_init_args
 
 
-@skip_on_missing_imports(["openai"], "openai")
+@run_for_optional_imports(["openai"], "openai")
 def test_log_oai_client(db_connection):
     cur = db_connection.cursor()
 

{ag2-0.8.0b1 → ag2-0.8.1}/test/test_notebook.py

@@ -11,7 +11,7 @@ import sys
 
 import pytest
 
-from autogen.import_utils import skip_on_missing_imports
+from autogen.import_utils import run_for_optional_imports
 
 here = os.path.abspath(os.path.dirname(__file__))
 
@@ -44,102 +44,102 @@ def run_notebook(input_nb, output_nb="executed_openai_notebook.ipynb", save=Fals
                 nbformat.write(nb, nb_executed_file)
 
 
-@pytest.mark.openai
+@run_for_optional_imports("openai", "openai")
 @pytest.mark.skipif(
     not sys.version.startswith("3.13"),
     reason="do not run if py!=3.13",
 )
-@skip_on_missing_imports(["openai"])
+@run_for_optional_imports(["openai"], "openai")
 def test_agentchat_auto_feedback_from_code(save=False):
     run_notebook("agentchat_auto_feedback_from_code_execution.ipynb", save=save)
 
 
-@pytest.mark.openai
+@run_for_optional_imports("openai", "openai")
 @pytest.mark.skipif(
     not sys.version.startswith("3.11"),
     reason="do not run if py!=3.11",
 )
-@skip_on_missing_imports(["openai"])
+@run_for_optional_imports(["openai"], "openai")
 def _test_oai_completion(save=False):
     run_notebook("oai_completion.ipynb", save=save)
 
 
-@pytest.mark.openai
+@run_for_optional_imports("openai", "openai")
 @pytest.mark.skipif(
     not sys.version.startswith("3.12"),
     reason="do not run if py!=3.12",
 )
-@skip_on_missing_imports(["openai"])
+@run_for_optional_imports(["openai"], "openai")
 def test_agentchat_function_call(save=False):
     run_notebook("agentchat_function_call.ipynb", save=save)
 
 
-@pytest.mark.openai
+@run_for_optional_imports("openai", "openai")
 @pytest.mark.skipif(
     not sys.version.startswith("3.10"),
     reason="do not run if py!=3.10",
 )
-@skip_on_missing_imports(["openai"])
+@run_for_optional_imports(["openai"], "openai")
 def test_agentchat_function_call_currency_calculator(save=False):
     run_notebook("agentchat_function_call_currency_calculator.ipynb", save=save)
 
 
-@pytest.mark.openai
+@run_for_optional_imports("openai", "openai")
 @pytest.mark.skipif(
     not sys.version.startswith("3.13"),
     reason="do not run if py!=3.13",
 )
-@skip_on_missing_imports(["openai"])
+@run_for_optional_imports(["openai"], "openai")
 def test_agentchat_function_call_async(save=False):
     run_notebook("agentchat_function_call_async.ipynb", save=save)
 
 
-@pytest.mark.openai
+@run_for_optional_imports("openai", "openai")
 @pytest.mark.skipif(
     not sys.version.startswith("3.12"),
     reason="do not run if py!=3.12",
 )
-@skip_on_missing_imports(["openai"])
+@run_for_optional_imports(["openai"], "openai")
 def _test_agentchat_MathChat(save=False):  # noqa: N802
     run_notebook("agentchat_MathChat.ipynb", save=save)
 
 
-@pytest.mark.openai
+@run_for_optional_imports("openai", "openai")
 @pytest.mark.skipif(
     not sys.version.startswith("3.10"),
     reason="do not run if py!=3.10",
 )
-@skip_on_missing_imports(["openai"])
+@run_for_optional_imports(["openai"], "openai")
 def _test_oai_chatgpt_gpt4(save=False):
     run_notebook("oai_chatgpt_gpt4.ipynb", save=save)
 
 
-@pytest.mark.openai
+@run_for_optional_imports("openai", "openai")
 @pytest.mark.skipif(
     not sys.version.startswith("3.12"),
     reason="do not run if py!=3.12",
 )
-@skip_on_missing_imports(["openai"])
+@run_for_optional_imports(["openai"], "openai")
 def test_agentchat_groupchat_finite_state_machine(save=False):
     run_notebook("agentchat_groupchat_finite_state_machine.ipynb", save=save)
 
 
-@pytest.mark.openai
+@run_for_optional_imports("openai", "openai")
 @pytest.mark.skipif(
     not sys.version.startswith("3.11"),
     reason="do not run if py!=3.11",
 )
-@skip_on_missing_imports(["openai"])
+@run_for_optional_imports(["openai"], "openai")
 def test_agentchat_cost_token_tracking(save=False):
     run_notebook("agentchat_cost_token_tracking.ipynb", save=save)
 
 
-@pytest.mark.openai
+@run_for_optional_imports("openai", "openai")
 @pytest.mark.skipif(
     not sys.version.startswith("3.11"),
     reason="do not run if py!=3.11",
 )
-@skip_on_missing_imports(["openai"])
+@run_for_optional_imports(["openai"], "openai")
 def test_agentchat_groupchat_stateflow(save=False):
     run_notebook("agentchat_groupchat_stateflow.ipynb", save=save)
 

{ag2-0.8.0b1 → ag2-0.8.1}/test/test_retrieve_utils.py

@@ -13,7 +13,7 @@ from contextlib import suppress
 
 import pytest
 
-from autogen.import_utils import optional_import_block, skip_on_missing_imports
+from autogen.import_utils import optional_import_block, run_for_optional_imports
 from autogen.retrieve_utils import (
     create_vector_db_from_dir,
     extract_text_from_pdf,
@@ -36,7 +36,7 @@ simplify the process of building applications that leverage the power of LLMs, a
 integration, testing, and deployment."""
 
 
-@skip_on_missing_imports(["bs4", "chromadb", "markdownify", "pypdf"], "retrievechat")
+@run_for_optional_imports(["bs4", "chromadb", "markdownify", "pypdf"], "retrievechat")
 class TestRetrieveUtils:
     def test_split_text_to_chunks(self):
         long_text = "A" * 10000
@@ -127,7 +127,10 @@ class TestRetrieveUtils:
         results = query_vector_db(["autogen"], client=client)
         assert isinstance(results, dict) and any("autogen" in res[0].lower() for res in results.get("documents", []))
 
-    @skip_on_missing_imports(["lancedb"], "unknown")
+    @pytest.mark.skip(
+        reason="This test is failing due to lancedb  missing in project install, TODO: add lancedb installation to CI?"
+    )
+    @run_for_optional_imports(["lancedb"], "unknown")
     def test_custom_vector_db(self):
         with optional_import_block() as result:
             import lancedb
@@ -231,7 +234,10 @@ class TestRetrieveUtils:
         print(results["ids"][0])
         assert len(results["ids"][0]) > 0
 
-    @skip_on_missing_imports(["unstructured"], "unknown")
+    @pytest.mark.skip(
+        reason="This test is failing due to unstructured  missing in project install, TODO: add unstructured installation to CI?"
+    )
+    @run_for_optional_imports(["unstructured"], "unknown")
     def test_unstructured(self):
         pdf_file_path = os.path.join(test_dir, "example.pdf")
         txt_file_path = os.path.join(test_dir, "example.txt")

{ag2-0.8.0b1 → ag2-0.8.1}/test/test_token_count.py

@@ -8,7 +8,7 @@
 
 import pytest
 
-from autogen.import_utils import skip_on_missing_imports
+from autogen.import_utils import run_for_optional_imports
 from autogen.token_count_utils import (
     _num_token_from_messages,
     count_token,
@@ -89,7 +89,7 @@ def test_num_token_from_messages(model: str, expected_count: int) -> None:
     assert _num_token_from_messages(messages=messages, model=model) == expected_count
 
 
-@skip_on_missing_imports("PIL", "unknown")
+@run_for_optional_imports("PIL", "unknown")
 def test_num_tokens_from_gpt_image():
     # mock num_tokens_from_gpt_image function
     base64_encoded_image = (

ag2-0.8.0b1/ag2.egg-info/requires.txt

@@ -1,145 +0,0 @@
-pyautogen==0.8.0b1
-
-[anthropic]
-pyautogen[anthropic]==0.8.0b1
-
-[autobuild]
-pyautogen[autobuild]==0.8.0b1
-
-[bedrock]
-pyautogen[bedrock]==0.8.0b1
-
-[blendsearch]
-pyautogen[blendsearch]==0.8.0b1
-
-[browser-use]
-pyautogen[browser-use]==0.8.0b1
-
-[captainagent]
-pyautogen[captainagent]==0.8.0b1
-
-[cerebras]
-pyautogen[cerebras]==0.8.0b1
-
-[cohere]
-pyautogen[cohere]==0.8.0b1
-
-[commsagent-discord]
-pyautogen[commsagent-discord]==0.8.0b1
-
-[commsagent-slack]
-pyautogen[commsagent-slack]==0.8.0b1
-
-[commsagent-telegram]
-pyautogen[commsagent-telegram]==0.8.0b1
-
-[cosmosdb]
-pyautogen[cosmosdb]==0.8.0b1
-
-[crawl4ai]
-pyautogen[crawl4ai]==0.8.0b1
-
-[dev]
-pyautogen[dev]==0.8.0b1
-
-[docs]
-pyautogen[docs]==0.8.0b1
-
-[flaml]
-pyautogen[flaml]==0.8.0b1
-
-[gemini]
-pyautogen[gemini]==0.8.0b1
-
-[graph]
-pyautogen[graph]==0.8.0b1
-
-[graph-rag-falkor-db]
-pyautogen[graph-rag-falkor-db]==0.8.0b1
-
-[groq]
-pyautogen[groq]==0.8.0b1
-
-[interop]
-pyautogen[interop]==0.8.0b1
-
-[interop-crewai]
-pyautogen[interop-crewai]==0.8.0b1
-
-[interop-langchain]
-pyautogen[interop-langchain]==0.8.0b1
-
-[interop-pydantic-ai]
-pyautogen[interop-pydantic-ai]==0.8.0b1
-
-[jupyter-executor]
-pyautogen[jupyter-executor]==0.8.0b1
-
-[lint]
-pyautogen[lint]==0.8.0b1
-
-[lmm]
-pyautogen[lmm]==0.8.0b1
-
-[long-context]
-pyautogen[long-context]==0.8.0b1
-
-[mathchat]
-pyautogen[mathchat]==0.8.0b1
-
-[mistral]
-pyautogen[mistral]==0.8.0b1
-
-[neo4j]
-pyautogen[neo4j]==0.8.0b1
-
-[ollama]
-pyautogen[ollama]==0.8.0b1
-
-[openai]
-pyautogen[openai]==0.8.0b1
-
-[openai-realtime]
-pyautogen[openai-realtime]==0.8.0b1
-
-[rag]
-pyautogen[rag]==0.8.0b1
-
-[redis]
-pyautogen[redis]==0.8.0b1
-
-[retrievechat]
-pyautogen[retrievechat]==0.8.0b1
-
-[retrievechat-couchbase]
-pyautogen[retrievechat-couchbase]==0.8.0b1
-
-[retrievechat-mongodb]
-pyautogen[retrievechat-mongodb]==0.8.0b1
-
-[retrievechat-pgvector]
-pyautogen[retrievechat-pgvector]==0.8.0b1
-
-[retrievechat-qdrant]
-pyautogen[retrievechat-qdrant]==0.8.0b1
-
-[teachable]
-pyautogen[teachable]==0.8.0b1
-
-[test]
-pyautogen[test]==0.8.0b1
-
-[together]
-pyautogen[together]==0.8.0b1
-
-[twilio]
-pyautogen[twilio]==0.8.0b1
-
-[types]
-pyautogen[types]==0.8.0b1
-
-[websockets]
-pyautogen[websockets]==0.8.0b1
-
-[websurfer]
-pyautogen[websurfer]==0.8.0b1