Add provider configuration support to create session (#25)

* Add provider configuration support to create session

* Add tests similar to existing ones

* Factor out common logic

* Reformat

---------

Co-authored-by: Steve Sanderson <SteveSandersonMS@users.noreply.github.com>

Author: patniko

python/copilot/client.py

@@ -242,6 +242,11 @@ async def create_session(self, config: Optional[SessionConfig] = None) -> Copilo
         if streaming is not None:
             payload["streaming"] = streaming
 
+        # Add provider configuration if provided
+        provider = cfg.get("provider")
+        if provider:
+            payload["provider"] = self._convert_provider_to_wire_format(provider)
+
         if not self._client:
             raise RuntimeError("Client not connected")
         response = await self._client.request("session.create", payload)
@@ -293,15 +298,7 @@ async def resume_session(
 
         provider = cfg.get("provider")
         if provider:
-            # Convert snake_case to camelCase for the wire format
-            wire_provider = {"type": provider.get("type")}
-            if "base_url" in provider:
-                wire_provider["baseUrl"] = provider["base_url"]
-            if "api_key" in provider:
-                wire_provider["apiKey"] = provider["api_key"]
-            if "wire_api" in provider:
-                wire_provider["wireApi"] = provider["wire_api"]
-            payload["provider"] = wire_provider
+            payload["provider"] = self._convert_provider_to_wire_format(provider)
 
         # Add streaming option if provided
         streaming = cfg.get("streaming")
@@ -339,6 +336,26 @@ async def ping(self, message: Optional[str] = None) -> dict:
 
         return await self._client.request("ping", {"message": message})
 
+    def _convert_provider_to_wire_format(self, provider: Dict[str, Any]) -> Dict[str, Any]:
+        """Convert provider config from snake_case to camelCase wire format."""
+        wire_provider: Dict[str, Any] = {"type": provider.get("type")}
+        if "base_url" in provider:
+            wire_provider["baseUrl"] = provider["base_url"]
+        if "api_key" in provider:
+            wire_provider["apiKey"] = provider["api_key"]
+        if "wire_api" in provider:
+            wire_provider["wireApi"] = provider["wire_api"]
+        if "bearer_token" in provider:
+            wire_provider["bearerToken"] = provider["bearer_token"]
+        if "azure" in provider:
+            azure = provider["azure"]
+            wire_azure: Dict[str, Any] = {}
+            if "api_version" in azure:
+                wire_azure["apiVersion"] = azure["api_version"]
+            if wire_azure:
+                wire_provider["azure"] = wire_azure
+        return wire_provider
+
     async def _start_cli_server(self) -> None:
         """Start the CLI server process"""
         cli_path = self.options["cli_path"]

python/copilot/types.py

@@ -128,6 +128,8 @@ class SessionConfig(TypedDict, total=False):
     available_tools: list[str]
     # List of tool names to disable (ignored if available_tools is set)
     excluded_tools: list[str]
+    # Custom provider configuration (BYOK - Bring Your Own Key)
+    provider: ProviderConfig
     # Enable streaming of assistant message and reasoning chunks
     # When True, assistant.message_delta and assistant.reasoning_delta events
     # with delta_content are sent as the response is generated

python/e2e/test_session.py

@@ -208,6 +208,33 @@ def get_secret_number_handler(invocation):
         answer = await get_final_assistant_message(session)
         assert "54321" in answer.data.content
 
+    async def test_should_create_session_with_custom_provider(self, ctx: E2ETestContext):
+        session = await ctx.client.create_session(
+            {
+                "provider": {
+                    "type": "openai",
+                    "base_url": "https://api.openai.com/v1",
+                    "api_key": "fake-key",
+                }
+            }
+        )
+        assert session.session_id
+
+    async def test_should_create_session_with_azure_provider(self, ctx: E2ETestContext):
+        session = await ctx.client.create_session(
+            {
+                "provider": {
+                    "type": "azure",
+                    "base_url": "https://my-resource.openai.azure.com",
+                    "api_key": "fake-key",
+                    "azure": {
+                        "api_version": "2024-02-15-preview",
+                    },
+                }
+            }
+        )
+        assert session.session_id
+
     async def test_should_resume_session_with_custom_provider(self, ctx: E2ETestContext):
         session = await ctx.client.create_session()
         session_id = session.session_id