add(test): context management, sub-agent, chat, and provider-specific smoke tests

Author: Tom-Ryder

Tests/AgentRunKitTests/Smoke/AnthropicSmokeTests.swift

@@ -6,6 +6,11 @@ private let apiKey = ProcessInfo.processInfo.environment["ANTHROPIC_API_KEY"] ??
 private let hasAPIKey = !apiKey.isEmpty
 private let model = ProcessInfo.processInfo.environment["SMOKE_ANTHROPIC_MODEL"] ?? "claude-sonnet-4-6"
 
+private let cachingSystemPrompt = String(
+    repeating: "You are a helpful Swift programming assistant. Follow best practices. ",
+    count: 150
+)
+
 @Suite(.enabled(if: hasAPIKey, "Requires ANTHROPIC_API_KEY environment variable"))
 struct AnthropicSmokeTests {
     let client = AnthropicClient(apiKey: apiKey, model: model, maxTokens: 1024)
@@ -46,6 +51,45 @@ struct AnthropicSmokeTests {
         try await assertSmokeStreamingTokenUsage(client: client)
     }
 
+    @Test func chatStreamWithTools() async throws {
+        try await assertSmokeChatStreamWithTools(client: client)
+    }
+
+    @Test func cachingEnabled() async throws {
+        let cachingClient = AnthropicClient(
+            apiKey: apiKey,
+            model: model,
+            maxTokens: 1024,
+            cachingEnabled: true
+        )
+
+        let messages: [ChatMessage] = [
+            .system(cachingSystemPrompt),
+            .user("What is Swift?"),
+        ]
+
+        let response1 = try await cachingClient.generate(messages: messages, tools: [smokeWeatherTool])
+        let usage1 = response1.tokenUsage
+        #expect(usage1 != nil)
+        #expect((usage1?.cacheWrite ?? 0) > 0)
+
+        let response2 = try await cachingClient.generate(messages: messages, tools: [smokeWeatherTool])
+        let usage2 = response2.tokenUsage
+        #expect(usage2 != nil)
+        #expect((usage2?.cacheRead ?? 0) > 0)
+    }
+
+    @Test func nonInterleavedReasoning() async throws {
+        let nonInterleavedClient = AnthropicClient(
+            apiKey: apiKey,
+            model: model,
+            maxTokens: 16384,
+            reasoningConfig: .budget(4096),
+            interleavedThinking: false
+        )
+        try await assertSmokeReasoningGenerate(client: nonInterleavedClient)
+    }
+
     @Test func reasoningStream() async throws {
         let thinkingClient = AnthropicClient(
             apiKey: apiKey,

Tests/AgentRunKitTests/Smoke/ContextManagementSmokeTests.swift

@@ -0,0 +1,49 @@
+@testable import AgentRunKit
+import Foundation
+import Testing
+
+private let apiKey = ProcessInfo.processInfo.environment["OPENROUTER_API_KEY"] ?? ""
+private let hasAPIKey = !apiKey.isEmpty
+private let model = ProcessInfo.processInfo.environment["SMOKE_OPENROUTER_MODEL"] ?? "google/gemini-3-flash-preview"
+
+@Suite(.enabled(if: hasAPIKey, "Requires OPENROUTER_API_KEY environment variable"))
+struct ContextManagementSmokeTests {
+    let client = OpenAIClient(
+        apiKey: apiKey,
+        model: model,
+        maxTokens: 1024,
+        baseURL: OpenAIClient.openRouterBaseURL
+    )
+
+    let budgetClient = OpenAIClient(
+        apiKey: apiKey,
+        model: model,
+        maxTokens: 1024,
+        contextWindowSize: 500,
+        baseURL: OpenAIClient.openRouterBaseURL
+    )
+
+    @Test func observationPruning() async throws {
+        try await assertSmokeObservationPruning(client: budgetClient)
+    }
+
+    @Test func llmSummarization() async throws {
+        try await assertSmokeLLMSummarization(client: budgetClient)
+    }
+
+    @Test func toolResultTruncation() async throws {
+        try await assertSmokeToolResultTruncation(client: client)
+    }
+
+    @Test func maxMessages() async throws {
+        try await assertSmokeMaxMessages(client: client)
+    }
+
+    @Test func budgetEvents() async throws {
+        try await assertSmokeBudgetEvents(client: budgetClient)
+    }
+
+    @Test func iterationCompleted() async throws {
+        try await assertSmokeIterationCompleted(client: client)
+    }
+}

Tests/AgentRunKitTests/Smoke/GeminiSmokeTests.swift

@@ -50,6 +50,14 @@ struct GeminiSmokeTests {
         try await assertSmokeStreamingTokenUsage(client: client)
     }
 
+    @Test func chatStreamWithTools() async throws {
+        try await assertSmokeChatStreamWithTools(client: client)
+    }
+
+    @Test func nestedStructuredOutput() async throws {
+        try await assertSmokeNestedStructuredOutput(client: client)
+    }
+
     @Test func reasoningStream() async throws {
         let thinkingClient = GeminiClient(
             apiKey: apiKey,

Tests/AgentRunKitTests/Smoke/OpenAISmokeTests.swift

@@ -54,4 +54,12 @@ struct OpenAISmokeTests {
     @Test func streamingTokenUsage() async throws {
         try await assertSmokeStreamingTokenUsage(client: client)
     }
+
+    @Test func chatStreamWithTools() async throws {
+        try await assertSmokeChatStreamWithTools(client: client)
+    }
+
+    @Test func nestedStructuredOutput() async throws {
+        try await assertSmokeNestedStructuredOutput(client: client)
+    }
 }

Tests/AgentRunKitTests/Smoke/OpenRouterSmokeTests.swift

@@ -54,4 +54,12 @@ struct OpenRouterSmokeTests {
     @Test func streamingTokenUsage() async throws {
         try await assertSmokeStreamingTokenUsage(client: client)
     }
+
+    @Test func chatStreamWithTools() async throws {
+        try await assertSmokeChatStreamWithTools(client: client)
+    }
+
+    @Test func nestedStructuredOutput() async throws {
+        try await assertSmokeNestedStructuredOutput(client: client)
+    }
 }

Tests/AgentRunKitTests/Smoke/ResponsesAPISmokeTests.swift

@@ -58,6 +58,14 @@ struct ResponsesAPISmokeTests {
         try await assertSmokeStreamingTokenUsage(client: makeClient())
     }
 
+    @Test func chatStreamWithTools() async throws {
+        try await assertSmokeChatStreamWithTools(client: makeClient())
+    }
+
+    @Test func nestedStructuredOutput() async throws {
+        try await assertSmokeNestedStructuredOutput(client: makeClient())
+    }
+
     @Test func conversationPersistence() async throws {
         let persistentClient = ResponsesAPIClient(
             apiKey: apiKey,