@@ -350,7 +350,6 @@ async def process_request(
350350 api_key : Optional [str ] = None ,
351351 api_base : Optional [str ] = None ,
352352 extra_headers : Optional [Dict [str , str ]] = None ,
353- is_copilot : bool = False ,
354353 ) -> PipelineResult :
355354 """Process a request through all pipeline steps"""
356355 self .context .metadata ["extra_headers" ] = extra_headers
@@ -362,7 +361,9 @@ async def process_request(
362361 self .context .sensitive .api_base = api_base
363362
364363 # For Copilot provider=openai. Use a flag to not clash with other places that may use that.
365- provider_db = "copilot" if is_copilot else provider
364+ provider_db = provider
365+ if self .context .client == ClientType .COPILOT :
366+ provider_db = "copilot"
366367
367368 for step in self .pipeline_steps :
368369 result = await step .process (current_request , self .context )
@@ -419,9 +420,13 @@ async def process_request(
419420 api_key : Optional [str ] = None ,
420421 api_base : Optional [str ] = None ,
421422 extra_headers : Optional [Dict [str , str ]] = None ,
422- is_copilot : bool = False ,
423423 ) -> PipelineResult :
424424 """Create a new pipeline instance and process the request"""
425425 return await self .instance .process_request (
426- request , provider , model , api_key , api_base , extra_headers , is_copilot
426+ request ,
427+ provider ,
428+ model ,
429+ api_key ,
430+ api_base ,
431+ extra_headers ,
427432 )
0 commit comments