first commit

.env.example

@@ -0,0 +1,3 @@
+OPENAI_API_KEY=<OpenAI_APIキー>
+GOOGLE_CSE_ID=<Googleカスタム検索_検索エンジンID>
+GOOGLE_API_KEY=<Googleカスタム検索_APIキー>

.gitignore

@@ -0,0 +1 @@
+.env

app.py

@@ -0,0 +1,52 @@
+import os
+import dotenv
+import sys
+import gradio as gr
+
+from langchain.agents import ZeroShotAgent, Tool, AgentExecutor, load_tools
+from langchain import OpenAI, SerpAPIWrapper, LLMChain
+
+dotenv.load_dotenv()
+
+OPENAI_API_KEY=os.environ["OPENAI_API_KEY"]
+GOOGLE_CSE_ID=os.environ["GOOGLE_CSE_ID"]
+GOOGLE_API_KEY=os.environ["GOOGLE_API_KEY"]
+
+def search_ang_generate(question):
+    # ツールの準備
+    tools = load_tools(["google-search"], llm=OpenAI())
+
+    # プロンプトテンプレートの準備
+    prefix = """次の質問にできる限り答えてください。次のツールにアクセスできます:"""
+    suffix = """始めましょう!
+
+    Question: {input}
+    {agent_scratchpad}"""
+
+    prompt = ZeroShotAgent.create_prompt(
+        tools,
+        prefix=prefix,
+        suffix=suffix,
+        input_variables=["input", "agent_scratchpad"]
+    )
+
+    # エージェントの準備
+    llm_chain = LLMChain(llm=OpenAI(temperature=0), prompt=prompt)
+    agent = ZeroShotAgent(llm_chain=llm_chain, tools=tools)
+    # agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=tools)
+    agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=tools, verbose=True)
+
+    result = agent_executor.run(question)
+    
+    return result
+
+def main():
+    if len(sys.argv) > 1:
+        question = sys.argv[1]
+        result = search_ang_generate(question)
+        print(result)
+    else:
+        gr.Interface(fn=search_ang_generate, inputs="text", outputs="text").launch()
+
+if __name__ == "__main__":
+	main()

requirements.txt

@@ -0,0 +1,5 @@
+langchain
+openai
+python-dotenv
+google-api-python-client
+gradio