@@ -357,7 +357,6 @@ async def process_request(
357357 api_key : Optional [str ] = None ,
358358 api_base : Optional [str ] = None ,
359359 extra_headers : Optional [Dict [str , str ]] = None ,
360- is_copilot : bool = False ,
361360 ) -> PipelineResult :
362361 """Process a request through all pipeline steps"""
363362 self .context .metadata ["extra_headers" ] = extra_headers
@@ -369,7 +368,9 @@ async def process_request(
369368 self .context .sensitive .api_base = api_base
370369
371370 # For Copilot provider=openai. Use a flag to not clash with other places that may use that.
372- provider_db = "copilot" if is_copilot else provider
371+ provider_db = provider
372+ if self .context .client == ClientType .COPILOT :
373+ provider_db = "copilot"
373374
374375 for step in self .pipeline_steps :
375376 result = await step .process (current_request , self .context )
@@ -426,9 +427,13 @@ async def process_request(
426427 api_key : Optional [str ] = None ,
427428 api_base : Optional [str ] = None ,
428429 extra_headers : Optional [Dict [str , str ]] = None ,
429- is_copilot : bool = False ,
430430 ) -> PipelineResult :
431431 """Create a new pipeline instance and process the request"""
432432 return await self .instance .process_request (
433- request , provider , model , api_key , api_base , extra_headers , is_copilot
433+ request ,
434+ provider ,
435+ model ,
436+ api_key ,
437+ api_base ,
438+ extra_headers ,
434439 )
0 commit comments