Add config dir to session options (#15)

* Add config dir to session options

* add tests

* Use join

Author: devm33

dotnet/src/Client.cs

@@ -343,7 +343,8 @@ public async Task<CopilotSession> CreateSessionAsync(SessionConfig? config = nul
             config?.OnPermissionRequest != null ? true : null,
             config?.Streaming == true ? true : null,
             config?.McpServers,
-            config?.CustomAgents);
+            config?.CustomAgents,
+            config?.ConfigDir);
 
         var response = await connection.Rpc.InvokeWithCancellationAsync<CreateSessionResponse>(
             "session.create", [request], cancellationToken);
@@ -925,7 +926,8 @@ private record CreateSessionRequest(
         bool? RequestPermission,
         bool? Streaming,
         Dictionary<string, object>? McpServers,
-        List<CustomAgentConfig>? CustomAgents);
+        List<CustomAgentConfig>? CustomAgents,
+        string? ConfigDir);
 
     private record ToolDefinition(
         string Name,

dotnet/src/Types.cs

@@ -293,6 +293,13 @@ public class SessionConfig
 {
     public string? SessionId { get; set; }
     public string? Model { get; set; }
+
+    /// <summary>
+    /// Override the default configuration directory location.
+    /// When specified, the session will use this directory for storing config and state.
+    /// </summary>
+    public string? ConfigDir { get; set; }
+
     public ICollection<AIFunction>? Tools { get; set; }
     public SystemMessageConfig? SystemMessage { get; set; }
     public List<string>? AvailableTools { get; set; }

dotnet/test/SessionTests.cs

@@ -380,4 +380,19 @@ public async Task SendAndWait_Throws_On_Timeout()
 
         Assert.Contains("timed out", ex.Message);
     }
+
+    [Fact]
+    public async Task Should_Create_Session_With_Custom_Config_Dir()
+    {
+        var customConfigDir = Path.Join(Ctx.HomeDir, "custom-config");
+        var session = await Client.CreateSessionAsync(new SessionConfig { ConfigDir = customConfigDir });
+
+        Assert.Matches(@"^[a-f0-9-]+$", session.SessionId);
+
+        // Session should work normally with custom config dir
+        await session.SendAsync(new MessageOptions { Prompt = "What is 1+1?" });
+        var assistantMessage = await TestHelper.GetFinalAssistantMessageAsync(session);
+        Assert.NotNull(assistantMessage);
+        Assert.Contains("2", assistantMessage!.Data.Content);
+    }
 }

go/client.go

@@ -532,6 +532,10 @@ func (c *Client) CreateSession(config *SessionConfig) (*Session, error) {
 			}
 			params["customAgents"] = customAgents
 		}
+		// Add config directory override
+		if config.ConfigDir != "" {
+			params["configDir"] = config.ConfigDir
+		}
 	}
 
 	result, err := c.client.Request("session.create", params)

go/e2e/session_test.go

@@ -718,6 +718,38 @@ func TestSession(t *testing.T) {
 			t.Errorf("Expected assistant message to contain '300', got %v", assistantMessage.Data.Content)
 		}
 	})
+
+	t.Run("should create session with custom config dir", func(t *testing.T) {
+		ctx.ConfigureForTest(t)
+
+		customConfigDir := ctx.HomeDir + "/custom-config"
+		session, err := client.CreateSession(&copilot.SessionConfig{
+			ConfigDir: customConfigDir,
+		})
+		if err != nil {
+			t.Fatalf("Failed to create session with custom config dir: %v", err)
+		}
+
+		matched, _ := regexp.MatchString(`^[a-f0-9-]+$`, session.SessionID)
+		if !matched {
+			t.Errorf("Expected session ID to match UUID pattern, got %q", session.SessionID)
+		}
+
+		// Session should work normally with custom config dir
+		_, err = session.Send(copilot.MessageOptions{Prompt: "What is 1+1?"})
+		if err != nil {
+			t.Fatalf("Failed to send message: %v", err)
+		}
+
+		assistantMessage, err := testharness.GetFinalAssistantMessage(session, 60*time.Second)
+		if err != nil {
+			t.Fatalf("Failed to get assistant message: %v", err)
+		}
+
+		if assistantMessage.Data.Content == nil || !strings.Contains(*assistantMessage.Data.Content, "2") {
+			t.Errorf("Expected assistant message to contain '2', got %v", assistantMessage.Data.Content)
+		}
+	})
 }
 
 func getSystemMessage(exchange testharness.ParsedHttpExchange) string {

go/types.go

@@ -138,6 +138,9 @@ type SessionConfig struct {
 	SessionID string
 	// Model to use for this session
 	Model string
+	// ConfigDir overrides the default configuration directory location.
+	// When specified, the session will use this directory for storing config and state.
+	ConfigDir string
 	// Tools exposes caller-implemented tools to the CLI
 	Tools []Tool
 	// SystemMessage configures system message customization

nodejs/src/client.ts

@@ -446,6 +446,7 @@ export class CopilotClient {
             streaming: config.streaming,
             mcpServers: config.mcpServers,
             customAgents: config.customAgents,
+            configDir: config.configDir,
         });
 
         const sessionId = (response as { sessionId: string }).sessionId;

nodejs/src/types.ts

@@ -324,6 +324,12 @@ export interface SessionConfig {
      */
     model?: string;
 
+    /**
+     * Override the default configuration directory location.
+     * When specified, the session will use this directory for storing config and state.
+     */
+    configDir?: string;
+
     /**
      * Tools exposed to the CLI server
      */

nodejs/test/e2e/session.test.ts

@@ -321,6 +321,20 @@ describe("Sessions", async () => {
         // Verify the assistant response contains the expected answer
         expect(assistantMessage?.data.content).toContain("300");
     });
+
+    it("should create session with custom config dir", async () => {
+        const customConfigDir = `${homeDir}/custom-config`;
+        const session = await client.createSession({
+            configDir: customConfigDir,
+        });
+
+        expect(session.sessionId).toMatch(/^[a-f0-9-]+$/);
+
+        // Session should work normally with custom config dir
+        await session.send({ prompt: "What is 1+1?" });
+        const assistantMessage = await getFinalAssistantMessage(session);
+        expect(assistantMessage.data.content).toContain("2");
+    });
 });
 
 function getSystemMessage(exchange: ParsedHttpExchange): string | undefined {

python/copilot/client.py

@@ -400,6 +400,11 @@ async def create_session(self, config: Optional[SessionConfig] = None) -> Copilo
                 self._convert_custom_agent_to_wire_format(agent) for agent in custom_agents
             ]
 
+        # Add config directory override if provided
+        config_dir = cfg.get("config_dir")
+        if config_dir:
+            payload["configDir"] = config_dir
+
         if not self._client:
             raise RuntimeError("Client not connected")
         response = await self._client.request("session.create", payload)