ag2 0.6.1__tar.gz → 0.7.0__tar.gz

{ag2-0.6.1/ag2.egg-info → ag2-0.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ag2
-Version: 0.6.1
+Version: 0.7.0
 Summary: Alias package for pyautogen
 Home-page: https://github.com/ag2ai/ag2
 Author: Chi Wang & Qingyun Wu
@@ -12,41 +12,41 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9,<3.14
 Description-Content-Type: text/markdown
-Provides-Extra: anthropic
-Provides-Extra: autobuild
-Provides-Extra: bedrock
+Provides-Extra: test
 Provides-Extra: blendsearch
-Provides-Extra: captainagent
-Provides-Extra: cerebras
-Provides-Extra: cohere
-Provides-Extra: cosmosdb
-Provides-Extra: gemini
-Provides-Extra: graph
-Provides-Extra: graph-rag-falkor-db
-Provides-Extra: groq
-Provides-Extra: interop
-Provides-Extra: interop-crewai
-Provides-Extra: interop-langchain
-Provides-Extra: interop-pydantic-ai
-Provides-Extra: jupyter-executor
-Provides-Extra: lmm
-Provides-Extra: long-context
 Provides-Extra: mathchat
-Provides-Extra: mistral
-Provides-Extra: neo4j
-Provides-Extra: ollama
-Provides-Extra: redis
 Provides-Extra: retrievechat
-Provides-Extra: retrievechat-mongodb
 Provides-Extra: retrievechat-pgvector
+Provides-Extra: retrievechat-mongodb
 Provides-Extra: retrievechat-qdrant
+Provides-Extra: graph-rag-falkor-db
+Provides-Extra: autobuild
+Provides-Extra: captainagent
 Provides-Extra: teachable
-Provides-Extra: test
+Provides-Extra: lmm
+Provides-Extra: graph
+Provides-Extra: gemini
 Provides-Extra: together
-Provides-Extra: twilio
-Provides-Extra: types
-Provides-Extra: websockets
 Provides-Extra: websurfer
+Provides-Extra: redis
+Provides-Extra: cosmosdb
+Provides-Extra: websockets
+Provides-Extra: jupyter-executor
+Provides-Extra: types
+Provides-Extra: long-context
+Provides-Extra: anthropic
+Provides-Extra: cerebras
+Provides-Extra: mistral
+Provides-Extra: groq
+Provides-Extra: cohere
+Provides-Extra: ollama
+Provides-Extra: bedrock
+Provides-Extra: twilio
+Provides-Extra: interop-crewai
+Provides-Extra: interop-langchain
+Provides-Extra: interop-pydantic-ai
+Provides-Extra: interop
+Provides-Extra: neo4j
 License-File: LICENSE
 License-File: NOTICE.md
 

{ag2-0.6.1 → ag2-0.7.0/ag2.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ag2
-Version: 0.6.1
+Version: 0.7.0
 Summary: Alias package for pyautogen
 Home-page: https://github.com/ag2ai/ag2
 Author: Chi Wang & Qingyun Wu
@@ -12,41 +12,41 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9,<3.14
 Description-Content-Type: text/markdown
-Provides-Extra: anthropic
-Provides-Extra: autobuild
-Provides-Extra: bedrock
+Provides-Extra: test
 Provides-Extra: blendsearch
-Provides-Extra: captainagent
-Provides-Extra: cerebras
-Provides-Extra: cohere
-Provides-Extra: cosmosdb
-Provides-Extra: gemini
-Provides-Extra: graph
-Provides-Extra: graph-rag-falkor-db
-Provides-Extra: groq
-Provides-Extra: interop
-Provides-Extra: interop-crewai
-Provides-Extra: interop-langchain
-Provides-Extra: interop-pydantic-ai
-Provides-Extra: jupyter-executor
-Provides-Extra: lmm
-Provides-Extra: long-context
 Provides-Extra: mathchat
-Provides-Extra: mistral
-Provides-Extra: neo4j
-Provides-Extra: ollama
-Provides-Extra: redis
 Provides-Extra: retrievechat
-Provides-Extra: retrievechat-mongodb
 Provides-Extra: retrievechat-pgvector
+Provides-Extra: retrievechat-mongodb
 Provides-Extra: retrievechat-qdrant
+Provides-Extra: graph-rag-falkor-db
+Provides-Extra: autobuild
+Provides-Extra: captainagent
 Provides-Extra: teachable
-Provides-Extra: test
+Provides-Extra: lmm
+Provides-Extra: graph
+Provides-Extra: gemini
 Provides-Extra: together
-Provides-Extra: twilio
-Provides-Extra: types
-Provides-Extra: websockets
 Provides-Extra: websurfer
+Provides-Extra: redis
+Provides-Extra: cosmosdb
+Provides-Extra: websockets
+Provides-Extra: jupyter-executor
+Provides-Extra: types
+Provides-Extra: long-context
+Provides-Extra: anthropic
+Provides-Extra: cerebras
+Provides-Extra: mistral
+Provides-Extra: groq
+Provides-Extra: cohere
+Provides-Extra: ollama
+Provides-Extra: bedrock
+Provides-Extra: twilio
+Provides-Extra: interop-crewai
+Provides-Extra: interop-langchain
+Provides-Extra: interop-pydantic-ai
+Provides-Extra: interop
+Provides-Extra: neo4j
 License-File: LICENSE
 License-File: NOTICE.md
 

{ag2-0.6.1 → ag2-0.7.0}/ag2.egg-info/SOURCES.txt

@@ -2,7 +2,6 @@ LICENSE
 NOTICE.md
 README.md
 pyproject.toml
-setup.py
 setup_ag2.py
 ag2.egg-info/PKG-INFO
 ag2.egg-info/SOURCES.txt
@@ -11,7 +10,6 @@ ag2.egg-info/requires.txt
 ag2.egg-info/top_level.txt
 test/test_browser_utils.py
 test/test_code_utils.py
-test/test_function_utils.py
 test/test_graph_utils.py
 test/test_logging.py
 test/test_notebook.py

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

@@ -0,0 +1,106 @@
+pyautogen==0.7.0
+
+[anthropic]
+pyautogen[anthropic]==0.7.0
+
+[autobuild]
+pyautogen[autobuild]==0.7.0
+
+[bedrock]
+pyautogen[bedrock]==0.7.0
+
+[blendsearch]
+pyautogen[blendsearch]==0.7.0
+
+[captainagent]
+pyautogen[captainagent]==0.7.0
+
+[cerebras]
+pyautogen[cerebras]==0.7.0
+
+[cohere]
+pyautogen[cohere]==0.7.0
+
+[cosmosdb]
+pyautogen[cosmosdb]==0.7.0
+
+[gemini]
+pyautogen[gemini]==0.7.0
+
+[graph]
+pyautogen[graph]==0.7.0
+
+[graph-rag-falkor-db]
+pyautogen[graph-rag-falkor-db]==0.7.0
+
+[groq]
+pyautogen[groq]==0.7.0
+
+[interop]
+pyautogen[interop]==0.7.0
+
+[interop-crewai]
+pyautogen[interop-crewai]==0.7.0
+
+[interop-langchain]
+pyautogen[interop-langchain]==0.7.0
+
+[interop-pydantic-ai]
+pyautogen[interop-pydantic-ai]==0.7.0
+
+[jupyter-executor]
+pyautogen[jupyter-executor]==0.7.0
+
+[lmm]
+pyautogen[lmm]==0.7.0
+
+[long-context]
+pyautogen[long-context]==0.7.0
+
+[mathchat]
+pyautogen[mathchat]==0.7.0
+
+[mistral]
+pyautogen[mistral]==0.7.0
+
+[neo4j]
+pyautogen[neo4j]==0.7.0
+
+[ollama]
+pyautogen[ollama]==0.7.0
+
+[redis]
+pyautogen[redis]==0.7.0
+
+[retrievechat]
+pyautogen[retrievechat]==0.7.0
+
+[retrievechat-mongodb]
+pyautogen[retrievechat-mongodb]==0.7.0
+
+[retrievechat-pgvector]
+pyautogen[retrievechat-pgvector]==0.7.0
+
+[retrievechat-qdrant]
+pyautogen[retrievechat-qdrant]==0.7.0
+
+[teachable]
+pyautogen[teachable]==0.7.0
+
+[test]
+pyautogen[test]==0.7.0
+
+[together]
+pyautogen[together]==0.7.0
+
+[twilio]
+pyautogen[twilio]==0.7.0
+
+[types]
+pyautogen[types]==0.7.0
+
+[websockets]
+pyautogen[websockets]==0.7.0
+
+[websurfer]
+pyautogen[websurfer]==0.7.0

ag2-0.7.0/pyproject.toml

@@ -0,0 +1,311 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[metadata]
+license_file = "LICENSE"
+description-file = "README.md"
+
+[project]
+name = "pyautogen"
+description = "A programming framework for agentic AI"
+readme = "README.md"
+authors = [
+    {name = "Chi Wang", email = "support@ag2.ai"},
+    {name = "Qingyun Wu", email = "support@ag2.ai"},
+]
+
+keywords = [
+    "ai",
+    "agent",
+    "autogen",
+    "ag2",
+    "pyautogen",
+    "ag2.ai",
+    "ag2ai",
+    "agentic"
+]
+
+requires-python = ">=3.9,<3.14"
+
+dynamic = ["version"]
+
+classifiers = [
+    "Development Status :: 5 - Production/Stable",
+    "License :: OSI Approved :: Apache Software License",
+    "Programming Language :: Python",
+    "Programming Language :: Python :: Implementation :: CPython",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Operating System :: OS Independent",
+    "Topic :: Software Development :: Libraries :: Application Frameworks",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+    "Topic :: Software Development :: Libraries",
+    "Topic :: Software Development",
+    "Intended Audience :: Developers",
+    "Intended Audience :: Information Technology",
+]
+
+dependencies = [
+    "openai>=1.58",
+    "diskcache",
+    "termcolor",
+    "flaml",
+    # numpy is installed by flaml, but we want to pin the version to below 2.x (see https://github.com/microsoft/autogen/issues/1960)
+    "numpy>=2.1; python_version>='3.13'",  # numpy 2.1+ required for Python 3.13
+    "numpy>=1.24.0,<2.0.0; python_version<'3.13'",  # numpy 1.24+ for older Python versions
+    "python-dotenv",
+    "tiktoken",
+    # Disallowing 2.6.0 can be removed when this is fixed https://github.com/pydantic/pydantic/issues/8705
+    "pydantic>=1.10,<3,!=2.6.0",  # could be both V1 and V2
+    "docker",
+    "packaging",
+    "websockets>=14,<15",
+    "asyncer==0.0.8",
+    "fast_depends>=2.4.12,<3",
+]
+
+[project.optional-dependencies]
+# public distributions
+jupyter-executor = [
+    "jupyter-kernel-gateway",
+    "websocket-client",
+    "requests",
+    "jupyter-client>=8.6.0",
+    "ipykernel>=6.29.0",
+]
+
+retrievechat = [
+    "protobuf==4.25.3",
+    "chromadb==0.5.3",
+    "sentence_transformers",
+    "pypdf",
+    "ipython",
+    "beautifulsoup4",
+    "markdownify",
+]
+
+retrievechat-pgvector = [
+    "pyautogen[retrievechat]",
+    "pgvector>=0.2.5",
+    "psycopg[binary]>=3.1.18; platform_system=='Windows' or platform_system=='Darwin'",
+    "psycopg>=3.1.18; platform_system=='Linux'",
+]
+
+retrievechat-mongodb = [
+    "pyautogen[retrievechat]",
+    "pymongo>=4.0.0",
+]
+
+retrievechat-qdrant = [
+    "pyautogen[retrievechat]",
+    "qdrant_client",
+    "fastembed>=0.3.1",
+]
+
+graph-rag-falkor-db = [
+    "graphrag_sdk==0.3.3",
+    "falkordb>=1.0.10"
+]
+
+neo4j = [
+    "docx2txt==0.8",
+    "llama-index==0.12.5",
+    "llama-index-graph-stores-neo4j==0.4.2",
+    "llama-index-core==0.12.5",
+    "llama-index-readers-web==0.3.3",
+]
+
+# used for agentchat_realtime_swarm notebook and realtime agent twilio demo
+twilio = [
+    "fastapi>=0.115.0,<1",
+    "uvicorn>=0.30.6,<1",
+    "twilio>=9.3.2"
+]
+
+interop-crewai = [
+    "crewai[tools]>=0.86,<1; python_version>='3.10' and python_version<'3.13'",
+    "weaviate-client==4.10.2; python_version>='3.10' and python_version<'3.13'",
+]
+interop-langchain = ["langchain-community>=0.3.12,<1"]
+interop-pydantic-ai = ["pydantic-ai==0.0.13"]
+interop =[
+    "pyautogen[interop-crewai, interop-langchain, interop-pydantic-ai]",
+]
+
+types = [
+    "mypy==1.9.0",
+    "pyautogen[test, jupyter-executor, interop]",
+]
+
+# pysqlite3-binary used so it doesn't need to compile pysqlite3
+autobuild = ["chromadb", "sentence-transformers", "huggingface-hub", "pysqlite3-binary"]
+
+blendsearch = ["flaml[blendsearch]"]
+mathchat = ["sympy", "pydantic==1.10.9", "wolframalpha"]
+captainagent = ["pyautogen[autobuild]", "pandas"]
+teachable = ["chromadb"]
+lmm = ["replicate", "pillow"]
+graph = ["networkx", "matplotlib"]
+gemini = [
+    "google-generativeai>=0.5,<1",
+    "google-cloud-aiplatform",
+    "google-auth",
+    "pillow",
+    "pydantic",
+    "jsonschema",
+]
+together = ["together>=1.2"]
+websurfer = ["beautifulsoup4", "markdownify", "pdfminer.six", "pathvalidate"]
+redis = ["redis"]
+cosmosdb = ["azure-cosmos>=4.2.0"]
+websockets = ["websockets>=14.0,<15"]
+long-context = ["llmlingua<0.3"]
+anthropic = ["anthropic[vertex]>=0.23.1"]
+cerebras = ["cerebras_cloud_sdk>=1.0.0"]
+mistral = ["mistralai>=1.0.1"]
+groq = ["groq>=0.9.0"]
+cohere = ["cohere>=5.5.8"]
+ollama = ["ollama>=0.3.3", "fix_busted_json>=0.0.18"]
+bedrock = ["boto3>=1.34.149"]
+
+# dev dependencies
+test = [
+    "ipykernel",
+    "nbconvert",
+    "nbformat",
+    "pre-commit",
+    "pytest-cov>=5",
+    "pytest-asyncio",
+    "pytest>=8,<9",
+    "pandas",
+    "fastapi>=0.115.0,<1",
+]
+
+[project.urls]
+Homepage = "https://ag2.ai/"
+Documentation = "https://docs.ag2.ai/docs/Home"
+Tracker = "https://github.com/ag2ai/ag2/issues"
+Source = "https://github.com/ag2ai/ag2"
+Discord = "https://discord.gg/pAbnFJrkgZ"
+
+[tool.hatch.version]
+path = "autogen/version.py"
+
+[tool.hatch.build]
+skip-excluded-dirs = true
+exclude = ["/test", "/notebook"]
+
+[tool.hatch.build.targets.wheel]
+packages = ["autogen"]
+only-include = ["autogen", "autogen/agentchat/contrib/captainagent/tools"]
+
+[tool.hatch.build.targets.sdist]
+exclude = ["test", "notebook"]
+
+[tool.hatch.build.targets.wheel.sources]
+"autogen" = "autogen"
+"autogen/agentchat/contrib/captainagent/tools" = "autogen/agentchat/contrib/captainagent/tools"
+
+
+[tool.pytest.ini_options]
+addopts = '--cov=. --cov-append --cov-branch --cov-report=xml -m "not conda"'
+markers = ["conda: test related to conda forge distribution"]
+
+[tool.black]
+# https://github.com/psf/black
+line-length = 120
+exclude = "(.eggs|.git|.hg|.mypy_cache|.venv|_build|buck-out|build|dist)"
+
+
+[tool.ruff]
+line-length = 120
+
+[tool.ruff.lint]
+# Enable Pyflakes `E` and `F` codes by default.
+select = [
+    "E",
+    "W", # see: https://pypi.org/project/pycodestyle
+    "F", # see: https://pypi.org/project/pyflakes
+    #    "D",  # see: https://pypi.org/project/pydocstyle
+    #    "N",  # see: https://pypi.org/project/pep8-naming
+    #    "S",  # see: https://pypi.org/project/flake8-bandit
+    "I", # see: https://pypi.org/project/isort/
+]
+ignore = ["E501", "F401", "F403", "C901"]
+
+# Exclude a variety of commonly ignored directories.
+exclude = [
+    ".eggs",
+    ".git",
+    ".mypy_cache",
+    ".ruff_cache",
+    "__pypackages__",
+    "_build",
+    "build",
+    "dist",
+    "docs",
+    # This file needs to be either upgraded or removed and therefore should be
+    # ignore from type checking for now
+    "math_utils\\.py$",
+    "**/cap/py/autogencap/proto/*",
+]
+unfixable = ["F401"]
+
+[tool.ruff.lint.mccabe]
+# Unlike Flake8, default to a complexity level of 10.
+max-complexity = 10
+
+[tool.mypy]
+files = [
+    "autogen/logger",
+    "autogen/exception_utils.py",
+    "autogen/coding",
+    "autogen/oai/openai_utils.py",
+    "autogen/_pydantic.py",
+    "autogen/io",
+    "autogen/tools",
+    "autogen/interop",
+    "autogen/agentchat/realtime_agent",
+    "autogen/messages",
+    "test/test_pydantic.py",
+    "test/io",
+    "test/tools",
+    "test/interop",
+    "test/agentchat/realtime_agent",
+    "test/messages",
+]
+exclude = [
+    "autogen/math_utils\\.py",
+    "autogen/oai/completion\\.py",
+    "autogen/agentchat/contrib/compressible_agent\\.py",
+    "autogen/agentchat/contrib/math_user_proxy_agent.py",
+    "autogen/oai/openai_utils.py",
+]
+
+strict = true
+ignore_missing_imports = true
+install_types = true
+non_interactive = true
+plugins = ["pydantic.mypy"]
+
+# remove after all files in the repo are fixed
+follow_imports = "silent"
+
+# from https://blog.wolt.com/engineering/2021/09/30/professional-grade-mypy-configuration/
+disallow_untyped_defs = true
+no_implicit_optional = true
+check_untyped_defs = true
+warn_return_any = true
+show_error_codes = true
+
+warn_unused_ignores = false
+
+disallow_incomplete_defs = true
+disallow_untyped_decorators = true
+disallow_any_unimported = true

{ag2-0.6.1 → ag2-0.7.0}/test/test_browser_utils.py

@@ -10,11 +10,10 @@ import hashlib
 import math
 import os
 import re
-import sys
+from tempfile import TemporaryDirectory
 
 import pytest
 import requests
-from agentchat.test_assistant_agent import KEY_LOC  # noqa: E402
 
 BLOG_POST_URL = "https://docs.ag2.ai/blog/2023-04-21-LLM-tuning-math"
 BLOG_POST_TITLE = "Does Model and Inference Parameter Matter in LLM Applications? - A Case Study for MATH - AG2"
@@ -49,26 +48,26 @@ else:
     skip_bing = False
 
 
-def _rm_folder(path):
-    """Remove all the regular files in a folder, then deletes the folder. Assumes a flat file structure, with no subdirectories."""
-    for fname in os.listdir(path):
-        fpath = os.path.join(path, fname)
-        if os.path.isfile(fpath):
-            os.unlink(fpath)
-    os.rmdir(path)
+# def _rm_folder(path):
+#     """Remove all the regular files in a folder, then deletes the folder. Assumes a flat file structure, with no subdirectories."""
+#     for fname in os.listdir(path):
+#         fpath = os.path.join(path, fname)
+#         if os.path.isfile(fpath):
+#             os.unlink(fpath)
+#     os.rmdir(path)
+
+
+@pytest.fixture
+def downloads_folder():
+    with TemporaryDirectory() as downloads_folder:
+        yield downloads_folder
 
 
 @pytest.mark.skipif(
     skip_all,
     reason="do not run if dependency is not installed",
 )
-def test_simple_text_browser():
-    # Create a downloads folder (removing any leftover ones from prior tests)
-    downloads_folder = os.path.join(KEY_LOC, "downloads")
-    if os.path.isdir(downloads_folder):
-        _rm_folder(downloads_folder)
-    os.mkdir(downloads_folder)
-
+def test_simple_text_browser(downloads_folder: str):
     # Instantiate the browser
     user_agent = "python-requests/" + requests.__version__
     viewport_size = 1024
@@ -151,9 +150,6 @@ def test_simple_text_browser():
     viewport = browser.visit_page(PDF_URL)
     assert PDF_STRING in viewport
 
-    # Clean up
-    _rm_folder(downloads_folder)
-
 
 @pytest.mark.skipif(
     skip_all or skip_bing,