ag2 0.9.1a1__py3-none-any.whl → 0.9.2__py3-none-any.whl

autogen/types.py

@@ -0,0 +1,29 @@
+# Copyright (c) 2023 - 2025, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# Portions derived from  https://github.com/microsoft/autogen are under the MIT License.
+# SPDX-License-Identifier: MIT
+
+from typing import Any, Literal, TypedDict, Union
+
+MessageContentType = Union[str, list[Union[dict[str, Any], str]], None]
+
+
+class UserMessageTextContentPart(TypedDict):
+    """Represents a text content part of a user message"""
+
+    type: Literal["text"]
+    """The type of the content part. Always "text" for text content parts."""
+    text: str
+    """The text content of the part."""
+
+
+class UserMessageImageContentPart(TypedDict):
+    """Represents an image content part of a user message"""
+
+    type: Literal["image_url"]
+    """The type of the content part. Always "image_url" for image content parts."""
+    # Ignoring the other "detail param for now"
+    image_url: dict[Literal["url"], str]
+    """The URL of the image."""

autogen/version.py

@@ -0,0 +1,7 @@
+# Copyright (c) 2023 - 2025, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
+#
+# SPDX-License-Identifier: Apache-2.0
+
+__all__ = ["__version__"]
+
+__version__ = "0.9.2"

templates/client_template/main.jinja2

@@ -0,0 +1,69 @@
+{{imports}}
+
+import argparse
+import os
+from typing import *
+
+from autogen.mcp.mcp_proxy import MCPProxy
+from autogen.mcp.mcp_proxy.security import BaseSecurity
+from autogen.mcp.mcp_proxy.security import {% for i in security_classes %}{{i}}{% if not loop.last %}, {% endif %}{% endfor %}
+
+app = MCPProxy(
+    {% if info %}
+    {% for key,value in info.items() %}
+    {% set info_value= value.__repr__() %}
+    {% if not key.startswith('x-') %}
+    {{ key }} = {{info_value}},
+    {% endif %}
+    {% endfor %}
+    {% endif %}
+    )
+
+
+{% for operation in operations %}
+@app.{{operation.type}}('{{operation.path}}'
+    {% if operation.description %}
+    , description=""" {{operation.description}} """
+    {% endif %}
+    {% if operation.tags %}
+    , tags={{operation.tags}}
+    {% endif %}
+    {% if operation.security %}
+    , security=[{% for security in operation.security %}
+    {% for key, value in security.items() %}
+    {{security_parameters[key]}},
+    {% endfor %}
+    {% endfor %}]
+    {% endif %}
+    )
+def {{operation.function_name}}({{operation.snake_case_arguments}}
+    ):
+    {%- if operation.summary %}
+    """
+    {{ operation.summary }}
+    """
+    {%- endif %}
+    raise RuntimeError("Should be patched by MCPProxy and never executed")
+{% endfor %}
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Math Server")
+    parser.add_argument("transport", choices=["stdio", "sse"], help="Transport mode (stdio or sse)")
+    args = parser.parse_args()
+
+    if "CONFIG_PATH" in os.environ:
+        config_path = os.environ["CONFIG_PATH"]
+        app.load_configuration(config_path)
+
+    if "CONFIG" in os.environ:
+        config = os.environ["CONFIG"]
+        app.load_configuration_from_string(config)
+
+    if "SECURITY" in os.environ:
+        security_params = BaseSecurity.parse_security_parameters_from_env(
+            os.environ,
+        )
+
+        app.set_security_params(security_params)
+
+    app.mcp.run(transport=args.transport)

templates/config_template/config.jinja2

@@ -0,0 +1,7 @@
+{
+  "server": {
+    "url": "{{ server_url }}"
+  },
+  "authentication": {{ authentications | tojson(indent=2) }},
+  "operations": {{ operations | tojson(indent=2) }}
+}

templates/main.jinja2

@@ -0,0 +1,61 @@
+{{imports}}
+
+import argparse
+import os
+from typing import *
+
+from autogen.mcp.mcp_proxy import MCPProxy
+from autogen.mcp.mcp_proxy.security import BaseSecurity
+from autogen.mcp.mcp_proxy.security import {% for i in security_classes %}{{i}}{% if not loop.last %}, {% endif %}{% endfor %}
+
+app = MCPProxy(
+    {% if info %}
+    {% for key,value in info.items() %}
+    {% set info_value= value.__repr__() %}
+    {% if not key.startswith('x-') %}
+    {{ key }} = {{info_value}},
+    {% endif %}
+    {% endfor %}
+    {% endif %}
+    )
+
+
+{% for operation in operations %}
+@app.{{operation.type}}('{{operation.path}}'
+    {% if operation.description %}
+    , description=""" {{operation.description}} """
+    {% endif %}
+    {% if operation.tags %}
+    , tags={{operation.tags}}
+    {% endif %}
+    {% if operation.security %}
+    , security=[{% for security in operation.security %}
+    {% for key, value in security.items() %}
+    {{security_parameters[key]}},
+    {% endfor %}
+    {% endfor %}]
+    {% endif %}
+    )
+def {{operation.function_name}}({{operation.snake_case_arguments}}
+    ):
+    {%- if operation.summary %}
+    """
+    {{ operation.summary }}
+    """
+    {%- endif %}
+    raise RuntimeError("Should be patched by MCPProxy and never executed")
+{% endfor %}
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Math Server")
+    parser.add_argument("transport", choices=["stdio", "sse"], help="Transport mode (stdio or sse)")
+    args = parser.parse_args()
+
+    if "SECURITY" in os.environ:
+        security_params = BaseSecurity.parse_security_parameters_from_env(
+            os.environ,
+        )
+
+        app.set_security_params(security_params)
+
+    app._mcp.run(transport=args.transport)

ag2-0.9.1a1.dist-info/RECORD

@@ -1,6 +0,0 @@
-ag2-0.9.1a1.dist-info/LICENSE,sha256=GEFQVNayAR-S_rQD5l8hPdgvgyktVdy4Bx5-v90IfRI,11384
-ag2-0.9.1a1.dist-info/METADATA,sha256=u_gE4P7xBKKQBZEkpPyKOzSXZ2X8dLYXWi1JskiI10s,24603
-ag2-0.9.1a1.dist-info/NOTICE.md,sha256=07iCPQGbth4pQrgkSgZinJGT5nXddkZ6_MGYcBd2oiY,1134
-ag2-0.9.1a1.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-ag2-0.9.1a1.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-ag2-0.9.1a1.dist-info/RECORD,,

ag2-0.9.1a1.dist-info/top_level.txt

@@ -1 +0,0 @@
-