3rushi commited on
Commit
4ea678e
·
verified ·
1 Parent(s): b52bf3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -12,7 +12,12 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
12
 
13
  @tool
14
  def wikipedia_search(query: str) -> str:
15
- """Search Wikipedia and return a concise summary."""
 
 
 
 
 
16
  try:
17
  docs = WikipediaLoader(query=query, load_max_docs=2).load()
18
  return "\n\n".join(doc.page_content[:1500] for doc in docs)
@@ -21,7 +26,12 @@ def wikipedia_search(query: str) -> str:
21
 
22
  @tool
23
  def arxiv_search(query: str) -> str:
24
- """Search arXiv and return paper abstracts."""
 
 
 
 
 
25
  try:
26
  docs = ArxivLoader(query=query, load_max_docs=2).load()
27
  return "\n\n".join(
 
12
 
13
  @tool
14
  def wikipedia_search(query: str) -> str:
15
+ """
16
+ Search Wikipedia and return a concise summary.
17
+
18
+ Args:
19
+ query: The search query string.
20
+ """
21
  try:
22
  docs = WikipediaLoader(query=query, load_max_docs=2).load()
23
  return "\n\n".join(doc.page_content[:1500] for doc in docs)
 
26
 
27
  @tool
28
  def arxiv_search(query: str) -> str:
29
+ """
30
+ Search arXiv and return paper abstracts.
31
+
32
+ Args:
33
+ query: The search query string.
34
+ """
35
  try:
36
  docs = ArxivLoader(query=query, load_max_docs=2).load()
37
  return "\n\n".join(