ag2 0.3.2__tar.gz → 0.4__tar.gz

{ag2-0.3.2/ag2.egg-info → ag2-0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ag2
-Version: 0.3.2
+Version: 0.4
 Summary: A programming framework for agentic AI
 Home-page: https://github.com/ag2ai/ag2
 Author: Chi Wang & Qingyun Wu
@@ -10,12 +10,13 @@ Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.8,<3.13
+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
@@ -104,7 +105,7 @@ We adopt the Apache 2.0 license from v0.3. This enhances our commitment to open-
 
 :tada: Dec 31, 2023: [AutoGen: Enabling Next-Gen LLM Applications via Multi-Agent Conversation Framework](https://arxiv.org/abs/2308.08155) is selected by [TheSequence: My Five Favorite AI Papers of 2023](https://thesequence.substack.com/p/my-five-favorite-ai-papers-of-2023).
 
-<!-- :fire: Nov 24: pyautogen [v0.2](https://github.com/microsoft/autogen/releases/tag/v0.2.0) is released with many updates and new features compared to v0.1.1. It switches to using openai-python v1. Please read the [migration guide](https://ag2ai.github.io/ag2/docs/Installation#python). -->
+<!-- :fire: Nov 24: pyautogen [v0.2](https://github.com/ag2ai/ag2/releases/tag/v0.2.0) is released with many updates and new features compared to v0.1.1. It switches to using openai-python v1. Please read the [migration guide](https://ag2ai.github.io/ag2/docs/Installation#python). -->
 
 <!-- :fire: Nov 11: OpenAI's Assistants are available in AutoGen and interoperatable with other AutoGen agents! Checkout our [blogpost](https://ag2ai.github.io/ag2/blog/2023/11/13/OAI-assistants) for details and examples. -->
 
@@ -129,7 +130,7 @@ We adopt the Apache 2.0 license from v0.3. This enhances our commitment to open-
 
 ## What is AG2
 
-AG2 (formally AutoGen) is an open-source programming framework for building AI agents and facilitating cooperation among multiple agents to solve tasks. AG2 aims to streamline the development and research of agentic AI, much like PyTorch does for Deep Learning. It offers features such as agents capable of interacting with each other, facilitates the use of various large language models (LLMs) and tool use support, autonomous and human-in-the-loop workflows, and multi-agent conversation patterns.
+AG2 (formerly AutoGen) is an open-source programming framework for building AI agents and facilitating cooperation among multiple agents to solve tasks. AG2 aims to streamline the development and research of agentic AI, much like PyTorch does for Deep Learning. It offers features such as agents capable of interacting with each other, facilitates the use of various large language models (LLMs) and tool use support, autonomous and human-in-the-loop workflows, and multi-agent conversation patterns.
 
 **Open Source Statement**: The project welcomes contributions from developers and organizations worldwide. Our goal is to foster a collaborative and inclusive community where diverse perspectives and expertise can drive innovation and enhance the project's capabilities. Whether you are an individual contributor or represent an organization, we invite you to join us in shaping the future of this project. Together, we can build something truly remarkable.
 
@@ -175,7 +176,7 @@ Find detailed instructions for users [here](https://ag2ai.github.io/ag2/docs/ins
 
 ### Option 2. Install AG2 Locally
 
-AG2 requires **Python version >= 3.8, < 3.13**. It can be installed from pip:
+AG2 requires **Python version >= 3.8, < 3.14**. It can be installed from pip:
 
 ```bash
 pip install ag2
@@ -240,7 +241,7 @@ The figure below shows an example conversation flow with AG2.
 ![Agent Chat Example](./website/static/img/chat_example.png)
 
 
-Alternatively, the [sample code](https://github.com/ag2ai/build-with-autogen/blob/main/samples/simple_chat.py) here allows a user to chat with an AG2 agent in ChatGPT style.
+Alternatively, the [sample code](https://github.com/ag2ai/build-with-ag2/blob/main/samples/simple_chat.py) here allows a user to chat with an AG2 agent in ChatGPT style.
 Please find more [code examples](https://ag2ai.github.io/ag2/docs/Examples#automated-multi-agent-chat) for this feature.
 
 <p align="right" style="font-size: 14px; color: #555; margin-top: 20px;">
@@ -299,12 +300,12 @@ In addition, you can find:
 ## CookBook
 
 Explore detailed implementations with sample code and applications to help you get started with AG2.
-[Cookbook](https://github.com/ag2ai/build-with-autogen)
+[Cookbook](https://github.com/ag2ai/build-with-ag2)
 
 
 ## Related Papers
 
-[AG2](https://arxiv.org/abs/2308.08155)
+[AutoGen](https://arxiv.org/abs/2308.08155)
 
 ```
 @inproceedings{wu2023autogen,
@@ -380,7 +381,7 @@ Explore detailed implementations with sample code and applications to help you g
 ## License
 This project is licensed under the [Apache License, Version 2.0 (Apache-2.0)](./LICENSE).
 
-This project is a spin-off of https://github.com/microsoft/autogen and contains code under two licenses:
+This project is a spin-off of [AutoGen](https://github.com/microsoft/autogen) and contains code under two licenses:
 
 - The original code from https://github.com/microsoft/autogen is licensed under the MIT License.  See the [LICENSE_original_MIT](./license_original/LICENSE_original_MIT) file for details.
 

{ag2-0.3.2 → ag2-0.4}/README.md

@@ -59,7 +59,7 @@ We adopt the Apache 2.0 license from v0.3. This enhances our commitment to open-
 
 :tada: Dec 31, 2023: [AutoGen: Enabling Next-Gen LLM Applications via Multi-Agent Conversation Framework](https://arxiv.org/abs/2308.08155) is selected by [TheSequence: My Five Favorite AI Papers of 2023](https://thesequence.substack.com/p/my-five-favorite-ai-papers-of-2023).
 
-<!-- :fire: Nov 24: pyautogen [v0.2](https://github.com/microsoft/autogen/releases/tag/v0.2.0) is released with many updates and new features compared to v0.1.1. It switches to using openai-python v1. Please read the [migration guide](https://ag2ai.github.io/ag2/docs/Installation#python). -->
+<!-- :fire: Nov 24: pyautogen [v0.2](https://github.com/ag2ai/ag2/releases/tag/v0.2.0) is released with many updates and new features compared to v0.1.1. It switches to using openai-python v1. Please read the [migration guide](https://ag2ai.github.io/ag2/docs/Installation#python). -->
 
 <!-- :fire: Nov 11: OpenAI's Assistants are available in AutoGen and interoperatable with other AutoGen agents! Checkout our [blogpost](https://ag2ai.github.io/ag2/blog/2023/11/13/OAI-assistants) for details and examples. -->
 
@@ -84,7 +84,7 @@ We adopt the Apache 2.0 license from v0.3. This enhances our commitment to open-
 
 ## What is AG2
 
-AG2 (formally AutoGen) is an open-source programming framework for building AI agents and facilitating cooperation among multiple agents to solve tasks. AG2 aims to streamline the development and research of agentic AI, much like PyTorch does for Deep Learning. It offers features such as agents capable of interacting with each other, facilitates the use of various large language models (LLMs) and tool use support, autonomous and human-in-the-loop workflows, and multi-agent conversation patterns.
+AG2 (formerly AutoGen) is an open-source programming framework for building AI agents and facilitating cooperation among multiple agents to solve tasks. AG2 aims to streamline the development and research of agentic AI, much like PyTorch does for Deep Learning. It offers features such as agents capable of interacting with each other, facilitates the use of various large language models (LLMs) and tool use support, autonomous and human-in-the-loop workflows, and multi-agent conversation patterns.
 
 **Open Source Statement**: The project welcomes contributions from developers and organizations worldwide. Our goal is to foster a collaborative and inclusive community where diverse perspectives and expertise can drive innovation and enhance the project's capabilities. Whether you are an individual contributor or represent an organization, we invite you to join us in shaping the future of this project. Together, we can build something truly remarkable.
 
@@ -130,7 +130,7 @@ Find detailed instructions for users [here](https://ag2ai.github.io/ag2/docs/ins
 
 ### Option 2. Install AG2 Locally
 
-AG2 requires **Python version >= 3.8, < 3.13**. It can be installed from pip:
+AG2 requires **Python version >= 3.8, < 3.14**. It can be installed from pip:
 
 ```bash
 pip install ag2
@@ -195,7 +195,7 @@ The figure below shows an example conversation flow with AG2.
 ![Agent Chat Example](./website/static/img/chat_example.png)
 
 
-Alternatively, the [sample code](https://github.com/ag2ai/build-with-autogen/blob/main/samples/simple_chat.py) here allows a user to chat with an AG2 agent in ChatGPT style.
+Alternatively, the [sample code](https://github.com/ag2ai/build-with-ag2/blob/main/samples/simple_chat.py) here allows a user to chat with an AG2 agent in ChatGPT style.
 Please find more [code examples](https://ag2ai.github.io/ag2/docs/Examples#automated-multi-agent-chat) for this feature.
 
 <p align="right" style="font-size: 14px; color: #555; margin-top: 20px;">
@@ -254,12 +254,12 @@ In addition, you can find:
 ## CookBook
 
 Explore detailed implementations with sample code and applications to help you get started with AG2.
-[Cookbook](https://github.com/ag2ai/build-with-autogen)
+[Cookbook](https://github.com/ag2ai/build-with-ag2)
 
 
 ## Related Papers
 
-[AG2](https://arxiv.org/abs/2308.08155)
+[AutoGen](https://arxiv.org/abs/2308.08155)
 
 ```
 @inproceedings{wu2023autogen,
@@ -335,7 +335,7 @@ Explore detailed implementations with sample code and applications to help you g
 ## License
 This project is licensed under the [Apache License, Version 2.0 (Apache-2.0)](./LICENSE).
 
-This project is a spin-off of https://github.com/microsoft/autogen and contains code under two licenses:
+This project is a spin-off of [AutoGen](https://github.com/microsoft/autogen) and contains code under two licenses:
 
 - The original code from https://github.com/microsoft/autogen is licensed under the MIT License.  See the [LICENSE_original_MIT](./license_original/LICENSE_original_MIT) file for details.
 

{ag2-0.3.2 → ag2-0.4/ag2.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ag2
-Version: 0.3.2
+Version: 0.4
 Summary: A programming framework for agentic AI
 Home-page: https://github.com/ag2ai/ag2
 Author: Chi Wang & Qingyun Wu
@@ -10,12 +10,13 @@ Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.8,<3.13
+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
@@ -104,7 +105,7 @@ We adopt the Apache 2.0 license from v0.3. This enhances our commitment to open-
 
 :tada: Dec 31, 2023: [AutoGen: Enabling Next-Gen LLM Applications via Multi-Agent Conversation Framework](https://arxiv.org/abs/2308.08155) is selected by [TheSequence: My Five Favorite AI Papers of 2023](https://thesequence.substack.com/p/my-five-favorite-ai-papers-of-2023).
 
-<!-- :fire: Nov 24: pyautogen [v0.2](https://github.com/microsoft/autogen/releases/tag/v0.2.0) is released with many updates and new features compared to v0.1.1. It switches to using openai-python v1. Please read the [migration guide](https://ag2ai.github.io/ag2/docs/Installation#python). -->
+<!-- :fire: Nov 24: pyautogen [v0.2](https://github.com/ag2ai/ag2/releases/tag/v0.2.0) is released with many updates and new features compared to v0.1.1. It switches to using openai-python v1. Please read the [migration guide](https://ag2ai.github.io/ag2/docs/Installation#python). -->
 
 <!-- :fire: Nov 11: OpenAI's Assistants are available in AutoGen and interoperatable with other AutoGen agents! Checkout our [blogpost](https://ag2ai.github.io/ag2/blog/2023/11/13/OAI-assistants) for details and examples. -->
 
@@ -129,7 +130,7 @@ We adopt the Apache 2.0 license from v0.3. This enhances our commitment to open-
 
 ## What is AG2
 
-AG2 (formally AutoGen) is an open-source programming framework for building AI agents and facilitating cooperation among multiple agents to solve tasks. AG2 aims to streamline the development and research of agentic AI, much like PyTorch does for Deep Learning. It offers features such as agents capable of interacting with each other, facilitates the use of various large language models (LLMs) and tool use support, autonomous and human-in-the-loop workflows, and multi-agent conversation patterns.
+AG2 (formerly AutoGen) is an open-source programming framework for building AI agents and facilitating cooperation among multiple agents to solve tasks. AG2 aims to streamline the development and research of agentic AI, much like PyTorch does for Deep Learning. It offers features such as agents capable of interacting with each other, facilitates the use of various large language models (LLMs) and tool use support, autonomous and human-in-the-loop workflows, and multi-agent conversation patterns.
 
 **Open Source Statement**: The project welcomes contributions from developers and organizations worldwide. Our goal is to foster a collaborative and inclusive community where diverse perspectives and expertise can drive innovation and enhance the project's capabilities. Whether you are an individual contributor or represent an organization, we invite you to join us in shaping the future of this project. Together, we can build something truly remarkable.
 
@@ -175,7 +176,7 @@ Find detailed instructions for users [here](https://ag2ai.github.io/ag2/docs/ins
 
 ### Option 2. Install AG2 Locally
 
-AG2 requires **Python version >= 3.8, < 3.13**. It can be installed from pip:
+AG2 requires **Python version >= 3.8, < 3.14**. It can be installed from pip:
 
 ```bash
 pip install ag2
@@ -240,7 +241,7 @@ The figure below shows an example conversation flow with AG2.
 ![Agent Chat Example](./website/static/img/chat_example.png)
 
 
-Alternatively, the [sample code](https://github.com/ag2ai/build-with-autogen/blob/main/samples/simple_chat.py) here allows a user to chat with an AG2 agent in ChatGPT style.
+Alternatively, the [sample code](https://github.com/ag2ai/build-with-ag2/blob/main/samples/simple_chat.py) here allows a user to chat with an AG2 agent in ChatGPT style.
 Please find more [code examples](https://ag2ai.github.io/ag2/docs/Examples#automated-multi-agent-chat) for this feature.
 
 <p align="right" style="font-size: 14px; color: #555; margin-top: 20px;">
@@ -299,12 +300,12 @@ In addition, you can find:
 ## CookBook
 
 Explore detailed implementations with sample code and applications to help you get started with AG2.
-[Cookbook](https://github.com/ag2ai/build-with-autogen)
+[Cookbook](https://github.com/ag2ai/build-with-ag2)
 
 
 ## Related Papers
 
-[AG2](https://arxiv.org/abs/2308.08155)
+[AutoGen](https://arxiv.org/abs/2308.08155)
 
 ```
 @inproceedings{wu2023autogen,
@@ -380,7 +381,7 @@ Explore detailed implementations with sample code and applications to help you g
 ## License
 This project is licensed under the [Apache License, Version 2.0 (Apache-2.0)](./LICENSE).
 
-This project is a spin-off of https://github.com/microsoft/autogen and contains code under two licenses:
+This project is a spin-off of [AutoGen](https://github.com/microsoft/autogen) and contains code under two licenses:
 
 - The original code from https://github.com/microsoft/autogen is licensed under the MIT License.  See the [LICENSE_original_MIT](./license_original/LICENSE_original_MIT) file for details.
 

{ag2-0.3.2 → ag2-0.4}/ag2.egg-info/SOURCES.txt

@@ -33,6 +33,7 @@ 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
@@ -43,7 +44,9 @@ 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/capabilities/__init__.py
 autogen/agentchat/contrib/capabilities/agent_capability.py

{ag2-0.3.2 → ag2-0.4}/ag2.egg-info/requires.txt

@@ -1,7 +1,6 @@
 diskcache
 docker
 flaml
-numpy<2,>=1.17.0
 openai>=1.3
 packaging
 pydantic!=2.6.0,<3,>=1.10
@@ -9,6 +8,12 @@ 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
 
@@ -24,6 +29,13 @@ boto3>=1.34.149
 [blendsearch]
 flaml[blendsearch]
 
+[captainagent]
+chromadb
+huggingface-hub
+pandas
+pysqlite3
+sentence-transformers
+
 [cerebras]
 cerebras_cloud_sdk>=1.0.0
 

{ag2-0.3.2 → ag2-0.4}/autogen/agentchat/__init__.py

@@ -7,12 +7,22 @@
 from .agent import Agent
 from .assistant_agent import AssistantAgent
 from .chat import ChatResult, initiate_chats
+
+# Imported last to avoid circular imports
+from .contrib.swarm_agent import (
+    AFTER_WORK,
+    ON_CONDITION,
+    AfterWorkOption,
+    SwarmAgent,
+    SwarmResult,
+    initiate_swarm_chat,
+)
 from .conversable_agent import ConversableAgent, register_function
 from .groupchat import GroupChat, GroupChatManager
 from .user_proxy_agent import UserProxyAgent
 from .utils import gather_usage_summary
 
-__all__ = (
+__all__ = [
     "Agent",
     "ConversableAgent",
     "AssistantAgent",
@@ -23,4 +33,10 @@ __all__ = (
     "initiate_chats",
     "gather_usage_summary",
     "ChatResult",
-)
+    "initiate_swarm_chat",
+    "SwarmAgent",
+    "SwarmResult",
+    "ON_CONDITION",
+    "AFTER_WORK",
+    "AfterWorkOption",
+]

{ag2-0.3.2 → ag2-0.4}/autogen/agentchat/contrib/agent_builder.py

@@ -203,7 +203,9 @@ Match roles in the role set to each expert in expert set.
             builder_filter_dict.update({"model": builder_model})
         if len(builder_model_tags) != 0:
             builder_filter_dict.update({"tags": builder_model_tags})
-        builder_config_list = autogen.config_list_from_json(config_file_or_env, filter_dict=builder_filter_dict)
+        builder_config_list = autogen.config_list_from_json(
+            config_file_or_env, file_location=config_file_location, filter_dict=builder_filter_dict
+        )
         if len(builder_config_list) == 0:
             raise RuntimeError(
                 f"Fail to initialize build manager: {builder_model}{builder_model_tags} does not exist in {config_file_or_env}. "
@@ -267,7 +269,7 @@ Match roles in the role set to each expert in expert set.
         description = agent_config["description"]
 
         # Path to the customize **ConversableAgent** class.
-        model_path = agent_config.get("model_path", None)
+        agent_path = agent_config.get("agent_path", None)
         filter_dict = {}
         if len(model_name_or_hf_repo) > 0:
             filter_dict.update({"model": model_name_or_hf_repo})
@@ -302,8 +304,8 @@ Match roles in the role set to each expert in expert set.
                 )
 
             model_class = autogen.AssistantAgent
-            if model_path:
-                module_path, model_class_name = model_path.replace("/", ".").rsplit(".", 1)
+            if agent_path:
+                module_path, model_class_name = agent_path.replace("/", ".").rsplit(".", 1)
                 module = importlib.import_module(module_path)
                 model_class = getattr(module, model_class_name)
                 if not issubclass(model_class, autogen.ConversableAgent):
@@ -313,7 +315,7 @@ Match roles in the role set to each expert in expert set.
             additional_config = {
                 k: v
                 for k, v in agent_config.items()
-                if k not in ["model", "name", "system_message", "description", "model_path", "tags"]
+                if k not in ["model", "name", "system_message", "description", "agent_path", "tags"]
             }
             agent = model_class(
                 name=agent_name, llm_config=current_config.copy(), description=description, **additional_config