Update and rename blog_gen_example.py to blog_gen_example.py

pull/307/head
pliny 1 year ago committed by Zack
parent 0508503c64
commit f530ebcc4b

@ -5,9 +5,7 @@ from 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."
)
raise ValueError("OPENAI_API_KEY environment variable not set.")
blog_topic = input("Enter the topic for the blog generation: ")

@ -1,23 +0,0 @@
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