From cea6572f0f6cb4877b1b206b8e6d5016fe590db4 Mon Sep 17 00:00:00 2001 From: ascender1729 Date: Wed, 14 May 2025 00:21:13 +0000 Subject: [PATCH] Add agent_xml_demo.py: Demo agent for XML output with OpenAI GPT-4o-mini --- examples/agent_xml_demo.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 examples/agent_xml_demo.py diff --git a/examples/agent_xml_demo.py b/examples/agent_xml_demo.py new file mode 100644 index 00000000..f2bd292f --- /dev/null +++ b/examples/agent_xml_demo.py @@ -0,0 +1,26 @@ +from swarms.structs.agent import Agent +from swarms.utils.litellm_wrapper import LiteLLM +import os + +def main(): + api_key = os.getenv("OPENAI_API_KEY") + llm = LiteLLM( + model_name="gpt-4o-mini", + openai_api_key=api_key, + temperature=0.1, + max_tokens=256, + system_prompt="You are an XML output agent." + ) + agent = Agent( + llm=llm, + agent_name="XMLAgent", + agent_description="Demo agent for XML output", + max_loops=1, + output_type="xml", + ) + result = agent.run("What is the capital of France?") + print("Agent XML output:") + print(result) + +if __name__ == "__main__": + main()