Update asyncio.gather to return exceptions.

eli64s · Jan 23, 2024 · 3beb6da · 3beb6da
1 parent 6ae121b
commit 3beb6da
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/readmeai/core/model.py b/readmeai/core/model.py
@@ -95,8 +95,8 @@ async def _batch_prompts(
 
         for batch in self._generate_batches(prompts, batch_size):
             batch_responses = await asyncio.gather(
-                *[self._process_batch(prompt) for prompt in batch]
-                # , return_exceptions=True
+                *[self._process_batch(prompt) for prompt in batch],
+                return_exceptions=True,
             )
             responses.extend(batch_responses)
 
@@ -145,7 +145,7 @@ async def _set_prompt_context(
                     {
                         "repo": self.config.git.repository,
                         "dependencies": dependencies,
-                        "files": summaries,
+                        "summaries": summaries,
                     },
                 ),
                 (