File tree Expand file tree Collapse file tree 3 files changed +7
-7
lines changed Expand file tree Collapse file tree 3 files changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -37,10 +37,10 @@ extension BedrockService {
3737 return BedrockAgentRuntimeClient ( config: config)
3838 }
3939 /// Retrieves information from a knowledge base for RAG applications
40- ///
40+ ///
4141 /// This method queries an Amazon Bedrock knowledge base to retrieve relevant information
4242 /// that can be used for Retrieval-Augmented Generation (RAG) applications.
43- ///
43+ ///
4444 /// - Parameters:
4545 /// - knowledgeBaseId: The unique identifier of the knowledge base to query
4646 /// - retrievalQuery: The query to search for in the knowledge base
Original file line number Diff line number Diff line change @@ -23,7 +23,7 @@ import Foundation
2323#endif
2424
2525/// Protocol for Amazon Bedrock Agent Runtime operations
26- ///
26+ ///
2727/// This protocol allows writing mocks for unit tests and provides a clean interface
2828/// for knowledge base retrieval operations.
2929public protocol BedrockAgentRuntimeProtocol : Sendable {
Original file line number Diff line number Diff line change @@ -23,7 +23,7 @@ public struct RetrieveRequest: Sendable {
2323 public let retrievalQuery : String
2424 /// The number of results to return
2525 public let numberOfResults : Int
26-
26+
2727 /// Creates a new retrieve request
2828 /// - Parameters:
2929 /// - knowledgeBaseId: The unique identifier of the knowledge base to query
@@ -38,9 +38,9 @@ public struct RetrieveRequest: Sendable {
3838 self . retrievalQuery = retrievalQuery
3939 self . numberOfResults = numberOfResults
4040 }
41-
41+
4242 internal var input : RetrieveInput {
43- return RetrieveInput (
43+ RetrieveInput (
4444 knowledgeBaseId: knowledgeBaseId,
4545 retrievalConfiguration: BedrockAgentRuntimeClientTypes . KnowledgeBaseRetrievalConfiguration (
4646 vectorSearchConfiguration: BedrockAgentRuntimeClientTypes . KnowledgeBaseVectorSearchConfiguration (
@@ -50,4 +50,4 @@ public struct RetrieveRequest: Sendable {
5050 retrievalQuery: BedrockAgentRuntimeClientTypes . KnowledgeBaseQuery ( text: retrievalQuery)
5151 )
5252 }
53- }
53+ }
You can’t perform that action at this time.
0 commit comments