mocks

Documentation

Index

• type IClient
  • func NewIClient(t interface{ ... }) *IClient
  • func (_m *IClient) DeleteAgentMemory(ctx context.Context, params *bedrockagentruntime.DeleteAgentMemoryInput, ...) (*bedrockagentruntime.DeleteAgentMemoryOutput, error)
  • func (_m *IClient) GenerateQuery(ctx context.Context, params *bedrockagentruntime.GenerateQueryInput, ...) (*bedrockagentruntime.GenerateQueryOutput, error)
  • func (_m *IClient) GetAgentMemory(ctx context.Context, params *bedrockagentruntime.GetAgentMemoryInput, ...) (*bedrockagentruntime.GetAgentMemoryOutput, error)
  • func (_m *IClient) InvokeAgent(ctx context.Context, params *bedrockagentruntime.InvokeAgentInput, ...) (*bedrockagentruntime.InvokeAgentOutput, error)
  • func (_m *IClient) InvokeFlow(ctx context.Context, params *bedrockagentruntime.InvokeFlowInput, ...) (*bedrockagentruntime.InvokeFlowOutput, error)
  • func (_m *IClient) InvokeInlineAgent(ctx context.Context, params *bedrockagentruntime.InvokeInlineAgentInput, ...) (*bedrockagentruntime.InvokeInlineAgentOutput, error)
  • func (_m *IClient) OptimizePrompt(ctx context.Context, params *bedrockagentruntime.OptimizePromptInput, ...) (*bedrockagentruntime.OptimizePromptOutput, error)
  • func (_m *IClient) Options() bedrockagentruntime.Options
  • func (_m *IClient) Rerank(ctx context.Context, params *bedrockagentruntime.RerankInput, ...) (*bedrockagentruntime.RerankOutput, error)
  • func (_m *IClient) Retrieve(ctx context.Context, params *bedrockagentruntime.RetrieveInput, ...) (*bedrockagentruntime.RetrieveOutput, error)
  • func (_m *IClient) RetrieveAndGenerate(ctx context.Context, params *bedrockagentruntime.RetrieveAndGenerateInput, ...) (*bedrockagentruntime.RetrieveAndGenerateOutput, error)
  • func (_m *IClient) RetrieveAndGenerateStream(ctx context.Context, ...) (*bedrockagentruntime.RetrieveAndGenerateStreamOutput, error)

Types

type IClient

type IClient struct {
	mock.Mock
}

IClient is an autogenerated mock type for the IClient type

func NewIClient

func NewIClient(t interface {
	mock.TestingT
	Cleanup(func())
}) *IClient

NewIClient creates a new instance of IClient. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. The first argument is typically a *testing.T value.

func (*IClient) DeleteAgentMemory

func (_m *IClient) DeleteAgentMemory(ctx context.Context, params *bedrockagentruntime.DeleteAgentMemoryInput, optFns ...func(*bedrockagentruntime.Options)) (*bedrockagentruntime.DeleteAgentMemoryOutput, error)

DeleteAgentMemory provides a mock function with given fields: ctx, params, optFns

func (*IClient) GenerateQuery

func (_m *IClient) GenerateQuery(ctx context.Context, params *bedrockagentruntime.GenerateQueryInput, optFns ...func(*bedrockagentruntime.Options)) (*bedrockagentruntime.GenerateQueryOutput, error)

GenerateQuery provides a mock function with given fields: ctx, params, optFns

func (*IClient) GetAgentMemory

func (_m *IClient) GetAgentMemory(ctx context.Context, params *bedrockagentruntime.GetAgentMemoryInput, optFns ...func(*bedrockagentruntime.Options)) (*bedrockagentruntime.GetAgentMemoryOutput, error)

GetAgentMemory provides a mock function with given fields: ctx, params, optFns

func (*IClient) InvokeAgent

func (_m *IClient) InvokeAgent(ctx context.Context, params *bedrockagentruntime.InvokeAgentInput, optFns ...func(*bedrockagentruntime.Options)) (*bedrockagentruntime.InvokeAgentOutput, error)

InvokeAgent provides a mock function with given fields: ctx, params, optFns

func (*IClient) InvokeFlow

func (_m *IClient) InvokeFlow(ctx context.Context, params *bedrockagentruntime.InvokeFlowInput, optFns ...func(*bedrockagentruntime.Options)) (*bedrockagentruntime.InvokeFlowOutput, error)

InvokeFlow provides a mock function with given fields: ctx, params, optFns

func (*IClient) InvokeInlineAgent

func (_m *IClient) InvokeInlineAgent(ctx context.Context, params *bedrockagentruntime.InvokeInlineAgentInput, optFns ...func(*bedrockagentruntime.Options)) (*bedrockagentruntime.InvokeInlineAgentOutput, error)

InvokeInlineAgent provides a mock function with given fields: ctx, params, optFns

func (*IClient) OptimizePrompt

func (_m *IClient) OptimizePrompt(ctx context.Context, params *bedrockagentruntime.OptimizePromptInput, optFns ...func(*bedrockagentruntime.Options)) (*bedrockagentruntime.OptimizePromptOutput, error)

OptimizePrompt provides a mock function with given fields: ctx, params, optFns

func (*IClient) Options

func (_m *IClient) Options() bedrockagentruntime.Options

Options provides a mock function with given fields:

func (*IClient) Rerank

func (_m *IClient) Rerank(ctx context.Context, params *bedrockagentruntime.RerankInput, optFns ...func(*bedrockagentruntime.Options)) (*bedrockagentruntime.RerankOutput, error)

Rerank provides a mock function with given fields: ctx, params, optFns

func (*IClient) Retrieve

func (_m *IClient) Retrieve(ctx context.Context, params *bedrockagentruntime.RetrieveInput, optFns ...func(*bedrockagentruntime.Options)) (*bedrockagentruntime.RetrieveOutput, error)

Retrieve provides a mock function with given fields: ctx, params, optFns

func (*IClient) RetrieveAndGenerate

func (_m *IClient) RetrieveAndGenerate(ctx context.Context, params *bedrockagentruntime.RetrieveAndGenerateInput, optFns ...func(*bedrockagentruntime.Options)) (*bedrockagentruntime.RetrieveAndGenerateOutput, error)

RetrieveAndGenerate provides a mock function with given fields: ctx, params, optFns

func (*IClient) RetrieveAndGenerateStream

func (_m *IClient) RetrieveAndGenerateStream(ctx context.Context, params *bedrockagentruntime.RetrieveAndGenerateStreamInput, optFns ...func(*bedrockagentruntime.Options)) (*bedrockagentruntime.RetrieveAndGenerateStreamOutput, error)

RetrieveAndGenerateStream provides a mock function with given fields: ctx, params, optFns