Update main.py
Browse files
main.py
CHANGED
@@ -4,4 +4,24 @@ app = FastAPI()
|
|
4 |
|
5 |
@app.get("/")
|
6 |
def read_root():
|
7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
|