fffiloni commited on
Commit
f5d0f7b
·
1 Parent(s): 497eb54

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -0
app.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import gradio as gr
3
+
4
+ from langchain.chat_models import ChatOpenAI
5
+ from langchain.agents import load_tools, initialize_agent
6
+ from langchain.agents import AgentType
7
+ from langchain.tools import AIPluginTool
8
+
9
+ tool = AIPluginTool.from_plugin_url("https://www.klarna.com/.well-known/ai-plugin.json")
10
+ llm = ChatOpenAI(temperature=0)
11
+ tools = load_tools(["requests_all"] )
12
+ tools += [tool]
13
+ agent_chain = initialize_agent( tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)
14
+
15
+ def run(prompt):
16
+ return agent_chain.run("what t shirts are available in klarna?")
17
+
18
+ with gr.Blocks() as demo:
19
+ with gr.Column():
20
+ prompt = gr.Textbox()
21
+ run_btn = gr.Button("Run")
22
+ response = gr.Textbox()
23
+ run_btn.click(fn=run,
24
+ inputs=[prompt],
25
+ outputs=[response]
26
+ )
27
+ demo.launch()