gpt-4対応

app.py

@@ -3,8 +3,10 @@ import dotenv
 import sys
 import gradio as gr
 
-from langchain.agents import ZeroShotAgent, Tool, AgentExecutor, load_tools
-from langchain import OpenAI, SerpAPIWrapper, LLMChain
+from langchain import LLMChain
+from langchain.agents import ZeroShotAgent, AgentExecutor, load_tools, Tool
+from langchain.chat_models import ChatOpenAI
+from langchain.utilities import GoogleSearchAPIWrapper
 
 dotenv.load_dotenv()
 
@@ -12,13 +14,20 @@ 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):
+def search_and_generate(question, prefix = "次の質問にできる限り答えてください。"):
     # ツールの準備
-    tools = load_tools(["google-search"], llm=OpenAI())
+    search = GoogleSearchAPIWrapper()
+    search_tool = Tool(
+        name="Search",
+        func=search.run,
+        description="useful for when you need to answer questions about current events"
+    ),
+
+    tools = load_tools(["google-search"], llm=ChatOpenAI())
 
     # プロンプトテンプレートの準備
-    prefix = """次の質問にできる限り答えてください。次のツールにアクセスできます:"""
-    suffix = """始めましょう!
+    prefix = f"{prefix} 次のツールにアクセスできます:"
+    suffix = """始めましょう。
 
     Question: {input}
     {agent_scratchpad}"""
@@ -31,9 +40,9 @@ def search_ang_generate(question):
     )
 
     # エージェントの準備
-    llm_chain = LLMChain(llm=OpenAI(temperature=0), prompt=prompt)
+    llm = ChatOpenAI(model_name="gpt-4")
+    llm_chain = LLMChain(llm=llm, 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)
@@ -43,10 +52,10 @@ def search_ang_generate(question):
 def main():
     if len(sys.argv) > 1:
         question = sys.argv[1]
-        result = search_ang_generate(question)
+        result = search_and_generate(question)
         print(result)
     else:
-        gr.Interface(fn=search_ang_generate, inputs="text", outputs="text").launch()
+        gr.Interface(fn=search_and_generate, inputs="text", outputs="text").launch()
 
 if __name__ == "__main__":
-	main()
+    main()