fix: summarize llm_model

ishaan-jaff · Jul 8, 2023 · 35d0690 · 35d0690
1 parent 852144d
commit 35d0690
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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'),