Create article_generator.py

article_generator.py

@@ -0,0 +1,18 @@
+from langchain.llms import OpenAI
+from langchain_experimental.plan_and_execute import PlanAndExecute, load_agent_executor, load_chat_planner
+import openai
+import os
+
+def generate_article(headline, api_key):
+    """見出しを基にしてLangchainのPlanAndExecuteモデルにより記事本文を生成する"""
+    openai.api_key = api_key
+    openai_llm = OpenAI(api_key=api_key)
+
+    # PlanAndExecuteの設定
+    agent_executor = load_agent_executor("simple", llm=openai_llm)
+    chat_planner = load_chat_planner("headline_to_article", llm=openai_llm)
+    plan_and_execute = PlanAndExecute(planner=chat_planner, executor=agent_executor)
+
+    # 記事生成
+    output = plan_and_execute.execute({"headline": headline})
+    return output["text"]