Skip to content

Smithy BiDi streaming support + Bedrock Migration #3675

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
sbaluja wants to merge 4 commits into
base: main
Choose a base branch
from
+2,669 −3,497
Open

Smithy BiDi streaming support + Bedrock Migration #3675

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
049bea7
Add support for bidirectional streaming in smithy clients
sbaluja Dec 30, 2025
17af789
Migrate bedrock clients to smithy
sbaluja Dec 30, 2025
268bb54
Use full namepsace for Event::Message
sbaluja Dec 30, 2025
9d67d27
Fix 'hides overloaded virtual function' error
sbaluja Dec 30, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 12 additions & 9 deletions ...p-sdk-bedrock-agent-runtime/include/aws/bedrock-agent-runtime/BedrockAgentRuntimeClient.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,26 +4,33 @@
*/

#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntimeErrorMarshaller.h>
#include <aws/bedrock-agent-runtime/BedrockAgentRuntimeServiceClientModel.h>
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/core/client/AWSClient.h>
#include <aws/core/client/AWSClientAsyncCRTP.h>
#include <aws/core/client/ClientConfiguration.h>
#include <aws/core/utils/json/JsonSerializer.h>
#include <smithy/client/AwsSmithyClient.h>
#include <smithy/client/serializer/JsonOutcomeSerializer.h>
#include <smithy/identity/auth/built-in/GenericAuthSchemeResolver.h>
#include <smithy/identity/auth/built-in/SigV4AuthScheme.h>

namespace Aws {
namespace BedrockAgentRuntime {
AWS_BEDROCKAGENTRUNTIME_API extern const char SERVICE_NAME[];
/**
* <p>Contains APIs related to model invocation and querying of knowledge
* bases.</p>
*/
class AWS_BEDROCKAGENTRUNTIME_API BedrockAgentRuntimeClient
: public Aws::Client::AWSJsonClient,
public Aws::Client::ClientWithAsyncTemplateMethods<BedrockAgentRuntimeClient> {
: Aws::Client::ClientWithAsyncTemplateMethods<BedrockAgentRuntimeClient>,
public smithy::client::AwsSmithyClientT<
Aws::BedrockAgentRuntime::SERVICE_NAME, Aws::BedrockAgentRuntime::BedrockAgentRuntimeClientConfiguration,
smithy::AuthSchemeResolverBase<>, Aws::Crt::Variant<smithy::SigV4AuthScheme>, BedrockAgentRuntimeEndpointProviderBase,
smithy::client::JsonOutcomeSerializer, smithy::client::JsonOutcome, Aws::Client::BedrockAgentRuntimeErrorMarshaller> {
public:
typedef Aws::Client::AWSJsonClient BASECLASS;
static const char* GetServiceName();
static const char* GetAllocationTag();
inline const char* GetServiceClientName() const override { return "Bedrock Agent Runtime"; }

typedef BedrockAgentRuntimeClientConfiguration ClientConfigurationType;
typedef BedrockAgentRuntimeEndpointProvider EndpointProviderType;
Expand Down Expand Up @@ -1058,10 +1065,6 @@ class AWS_BEDROCKAGENTRUNTIME_API BedrockAgentRuntimeClient

private:
friend class Aws::Client::ClientWithAsyncTemplateMethods<BedrockAgentRuntimeClient>;
void init(const BedrockAgentRuntimeClientConfiguration& clientConfiguration);

BedrockAgentRuntimeClientConfiguration m_clientConfiguration;
std::shared_ptr<BedrockAgentRuntimeEndpointProviderBase> m_endpointProvider;
};

} // namespace BedrockAgentRuntime
Expand Down
892 changes: 351 additions & 541 deletions generated/src/aws-cpp-sdk-bedrock-agent-runtime/source/BedrockAgentRuntimeClient.cpp
View file
Open in desktop

Large diffs are not rendered by default.

22 changes: 13 additions & 9 deletions generated/src/aws-cpp-sdk-bedrock-agent/include/aws/bedrock-agent/BedrockAgentClient.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,25 +4,33 @@
*/

#pragma once
#include <aws/bedrock-agent/BedrockAgentErrorMarshaller.h>
#include <aws/bedrock-agent/BedrockAgentServiceClientModel.h>
#include <aws/bedrock-agent/BedrockAgent_EXPORTS.h>
#include <aws/core/client/AWSClient.h>
#include <aws/core/client/AWSClientAsyncCRTP.h>
#include <aws/core/client/ClientConfiguration.h>
#include <aws/core/utils/json/JsonSerializer.h>
#include <smithy/client/AwsSmithyClient.h>
#include <smithy/client/serializer/JsonOutcomeSerializer.h>
#include <smithy/identity/auth/built-in/GenericAuthSchemeResolver.h>
#include <smithy/identity/auth/built-in/SigV4AuthScheme.h>

namespace Aws {
namespace BedrockAgent {
AWS_BEDROCKAGENT_API extern const char SERVICE_NAME[];
/**
* <p>Describes the API operations for creating and managing Amazon Bedrock
* agents.</p>
*/
class AWS_BEDROCKAGENT_API BedrockAgentClient : public Aws::Client::AWSJsonClient,
public Aws::Client::ClientWithAsyncTemplateMethods<BedrockAgentClient> {
class AWS_BEDROCKAGENT_API BedrockAgentClient
: Aws::Client::ClientWithAsyncTemplateMethods<BedrockAgentClient>,
public smithy::client::AwsSmithyClientT<Aws::BedrockAgent::SERVICE_NAME, Aws::BedrockAgent::BedrockAgentClientConfiguration,
smithy::AuthSchemeResolverBase<>, Aws::Crt::Variant<smithy::SigV4AuthScheme>,
BedrockAgentEndpointProviderBase, smithy::client::JsonOutcomeSerializer,
smithy::client::JsonOutcome, Aws::Client::BedrockAgentErrorMarshaller> {
public:
typedef Aws::Client::AWSJsonClient BASECLASS;
static const char* GetServiceName();
static const char* GetAllocationTag();
inline const char* GetServiceClientName() const override { return "Bedrock Agent"; }

typedef BedrockAgentClientConfiguration ClientConfigurationType;
typedef BedrockAgentEndpointProvider EndpointProviderType;
Expand Down Expand Up @@ -2175,10 +2183,6 @@ class AWS_BEDROCKAGENT_API BedrockAgentClient : public Aws::Client::AWSJsonClien

private:
friend class Aws::Client::ClientWithAsyncTemplateMethods<BedrockAgentClient>;
void init(const BedrockAgentClientConfiguration& clientConfiguration);

BedrockAgentClientConfiguration m_clientConfiguration;
std::shared_ptr<BedrockAgentEndpointProviderBase> m_endpointProvider;
};

} // namespace BedrockAgent
Expand Down
Loading
Loading