Pass in copilot extra headers when analyzing output packages

When querying Copilot, the LLM requires certain headers to be explicitly
set to accept a request. We were already parsing them and storing in
context and even using in the input pipeline, but forgot to use them in
the output pipeline to extract packages from the inbound (LLM-to-client)
snippets.

Fixes: #335

Author: jhrozek

src/codegate/pipeline/extract_snippets/output.py

@@ -5,7 +5,7 @@
 from litellm.types.utils import Delta, StreamingChoices
 
 from codegate.llm_utils.extractor import PackageExtractor
-from codegate.pipeline.base import CodeSnippet, PipelineContext, PipelineSensitiveData
+from codegate.pipeline.base import CodeSnippet, PipelineContext
 from codegate.pipeline.extract_snippets.extract_snippets import extract_snippets
 from codegate.pipeline.output import OutputPipelineContext, OutputPipelineStep
 from codegate.storage import StorageEngine
@@ -39,14 +39,15 @@ def _create_chunk(self, original_chunk: ModelResponse, content: str) -> ModelRes
             object="chat.completion.chunk",
         )
 
-    async def _snippet_comment(self, snippet: CodeSnippet, secrets: PipelineSensitiveData) -> str:
+    async def _snippet_comment(self, snippet: CodeSnippet, context: PipelineContext) -> str:
         """Create a comment for a snippet"""
         snippet.libraries = await PackageExtractor.extract_packages(
             content=snippet.code,
-            provider=secrets.provider,
-            model=secrets.model,
-            api_key=secrets.api_key,
-            base_url=secrets.api_base,
+            provider=context.sensitive.provider if context.sensitive else None,
+            model=context.sensitive.model if context.sensitive else None,
+            api_key=context.sensitive.api_key if context.sensitive else None,
+            base_url=context.sensitive.api_base if context.sensitive else None,
+            extra_headers=context.metadata.get("extra_headers", None),
         )
         # Check if any of the snippet libraries is a bad package
         storage_engine = StorageEngine()
@@ -126,7 +127,7 @@ async def process_chunk(
                 chunks.append(self._create_chunk(chunk, before))
                 complete_comment += before
 
-            comment = await self._snippet_comment(last_snippet, input_context.sensitive)
+            comment = await self._snippet_comment(last_snippet, input_context)
             complete_comment += comment
             chunks.append(
                 self._create_chunk(