pvanand commited on
Commit
00d6bbe
·
verified ·
1 Parent(s): fe8c198

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -171,7 +171,7 @@ def format_for_model(state):
171
 
172
  # Get files in the collection
173
 
174
- collection_files = get_collection_files(collection_id, user_id) if collection_id and user_id else "No files available"
175
  logging.info(f"fetching collection for userid{user_id} and collection_id {collection_id}|| Results {collection_files}")
176
  return prompt.invoke({
177
  "collection_files": collection_files,
 
171
 
172
  # Get files in the collection
173
 
174
+ collection_files = get_current_files()#get_collection_files(collection_id, user_id) if collection_id and user_id else "No files available"
175
  logging.info(f"fetching collection for userid{user_id} and collection_id {collection_id}|| Results {collection_files}")
176
  return prompt.invoke({
177
  "collection_files": collection_files,