tomaslucas commited on
Commit
e7d6b68
·
verified ·
1 Parent(s): 64da845

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -18
app.py CHANGED
@@ -43,23 +43,23 @@ def get_book() -> List[PotterBook]:
43
  libros = [PotterBook(**libro) for libro in datos]
44
  return libros
45
 
46
- @tool
47
- def filter_book(libro, campo, valor, operador='==') -> List:
48
- """
49
- Filtra libros según un campo, valor y operador especificados
50
- operadores permitidos: ==, >, <, >=, <=, contains
51
- Args:
52
- libro: name o text title of the book we are looking for
53
- campo: field of the book we are looking for
54
- valor: value to compair with the field
55
- operador: kind of compartive that we want to have over the field and de value
56
- """
57
- if operador == 'contains':
58
- return [libro for libro in libros if valor.lower() in str(getattr(libro, campo)).lower()]
59
- elif operador in ['==', '>', '<', '>=', '<=']:
60
- return [libro for libro in libros if eval(f"{getattr(libro, campo)} {operador} {valor}")]
61
- else:
62
- raise ValueError("Operador no válido")
63
 
64
 
65
  @tool
@@ -99,7 +99,7 @@ with open("prompts.yaml", 'r') as stream:
99
 
100
  agent = CodeAgent(
101
  model=model,
102
- tools=[final_answer, get_current_time_in_timezone, get_book, filter_book], ## add your tools here (don't remove final answer)
103
  max_steps=6,
104
  verbosity_level=1,
105
  grammar=None,
 
43
  libros = [PotterBook(**libro) for libro in datos]
44
  return libros
45
 
46
+ # @tool
47
+ # def filter_book(libro, campo, valor, operador='==') -> List:
48
+ # """
49
+ # Filtra libros según un campo, valor y operador especificados
50
+ # operadores permitidos: ==, >, <, >=, <=, contains
51
+ # Args:
52
+ # libro: name o text title of the book we are looking for
53
+ # campo: field of the book we are looking for
54
+ # valor: value to compair with the field
55
+ # operador: kind of compartive that we want to have over the field and de value
56
+ # """
57
+ # if operador == 'contains':
58
+ # return [libro for libro in libros if valor.lower() in str(getattr(libro, campo)).lower()]
59
+ # elif operador in ['==', '>', '<', '>=', '<=']:
60
+ # return [libro for libro in libros if eval(f"{getattr(libro, campo)} {operador} {valor}")]
61
+ # else:
62
+ # raise ValueError("Operador no válido")
63
 
64
 
65
  @tool
 
99
 
100
  agent = CodeAgent(
101
  model=model,
102
+ tools=[final_answer, get_current_time_in_timezone, get_book,], ## add your tools here (don't remove final answer)
103
  max_steps=6,
104
  verbosity_level=1,
105
  grammar=None,