geronimo-pericoli commited on
Commit
12c2f26
·
verified ·
1 Parent(s): 724e391

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -1
app.py CHANGED
@@ -12,6 +12,7 @@ from llama_index.llms.openai import OpenAI
12
  from llama_index.embeddings.openai import OpenAIEmbedding
13
  from typing import Optional, List, Dict, Any
14
  from pathlib import Path
 
15
  import json
16
  import os
17
  import asyncio
@@ -203,10 +204,25 @@ demo = gr.TabbedInterface(
203
  description="Muestra los retrievers disponibles, opcionalmente filtrados por fuente.",
204
  api_name="list_retrievers"
205
  )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
206
  ],
207
  [
208
  "Búsqueda ArXiv",
209
- "Lista de Retrievers"
 
210
  ]
211
  )
212
 
 
12
  from llama_index.embeddings.openai import OpenAIEmbedding
13
  from typing import Optional, List, Dict, Any
14
  from pathlib import Path
15
+ import aiohttp
16
  import json
17
  import os
18
  import asyncio
 
204
  description="Muestra los retrievers disponibles, opcionalmente filtrados por fuente.",
205
  api_name="list_retrievers"
206
  )
207
+ ],
208
+ gr.Interface(
209
+ fn=search_tavily,
210
+ inputs=[
211
+ gr.Textbox(label="Consulta de búsqueda", placeholder="Ej: últimas noticias sobre IA"),
212
+ gr.Slider(1, 30, value=7, step=1, label="Últimos N días (0 para sin límite)"),
213
+ gr.Slider(1, 10, value=1, step=1, label="Máximo de resultados"),
214
+ gr.Checkbox(label="Incluir respuesta directa", value=False)
215
+ ],
216
+ outputs=gr.JSON(label="Resultados de Tavily"),
217
+ title="Búsqueda Web (Tavily)",
218
+ description="Realiza búsquedas en web usando la API de Tavily.",
219
+ api_name="search_tavily"
220
+ )
221
  ],
222
  [
223
  "Búsqueda ArXiv",
224
+ "Lista de Retrievers",
225
+ "Búsqueda Web (Tavily)"
226
  ]
227
  )
228