ag2 0.9.1.post0__py3-none-any.whl → 0.9.3__py3-none-any.whl

autogen/version.py

@@ -4,4 +4,4 @@
 
 __all__ = ["__version__"]
 
-__version__ = "0.9.1post0"
+__version__ = "0.9.3"

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)