Mjlehtim commited on
Commit
5ad1d71
·
verified ·
1 Parent(s): 66a255f

as_retriever() added to deal with lacking 'collections' issue

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1061,11 +1061,11 @@ with strategies_container:
1061
  }
1062
 
1063
  if company_document:
1064
- store = create_vector_database_company()
1065
  if financial_document:
1066
- store = create_vector_database_financial()
1067
  if intercreditor_document:
1068
- store = create_vector_database_intercreditor()
1069
 
1070
  agent_settings = {
1071
  "max_iterations": 3,
 
1061
  }
1062
 
1063
  if company_document:
1064
+ store = create_vector_database_company().as_retriever()
1065
  if financial_document:
1066
+ store = create_vector_database_financial().as_retriever()
1067
  if intercreditor_document:
1068
+ store = create_vector_database_intercreditor().as_retriever()
1069
 
1070
  agent_settings = {
1071
  "max_iterations": 3,