@@ -167,7 +167,11 @@ def _build_contexts_with_sources(
167167 contexts_with_sources : List [ContextWithSources ] = []
168168 for idx , context in enumerate (contexts ):
169169 source_file = source_files [idx ] if idx < len (source_files ) else None
170- score = context_scores [idx ] if context_scores and idx < len (context_scores ) else None
170+ score = (
171+ context_scores [idx ]
172+ if context_scores and idx < len (context_scores )
173+ else None
174+ )
171175 contexts_with_sources .append (
172176 ContextWithSources (
173177 context = context ,
@@ -199,9 +203,7 @@ def _validate_merge_request(
199203 "`target_files_per_context` must be at least 2 when provided."
200204 )
201205 if max_files_per_context < 2 :
202- raise ValueError (
203- "`max_files_per_context` must be at least 2."
204- )
206+ raise ValueError ("`max_files_per_context` must be at least 2." )
205207 if len (contexts_with_sources ) < 2 :
206208 return False
207209 distinct_files = {
@@ -219,7 +221,9 @@ def _merge_cross_file_contexts(
219221 max_files_per_context : int = 3 ,
220222 ) -> List [ContextWithSources ]:
221223 if not self ._validate_merge_request (
222- contexts_with_sources , target_files_per_context , max_files_per_context
224+ contexts_with_sources ,
225+ target_files_per_context ,
226+ max_files_per_context ,
223227 ):
224228 return contexts_with_sources
225229 context_texts = [
@@ -241,7 +245,9 @@ async def _a_merge_cross_file_contexts(
241245 max_files_per_context : int = 3 ,
242246 ) -> List [ContextWithSources ]:
243247 if not self ._validate_merge_request (
244- contexts_with_sources , target_files_per_context , max_files_per_context
248+ contexts_with_sources ,
249+ target_files_per_context ,
250+ max_files_per_context ,
245251 ):
246252 return contexts_with_sources
247253 context_texts = [
@@ -793,9 +799,8 @@ def generate_goldens_from_contexts(
793799 qualified_synthetic_inputs
794800 ):
795801 context_source_files : List [str ] = []
796- if (
797- source_files is not None
798- and context_index < len (source_files )
802+ if source_files is not None and context_index < len (
803+ source_files
799804 ):
800805 context_source = source_files [context_index ]
801806 if isinstance (context_source , list ):
@@ -1109,9 +1114,7 @@ async def process_input(
11091114 context = context ,
11101115 expected_output = expected_output ,
11111116 source_file = (
1112- context_source_files [0 ]
1113- if context_source_files
1114- else None
1117+ context_source_files [0 ] if context_source_files else None
11151118 ),
11161119 additional_metadata = {
11171120 "evolutions" : evolutions_used ,
@@ -2257,27 +2260,25 @@ async def a_generate_conversational_goldens_from_docs(
22572260 )
22582261
22592262 # Generate conversational goldens from contexts
2260- goldens = (
2261- await self .a_generate_conversational_goldens_from_contexts (
2262- contexts = [item .context for item in contexts_with_sources ],
2263- include_expected_outcome = include_expected_outcome ,
2264- max_goldens_per_context = max_goldens_per_context ,
2265- source_files = [
2266- item .source_files for item in contexts_with_sources
2267- ],
2268- context_chunk_source_files = [
2269- item .chunk_source_files or []
2270- for item in contexts_with_sources
2271- ],
2272- target_files_per_context = context_construction_config .target_files_per_context ,
2273- _context_scores = [
2274- item .score if item .score is not None else 0.0
2275- for item in contexts_with_sources
2276- ],
2277- _progress = progress ,
2278- _pbar_id = pbar_id ,
2279- _reset_cost = False ,
2280- )
2263+ goldens = await self .a_generate_conversational_goldens_from_contexts (
2264+ contexts = [item .context for item in contexts_with_sources ],
2265+ include_expected_outcome = include_expected_outcome ,
2266+ max_goldens_per_context = max_goldens_per_context ,
2267+ source_files = [
2268+ item .source_files for item in contexts_with_sources
2269+ ],
2270+ context_chunk_source_files = [
2271+ item .chunk_source_files or []
2272+ for item in contexts_with_sources
2273+ ],
2274+ target_files_per_context = context_construction_config .target_files_per_context ,
2275+ _context_scores = [
2276+ item .score if item .score is not None else 0.0
2277+ for item in contexts_with_sources
2278+ ],
2279+ _progress = progress ,
2280+ _pbar_id = pbar_id ,
2281+ _reset_cost = False ,
22812282 )
22822283 if _reset_cost and self .cost_tracking and self .using_native_model :
22832284 print (f"💰 API cost: { self .synthesis_cost :.6f} " )
@@ -2434,9 +2435,8 @@ def generate_conversational_goldens_from_contexts(
24342435 qualified_synthetic_scenarios
24352436 ):
24362437 context_source_files : List [str ] = []
2437- if (
2438- source_files is not None
2439- and context_index < len (source_files )
2438+ if source_files is not None and context_index < len (
2439+ source_files
24402440 ):
24412441 context_source = source_files [context_index ]
24422442 if isinstance (context_source , list ):
0 commit comments