-
Notifications
You must be signed in to change notification settings - Fork 1
/
tools.py
34 lines (24 loc) · 881 Bytes
/
tools.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
from langchain_google_community import GoogleSearchAPIWrapper
from langchain_core.tools import Tool
def multiplier(a, b):
return a * b
def parsing_multiplier(string):
a, b = string.split(",")
return multiplier(int(a), int(b))
google = GoogleSearchAPIWrapper()
def top5_results(query):
return google.results(query, 5)
TOOL_PARSE_MULTIPLIER = Tool(
name="Multiplier",
func=parsing_multiplier,
description=(
"useful for when you need to multiply two numbers together. "
"The input to this tool should be a comma separated list of numbers of length two, representing the two numbers you want to multiply together. "
"For example, `1,2` would be the input if you wanted to multiply 1 by 2."
),
)
TOOL_GOOGLE = Tool(
name="google-search",
description="Search Google for recent results.",
func=top5_results,
)