Update asyncio.gather to return exceptions.

eli64s · eli64s · commit 3beb6da3fb6e · 2024-01-23T08:57:24.000-06:00
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,
                     },
                 ),
                 (

Original file line number	Diff line number	Diff line change
`@@ -95,8 +95,8 @@ async def _batch_prompts(`
`95`	`95`
`96`	`96`	`for batch in self._generate_batches(prompts, batch_size):`
`97`	`97`	`batch_responses = await asyncio.gather(`
`98`		`- *[self._process_batch(prompt) for prompt in batch]`
`99`		`- # , return_exceptions=True`
	`98`	`+ *[self._process_batch(prompt) for prompt in batch],`
	`99`	`+ return_exceptions=True,`
`100`	`100`	`)`
`101`	`101`	`responses.extend(batch_responses)`
`102`	`102`
`@@ -145,7 +145,7 @@ async def _set_prompt_context(`
`145`	`145`	`{`
`146`	`146`	`"repo": self.config.git.repository,`
`147`	`147`	`"dependencies": dependencies,`
`148`		`- "files": summaries,`
	`148`	`+ "summaries": summaries,`
`149`	`149`	`},`
`150`	`150`	`),`
`151`	`151`	`(`