ag2 0.3.2__py3-none-any.whl → 0.4__py3-none-any.whl

{ag2-0.3.2.dist-info → ag2-0.4.dist-info}/METADATA

@@ -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,20 +10,21 @@ 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
 License-File: LICENSE
 License-File: NOTICE.md
 Requires-Dist: diskcache
 Requires-Dist: docker
 Requires-Dist: flaml
-Requires-Dist: numpy (<2,>=1.17.0)
 Requires-Dist: openai (>=1.3)
 Requires-Dist: packaging
 Requires-Dist: pydantic (!=2.6.0,<3,>=1.10)
 Requires-Dist: python-dotenv
 Requires-Dist: termcolor
 Requires-Dist: tiktoken
+Requires-Dist: numpy (<2.0.0,>=1.24.0) ; python_version < "3.13"
+Requires-Dist: numpy (>=2.1) ; python_version >= "3.13"
 Provides-Extra: anthropic
 Requires-Dist: anthropic (>=0.23.1) ; extra == 'anthropic'
 Provides-Extra: autobuild
@@ -35,6 +36,12 @@ Provides-Extra: bedrock
 Requires-Dist: boto3 (>=1.34.149) ; extra == 'bedrock'
 Provides-Extra: blendsearch
 Requires-Dist: flaml[blendsearch] ; extra == 'blendsearch'
+Provides-Extra: captainagent
+Requires-Dist: chromadb ; extra == 'captainagent'
+Requires-Dist: huggingface-hub ; extra == 'captainagent'
+Requires-Dist: pandas ; extra == 'captainagent'
+Requires-Dist: pysqlite3 ; extra == 'captainagent'
+Requires-Dist: sentence-transformers ; extra == 'captainagent'
 Provides-Extra: cerebras
 Requires-Dist: cerebras-cloud-sdk (>=1.0.0) ; extra == 'cerebras'
 Provides-Extra: cohere
@@ -203,7 +210,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. -->
 
@@ -228,7 +235,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.
 
@@ -274,7 +281,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
@@ -339,7 +346,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;">
@@ -398,12 +405,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,
@@ -479,7 +486,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.dist-info → ag2-0.4.dist-info}/RECORD

@@ -11,21 +11,22 @@ autogen/retrieve_utils.py,sha256=CndQ0lr7dsRGLltudqp-O_dC_4NgO6fmhzka8iXLa-g,196
 autogen/runtime_logging.py,sha256=iAJO_oddNfItIghNlTv060Ld0NybK52SzL7qu3n0vRw,4837
 autogen/token_count_utils.py,sha256=z9HMEYf9DWr3IJah-PyzIwU_vr64_NF_6U0WCFI1L7s,10240
 autogen/types.py,sha256=zkSwYEfDePIsuk03IgutDPwb1QlFK-rAu23uYvQKrXE,601
-autogen/version.py,sha256=4AF2zyKy8W_AxDUAJXcwitR8PpTkqFi5Gtn0IR9b0jU,248
-autogen/agentchat/__init__.py,sha256=THTYvGvFyi-gKVAMnBdDgU9CKE8cWMqWcbFS-VrzOQs,773
+autogen/version.py,sha256=6gwzC7A47tDnPxQfQxXYWyKfO7d2lSjxsqVYAi87_pk,246
+autogen/agentchat/__init__.py,sha256=GhEKIablUdEaWlW8i7VpHwtzIVOl6N9sgbnAcT-X9jw,1091
 autogen/agentchat/agent.py,sha256=r3jjZbVGkms4a8FTOXzyiQQ7PcMPVBj8zwbSJ-siwXg,4764
 autogen/agentchat/assistant_agent.py,sha256=cEaYk15yyeRSE3LaguDThxzZFgTWiPJ1SxTvAvK6t68,5548
 autogen/agentchat/chat.py,sha256=F7WNQc_feDg0pXwkqG0ER74jy_zRzj6NO7MVNwEQcPM,12484
-autogen/agentchat/conversable_agent.py,sha256=oV3xQiZtn8zasMb8iH_jB-csc27wyKDMITLNkMvaaO4,139137
-autogen/agentchat/groupchat.py,sha256=S4hQ_pAxfk6n9_R0HhVM4fJ12hxEKP_nMhTeel2Yu_I,83723
+autogen/agentchat/conversable_agent.py,sha256=T-lnzd8BVTvgujjRsUJ1N1KrLo2ojQ7HG6l-CCnEVAA,139387
+autogen/agentchat/groupchat.py,sha256=2kzT4nf_Yexuq8YqcmJSX9TA2QoTv9yZXZ8W5Q3NpA8,83877
 autogen/agentchat/user_proxy_agent.py,sha256=c-RG-Sb_O3p3QnePkOVepVI5olyfKOlsM9SQaEOcthw,7307
 autogen/agentchat/utils.py,sha256=Y-FjMcwiJuO-WsgLDuXep_IU2absAMwQulCKs9QdHVc,8036
 autogen/agentchat/contrib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogen/agentchat/contrib/agent_builder.py,sha256=yY2eFSaewKmYS-vylNJVN_jtHG6zTdKETR50nrjyb9A,32641
+autogen/agentchat/contrib/agent_builder.py,sha256=z08C5-zpAXqKliX0vSdhphHr1w43YqENjYK6uhjKYdY,32699
 autogen/agentchat/contrib/agent_optimizer.py,sha256=dSsRIcYwZXxTWIGn_0gUl-zvDnWgOSQKyImzmkE6UPQ,22446
+autogen/agentchat/contrib/captainagent.py,sha256=ir079N3lk_xphZjrus8WZntZT2maCuJUj7anRbKaOkU,25500
 autogen/agentchat/contrib/gpt_assistant_agent.py,sha256=5aNsw8HJQKw5KumHRZekT2smnHR9e4PLRvIZzfsNPQw,25146
 autogen/agentchat/contrib/img_utils.py,sha256=XEnLKeWOXoD1IiWf8IT1A3QiUE8ZEdZrjxUaJwR1WVY,14276
-autogen/agentchat/contrib/llamaindex_conversable_agent.py,sha256=2g-yz9nNu1p3RxagwGYuU0bp_baje7vfV2gYNqFtiK0,4292
+autogen/agentchat/contrib/llamaindex_conversable_agent.py,sha256=BSfjG4fDXjG4jpBmyAso5k1dK8JJPYWoNuL4LiYl-EM,4535
 autogen/agentchat/contrib/llava_agent.py,sha256=pVKcwa_16Pg6lBfGJ6JXWUH-nAUtgW05q6KdhDESass,6330
 autogen/agentchat/contrib/math_user_proxy_agent.py,sha256=Q6YI8Z3M4Ndcp8ruMtpEr16rxLi4ZnIKKnYD8ujv33I,19938
 autogen/agentchat/contrib/multimodal_conversable_agent.py,sha256=vDYP6c3K1L6HaefG4gItGvlrbAa91KERJQTKZlMpe2c,5155
@@ -33,7 +34,9 @@ autogen/agentchat/contrib/qdrant_retrieve_user_proxy_agent.py,sha256=mgKHvKbx5dc
 autogen/agentchat/contrib/retrieve_assistant_agent.py,sha256=79igEA0r1eZYSuWKOCKkUwCV1c5yTWH295-L2-4ao4c,2424
 autogen/agentchat/contrib/retrieve_user_proxy_agent.py,sha256=kDEQ_HJ5nvbBmsCf-oyCnZGxzv3R0VawoLHKuNz5v4c,37069
 autogen/agentchat/contrib/society_of_mind_agent.py,sha256=lRF69C8ejcqPRGj5_fPtUs2NandyATw010OJB7OTdos,9087
+autogen/agentchat/contrib/swarm_agent.py,sha256=qeBV6XoBPdEGbJgSf2ZmpFEuDKmE1etee9X7hJNV7Bs,17032
 autogen/agentchat/contrib/text_analyzer_agent.py,sha256=WitXX--yoGh3Xq_jC_D85Unt7UCVPdhM4MzuVJxiRSg,3563
+autogen/agentchat/contrib/tool_retriever.py,sha256=WIOTdHOWR-JRscw9RJBRAuuSYqV97nhUhGw3yQ0J8sg,4304
 autogen/agentchat/contrib/web_surfer.py,sha256=kAqbfuyj_lK61VUhvVUM83CYs_Hsut6eKPy388LVwL8,14563
 autogen/agentchat/contrib/capabilities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogen/agentchat/contrib/capabilities/agent_capability.py,sha256=f6cBV4FcMvA7oZE1_o42RUUayC3KTkKOB7wW2fFEjso,771
@@ -97,16 +100,16 @@ autogen/oai/cerebras.py,sha256=7TrdvplPx5ATbFwSIvwqg_8Xmzkjsi2Hr9Q60w5uyZk,10669
 autogen/oai/client.py,sha256=HR8ATrTlNoSM8fr0fH9cmKR-lxr3Rd62kBI0aj-eoNg,52615
 autogen/oai/client_utils.py,sha256=W9RD6V3XPpelDqPwjG412go4qXfYWDPmKur68s-Siu0,7282
 autogen/oai/cohere.py,sha256=giMIRomeAEd3y0QsulmWYYLXXCfsqY0ojKG4UzwQ2SI,18513
-autogen/oai/completion.py,sha256=KwbFggV_b30uJfuJsw3oOpvVzIM05fNM1KHwjbLJHXY,53660
-autogen/oai/gemini.py,sha256=JcjJh9WoV7JDxkS2e4kercaApw_Ecp4q_T4t9GgrsBE,20599
+autogen/oai/completion.py,sha256=dT2LEjmRUMXx0UR7OKs6zYQH-cDtW5m0vWvwXXOjmvk,53652
+autogen/oai/gemini.py,sha256=YNrwGKwPVglfUhx2XPph9AbPN0b5IbPMaT0njWPEXjI,20595
 autogen/oai/groq.py,sha256=gpLe1UJZMtkJAks2mj40aksvMcQ0oBjgOxRuTVupG7s,11164
 autogen/oai/mistral.py,sha256=1xybwfihh6n2HyLlCNNebT01SYVIwNj9ClIms92ySak,11414
 autogen/oai/ollama.py,sha256=lKGvNdVoNhpvssYu-shrNPcMHczBX1kVktZe77c9zz0,25383
 autogen/oai/openai_utils.py,sha256=zZU1GFZsqy_dEYvt5UbCfRPh_i3nIFvGVnmF3kbTUvU,35162
 autogen/oai/together.py,sha256=AfEx2Dt4limeVvnCibtwvs9OuJkS1wcYFdA-P-gDzRw,13411
-ag2-0.3.2.dist-info/LICENSE,sha256=GEFQVNayAR-S_rQD5l8hPdgvgyktVdy4Bx5-v90IfRI,11384
-ag2-0.3.2.dist-info/METADATA,sha256=MVcmbzNczPQo86nvdU91GNfQ2aVC1notYGCZ9FQiK2A,24562
-ag2-0.3.2.dist-info/NOTICE.md,sha256=ucvou1bE6i2s40qyuU9RL0TeIVG01VhXoQ59EngtEz4,1317
-ag2-0.3.2.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-ag2-0.3.2.dist-info/top_level.txt,sha256=qRZ2MW8yuy0LtOLpeZndl4H1buusjxhwD6c9AXbVqKQ,8
-ag2-0.3.2.dist-info/RECORD,,
+ag2-0.4.dist-info/LICENSE,sha256=GEFQVNayAR-S_rQD5l8hPdgvgyktVdy4Bx5-v90IfRI,11384
+ag2-0.4.dist-info/METADATA,sha256=Mp-d5Xx4JhYxpYBcokQfGAk6OQk0CdyhTOckr6AEOOM,24943
+ag2-0.4.dist-info/NOTICE.md,sha256=ucvou1bE6i2s40qyuU9RL0TeIVG01VhXoQ59EngtEz4,1317
+ag2-0.4.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+ag2-0.4.dist-info/top_level.txt,sha256=qRZ2MW8yuy0LtOLpeZndl4H1buusjxhwD6c9AXbVqKQ,8
+ag2-0.4.dist-info/RECORD,,

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",
+]

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