AlexxRubio commited on
Commit
e6094a7
Β·
verified Β·
1 Parent(s): 2f79d72

Update retriever.py

Browse files
Files changed (1) hide show
  1. retriever.py +3 -2
retriever.py CHANGED
@@ -14,12 +14,13 @@ class GuestInfoRetrieverTool(Tool):
14
  }
15
  }
16
  output_type = "string"
17
-
 
18
  def __init__(self, docs):
19
  super().__init__()
20
  self.retriever = BM25Retriever.from_documents(docs)
21
 
22
-
23
  def forward(self, query: str):
24
  results = self.retriever.invoke(query)
25
  if results:
 
14
  }
15
  }
16
  output_type = "string"
17
+ # I tried doing super() init instead of self.retriever = false
18
+ # and it runs but it doesnt let me query still idk
19
  def __init__(self, docs):
20
  super().__init__()
21
  self.retriever = BM25Retriever.from_documents(docs)
22
 
23
+ # swtiched to invoke because the omne before just killed everything and wouldn't work
24
  def forward(self, query: str):
25
  results = self.retriever.invoke(query)
26
  if results: