Update Sources/BedrockService/BedrockService+AgentRuntime.swift

sebsto · Copilot · web-flow · commit 7061b0046bba · 2025-10-23T20:52:26.000+02:00
Co-authored-by: Copilot &lt;175728472+Copilot@users.noreply.github.com&gt;
diff --git a/Sources/BedrockService/BedrockService+AgentRuntime.swift b/Sources/BedrockService/BedrockService+AgentRuntime.swift
@@ -58,7 +58,8 @@ extension BedrockService {
 
         let input = RetrieveInput(
             knowledgeBaseId: knowledgeBaseId,
-            retrievalQuery: BedrockAgentRuntimeClientTypes.KnowledgeBaseQuery(text: retrievalQuery)
+            retrievalQuery: BedrockAgentRuntimeClientTypes.KnowledgeBaseQuery(text: retrievalQuery),
+            numberOfResults: numberOfResults
         )
 
         do {

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,8 @@ extension BedrockService {`
`58`	`58`
`59`	`59`	`let input = RetrieveInput(`
`60`	`60`	`knowledgeBaseId: knowledgeBaseId,`
`61`		`- retrievalQuery: BedrockAgentRuntimeClientTypes.KnowledgeBaseQuery(text: retrievalQuery)`
	`61`	`+ retrievalQuery: BedrockAgentRuntimeClientTypes.KnowledgeBaseQuery(text: retrievalQuery),`
	`62`	`+ numberOfResults: numberOfResults`
`62`	`63`	`)`
`63`	`64`
`64`	`65`	`do {`