Add agent_xml_demo.py: Demo agent for XML output with OpenAI GPT-4o-mini

pull/845/head
ascender1729 2 months ago
parent 7a62ef4b5b
commit cea6572f0f

@ -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()
Loading…
Cancel
Save