Update main.py

main.py

@@ -140,7 +140,7 @@ prompt = ChatPromptTemplate.from_messages([
     ("placeholder", "{messages}"),
 ])
 
-async def get_collection_files(collection_id: str, user_id: str) -> str:
+def get_collection_files(collection_id: str, user_id: str) -> str:
     """Get list of files in the specified collection"""
     try:
         # Get the full collection name
@@ -159,7 +159,7 @@ async def get_collection_files(collection_id: str, user_id: str) -> str:
         logging.error(f"Error getting collection files: {str(e)}")
         return f"Error getting files: {str(e)}"
 
-async def format_for_model(state):
+def format_for_model(state):
     # Get collection_id and user_id from the state's configurable
     config = state.get("configurable", {})
     collection_id = config.get("collection_id")