@@ -332,7 +332,6 @@ async def process_request(
332332 api_key : Optional [str ] = None ,
333333 api_base : Optional [str ] = None ,
334334 extra_headers : Optional [Dict [str , str ]] = None ,
335- is_copilot : bool = False ,
336335 ) -> PipelineResult :
337336 """Process a request through all pipeline steps"""
338337 self .context .metadata ["extra_headers" ] = extra_headers
@@ -344,7 +343,9 @@ async def process_request(
344343 self .context .sensitive .api_base = api_base
345344
346345 # For Copilot provider=openai. Use a flag to not clash with other places that may use that.
347- provider_db = "copilot" if is_copilot else provider
346+ provider_db = provider
347+ if self .context .client == ClientType .COPILOT :
348+ provider_db = "copilot"
348349
349350 for step in self .pipeline_steps :
350351 result = await step .process (current_request , self .context )
@@ -401,9 +402,13 @@ async def process_request(
401402 api_key : Optional [str ] = None ,
402403 api_base : Optional [str ] = None ,
403404 extra_headers : Optional [Dict [str , str ]] = None ,
404- is_copilot : bool = False ,
405405 ) -> PipelineResult :
406406 """Create a new pipeline instance and process the request"""
407407 return await self .instance .process_request (
408- request , provider , model , api_key , api_base , extra_headers , is_copilot
408+ request ,
409+ provider ,
410+ model ,
411+ api_key ,
412+ api_base ,
413+ extra_headers ,
409414 )
0 commit comments