zvl commited on
Commit
469d60b
·
verified ·
1 Parent(s): 75b7b6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -9,7 +9,7 @@ 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 buscar_datos_gob(term: str, page_size: int = 10, page: int = 0) -> dict:
13
  """
14
  Busca conjuntos de datos en la API de datos.gob.es.
15
 
@@ -17,15 +17,17 @@ def buscar_datos_gob(term: str, page_size: int = 10, page: int = 0) -> dict:
17
  term (str): Palabra clave para buscar conjuntos de datos.
18
  page_size (int): Número de resultados por página. Máximo 50. (Por defecto: 10)
19
  page (int): Página de resultados a consultar. (Por defecto: 0)
 
20
 
21
  Returns:
22
  dict: Diccionario con los conjuntos de datos encontrados o un mensaje de error.
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"
@@ -75,7 +77,7 @@ with open("prompts.yaml", 'r') as stream:
75
 
76
  agent = CodeAgent(
77
  model=model,
78
- tools=[final_answer, buscar_datos_gob], ## add your tools here (don't remove final answer)
79
  max_steps=6,
80
  verbosity_level=1,
81
  grammar=None,
 
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, sort: str = "title") -> dict:
13
  """
14
  Busca conjuntos de datos en la API de datos.gob.es.
15
 
 
17
  term (str): Palabra clave para buscar conjuntos de datos.
18
  page_size (int): Número de resultados por página. Máximo 50. (Por defecto: 10)
19
  page (int): Página de resultados a consultar. (Por defecto: 0)
20
+ sort (str): Campo por el cual ordenar los resultados (Por defecto: "title").
21
 
22
  Returns:
23
  dict: Diccionario con los conjuntos de datos encontrados o un mensaje de error.
24
  """
25
+ base_url = "https://datos.gob.es/apidata/catalog/dataset"
26
  params = {
27
  "q": term,
28
  "_pageSize": page_size,
29
+ "_page": page,
30
+ "_sort": sort # ✅ Ahora `sort` está definido
31
  }
32
  headers = {
33
  "Accept": "application/json"
 
77
 
78
  agent = CodeAgent(
79
  model=model,
80
+ tools=[final_answer, buscar_datos_gob, get_current_time_in_timezone, image_generation_tool], # Ahora se usa
81
  max_steps=6,
82
  verbosity_level=1,
83
  grammar=None,