zvl commited on
Commit
87b53a2
·
verified ·
1 Parent(s): ae7a494

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -6
app.py CHANGED
@@ -9,14 +9,32 @@ from Gradio_UI import GradioUI
9
 
10
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
11
  @tool
12
- def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
13
- #Keep this format for the description / args / args description but feel free to modify the tool
14
- """A tool that does nothing yet
 
15
  Args:
16
- arg1: the first argument
17
- arg2: the second argument
 
 
 
 
18
  """
19
- return "What magic will you build ?"
 
 
 
 
 
 
 
 
 
 
 
 
 
20
 
21
  @tool
22
  def get_current_time_in_timezone(timezone: str) -> str:
 
9
 
10
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
11
  @tool
12
+ def buscar_datos_gob(term: str, page_size: int = 10, page: int = 0) -> dict:
13
+ """
14
+ Busca conjuntos de datos en datos.gob.es que coincidan con el término proporcionado.
15
+
16
  Args:
17
+ term (str): Término de búsqueda.
18
+ page_size (int): Número de resultados por página (máximo 50).
19
+ page (int): Número de la página a consultar.
20
+
21
+ Returns:
22
+ dict: Resultados de la búsqueda.
23
  """
24
+ base_url = "http://datos.gob.es/apidata/catalog/dataset"
25
+ params = {
26
+ "q": term,
27
+ "_pageSize": page_size,
28
+ "_page": page
29
+ }
30
+ headers = {
31
+ "Accept": "application/json"
32
+ }
33
+ response = requests.get(base_url, params=params, headers=headers)
34
+ if response.status_code == 200:
35
+ return response.json()
36
+ else:
37
+ return {"error": f"Error en la solicitud: {response.status_code}"}
38
 
39
  @tool
40
  def get_current_time_in_timezone(timezone: str) -> str: