ag2 0.8.0b1__tar.gz → 0.8.2rc0__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ag2
-Version: 0.8.0b1
+Version: 0.8.2rc0
 Summary: Alias package for pyautogen
 Home-page: https://github.com/ag2ai/ag2
 Author: Chi Wang & Qingyun Wu
@@ -25,6 +25,7 @@ Provides-Extra: graph-rag-falkor-db
 Provides-Extra: rag
 Provides-Extra: crawl4ai
 Provides-Extra: browser-use
+Provides-Extra: google-search
 Provides-Extra: neo4j
 Provides-Extra: twilio
 Provides-Extra: interop-crewai
@@ -39,6 +40,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 +54,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.2rc0/ag2.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ag2
-Version: 0.8.0b1
+Version: 0.8.2rc0
 Summary: Alias package for pyautogen
 Home-page: https://github.com/ag2ai/ag2
 Author: Chi Wang & Qingyun Wu
@@ -25,6 +25,7 @@ Provides-Extra: graph-rag-falkor-db
 Provides-Extra: rag
 Provides-Extra: crawl4ai
 Provides-Extra: browser-use
+Provides-Extra: google-search
 Provides-Extra: neo4j
 Provides-Extra: twilio
 Provides-Extra: interop-crewai
@@ -39,6 +40,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 +54,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.2rc0/ag2.egg-info/requires.txt

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

{ag2-0.8.0b1 → ag2-0.8.2rc0}/pyproject.toml

@@ -60,8 +60,9 @@ dependencies = [
     "docker",
     "packaging",
     "asyncer==0.0.8",
-    "fast-depends>=2.4.12,<3",
+#    "fast-depends>=2.4.12,<3",  # integrated into the package
     "httpx>=0.28.1,<1",
+    "anyio>=3.0.0,<5.0.0"  # needed by the internal fast-depends
 ]
 
 [project.optional-dependencies]
@@ -76,11 +77,11 @@ flaml = [
 # public distributions
 
 openai = [
-    "openai>=1.58",
+    "openai>=1.66.2",
 ]
 
 openai-realtime = [
-    "autogen[openai]",
+    "pyautogen[openai]",
     "openai[realtime]",
 ]
 
@@ -139,6 +140,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",
 ]
 
 
@@ -152,12 +157,16 @@ browser-use = [
     "browser-use==0.1.37",
 ]
 
+google-search = [
+    "google-api-python-client>=2.163.0,<3.0",
+]
+
 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,13 +181,12 @@ 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]",
 ]
 
-# pysqlite3-binary used so it doesn't need to compile pysqlite3
-autobuild = ["chromadb", "sentence-transformers", "huggingface-hub", "pysqlite3-binary"]
+autobuild = ["chromadb", "sentence-transformers", "huggingface-hub"]
 
 blendsearch = ["flaml[blendsearch]"]
 mathchat = ["sympy", "wolframalpha"]
@@ -188,7 +196,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 +204,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 +222,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 +237,25 @@ 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",
+    "dirty-equals==0.9.0",
 ]
 
 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 +271,7 @@ types = [
 ]
 
 lint = [
-    "ruff==0.9.7",
+    "ruff==0.9.9",
     "codespell==2.4.1",
     "pyupgrade-directories==0.3.0",
 ]
@@ -265,7 +281,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 +310,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",
@@ -322,7 +346,12 @@ markers = [
     "interop",
     "browser_use",
     "crawl4ai",
+    "google_search",
     "websockets",
+    "commsagent_discord",
+    "commsagent_slack",
+    "commsagent-telegram",
+    "lmm",
 ]
 
 [tool.black]
@@ -379,6 +408,8 @@ select = [
 #    "UP",     # pyupgrade              https://docs.astral.sh/ruff/rules/#pydocstyle-d
 ]
 
+extend-select = ["D417"]
+
 ignore = ["E501", "F403", "C901",
     "E402",
     "E721",
@@ -404,12 +435,16 @@ 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",
+    "autogen/fast_depends",
     "autogen/import_utils.py",
     "autogen/interop",
     "autogen/io",
@@ -418,12 +453,15 @@ 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",
+#    "test/fast_depends",
     "test/interop",
     "test/io",
     "test/messages",

{ag2-0.8.0b1 → ag2-0.8.2rc0}/setup_ag2.py

@@ -41,6 +41,7 @@ setuptools.setup(
         "rag": ["pyautogen[rag]==" + __version__],
         "crawl4ai": ["pyautogen[crawl4ai]==" + __version__],
         "browser-use": ["pyautogen[browser-use]==" + __version__],
+        "google-search": ["pyautogen[google-search]==" + __version__],
         "neo4j": ["pyautogen[neo4j]==" + __version__],
         "twilio": ["pyautogen[twilio]==" + __version__],
         "interop-crewai": ["pyautogen[interop-crewai]==" + __version__],
@@ -55,6 +56,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 +70,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.2rc0}/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()