jerpint commited on
Commit
463196d
·
1 Parent(s): 6aad21a

remove response from fn; was unused

Browse files
Files changed (1) hide show
  1. buster/busterbot.py +3 -3
buster/busterbot.py CHANGED
@@ -148,10 +148,8 @@ class Buster:
148
 
149
  def add_sources(
150
  self,
151
- response,
152
  matched_documents: pd.DataFrame,
153
  ):
154
- logger.info(f"GPT Response:\n{response.text}")
155
  sources = (
156
  Source(
157
  source=dct["source"], title=dct["title"], url=dct["url"], question_similarity=dct["similarity"] * 100
@@ -208,7 +206,9 @@ class Buster:
208
  # generate a completion
209
  documents: str = self.prepare_documents(matched_documents, max_words=self.cfg.max_words)
210
  response: Response = self.completer.generate_response(user_input, documents)
211
- sources = self.add_sources(response, matched_documents)
 
 
212
 
213
  # check for relevance
214
  relevant = self.check_response_relevance(
 
148
 
149
  def add_sources(
150
  self,
 
151
  matched_documents: pd.DataFrame,
152
  ):
 
153
  sources = (
154
  Source(
155
  source=dct["source"], title=dct["title"], url=dct["url"], question_similarity=dct["similarity"] * 100
 
206
  # generate a completion
207
  documents: str = self.prepare_documents(matched_documents, max_words=self.cfg.max_words)
208
  response: Response = self.completer.generate_response(user_input, documents)
209
+ logger.info(f"GPT Response:\n{response.text}")
210
+
211
+ sources = self.add_sources(matched_documents)
212
 
213
  # check for relevance
214
  relevant = self.check_response_relevance(