Spaces:
Running
Running
Update app.py
Browse files
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 = "
|
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
|
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 sí se usa
|
81 |
max_steps=6,
|
82 |
verbosity_level=1,
|
83 |
grammar=None,
|