fix: tidy up

Author: iziang

aperag/service/agent_chat_service.py

@@ -174,25 +174,18 @@ async def handle_websocket_agent_chat(self, websocket: WebSocket, user: str, bot
 
             # Get default collections once for performance
             if bot_config.agent.collections:
-                user_collections = await self.db_ops.list_all_accessible_collections_for_user(user)
-                user_collections_ids = [collection['id'] for collection in user_collections]
-                # Create a dictionary for easy lookup
-                user_collections_dict = {collection['id']: collection for collection in user_collections}
-                
                 agent_collection_ids = [collection.id for collection in bot_config.agent.collections]
-                for collection_id in agent_collection_ids:
-                    if collection_id not in user_collections_ids:
-                        raise ResourceNotFoundException("Collection", collection_id)
-                    agent_collection = user_collections_dict[collection_id]
+                agent_collections = await self.db_ops.query_collections_by_ids(user, agent_collection_ids)
+                for agent_collection in agent_collections:
                     default_collections.append(view_models.Collection(
-                        id=agent_collection['id'],
-                        title=agent_collection['title'],
-                        description=agent_collection['description'],
-                        type=agent_collection['type'],
-                        status=agent_collection['status'],
-                        config=parseCollectionConfig(agent_collection['config']),
-                        created=agent_collection['gmt_created'].isoformat(),
-                        updated=agent_collection['gmt_updated'].isoformat(),
+                        id=agent_collection.id,
+                        title=agent_collection.title,
+                        description=agent_collection.description,
+                        type=agent_collection.type,
+                        status=agent_collection.status,
+                        config=parseCollectionConfig(agent_collection.config),
+                        created=agent_collection.gmt_created.isoformat(),
+                        updated=agent_collection.gmt_updated.isoformat(),
                     ))
 
         while True:

envs/env.template

@@ -7,11 +7,16 @@ POSTGRES_PASSWORD=postgres
 
 # Database Connection Pool Settings
 # Adjust these values based on your server resources and expected load
-DB_POOL_SIZE=20          # Base connection pool size
-DB_MAX_OVERFLOW=40       # Maximum overflow connections (total = pool_size + max_overflow)
-DB_POOL_TIMEOUT=60       # Connection timeout in seconds
-DB_POOL_RECYCLE=3600     # Recycle connections after 1 hour (in seconds)
-DB_POOL_PRE_PING=True    # Validate connections before use
+# Base connection pool size
+DB_POOL_SIZE=20
+# Maximum overflow connections (total = pool_size + max_overflow)
+DB_MAX_OVERFLOW=40
+# Connection timeout in seconds
+DB_POOL_TIMEOUT=60
+# Recycle connections after 1 hour (in seconds)
+DB_POOL_RECYCLE=3600
+# Validate connections before use
+DB_POOL_PRE_PING=True
 
 # Redis
 REDIS_HOST=127.0.0.1

web/src/app/workspace/collections/[collectionId]/documents/upload/document-upload.tsx

@@ -378,7 +378,7 @@ export const DocumentUpload = () => {
     <>
       <FileUpload
         maxFiles={1000}
-        maxSize={10 * 1024 * 1024}
+        maxSize={100 * 1024 * 1024}
         className="w-full gap-4"
         accept=".pdf,.doc,.docx,.txt,.md,.ppt,.pptx,.xls,.xlsx"
         value={documents.map((f) => f.file)}