Create blog_gen_example.py

pull/158/head
pliny 1 year ago committed by GitHub
parent c85275266a
commit 7554fbd0cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -0,0 +1,23 @@
import os
from swarms.swarms.blog_gen import BlogGen
def main():
api_key = os.getenv("OPENAI_API_KEY")
if not api_key:
raise ValueError("OPENAI_API_KEY environment variable not set.")
blog_topic = input("Enter the topic for the blog generation: ")
blog_generator = BlogGen(api_key, blog_topic)
blog_generator.TOPIC_SELECTION_SYSTEM_PROMPT = (
blog_generator.TOPIC_SELECTION_SYSTEM_PROMPT.replace(
"{{BLOG_TOPIC}}", blog_topic
)
)
blog_generator.run_workflow()
if __name__ == "__main__":
main()
Loading…
Cancel
Save