This commit is contained in:
Hyeong Joon Suh 2024-09-09 14:47:32 -07:00 committed by danswer-trial
parent 536079f746
commit eda4108d20
2 changed files with 4 additions and 3 deletions

View File

@ -50,7 +50,8 @@ def oneoff_standard_answers(
)
server_standard_answers = [
StandardAnswer.from_model(db_answer) for db_answer in matching_standard_answers
StandardAnswer.from_model(db_answer[0])
for db_answer in matching_standard_answers
]
return server_standard_answers

View File

@ -182,7 +182,7 @@ def find_matching_standard_answers(
id_in: list[int],
query: str,
db_session: Session,
) -> list[(StandardAnswer, bool)]:
) -> list[tuple[StandardAnswer, bool]]:
"""
Returns a list of tuples, where each tuple contains a StandardAnswer and a boolean
indicating if the query is a regex match.
@ -194,7 +194,7 @@ def find_matching_standard_answers(
)
possible_standard_answers = db_session.scalars(stmt).all()
matching_standard_answers: list[StandardAnswer] = []
matching_standard_answers: list[tuple[StandardAnswer, bool]] = []
for standard_answer in possible_standard_answers:
# Remove punctuation and split the keyword into individual words
keyword_words = "".join(