Spaces:
Runtime error
Runtime error
File size: 1,807 Bytes
e67043b |
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 |
from ..tool import Tool
from typing import Any
import arxiv
def build_tool(config) -> Tool:
tool = Tool(
"Arxiv",
"Look up for information from scientific articles on arxiv.org",
name_for_model="Arxiv",
description_for_model=(
"Search information from Arxiv.org "
"Useful for when you need to answer questions about Physics, Mathematics, "
"Computer Science, Quantitative Biology, Quantitative Finance, Statistics, "
"Electrical Engineering, and Economics "
"from scientific articles on arxiv.org. "
"Input should be a search query."
),
logo_url="https://your-app-url.com/.well-known/logo.png",
contact_email="[email protected]",
legal_info_url="[email protected]",
)
arxiv_exceptions: Any # :meta private:
top_k_results: int = 3
ARXIV_MAX_QUERY_LENGTH = 300
doc_content_chars_max: int = 4000
@tool.get("/get_arxiv_article_information")
def get_arxiv_article_information(query: str):
"""Run Arxiv search and get the article meta information."""
param = {"q": query}
try:
results = arxiv.Search( # type: ignore
query[:ARXIV_MAX_QUERY_LENGTH], max_results=top_k_results
).results()
except arxiv_exceptions as ex:
return f"Arxiv exception: {ex}"
docs = [
f"Published: {result.updated.date()}\nTitle: {result.title}\n"
f"Authors: {', '.join(a.name for a in result.authors)}\n"
f"Summary: {result.summary}"
for result in results
]
if docs:
return "\n\n".join(docs)[:doc_content_chars_max]
else:
return "No good Arxiv Result was found"
return tool
|