diff --git a/api/service/generation_service.py b/api/service/generation_service.py index 3133027..9b67049 100644 --- a/api/service/generation_service.py +++ b/api/service/generation_service.py @@ -64,7 +64,8 @@ class GenerationService: future_prompt += prompt assets_data = [] if assets is not None: - assets_data = await self.dao.assets.get_assets_by_ids(assets) + assets_db = await self.dao.assets.get_assets_by_ids(assets) + assets_data.extend(asset.data for asset in assets_db) generated_prompt = self.gemini.generate_text(future_prompt, assets_data) logger.info(future_prompt) logger.info(generated_prompt)