ag2 0.9.3__py3-none-any.whl → 0.9.4__py3-none-any.whl

{ag2-0.9.3.dist-info → ag2-0.9.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ag2
-Version: 0.9.3
+Version: 0.9.4
 Summary: A programming framework for agentic AI
 Project-URL: Homepage, https://ag2.ai/
 Project-URL: Documentation, https://docs.ag2.ai

{ag2-0.9.3.dist-info → ag2-0.9.4.dist-info}/RECORD

@@ -14,7 +14,7 @@ autogen/retrieve_utils.py,sha256=R3Yp5d8dH4o9ayLZrGn4rCjIaY4glOHIiyQjwClmdi8,200
 autogen/runtime_logging.py,sha256=yCmZODvwqYR91m8lX3Q4SoPcY-DK48NF4m56CP6Om3c,4692
 autogen/token_count_utils.py,sha256=n4wTFVNHwrfjZkrErFr8kNig2K-YCGgMLWsjDRS9D6g,10797
 autogen/types.py,sha256=qu-7eywhakW2AxQ5lYisLLeIg45UoOW-b3ErIuyRTuw,1000
-autogen/version.py,sha256=tN9tkN8TlaYs7tWFTkUc6inNMUFd2YXRCAXmqLFBOI4,193
+autogen/version.py,sha256=zukvVxK3WY_ypC-a8Iy7A_baGAoqr3cT-lhYYnSMhhM,193
 autogen/_website/__init__.py,sha256=c8B9TpO07x9neD0zsJWj6AaEdlcP-WvxrvVOGWLtamk,143
 autogen/_website/generate_api_references.py,sha256=yKqyeSP_NE27wwLYWsZbTYRceEoxzNxPXqn6vsIzEvk,14789
 autogen/_website/generate_mkdocs.py,sha256=TkmLnUDv1Ms5cGClXPmenA8nxmwg4kR0E-FHCVjw_og,43246
@@ -25,12 +25,12 @@ autogen/agentchat/__init__.py,sha256=d5jPpXeavynP9eh5uMIgJKbK3-3tywJBa6l3L9X4l34
 autogen/agentchat/agent.py,sha256=HePNJ5BXJTcZtaD2a8CoTeHAoLUUy3scM6Ihm-NsSWk,5828
 autogen/agentchat/assistant_agent.py,sha256=XTJvD66r4qYkdNAJJLr1CC-wTYFJWvhmD5_G0WbbX2I,5741
 autogen/agentchat/chat.py,sha256=6Gx2t1-Xa8kP6ZoUihHBNGOqNlrGhhqLPKrckL0n-RI,14003
-autogen/agentchat/conversable_agent.py,sha256=F5OtTxBYb4Cqw5cp6V1gt8tL6EVVFuAKmbCR9SVosT8,191085
-autogen/agentchat/groupchat.py,sha256=X54uAoAFYxMvxDVCMQJdlVnviOWvjqWp7bbt14PQUmk,85373
+autogen/agentchat/conversable_agent.py,sha256=-BNwxfqSaHQi428a6xgz_53ThPC8LDsdHzHiWzNsPtc,192671
+autogen/agentchat/groupchat.py,sha256=vdK8zUX2CG8fR5pkPGWX7Zn7DFwLvld0bpaRv3vOBEo,89471
 autogen/agentchat/user_proxy_agent.py,sha256=-gbDblRvE09FGuVB6-y5ZT9Cpvv--rM3FMi8PnPIFBA,7445
 autogen/agentchat/utils.py,sha256=2ZweUqe4ynxji0jnvd0r9Uxg3n0elif4a-jZOyeeqcg,8238
 autogen/agentchat/contrib/__init__.py,sha256=tOTe4nwbKj7elHpftAy3zS_embMDzncrKL98XKhY6-c,168
-autogen/agentchat/contrib/agent_optimizer.py,sha256=9EUhacAFOC5PZHWclj9wZA1bgjyVy3g3Y3TDbn-7BhI,22265
+autogen/agentchat/contrib/agent_optimizer.py,sha256=Y0H9unXXCOmBo6P-njtH3L_vmMCdwp0B9Kiihg8HBy4,22431
 autogen/agentchat/contrib/gpt_assistant_agent.py,sha256=0zne7MZBVYlxvYdI-Iu2BRziPSYQPZ1fg0XjfFzt9BA,25051
 autogen/agentchat/contrib/img_utils.py,sha256=BNc1ELE2dSzmFCc_9zwcdazmUxDZZrQv8jgXJZaheCM,15121
 autogen/agentchat/contrib/llamaindex_conversable_agent.py,sha256=sAbBmYIBB4RJt-GhgNzA5QSX4CP0q2eusQNa6K2_LA4,4595
@@ -132,7 +132,8 @@ autogen/agentchat/group/context_expression.py,sha256=zFWPV3yfV-0ayYXfrhRq6iWQZnR
 autogen/agentchat/group/context_str.py,sha256=EHjpDr8MLMl5AMXktMi9Wp4BIL_1hbIeJPMEXlLkTjs,1270
 autogen/agentchat/group/context_variables.py,sha256=d2Q31aoV2o_5QSd3Oh1fYDIV0fDaaHOOsjNQ92TC_H0,5649
 autogen/agentchat/group/group_tool_executor.py,sha256=lCUg0Z_R8u0mRsWE350p6pGNj-6fJUMGqWqs3xmznCQ,9104
-autogen/agentchat/group/group_utils.py,sha256=W3zUghHffO0_seZefB68-U0jIufFDxAkd1NuutuaGmM,27240
+autogen/agentchat/group/group_utils.py,sha256=nNHtAAS2tAvaA3VsXPYDBIV6vBve8PoVvYMs0Ps7GzI,27586
+autogen/agentchat/group/guardrails.py,sha256=OMqQgXtscfYHS6ET1x2HT6Cpi2gtnkEbNKQgWqnZtDw,5964
 autogen/agentchat/group/handoffs.py,sha256=ergiOsXC4z8N44LTUh-abIPrErjkXZWFRMXflinNq1Y,11743
 autogen/agentchat/group/llm_condition.py,sha256=wfuEET1VhyVVGedYxcyuhX_Vus6uZHxUl_SPpu4YIsc,2951
 autogen/agentchat/group/multi_agent_chat.py,sha256=oKAWiziaSZ0otwfGhSWaR26dYbfRQj_vau_P5Z1dvfY,7688
@@ -149,7 +150,7 @@ autogen/agentchat/group/patterns/round_robin.py,sha256=nS7nsQJKCq9lD0wyCx__gkWK3
 autogen/agentchat/group/targets/__init__.py,sha256=AJNSbl9iMe2hiDmZojTp8h889o5OYN3V7f2_2nr8px4,145
 autogen/agentchat/group/targets/group_chat_target.py,sha256=4_bzVg1ODSIx7UfP8O9OdvTlRNtFuqZt77YqKFIwh74,5364
 autogen/agentchat/group/targets/group_manager_target.py,sha256=Cp4Wbmp3a7fZBTN-s5eQEDRNyKWq9KYihRlzx1blnZQ,6276
-autogen/agentchat/group/targets/transition_target.py,sha256=vdfNu6XZre18nwF0Vazn_5C5_qPYMUt76xGC0W5gA98,16179
+autogen/agentchat/group/targets/transition_target.py,sha256=punRNX630N4HQrMmpVAeSCkdHDEd3CG51KHwlUM3JLo,16742
 autogen/agentchat/group/targets/transition_utils.py,sha256=fzRnaJtUEnNswkmamQH08X43xX2kV0SWGIWEV72TtIo,219
 autogen/agentchat/realtime/__init__.py,sha256=c8B9TpO07x9neD0zsJWj6AaEdlcP-WvxrvVOGWLtamk,143
 autogen/agentchat/realtime/experimental/__init__.py,sha256=leYemaQJXulYnp5atRJZE247EL5VJtdDoF_p1XJFQzM,619
@@ -198,7 +199,7 @@ autogen/agents/experimental/slack/slack.py,sha256=QzT0awsZpAVCs5mmXeHT-V6uhKik8n
 autogen/agents/experimental/telegram/__init__.py,sha256=Y-HQJXmeux9QVY-Jjy22WO8cPWryxwaKEbhcypdUImY,209
 autogen/agents/experimental/telegram/telegram.py,sha256=q-1i1xPVYxOUjABUvKRo5B9sPvF3p2bAQLQi9POndMM,3097
 autogen/agents/experimental/websurfer/__init__.py,sha256=I3D0sIBDW9zxSOM0wNXdFfMZQIpiutet-9ikrm0WD0s,212
-autogen/agents/experimental/websurfer/websurfer.py,sha256=SHhXfCxuV28zAzLx1kX5d9Cv2S0ikr_-suvqWig18kg,2596
+autogen/agents/experimental/websurfer/websurfer.py,sha256=MT6VIXmPscjY8LdweXL_txTNsNkpHtwSqR_475QYMeM,2908
 autogen/agents/experimental/wikipedia/__init__.py,sha256=ytX_85hqagdyl3I72kYQM2-3bR7rG_SI5cyx8pKSG2I,212
 autogen/agents/experimental/wikipedia/wikipedia.py,sha256=51fjVlmNDZfq3JrGIkknPXz41y_VJcMXfHNnnW58L0s,3942
 autogen/cache/__init__.py,sha256=HF7qJyJpzqTYXSwUyVUxAFTY0eakw5OFwhyg_KO3QwE,375
@@ -226,10 +227,10 @@ autogen/coding/jupyter/jupyter_client.py,sha256=ROXAWOKG_EJ_oFNuyqUd_3uOBPUTRoTh
 autogen/coding/jupyter/jupyter_code_executor.py,sha256=Z2vZvou6QzpMBg0IgOzVRoCADswd15mvfkktIjGhUMY,6374
 autogen/coding/jupyter/local_jupyter_server.py,sha256=7b8yi5qK8ms2e5-PRCrzmXKGp1iC5KgpMU8xiqQ9u8o,6589
 autogen/events/__init__.py,sha256=XwCA6Rsq9AyjgeecGuiwHcAvDQMmKXgGomw5iLDIU5Q,358
-autogen/events/agent_events.py,sha256=G4SO8DlXyeiBvUVW0z8BuOGpUpcUXjF6mzbDxW9Nn-w,30789
+autogen/events/agent_events.py,sha256=LSCOMwA-nlBSboE5jOh-Da2Pm7hghiwjfzYTz_bS30k,31112
 autogen/events/base_event.py,sha256=5K1wzDBAio9wLxahErSvE0-UbFfMuSTxBp6EI8SbPGU,3475
 autogen/events/client_events.py,sha256=2skE5f9BxQmwJv22vNe-s1ReDXE-DsadeAmMWlrmltc,5466
-autogen/events/helpers.py,sha256=CMcetNzfCE-dWTTJ_9B09of2slRG43yYnHXvFfuGz2Q,1185
+autogen/events/helpers.py,sha256=JNwnEfaFylWEaQivURXQ5_7DjsjmCZyiWRI4mQH0ueY,1528
 autogen/events/print_event.py,sha256=_B_60nnB0yFwWpv-LT2c-Zp_b-moREtafP4lDI8uJzk,1275
 autogen/extensions/__init__.py,sha256=tOTe4nwbKj7elHpftAy3zS_embMDzncrKL98XKhY6-c,168
 autogen/fast_depends/__init__.py,sha256=4KNQsBQmoDf5RIZsnOuMFy7t6-Y1hnKjIplGdbo_k8Y,465
@@ -276,17 +277,18 @@ autogen/logger/logger_utils.py,sha256=H9hcsRyEcUcfxTYWf5cRjtNghF4h3FT8sr4IIuqQum
 autogen/logger/sqlite_logger.py,sha256=sRwMx42zh85QWLz1BqKyVySI8OwEB_NjM3ObLOW-mcI,18685
 autogen/mcp/__init__.py,sha256=6BDDmw0sjLZRjyHnd-Gfh9BE-pTKTv5bkow9W6odHtQ,213
 autogen/mcp/__main__.py,sha256=C7nXbWxG3yGsWKRgFnhSsbQOmshLnz6ZOcCxK2TWWio,2487
+autogen/mcp/helpers.py,sha256=J5_J6n3jMJUEJH5K8k9BeUb6ymQgRUI0hC1gbY7rlwM,1533
 autogen/mcp/mcp_client.py,sha256=7c_lHgBJEs77TFYjLcTlVrEu_0z4EafPPY3PgteY87c,7400
 autogen/mcp/mcp_proxy/__init__.py,sha256=3HTU-TqHLk4XSXeBV1UFd9XkQ1B0yOuXXyGseXvDVec,518
 autogen/mcp/mcp_proxy/fastapi_code_generator_helpers.py,sha256=dx-w2tGVMnh8pzY2NuXlMD7oIF7_5Gvc5oSbHIySEpc,2110
-autogen/mcp/mcp_proxy/mcp_proxy.py,sha256=k2_FZyvCtF0h1CxL_VRL0rWMEuzmU42SpB6JCNbk9zc,22112
+autogen/mcp/mcp_proxy/mcp_proxy.py,sha256=NfjNsUnqpPQ_FrKJPiPUL7aDE9SZ_wNB0ZHAsiJHq9I,22131
 autogen/mcp/mcp_proxy/operation_grouping.py,sha256=1n4o5qhkQd2hVr7OaOMsRhInDveDGkCozLudsBVusC4,6349
 autogen/mcp/mcp_proxy/operation_renaming.py,sha256=G5J4VdxUAwSvOo-DsqIUbCfV9TnH3riaHLI6IpctDF8,3956
 autogen/mcp/mcp_proxy/patch_fastapi_code_generator.py,sha256=vBr8P890nsFvK4iuFicrdNskooHlBQeTQ6jbb5kcPAk,3490
 autogen/mcp/mcp_proxy/security.py,sha256=HUcdIE1CCmvuWBPUlvjoKOJV-05sve6pcKHalgm9Z8E,13611
 autogen/mcp/mcp_proxy/security_schema_visitor.py,sha256=UafLMd5zcz0COcc5Vfvt6bP3LQ3jmaBwQIlWhovN9H4,1345
 autogen/messages/__init__.py,sha256=ZuLvvIQRkNE5fotPe6MSS_YzOUkmfIqGSfOZZOZQ3go,321
-autogen/messages/agent_messages.py,sha256=ZnRkdWcKS20lW2Njt740m2YNfKanHt8C3k84KKbzR8E,30414
+autogen/messages/agent_messages.py,sha256=gzsJdzk2slAfSn8ZcXNW6hPegrYPNIEphdhHaqP19Zg,30451
 autogen/messages/base_message.py,sha256=MbYEXM0dWHl31y7o08PGeJNcpK67hpoQpUMgnZ8qTGE,3808
 autogen/messages/client_messages.py,sha256=9yVn4zug44sLIZQFKiCjOK2ML7uFAADflmGudgPV850,5653
 autogen/messages/print_message.py,sha256=51UpH-rkLrRxz_QxAhEMdjVAipiwkLirOuGtkuTvQL4,1393
@@ -297,8 +299,8 @@ autogen/oai/cerebras.py,sha256=8hiSBq88l2yTXUJPV7AvGXRCtwvW0Y9hIYUnYK2S2os,12462
 autogen/oai/client.py,sha256=BB_6Heny6_7lq8q7ZAPKohHAK63zs9UGzRoUknTxjYY,65051
 autogen/oai/client_utils.py,sha256=lVbHyff7OnpdM-tXskC23xLdFccj2AalTdWA4DxzxS4,7543
 autogen/oai/cohere.py,sha256=pRcQWjbzKbZ1RfC1vk9WGjgndwjHbIaOVoKEYdV2L6c,19421
-autogen/oai/gemini.py,sha256=45-MiC-m43KgqLHSBsCw7qUymI3Zic3NUeefZ-HeQtI,42398
-autogen/oai/gemini_types.py,sha256=lIrQGcret7213YR07uFfrw1CnFiZ9VwPyb55KyS0GPI,5864
+autogen/oai/gemini.py,sha256=bc_RQwtoGi7DnwQwPie7ZdvsqpD1AjYwzjVwnIKP39U,43168
+autogen/oai/gemini_types.py,sha256=i4wT8ytD2cO9nCUWm2FKLKVVxyd31wMhOpnrvnIiKIc,5888
 autogen/oai/groq.py,sha256=pQWtaAY_AjT30XKbZNHXDzWsawBys3yFWlfy6K4Nqr8,12431
 autogen/oai/mistral.py,sha256=SlOYPdnNLHuTEHBGCzsemG9sLEgphdUukRurERdMsvI,12677
 autogen/oai/ollama.py,sha256=t0fIgDCoIfsQZ3hhpseam5N-fbpI7-fw82bG55mA8nU,29103
@@ -320,15 +322,17 @@ autogen/tools/toolkit.py,sha256=1tOmTGJ96RhkJrrtAViKUyEcwacA6ztoIbYbnf8NgTU,2558
 autogen/tools/contrib/__init__.py,sha256=DWEjPK6xCR2ihAXXdquQZmiuqRLA3Pqb8QV8W1RtS3k,202
 autogen/tools/contrib/time/__init__.py,sha256=dplie5aBJZ8VoKy6EKcQMLTtSgcCkNDYzpdsC2I0YWk,195
 autogen/tools/contrib/time/time.py,sha256=tPi49vOUwfvujbYA-zS00CWcLW-y18CPyQ1gnJG6iRg,1271
-autogen/tools/experimental/__init__.py,sha256=G399sJCbL3hV2HSU3eGnrrQ4i3a_P55LobmtcYYUrqU,1466
+autogen/tools/experimental/__init__.py,sha256=jGyt9GVoJO6VsZdmBLzwTxXZGBbWEwdLrUjn-rWQ6os,1588
 autogen/tools/experimental/browser_use/__init__.py,sha256=kfxCajXcVMDH6CZq-lWh2p8PKxOwT9yjC_Za0jr4zUg,290
-autogen/tools/experimental/browser_use/browser_use.py,sha256=VEYJwUna_DM_HxeI5kgjhtTUzPBYlwrcGk5UOy85gDo,5426
+autogen/tools/experimental/browser_use/browser_use.py,sha256=KfU4MI_BWaHepv0bDMf9HTDUaHTJThuBJI8R_BPpjmg,5561
 autogen/tools/experimental/crawl4ai/__init__.py,sha256=UjFJLSZ9P5xT6WCV0RDPtwt4MHuwPdK90TU7ByXhLWs,207
 autogen/tools/experimental/crawl4ai/crawl4ai.py,sha256=MsOLtbPHRpRrCnRJPQVVVNwmsBcgsWLSHNXDOF-47ws,6088
 autogen/tools/experimental/deep_research/__init__.py,sha256=9SFcDEj2OHxNSlXP11lf1uHENlfUeO47ROcOSD9GCDs,220
 autogen/tools/experimental/deep_research/deep_research.py,sha256=1ZwZjVR1gKHvlWaT7I9_8qrEPTCYhManMci4ZHApIdE,14970
 autogen/tools/experimental/duckduckgo/__init__.py,sha256=hstpKQfvd5YD7X0iMuvJFlui3MtJVGbz5XpkOSaoy18,232
 autogen/tools/experimental/duckduckgo/duckduckgo_search.py,sha256=NCByIKfJz9CXj_IuRQdM8cIEwP_p_ZIt9aFpOjVfzYc,3532
+autogen/tools/experimental/firecrawl/__init__.py,sha256=L-_y07Pzq4TeIIIoYce-catqiZI8f1UAAQpUoL-iYmA,215
+autogen/tools/experimental/firecrawl/firecrawl_tool.py,sha256=4Qj5rFGdQelWJ0OUEVquMZKkxZg5thE-A6Eo9nRPbsU,34803
 autogen/tools/experimental/google/__init__.py,sha256=_j5i9rzajzilQetKCPh5QkX60p1iOri2Yw69zIKAdU0,461
 autogen/tools/experimental/google/model.py,sha256=hm3ynXSJAjeld85iSHo9QgIehoWEqeVhbSPC3EaHm0M,515
 autogen/tools/experimental/google/toolkit_protocol.py,sha256=Jg9aSy4sKDr-yYVxevYr7zbEcmA9EwFWrJBOogLfKJU,485
@@ -353,14 +357,16 @@ autogen/tools/experimental/perplexity/__init__.py,sha256=VxWG2HfcxwBhLUV4LnAlte9
 autogen/tools/experimental/perplexity/perplexity_search.py,sha256=UZtyS9Pde3_VXufuZVuiBw87MiOiAWwVyXqstzE7Rlk,9976
 autogen/tools/experimental/reliable/__init__.py,sha256=GLvfvsFLfBANFVcOptIa_Tm71YsHt5U7X5N9quqDZW0,479
 autogen/tools/experimental/reliable/reliable.py,sha256=5s3aD5u-6JJrXqbSQBcW2qxNfyvx3MQ--IcZ6wMdsvE,64687
+autogen/tools/experimental/searxng/__init__.py,sha256=IiScuchBMO0wDPUo58s41aQbr0CrTkumHeZj-eSzZJo,223
+autogen/tools/experimental/searxng/searxng_search.py,sha256=Mi_FskvurZqZxFvDfrq59AxyeA8QE02A9lTeDufeNH8,4909
 autogen/tools/experimental/tavily/__init__.py,sha256=rvztagn7FpWiXZwJtZmKbIFBteBenaedwsiUViDyC4Y,220
 autogen/tools/experimental/tavily/tavily_search.py,sha256=S1Aj519kaEIwOAJXTc9Y__GHyY--vvc-P-5nh3ksCOQ,7810
 autogen/tools/experimental/web_search_preview/__init__.py,sha256=8vd1XWS14883MQ4U4dHvMP8aAKAtMILLnqRJNzgNh6A,233
 autogen/tools/experimental/web_search_preview/web_search_preview.py,sha256=bLKZSl0670uh82C3OCceRD0WaRfEPtQ4o06PTsQ3kKk,4818
 autogen/tools/experimental/wikipedia/__init__.py,sha256=jS_gUMyz-uCWr75yKv2esIiGz0ijNDNokt6wovQULLE,274
 autogen/tools/experimental/wikipedia/wikipedia.py,sha256=Cw7luG34J6if_5D40C_039BTA6l_LvOiEg6FY5UysC4,11545
-templates/main.jinja2,sha256=UWDs5BU-h-L9daFPIy4OAkWFMu2-g5p1R759-yF6Oks,1729
-templates/client_template/main.jinja2,sha256=HLXxq-DAqHyh_8PZ9uk6OVHB0RC6FZqKdIbnJQNJrLU,1978
+templates/main.jinja2,sha256=mu7z_4NjI1XMEwQx9KaFQixi4ImfwY0FZQaYUJVBnSc,1765
+templates/client_template/main.jinja2,sha256=9_1pmPNLlPZpfcE0KqXJ6uBVzQtmFqWbxs2xyzCOfxY,2115
 templates/config_template/config.jinja2,sha256=A-p-YBsnEm1iftBIhuIBfIyCY-qw34KWcxalmM6KPWc,167
 autogen/agentchat/contrib/captainagent/tools/README.md,sha256=454O-irP4gjSdYnFgoE0i3BfueXO0qFONxe3GMxMpHg,1677
 autogen/agentchat/contrib/captainagent/tools/__init__.py,sha256=tOTe4nwbKj7elHpftAy3zS_embMDzncrKL98XKhY6-c,168
@@ -399,8 +405,8 @@ autogen/agentchat/contrib/captainagent/tools/math/modular_inverse_sum.py,sha256=
 autogen/agentchat/contrib/captainagent/tools/math/simplify_mixed_numbers.py,sha256=iqgpFJdyBHPPNCqkehSIbeuV8Rabr2eDMilT23Wx7PI,1687
 autogen/agentchat/contrib/captainagent/tools/math/sum_of_digit_factorials.py,sha256=-6T5r6Er4mONPldRxv3F9tLoE7Og3qmeSeTC7Du_tTg,596
 autogen/agentchat/contrib/captainagent/tools/math/sum_of_primes_below.py,sha256=Xig7K3A3DRnbv-UXfyo5bybGZUQYAQsltthfTYW5eV8,509
-ag2-0.9.3.dist-info/METADATA,sha256=tFxYI45V9ZmkhlvptG5JWYFlC61We-KYQXvQdgWrmiI,35268
-ag2-0.9.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ag2-0.9.3.dist-info/licenses/LICENSE,sha256=GEFQVNayAR-S_rQD5l8hPdgvgyktVdy4Bx5-v90IfRI,11384
-ag2-0.9.3.dist-info/licenses/NOTICE.md,sha256=07iCPQGbth4pQrgkSgZinJGT5nXddkZ6_MGYcBd2oiY,1134
-ag2-0.9.3.dist-info/RECORD,,
+ag2-0.9.4.dist-info/METADATA,sha256=YnCG5kxaDTMI72s4N3OPKiuuepUwQVYLdKjgte3l6ZA,35268
+ag2-0.9.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ag2-0.9.4.dist-info/licenses/LICENSE,sha256=GEFQVNayAR-S_rQD5l8hPdgvgyktVdy4Bx5-v90IfRI,11384
+ag2-0.9.4.dist-info/licenses/NOTICE.md,sha256=07iCPQGbth4pQrgkSgZinJGT5nXddkZ6_MGYcBd2oiY,1134
+ag2-0.9.4.dist-info/RECORD,,

autogen/agentchat/contrib/agent_optimizer.py

@@ -140,7 +140,7 @@ History:
 
 According to the information I provide, please take one of four actions to manipulate list B using the functions you know.
 Instead of returning TERMINATE directly or taking no action, you should try your best to optimize the function list. Only take no action if you really think the current list is optimal, as more actions will harm performance in future tasks.
-Even adding a general function that can substitute the assistant’s repeated suggestions of Python code with the same functionality could also be helpful.
+Even adding a general function that can substitute the assistant's repeated suggestions of Python code with the same functionality could also be helpful.
 """
 
 
@@ -176,14 +176,15 @@ class AgentOptimizer:
     def __init__(
         self,
         max_actions_per_step: int,
-        llm_config: Union[LLMConfig, dict[str, Any]],
+        llm_config: Optional[Union[LLMConfig, dict[str, Any]]] = None,
         optimizer_model: Optional[str] = "gpt-4-1106-preview",
     ):
         """(These APIs are experimental and may change in the future.)
 
         Args:
             max_actions_per_step (int): the maximum number of actions that the optimizer can take in one step.
-            llm_config (LLMConfig or dict): llm inference configuration.
+            llm_config (LLMConfig or dict or None): llm inference configuration.
+                If None, the current LLMConfig from context is used.
                 Please refer to [OpenAIWrapper.create](https://docs.ag2.ai/latest/docs/api-reference/autogen/OpenAIWrapper/#autogen.OpenAIWrapper.create) for available options.
                 When using OpenAI or Azure OpenAI endpoints, please specify a non-empty 'model' either in `llm_config` or in each config of 'config_list' in `llm_config`.
             optimizer_model: the model used for the optimizer.
@@ -203,6 +204,8 @@ class AgentOptimizer:
         self._failure_functions_performance = []
         self._best_performance = -1
 
+        if llm_config is None:
+            llm_config = LLMConfig.current
         assert isinstance(llm_config, (dict, LLMConfig)), "llm_config must be a dict or LLMConfig"
         llm_config = copy.deepcopy(llm_config)
         self.llm_config = llm_config

autogen/agentchat/conversable_agent.py

@@ -77,13 +77,13 @@ from .chat import (
     initiate_chats,
 )
 from .group.context_variables import ContextVariables
+from .group.guardrails import Guardrail
 from .group.handoffs import Handoffs
 from .utils import consolidate_chat_info, gather_usage_summary
 
 if TYPE_CHECKING:
     from .group.on_condition import OnCondition
     from .group.on_context_condition import OnContextCondition
-
 __all__ = ("ConversableAgent",)
 
 logger = logging.getLogger(__name__)
@@ -224,6 +224,8 @@ class ConversableAgent(LLMAgent):
             handoffs (Handoffs): Handoffs object containing all handoff transition conditions.
         """
         self.handoffs = handoffs if handoffs is not None else Handoffs()
+        self.input_guardrails: list["Guardrail"] = []
+        self.output_guardrails: list["Guardrail"] = []
 
         # we change code_execution_config below and we have to make sure we don't change the input
         # in case of UserProxyAgent, without this we could even change the default value {}
@@ -1471,7 +1473,11 @@ class ConversableAgent(LLMAgent):
                 self.send(msg2send, recipient, request_reply=True, silent=silent)
 
             else:  # No breaks in the for loop, so we have reached max turns
-                iostream.send(TerminationEvent(termination_reason=f"Maximum turns ({max_turns}) reached"))
+                iostream.send(
+                    TerminationEvent(
+                        termination_reason=f"Maximum turns ({max_turns}) reached", sender=self, recipient=recipient
+                    )
+                )
         else:
             self._prepare_chat(recipient, clear_history)
             if isinstance(message, Callable):
@@ -1651,7 +1657,11 @@ class ConversableAgent(LLMAgent):
                     break
                 await self.a_send(msg2send, recipient, request_reply=True, silent=silent)
             else:  # No breaks in the for loop, so we have reached max turns
-                iostream.send(TerminationEvent(termination_reason=f"Maximum turns ({max_turns}) reached"))
+                iostream.send(
+                    TerminationEvent(
+                        termination_reason=f"Maximum turns ({max_turns}) reached", sender=self, recipient=recipient
+                    )
+                )
         else:
             self._prepare_chat(recipient, clear_history)
             if isinstance(message, Callable):
@@ -2587,7 +2597,7 @@ class ConversableAgent(LLMAgent):
             self._consecutive_auto_reply_counter[sender] = 0
 
             if termination_reason:
-                iostream.send(TerminationEvent(termination_reason=termination_reason))
+                iostream.send(TerminationEvent(termination_reason=termination_reason, sender=self, recipient=sender))
 
             return True, None
 
@@ -2727,7 +2737,7 @@ class ConversableAgent(LLMAgent):
             self._consecutive_auto_reply_counter[sender] = 0
 
             if termination_reason:
-                iostream.send(TerminationEvent(termination_reason=termination_reason))
+                iostream.send(TerminationEvent(termination_reason=termination_reason, sender=self, recipient=sender))
 
             return True, None
 
@@ -3994,6 +4004,42 @@ class ConversableAgent(LLMAgent):
         """
         self.handoffs.add_many(conditions)
 
+    def register_input_guardrail(self, guardrail: "Guardrail") -> None:
+        """
+        Register a guardrail to be used for input validation.
+
+        Args:
+            guardrail: The guardrail to register.
+        """
+        self.input_guardrails.append(guardrail)
+
+    def register_input_guardrails(self, guardrails: list["Guardrail"]) -> None:
+        """
+        Register multiple guardrails to be used for input validation.
+
+        Args:
+            guardrails: List of guardrails to register.
+        """
+        self.input_guardrails.extend(guardrails)
+
+    def register_output_guardrail(self, guardrail: "Guardrail") -> None:
+        """
+        Register a guardrail to be used for output validation.
+
+        Args:
+            guardrail: The guardrail to register.
+        """
+        self.output_guardrails.append(guardrail)
+
+    def register_output_guardrails(self, guardrails: list["Guardrail"]) -> None:
+        """
+        Register multiple guardrails to be used for output validation.
+
+        Args:
+            guardrails: List of guardrails to register.
+        """
+        self.output_guardrails.extend(guardrails)
+
 
 @export_module("autogen")
 def register_function(

autogen/agentchat/group/group_utils.py

@@ -137,13 +137,7 @@ def _run_oncontextconditions(
         if is_available and (
             on_condition.condition is None or on_condition.condition.evaluate(agent.context_variables)
         ):
-            # Condition has been met, we'll set the Tool Executor's next target
-            # attribute and that will be picked up on the next iteration when
-            # _determine_next_agent is called
-            for agent in agent._group_manager.groupchat.agents:  # type: ignore[attr-defined]
-                if isinstance(agent, GroupToolExecutor):
-                    agent.set_next_target(on_condition.target)
-                    break
+            on_condition.target.activate_target(agent._group_manager.groupchat)  # type: ignore[attr-defined]
 
             transfer_name = on_condition.target.display_name()
 
@@ -212,6 +206,18 @@ def ensure_handoff_agents_in_group(agents: list["ConversableAgent"]) -> None:
                 raise ValueError("Agent in after work target Hand-offs must be in the agents list")
 
 
+def ensure_guardrail_agents_in_group(agents: list["ConversableAgent"]) -> None:
+    """Ensure the agents in handoffs are in the group chat."""
+    agent_names = [agent.name for agent in agents]
+    for agent in agents:
+        for guardrail in agent.input_guardrails + agent.output_guardrails:
+            if (
+                isinstance(guardrail.target, (AgentTarget, AgentNameTarget))
+                and guardrail.target.agent_name not in agent_names
+            ):
+                raise ValueError("Agent in guardrail's target must be in the agents list")
+
+
 def prepare_exclude_transit_messages(agents: list["ConversableAgent"]) -> None:
     """Preparation for excluding transit messages by getting all tool names and registering a hook on agents to remove those messages."""
     # get all transit functions names
@@ -254,6 +260,9 @@ def prepare_group_agents(
     # Ensure all agents in hand-off after-works are in the passed in agents list
     ensure_handoff_agents_in_group(agents)
 
+    # Ensure all agents in guardrails are in the passed in agents list
+    ensure_guardrail_agents_in_group(agents)
+
     # Create Tool Executor for the group
     tool_execution = GroupToolExecutor()
 

autogen/agentchat/group/guardrails.py

@@ -0,0 +1,171 @@
+# Copyright (c) 2023 - 2025, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import json
+import re
+from abc import ABC, abstractmethod
+from typing import TYPE_CHECKING, Any, Optional, Union
+
+from pydantic import BaseModel, Field
+
+from ...oai.client import OpenAIWrapper
+
+if TYPE_CHECKING:
+    from ...llm_config import LLMConfig
+    from .targets.transition_target import TransitionTarget
+
+
+class GuardrailResult(BaseModel):
+    """Represents the outcome of a guardrail check."""
+
+    activated: bool
+    justification: str = Field(default="No justification provided")
+
+    def __str__(self) -> str:
+        return f"Guardrail Result: {self.activated}\nJustification: {self.justification}"
+
+    @staticmethod
+    def parse(text: str) -> "GuardrailResult":
+        """Parses a JSON string into a GuardrailResult object.
+
+        Args:
+            text (str): The JSON string to parse.
+
+        Returns:
+            GuardrailResult: The parsed GuardrailResult object.
+        """
+        try:
+            data = json.loads(text)
+            return GuardrailResult(**data)
+        except (json.JSONDecodeError, ValueError) as e:
+            raise ValueError(f"Failed to parse GuardrailResult from text: {text}") from e
+
+
+class Guardrail(ABC):
+    """Abstract base class for guardrails."""
+
+    def __init__(
+        self, name: str, condition: str, target: "TransitionTarget", activation_message: Optional[str] = None
+    ) -> None:
+        self.name = name
+        self.condition = condition
+        self.target = target
+        self.activation_message = (
+            activation_message if activation_message else f"Guardrail '{name}' has been activated."
+        )
+
+    @abstractmethod
+    def check(
+        self,
+        context: Union[str, list[dict[str, Any]]],
+    ) -> GuardrailResult:
+        """Checks the text against the guardrail and returns a GuardrailResult.
+
+        Args:
+            context (Union[str, list[dict[str, Any]]]): The context to check against the guardrail.
+
+        Returns:
+            GuardrailResult: The result of the guardrail check.
+        """
+        pass
+
+
+class LLMGuardrail(Guardrail):
+    """Guardrail that uses an LLM to check the context."""
+
+    def __init__(
+        self,
+        name: str,
+        condition: str,
+        target: "TransitionTarget",
+        llm_config: "LLMConfig",
+        activation_message: Optional[str] = None,
+    ) -> None:
+        super().__init__(name, condition, target, activation_message)
+
+        if not llm_config:
+            raise ValueError("LLMConfig is required.")
+
+        self.llm_config = llm_config.deepcopy()
+        setattr(self.llm_config, "response_format", GuardrailResult)
+        self.client = OpenAIWrapper(**self.llm_config.model_dump())
+
+        self.check_prompt = f"""You are a guardrail that checks if a condition is met in the conversation you are given.
+You will activate the guardrail only if the condition is met.
+
+**Condition: {self.condition}**"""
+
+    def check(
+        self,
+        context: Union[str, list[dict[str, Any]]],
+    ) -> GuardrailResult:
+        """Checks the context against the guardrail using an LLM.
+
+        Args:
+            context (Union[str, list[dict[str, Any]]]): The context to check against the guardrail.
+
+        Returns:
+            GuardrailResult: The result of the guardrail check.
+        """
+        # Set the check prompt as the system message
+        check_messages = [{"role": "system", "content": self.check_prompt}]
+        # If context is a string, wrap it in a user message and append it
+        if isinstance(context, str):
+            check_messages.append({"role": "user", "content": context})
+        # If context is a list of messages, append them
+        elif isinstance(context, list):
+            check_messages.extend(context)
+        else:
+            raise ValueError("Context must be a string or a list of messages.")
+        # Call the LLM with the check messages
+        response = self.client.create(messages=check_messages)
+        return GuardrailResult.parse(response.choices[0].message.content)  # type: ignore
+
+
+class RegexGuardrail(Guardrail):
+    """Guardrail that checks the context against a regular expression."""
+
+    def __init__(
+        self,
+        name: str,
+        condition: str,
+        target: "TransitionTarget",
+        activation_message: Optional[str] = None,
+    ) -> None:
+        super().__init__(name, condition, target, activation_message)
+        # Compile the regular expression condition
+        try:
+            self.regex = re.compile(condition)
+        except re.error as e:
+            raise ValueError(f"Invalid regex pattern '{condition}': {str(e)}")
+
+    def check(
+        self,
+        context: Union[str, list[dict[str, Any]]],
+    ) -> GuardrailResult:
+        """Checks the context against the guardrail using a regular expression.
+
+        Args:
+            context (Union[str, list[dict[str, Any]]]): The context to check against the guardrail.
+
+        Returns:
+            GuardrailResult: The result of the guardrail check.
+        """
+        # Create a list of the messages to check
+        if isinstance(context, str):
+            messages = [context]
+        elif isinstance(context, list):
+            messages = [message.get("content", "") for message in context]
+        else:
+            raise ValueError("Context must be a string or a list of messages.")
+
+        # Check each message against the regex
+        for message in messages:
+            match = self.regex.search(message)
+            # If a match is found, activate the guardrail and return the result
+            if match:
+                activated = True
+                justification = f"Match found -> {match.group(0)}"
+                return GuardrailResult(activated=activated, justification=justification)
+        return GuardrailResult(activated=False, justification="No match found in the context.")

autogen/agentchat/group/targets/transition_target.py

@@ -68,6 +68,16 @@ class TransitionTarget(BaseModel):
         """Create a wrapper agent for the target if needed."""
         raise NotImplementedError("Requires subclasses to implement.")
 
+    def activate_target(self, groupchat: "GroupChat") -> None:
+        """Activate the target in the groupchat, setting the next target for GroupToolExecutor.
+
+        The Tool Executor's next target attribute will be picked up on the next iteration when _determine_next_agent is called"""
+        for agent in groupchat.agents:  # type: ignore[attr-defined]
+            # get the GroupToolExecutor agent
+            if type(agent).__name__ == "GroupToolExecutor":
+                agent.set_next_target(self)  # type: ignore[attr-defined]
+                return
+
 
 class AgentTarget(TransitionTarget):
     """Target that represents a direct agent reference."""