mrchuy commited on
Commit
ac9159c
·
1 Parent(s): 5e93919

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +21 -1
main.py CHANGED
@@ -4,4 +4,24 @@ app = FastAPI()
4
 
5
  @app.get("/")
6
  def read_root():
7
- return {"Hello": "langchain {}".format(langchain.__version__)}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
 
5
  @app.get("/")
6
  def read_root():
7
+ run_something()
8
+ return {"Hello": "langchain {}".format(langchain.__version__)}
9
+
10
+ from langchain.llms.fake import FakeListLLM
11
+
12
+ from langchain.agents import load_tools
13
+ from langchain.agents import initialize_agent
14
+ from langchain.agents import AgentType
15
+
16
+ def run_something()
17
+ tools = load_tools(["python_repl"])
18
+ responses = ["Action: Python REPL\nAction Input: print(2 + 2)", "Final Answer: 4"]
19
+ llm = FakeListLLM(responses=responses)
20
+
21
+ agent = initialize_agent(
22
+ tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True
23
+ )
24
+ agent.run("whats 2 + 2")
25
+
26
+
27
+