zypchn commited on
Commit
d94aa26
·
verified ·
1 Parent(s): d4958cf

add debug tool

Browse files
Files changed (1) hide show
  1. src/tools.py +9 -0
src/tools.py CHANGED
@@ -35,6 +35,15 @@ class PokemonAdvisorTools():
35
  print(len(knowledge_base))
36
  print(len(knowledge_base_latest))
37
 
 
 
 
 
 
 
 
 
 
38
  def list_card_names(self, name_query: str) -> list:
39
  """
40
  Retrieves a list of card names from the database.
 
35
  print(len(knowledge_base))
36
  print(len(knowledge_base_latest))
37
 
38
+ def get_data_shape(self) -> int:
39
+ """
40
+ Retrieves the number of data points.
41
+ Use this when user asks about the number of data points.
42
+ Returns:
43
+ int: lenght of knowledge base
44
+ """
45
+ return len(self.knowledge_base)
46
+
47
  def list_card_names(self, name_query: str) -> list:
48
  """
49
  Retrieves a list of card names from the database.