Spaces:
Sleeping
Sleeping
File size: 3,026 Bytes
b3f9415 92357df b3f9415 92357df b3f9415 b16840f b3f9415 |
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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
import os
from langchain_community.tools import DuckDuckGoSearchRun
from langchain_community.tools.tavily_search import TavilySearchResults
from langchain_community.document_loaders import WikipediaLoader
from langchain_community.document_loaders import ArxivLoader
from langchain_core.tools import tool
@tool
def calculator(a: float, b: float, type: str) -> float:
"""Performs mathematical calculations, addition, subtraction, multiplication, division, modulus.
Args:
a (float): first float number
b (float): second float number
type (str): the type of calculation to perform, can be addition, subtraction, multiplication, division, modulus
"""
if type == "addition":
return a + b
elif type == "subtraction":
return a - b
elif type == "multiplication":
return a * b
elif type == "division":
if b == 0:
raise ValueError("Cannot divide by zero.")
return a / b
elif type == "modulus":
a % b
else:
TypeError(f"{type} is not an option for type, choose one of addition, subtraction, multiplication, division, modulus")
@tool
def duck_web_search(query: str) -> str:
"""Use DuckDuckGo to search the web.
Args:
query: The search query.
"""
search = DuckDuckGoSearchRun().invoke(query=query)
return {"duckduckgo_web_search": search}
@tool
def wiki_search(query: str) -> str:
"""Search Wikipedia for a query and return maximum 3 results.
Args:
query: The search query."""
documents = WikipediaLoader(query=query, load_max_docs=3).load()
processed_documents = "\n\n---\n\n".join(
[
f'Document title: {document.metadata.get("title", "")}. Summary: {document.metadata.get("summary", "")}. Documents details: {document.page_content}'
for document in documents
])
return {"wiki_results": processed_documents}
@tool
def arxiv_search(query: str) -> str:
"""Search Arxiv for a query and return maximum 3 result.
Args:
query: The search query."""
documents = ArxivLoader(query=query, load_max_docs=3).load()
processed_documents = "\n\n---\n\n".join(
[
f'Document title: {document.metadata.get("title", "")}. Summary: {document.metadata.get("summary", "")}. Documents details: {document.page_content}'
for document in documents
])
return {"arxiv_results": processed_documents}
@tool
def tavily_web_search(query: str) -> str:
"""Search the web using Tavily for a query and return maximum 3 results.
Args:
query: The search query."""
search_engine = TavilySearchResults(max_results=3)
search_documents = search_engine.invoke(input=query)
web_results = "\n\n---\n\n".join(
[
f'Document title: {document["title"]}. Contents: {document["content"]}. Relevance Score: {document["score"]}'
for document in search_documents
])
return {"web_results": web_results} |