diff --git a/backend/main.py b/backend/main.py index 628bdc9..06ef15a 100644 --- a/backend/main.py +++ b/backend/main.py @@ -150,7 +150,7 @@ def test(document: Document): print("found summary in db") return {"summary": indexed_summaries[2], "summary_list": indexed_summaries[3]} # get model - llm_model = db.execute(text("SELECT * FROM config ORDER BY id DESC LIMIT 1")).first()[2] + llm_model = db.execute(text("SELECT * FROM config ORDER BY id DESC LIMIT 1")).first()[3] s = Summarize(document=document, model=llm_model) summary = s.get_summary() entry = Summary(file_name=document.document, summary=summary.get('summary'),