ag2 0.4.1__tar.gz → 0.5.0__tar.gz

{ag2-0.4.1/ag2.egg-info → ag2-0.5.0}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: ag2
-Version: 0.4.1
-Summary: A programming framework for agentic AI
+Version: 0.5.0
+Summary: Alias package for pyautogen
 Home-page: https://github.com/ag2ai/ag2
 Author: Chi Wang & Qingyun Wu
 Author-email: support@ag2.ai
@@ -12,35 +12,6 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8,<3.14
 Description-Content-Type: text/markdown
-Provides-Extra: anthropic
-Provides-Extra: autobuild
-Provides-Extra: bedrock
-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: jupyter-executor
-Provides-Extra: lmm
-Provides-Extra: long-context
-Provides-Extra: mathchat
-Provides-Extra: mistral
-Provides-Extra: ollama
-Provides-Extra: redis
-Provides-Extra: retrievechat
-Provides-Extra: retrievechat-mongodb
-Provides-Extra: retrievechat-pgvector
-Provides-Extra: retrievechat-qdrant
-Provides-Extra: teachable
-Provides-Extra: test
-Provides-Extra: together
-Provides-Extra: types
-Provides-Extra: websockets
-Provides-Extra: websurfer
 License-File: LICENSE
 License-File: NOTICE.md
 
@@ -71,19 +42,11 @@ License-File: NOTICE.md
 > We invite collaborators from all organizations and individuals to join the development.
 
 
-:fire: :tada: AG2 is available via `ag2` (or its alias  `autogen` or  `pyautogen`)  on PyPI! Starting with version 0.3.2, you can now install AG2 using:
-```
-pip install ag2
-```
-or
+:fire: :tada: AG2 is available via `pyautogen` (or its alias  `autogen` or  `ag2`)  on PyPI!
+
 ```
 pip install pyautogen
 ```
-or
-```
-pip install autogen
-```
-
 
 📄 **License:**
 We adopt the Apache 2.0 license from v0.3. This enhances our commitment to open-source collaboration while providing additional protections for contributors and users alike.

{ag2-0.4.1 → ag2-0.5.0}/README.md

@@ -25,19 +25,11 @@
 > We invite collaborators from all organizations and individuals to join the development.
 
 
-:fire: :tada: AG2 is available via `ag2` (or its alias  `autogen` or  `pyautogen`)  on PyPI! Starting with version 0.3.2, you can now install AG2 using:
-```
-pip install ag2
-```
-or
+:fire: :tada: AG2 is available via `pyautogen` (or its alias  `autogen` or  `ag2`)  on PyPI!
+
 ```
 pip install pyautogen
 ```
-or
-```
-pip install autogen
-```
-
 
 📄 **License:**
 We adopt the Apache 2.0 license from v0.3. This enhances our commitment to open-source collaboration while providing additional protections for contributors and users alike.

{ag2-0.4.1 → ag2-0.5.0/ag2.egg-info}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: ag2
-Version: 0.4.1
-Summary: A programming framework for agentic AI
+Version: 0.5.0
+Summary: Alias package for pyautogen
 Home-page: https://github.com/ag2ai/ag2
 Author: Chi Wang & Qingyun Wu
 Author-email: support@ag2.ai
@@ -12,35 +12,6 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8,<3.14
 Description-Content-Type: text/markdown
-Provides-Extra: anthropic
-Provides-Extra: autobuild
-Provides-Extra: bedrock
-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: jupyter-executor
-Provides-Extra: lmm
-Provides-Extra: long-context
-Provides-Extra: mathchat
-Provides-Extra: mistral
-Provides-Extra: ollama
-Provides-Extra: redis
-Provides-Extra: retrievechat
-Provides-Extra: retrievechat-mongodb
-Provides-Extra: retrievechat-pgvector
-Provides-Extra: retrievechat-qdrant
-Provides-Extra: teachable
-Provides-Extra: test
-Provides-Extra: together
-Provides-Extra: types
-Provides-Extra: websockets
-Provides-Extra: websurfer
 License-File: LICENSE
 License-File: NOTICE.md
 
@@ -71,19 +42,11 @@ License-File: NOTICE.md
 > We invite collaborators from all organizations and individuals to join the development.
 
 
-:fire: :tada: AG2 is available via `ag2` (or its alias  `autogen` or  `pyautogen`)  on PyPI! Starting with version 0.3.2, you can now install AG2 using:
-```
-pip install ag2
-```
-or
+:fire: :tada: AG2 is available via `pyautogen` (or its alias  `autogen` or  `ag2`)  on PyPI!
+
 ```
 pip install pyautogen
 ```
-or
-```
-pip install autogen
-```
-
 
 📄 **License:**
 We adopt the Apache 2.0 license from v0.3. This enhances our commitment to open-source collaboration while providing additional protections for contributors and users alike.

ag2-0.5.0/ag2.egg-info/SOURCES.txt

@@ -0,0 +1,20 @@
+LICENSE
+NOTICE.md
+README.md
+pyproject.toml
+setup.py
+setup_ag2.py
+ag2.egg-info/PKG-INFO
+ag2.egg-info/SOURCES.txt
+ag2.egg-info/dependency_links.txt
+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
+test/test_pydantic.py
+test/test_retrieve_utils.py
+test/test_token_count.py

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

@@ -0,0 +1 @@
+pyautogen==0.5.0

ag2-0.5.0/ag2.egg-info/top_level.txt

@@ -0,0 +1 @@
+

{ag2-0.4.1 → ag2-0.5.0}/setup.py

@@ -60,8 +60,12 @@ retrieve_chat = [
 
 retrieve_chat_pgvector = [*retrieve_chat, "pgvector>=0.2.5"]
 
-graph_rag_falkor_db = [
-    "graphrag_sdk==0.3.3",
+graph_rag_falkor_db = ["graphrag_sdk==0.3.3", "falkordb>=1.0.10"]
+
+neo4j_graph_rag = [
+    "llama_index==0.11.8",
+    "llama-index-graph-stores-neo4j=0.3.0",
+    "llma-index-core==0.11.8",
 ]
 
 if current_os in ["Windows", "Darwin"]:
@@ -113,17 +117,8 @@ extra_require = {
     "bedrock": ["boto3>=1.34.149"],
 }
 
-
-if "--name" in sys.argv:
-    index = sys.argv.index("--name")
-    sys.argv.pop(index)  # Removes --name
-    package_name = sys.argv.pop(index)  # Removes the value after --name
-else:
-    package_name = "ag2"
-
-
 setuptools.setup(
-    name=package_name,
+    name="pyautogen",
     version=__version__,
     author="Chi Wang & Qingyun Wu",
     author_email="support@ag2.ai",

ag2-0.5.0/setup_ag2.py

@@ -0,0 +1,37 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import os
+
+import setuptools
+
+here = os.path.abspath(os.path.dirname(__file__))
+
+with open("README.md", "r", encoding="UTF-8") as fh:
+    long_description = fh.read()
+
+# Get the code version
+version = {}
+with open(os.path.join(here, "autogen/version.py")) as fp:
+    exec(fp.read(), version)
+__version__ = version["__version__"]
+
+setuptools.setup(
+    name="ag2",
+    version=__version__,
+    description="Alias package for pyautogen",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    install_requires=["pyautogen==" + __version__],
+    url="https://github.com/ag2ai/ag2",
+    author="Chi Wang & Qingyun Wu",
+    author_email="support@ag2.ai",
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: Apache Software License",
+        "Operating System :: OS Independent",
+    ],
+    license="Apache Software License 2.0",
+    python_requires=">=3.8,<3.14",
+)

{ag2-0.4.1 → ag2-0.5.0}/test/test_logging.py

@@ -5,8 +5,10 @@
 # Portions derived from  https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
 import json
+import os
 import sqlite3
 import uuid
+from pathlib import Path
 from typing import Any, Callable
 from unittest.mock import Mock, patch
 
@@ -232,12 +234,16 @@ def test_log_oai_client(db_connection):
 
 def test_to_dict():
     from autogen import Agent
+    from autogen.coding import LocalCommandLineCodeExecutor
+
+    agent_executor = LocalCommandLineCodeExecutor(work_dir=Path("."))
 
     agent1 = autogen.ConversableAgent(
         "alice",
         human_input_mode="NEVER",
         llm_config=False,
         default_auto_reply="This is alice speaking.",
+        code_execution_config={"executor": agent_executor},
     )
 
     agent2 = autogen.ConversableAgent(
@@ -256,6 +262,7 @@ def test_to_dict():
             self.d = None
             self.test_function = lambda x, y: x + y
             self.extra_key = "remove this key"
+            self.path = Path("/to/something")
 
     class Bar(object):
         def init(self):
@@ -270,6 +277,7 @@ def test_to_dict():
     bar = Bar()
     bar.build()
 
+    expected_path = "\\to\\something" if os.name == "nt" else "/to/something"
     expected_foo_val_field = [
         {
             "a": 1.234,
@@ -277,6 +285,7 @@ def test_to_dict():
             "c": {"some_key": [7, 8, 9]},
             "d": None,
             "test_function": "self.test_function = lambda x, y: x + y",
+            "path": expected_path,
         }
     ]
 

ag2-0.4.1/ag2.egg-info/SOURCES.txt

@@ -1,171 +0,0 @@
-LICENSE
-NOTICE.md
-README.md
-pyproject.toml
-setup.py
-ag2.egg-info/PKG-INFO
-ag2.egg-info/SOURCES.txt
-ag2.egg-info/dependency_links.txt
-ag2.egg-info/requires.txt
-ag2.egg-info/top_level.txt
-autogen/__init__.py
-autogen/_pydantic.py
-autogen/browser_utils.py
-autogen/code_utils.py
-autogen/exception_utils.py
-autogen/formatting_utils.py
-autogen/function_utils.py
-autogen/graph_utils.py
-autogen/math_utils.py
-autogen/retrieve_utils.py
-autogen/runtime_logging.py
-autogen/token_count_utils.py
-autogen/types.py
-autogen/version.py
-autogen/agentchat/__init__.py
-autogen/agentchat/agent.py
-autogen/agentchat/assistant_agent.py
-autogen/agentchat/chat.py
-autogen/agentchat/conversable_agent.py
-autogen/agentchat/groupchat.py
-autogen/agentchat/user_proxy_agent.py
-autogen/agentchat/utils.py
-autogen/agentchat/contrib/__init__.py
-autogen/agentchat/contrib/agent_builder.py
-autogen/agentchat/contrib/agent_optimizer.py
-autogen/agentchat/contrib/captainagent.py
-autogen/agentchat/contrib/gpt_assistant_agent.py
-autogen/agentchat/contrib/img_utils.py
-autogen/agentchat/contrib/llamaindex_conversable_agent.py
-autogen/agentchat/contrib/llava_agent.py
-autogen/agentchat/contrib/math_user_proxy_agent.py
-autogen/agentchat/contrib/multimodal_conversable_agent.py
-autogen/agentchat/contrib/qdrant_retrieve_user_proxy_agent.py
-autogen/agentchat/contrib/retrieve_assistant_agent.py
-autogen/agentchat/contrib/retrieve_user_proxy_agent.py
-autogen/agentchat/contrib/society_of_mind_agent.py
-autogen/agentchat/contrib/swarm_agent.py
-autogen/agentchat/contrib/text_analyzer_agent.py
-autogen/agentchat/contrib/tool_retriever.py
-autogen/agentchat/contrib/web_surfer.py
-autogen/agentchat/contrib/agent_eval/agent_eval.py
-autogen/agentchat/contrib/agent_eval/criterion.py
-autogen/agentchat/contrib/agent_eval/critic_agent.py
-autogen/agentchat/contrib/agent_eval/quantifier_agent.py
-autogen/agentchat/contrib/agent_eval/subcritic_agent.py
-autogen/agentchat/contrib/agent_eval/task.py
-autogen/agentchat/contrib/capabilities/__init__.py
-autogen/agentchat/contrib/capabilities/agent_capability.py
-autogen/agentchat/contrib/capabilities/generate_images.py
-autogen/agentchat/contrib/capabilities/teachability.py
-autogen/agentchat/contrib/capabilities/text_compressors.py
-autogen/agentchat/contrib/capabilities/transform_messages.py
-autogen/agentchat/contrib/capabilities/transforms.py
-autogen/agentchat/contrib/capabilities/transforms_util.py
-autogen/agentchat/contrib/capabilities/vision_capability.py
-autogen/agentchat/contrib/captainagent/tools/__init__.py
-autogen/agentchat/contrib/captainagent/tools/requirements.txt
-autogen/agentchat/contrib/captainagent/tools/tool_description.tsv
-autogen/agentchat/contrib/captainagent/tools/data_analysis/calculate_correlation.py
-autogen/agentchat/contrib/captainagent/tools/data_analysis/calculate_skewness_and_kurtosis.py
-autogen/agentchat/contrib/captainagent/tools/data_analysis/detect_outlier_iqr.py
-autogen/agentchat/contrib/captainagent/tools/data_analysis/detect_outlier_zscore.py
-autogen/agentchat/contrib/captainagent/tools/data_analysis/explore_csv.py
-autogen/agentchat/contrib/captainagent/tools/data_analysis/shapiro_wilk_test.py
-autogen/agentchat/contrib/captainagent/tools/information_retrieval/arxiv_download.py
-autogen/agentchat/contrib/captainagent/tools/information_retrieval/arxiv_search.py
-autogen/agentchat/contrib/captainagent/tools/information_retrieval/extract_pdf_image.py
-autogen/agentchat/contrib/captainagent/tools/information_retrieval/extract_pdf_text.py
-autogen/agentchat/contrib/captainagent/tools/information_retrieval/get_wikipedia_text.py
-autogen/agentchat/contrib/captainagent/tools/information_retrieval/get_youtube_caption.py
-autogen/agentchat/contrib/captainagent/tools/information_retrieval/image_qa.py
-autogen/agentchat/contrib/captainagent/tools/information_retrieval/optical_character_recognition.py
-autogen/agentchat/contrib/captainagent/tools/information_retrieval/perform_web_search.py
-autogen/agentchat/contrib/captainagent/tools/information_retrieval/scrape_wikipedia_tables.py
-autogen/agentchat/contrib/captainagent/tools/information_retrieval/transcribe_audio_file.py
-autogen/agentchat/contrib/captainagent/tools/information_retrieval/youtube_download.py
-autogen/agentchat/contrib/captainagent/tools/math/calculate_circle_area_from_diameter.py
-autogen/agentchat/contrib/captainagent/tools/math/calculate_day_of_the_week.py
-autogen/agentchat/contrib/captainagent/tools/math/calculate_fraction_sum.py
-autogen/agentchat/contrib/captainagent/tools/math/calculate_matrix_power.py
-autogen/agentchat/contrib/captainagent/tools/math/calculate_reflected_point.py
-autogen/agentchat/contrib/captainagent/tools/math/complex_numbers_product.py
-autogen/agentchat/contrib/captainagent/tools/math/compute_currency_conversion.py
-autogen/agentchat/contrib/captainagent/tools/math/count_distinct_permutations.py
-autogen/agentchat/contrib/captainagent/tools/math/evaluate_expression.py
-autogen/agentchat/contrib/captainagent/tools/math/find_continuity_point.py
-autogen/agentchat/contrib/captainagent/tools/math/fraction_to_mixed_numbers.py
-autogen/agentchat/contrib/captainagent/tools/math/modular_inverse_sum.py
-autogen/agentchat/contrib/captainagent/tools/math/simplify_mixed_numbers.py
-autogen/agentchat/contrib/captainagent/tools/math/sum_of_digit_factorials.py
-autogen/agentchat/contrib/captainagent/tools/math/sum_of_primes_below.py
-autogen/agentchat/contrib/graph_rag/__init__.py
-autogen/agentchat/contrib/graph_rag/document.py
-autogen/agentchat/contrib/graph_rag/falkor_graph_query_engine.py
-autogen/agentchat/contrib/graph_rag/falkor_graph_rag_capability.py
-autogen/agentchat/contrib/graph_rag/graph_query_engine.py
-autogen/agentchat/contrib/graph_rag/graph_rag_capability.py
-autogen/agentchat/contrib/vectordb/__init__.py
-autogen/agentchat/contrib/vectordb/base.py
-autogen/agentchat/contrib/vectordb/chromadb.py
-autogen/agentchat/contrib/vectordb/mongodb.py
-autogen/agentchat/contrib/vectordb/pgvectordb.py
-autogen/agentchat/contrib/vectordb/qdrant.py
-autogen/agentchat/contrib/vectordb/utils.py
-autogen/cache/__init__.py
-autogen/cache/abstract_cache_base.py
-autogen/cache/cache.py
-autogen/cache/cache_factory.py
-autogen/cache/cosmos_db_cache.py
-autogen/cache/disk_cache.py
-autogen/cache/in_memory_cache.py
-autogen/cache/redis_cache.py
-autogen/coding/__init__.py
-autogen/coding/base.py
-autogen/coding/docker_commandline_code_executor.py
-autogen/coding/factory.py
-autogen/coding/func_with_reqs.py
-autogen/coding/local_commandline_code_executor.py
-autogen/coding/markdown_code_extractor.py
-autogen/coding/utils.py
-autogen/coding/jupyter/__init__.py
-autogen/coding/jupyter/base.py
-autogen/coding/jupyter/docker_jupyter_server.py
-autogen/coding/jupyter/embedded_ipython_code_executor.py
-autogen/coding/jupyter/jupyter_client.py
-autogen/coding/jupyter/jupyter_code_executor.py
-autogen/coding/jupyter/local_jupyter_server.py
-autogen/extensions/__init__.py
-autogen/io/__init__.py
-autogen/io/base.py
-autogen/io/console.py
-autogen/io/websockets.py
-autogen/logger/__init__.py
-autogen/logger/base_logger.py
-autogen/logger/file_logger.py
-autogen/logger/logger_factory.py
-autogen/logger/logger_utils.py
-autogen/logger/sqlite_logger.py
-autogen/oai/__init__.py
-autogen/oai/anthropic.py
-autogen/oai/bedrock.py
-autogen/oai/cerebras.py
-autogen/oai/client.py
-autogen/oai/client_utils.py
-autogen/oai/cohere.py
-autogen/oai/completion.py
-autogen/oai/gemini.py
-autogen/oai/groq.py
-autogen/oai/mistral.py
-autogen/oai/ollama.py
-autogen/oai/openai_utils.py
-autogen/oai/together.py
-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
-test/test_pydantic.py
-test/test_retrieve_utils.py
-test/test_token_count.py

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

@@ -1,167 +0,0 @@
-diskcache
-docker
-flaml
-openai>=1.3
-packaging
-pydantic!=2.6.0,<3,>=1.10
-python-dotenv
-termcolor
-tiktoken
-
-[:python_version < "3.13"]
-numpy<2.0.0,>=1.24.0
-
-[:python_version >= "3.13"]
-numpy>=2.1
-
-[anthropic]
-anthropic>=0.23.1
-
-[autobuild]
-chromadb
-huggingface-hub
-pysqlite3-binary
-sentence-transformers
-
-[bedrock]
-boto3>=1.34.149
-
-[blendsearch]
-flaml[blendsearch]
-
-[captainagent]
-chromadb
-huggingface-hub
-pandas
-pysqlite3-binary
-sentence-transformers
-
-[cerebras]
-cerebras_cloud_sdk>=1.0.0
-
-[cohere]
-cohere>=5.5.8
-
-[cosmosdb]
-azure-cosmos>=4.2.0
-
-[gemini]
-google-auth
-google-cloud-aiplatform
-google-generativeai<1,>=0.5
-pillow
-pydantic
-
-[graph]
-matplotlib
-networkx
-
-[graph_rag_falkor_db]
-graphrag_sdk==0.3.3
-
-[groq]
-groq>=0.9.0
-
-[jupyter-executor]
-ipykernel>=6.29.0
-jupyter-client>=8.6.0
-jupyter-kernel-gateway
-requests
-websocket-client
-
-[lmm]
-pillow
-replicate
-
-[long-context]
-llmlingua<0.3
-
-[mathchat]
-pydantic==1.10.9
-sympy
-wolframalpha
-
-[mistral]
-mistralai>=1.0.1
-
-[ollama]
-fix_busted_json>=0.0.18
-ollama>=0.3.3
-
-[redis]
-redis
-
-[retrievechat]
-beautifulsoup4
-chromadb==0.5.3
-ipython
-markdownify
-protobuf==4.25.3
-pypdf
-sentence_transformers
-
-[retrievechat-mongodb]
-beautifulsoup4
-chromadb==0.5.3
-ipython
-markdownify
-protobuf==4.25.3
-pymongo>=4.0.0
-pypdf
-sentence_transformers
-
-[retrievechat-pgvector]
-beautifulsoup4
-chromadb==0.5.3
-ipython
-markdownify
-pgvector>=0.2.5
-protobuf==4.25.3
-psycopg>=3.1.18
-pypdf
-sentence_transformers
-
-[retrievechat-qdrant]
-beautifulsoup4
-chromadb==0.5.3
-fastembed>=0.3.1
-ipython
-markdownify
-protobuf==4.25.3
-pypdf
-qdrant_client
-sentence_transformers
-
-[teachable]
-chromadb
-
-[test]
-ipykernel
-nbconvert
-nbformat
-pandas
-pre-commit
-pytest-asyncio
-pytest-cov>=5
-pytest<8,>=6.1.1
-
-[together]
-together>=1.2
-
-[types]
-ipykernel>=6.29.0
-jupyter-client>=8.6.0
-jupyter-kernel-gateway
-mypy==1.9.0
-pytest<8,>=6.1.1
-requests
-websocket-client
-
-[websockets]
-websockets<13,>=12.0
-
-[websurfer]
-beautifulsoup4
-markdownify
-pathvalidate
-pdfminer.six

ag2-0.4.1/ag2.egg-info/top_level.txt

@@ -1 +0,0 @@
-autogen

ag2-0.4.1/autogen/__init__.py

@@ -1,17 +0,0 @@
-# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
-#
-# SPDX-License-Identifier: Apache-2.0
-#
-# Portions derived from  https://github.com/microsoft/autogen are under the MIT License.
-# SPDX-License-Identifier: MIT
-import logging
-
-from .agentchat import *
-from .code_utils import DEFAULT_MODEL, FAST_MODEL
-from .exception_utils import *
-from .oai import *
-from .version import __version__
-
-# Set the root logger.
-logger = logging.getLogger(__name__)
-logger.setLevel(logging.INFO)