@effect-aws/client-bedrock-runtime 1.9.2

package/src/BedrockRuntimeService.ts

@@ -0,0 +1,256 @@
+/**
+ * @since 1.0.0
+ */
+import {
+  ApplyGuardrailCommand,
+  type ApplyGuardrailCommandInput,
+  type ApplyGuardrailCommandOutput,
+  type BedrockRuntimeClient,
+  type BedrockRuntimeClientConfig,
+  ConverseCommand,
+  type ConverseCommandInput,
+  type ConverseCommandOutput,
+  ConverseStreamCommand,
+  type ConverseStreamCommandInput,
+  type ConverseStreamCommandOutput,
+  GetAsyncInvokeCommand,
+  type GetAsyncInvokeCommandInput,
+  type GetAsyncInvokeCommandOutput,
+  InvokeModelCommand,
+  type InvokeModelCommandInput,
+  type InvokeModelCommandOutput,
+  InvokeModelWithResponseStreamCommand,
+  type InvokeModelWithResponseStreamCommandInput,
+  type InvokeModelWithResponseStreamCommandOutput,
+  ListAsyncInvokesCommand,
+  type ListAsyncInvokesCommandInput,
+  type ListAsyncInvokesCommandOutput,
+  StartAsyncInvokeCommand,
+  type StartAsyncInvokeCommandInput,
+  type StartAsyncInvokeCommandOutput,
+} from "@aws-sdk/client-bedrock-runtime";
+import type { HttpHandlerOptions, SdkError, ServiceLogger } from "@effect-aws/commons";
+import { Service } from "@effect-aws/commons";
+import { Effect, Layer } from "effect";
+import * as Instance from "./BedrockRuntimeClientInstance.js";
+import * as BedrockRuntimeServiceConfig from "./BedrockRuntimeServiceConfig.js";
+import type {
+  AccessDeniedError,
+  ConflictError,
+  InternalServerError,
+  ModelError,
+  ModelNotReadyError,
+  ModelStreamError,
+  ModelTimeoutError,
+  ResourceNotFoundError,
+  ServiceQuotaExceededError,
+  ServiceUnavailableError,
+  ThrottlingError,
+  ValidationError,
+} from "./Errors.js";
+import { AllServiceErrors } from "./Errors.js";
+
+const commands = {
+  ApplyGuardrailCommand,
+  ConverseCommand,
+  ConverseStreamCommand,
+  GetAsyncInvokeCommand,
+  InvokeModelCommand,
+  InvokeModelWithResponseStreamCommand,
+  ListAsyncInvokesCommand,
+  StartAsyncInvokeCommand,
+};
+
+interface BedrockRuntimeService$ {
+  readonly _: unique symbol;
+
+  /**
+   * @see {@link ApplyGuardrailCommand}
+   */
+  applyGuardrail(
+    args: ApplyGuardrailCommandInput,
+    options?: HttpHandlerOptions,
+  ): Effect.Effect<
+    ApplyGuardrailCommandOutput,
+    | SdkError
+    | AccessDeniedError
+    | InternalServerError
+    | ResourceNotFoundError
+    | ServiceQuotaExceededError
+    | ThrottlingError
+    | ValidationError
+  >;
+
+  /**
+   * @see {@link ConverseCommand}
+   */
+  converse(
+    args: ConverseCommandInput,
+    options?: HttpHandlerOptions,
+  ): Effect.Effect<
+    ConverseCommandOutput,
+    | SdkError
+    | AccessDeniedError
+    | InternalServerError
+    | ModelError
+    | ModelNotReadyError
+    | ModelTimeoutError
+    | ResourceNotFoundError
+    | ServiceUnavailableError
+    | ThrottlingError
+    | ValidationError
+  >;
+
+  /**
+   * @see {@link ConverseStreamCommand}
+   */
+  converseStream(
+    args: ConverseStreamCommandInput,
+    options?: HttpHandlerOptions,
+  ): Effect.Effect<
+    ConverseStreamCommandOutput,
+    | SdkError
+    | AccessDeniedError
+    | InternalServerError
+    | ModelError
+    | ModelNotReadyError
+    | ModelTimeoutError
+    | ResourceNotFoundError
+    | ServiceUnavailableError
+    | ThrottlingError
+    | ValidationError
+  >;
+
+  /**
+   * @see {@link GetAsyncInvokeCommand}
+   */
+  getAsyncInvoke(
+    args: GetAsyncInvokeCommandInput,
+    options?: HttpHandlerOptions,
+  ): Effect.Effect<
+    GetAsyncInvokeCommandOutput,
+    SdkError | AccessDeniedError | InternalServerError | ThrottlingError | ValidationError
+  >;
+
+  /**
+   * @see {@link InvokeModelCommand}
+   */
+  invokeModel(
+    args: InvokeModelCommandInput,
+    options?: HttpHandlerOptions,
+  ): Effect.Effect<
+    InvokeModelCommandOutput,
+    | SdkError
+    | AccessDeniedError
+    | InternalServerError
+    | ModelError
+    | ModelNotReadyError
+    | ModelTimeoutError
+    | ResourceNotFoundError
+    | ServiceQuotaExceededError
+    | ServiceUnavailableError
+    | ThrottlingError
+    | ValidationError
+  >;
+
+  /**
+   * @see {@link InvokeModelWithResponseStreamCommand}
+   */
+  invokeModelWithResponseStream(
+    args: InvokeModelWithResponseStreamCommandInput,
+    options?: HttpHandlerOptions,
+  ): Effect.Effect<
+    InvokeModelWithResponseStreamCommandOutput,
+    | SdkError
+    | AccessDeniedError
+    | InternalServerError
+    | ModelError
+    | ModelNotReadyError
+    | ModelStreamError
+    | ModelTimeoutError
+    | ResourceNotFoundError
+    | ServiceQuotaExceededError
+    | ServiceUnavailableError
+    | ThrottlingError
+    | ValidationError
+  >;
+
+  /**
+   * @see {@link ListAsyncInvokesCommand}
+   */
+  listAsyncInvokes(
+    args: ListAsyncInvokesCommandInput,
+    options?: HttpHandlerOptions,
+  ): Effect.Effect<
+    ListAsyncInvokesCommandOutput,
+    SdkError | AccessDeniedError | InternalServerError | ThrottlingError | ValidationError
+  >;
+
+  /**
+   * @see {@link StartAsyncInvokeCommand}
+   */
+  startAsyncInvoke(
+    args: StartAsyncInvokeCommandInput,
+    options?: HttpHandlerOptions,
+  ): Effect.Effect<
+    StartAsyncInvokeCommandOutput,
+    | SdkError
+    | AccessDeniedError
+    | ConflictError
+    | InternalServerError
+    | ResourceNotFoundError
+    | ServiceQuotaExceededError
+    | ServiceUnavailableError
+    | ThrottlingError
+    | ValidationError
+  >;
+}
+
+/**
+ * @since 1.0.0
+ * @category constructors
+ */
+export const makeBedrockRuntimeService = Effect.gen(function*() {
+  const client = yield* Instance.BedrockRuntimeClientInstance;
+
+  return Service.fromClientAndCommands<BedrockRuntimeService$>(client, commands, AllServiceErrors);
+});
+
+/**
+ * @since 1.0.0
+ * @category models
+ */
+export class BedrockRuntimeService extends Effect.Tag("@effect-aws/client-bedrock-runtime/BedrockRuntimeService")<
+  BedrockRuntimeService,
+  BedrockRuntimeService$
+>() {
+  static readonly defaultLayer = Layer.effect(this, makeBedrockRuntimeService).pipe(Layer.provide(Instance.layer));
+  static readonly layer = (config: BedrockRuntimeService.Config) =>
+    Layer.effect(this, makeBedrockRuntimeService).pipe(
+      Layer.provide(Instance.layer),
+      Layer.provide(BedrockRuntimeServiceConfig.setBedrockRuntimeServiceConfig(config)),
+    );
+  static readonly baseLayer = (
+    evaluate: (defaultConfig: BedrockRuntimeClientConfig) => BedrockRuntimeClient,
+  ) =>
+    Layer.effect(this, makeBedrockRuntimeService).pipe(
+      Layer.provide(
+        Layer.effect(
+          Instance.BedrockRuntimeClientInstance,
+          Effect.map(BedrockRuntimeServiceConfig.toBedrockRuntimeClientConfig, evaluate),
+        ),
+      ),
+    );
+}
+
+/**
+ * @since 1.0.0
+ */
+export declare namespace BedrockRuntimeService {
+  /**
+   * @since 1.0.0
+   */
+  export interface Config extends Omit<BedrockRuntimeClientConfig, "logger"> {
+    readonly logger?: ServiceLogger.ServiceLoggerConstructorProps | true;
+  }
+}

package/src/BedrockRuntimeServiceConfig.ts

@@ -0,0 +1,52 @@
+/**
+ * @since 1.0.0
+ */
+import type { BedrockRuntimeClientConfig } from "@aws-sdk/client-bedrock-runtime";
+import { ServiceLogger } from "@effect-aws/commons";
+import { Effect, FiberRef, Layer } from "effect";
+import { dual } from "effect/Function";
+import { globalValue } from "effect/GlobalValue";
+import type { BedrockRuntimeService } from "./BedrockRuntimeService.js";
+
+/**
+ * @since 1.0.0
+ * @category bedrock-runtime service config
+ */
+const currentBedrockRuntimeServiceConfig = globalValue(
+  "@effect-aws/client-bedrock-runtime/currentBedrockRuntimeServiceConfig",
+  () => FiberRef.unsafeMake<BedrockRuntimeService.Config>({}),
+);
+
+/**
+ * @since 1.0.0
+ * @category bedrock-runtime service config
+ */
+export const withBedrockRuntimeServiceConfig: {
+  (config: BedrockRuntimeService.Config): <A, E, R>(effect: Effect.Effect<A, E, R>) => Effect.Effect<A, E, R>;
+  <A, E, R>(effect: Effect.Effect<A, E, R>, config: BedrockRuntimeService.Config): Effect.Effect<A, E, R>;
+} = dual(
+  2,
+  <A, E, R>(effect: Effect.Effect<A, E, R>, config: BedrockRuntimeService.Config): Effect.Effect<A, E, R> =>
+    Effect.locally(effect, currentBedrockRuntimeServiceConfig, config),
+);
+
+/**
+ * @since 1.0.0
+ * @category bedrock-runtime service config
+ */
+export const setBedrockRuntimeServiceConfig = (config: BedrockRuntimeService.Config) =>
+  Layer.locallyScoped(currentBedrockRuntimeServiceConfig, config);
+
+/**
+ * @since 1.0.0
+ * @category adapters
+ */
+export const toBedrockRuntimeClientConfig: Effect.Effect<BedrockRuntimeClientConfig> = Effect.gen(function*() {
+  const { logger: serviceLogger, ...config } = yield* FiberRef.get(currentBedrockRuntimeServiceConfig);
+
+  const logger = serviceLogger === true
+    ? yield* ServiceLogger.toClientLogger(ServiceLogger.defaultServiceLogger)
+    : (serviceLogger ? yield* ServiceLogger.toClientLogger(ServiceLogger.make(serviceLogger)) : undefined);
+
+  return { logger, ...config };
+});

package/src/Errors.ts

@@ -0,0 +1,47 @@
+import type {
+  AccessDeniedException,
+  ConflictException,
+  InternalServerException,
+  ModelErrorException,
+  ModelNotReadyException,
+  ModelStreamErrorException,
+  ModelTimeoutException,
+  ResourceNotFoundException,
+  ServiceQuotaExceededException,
+  ServiceUnavailableException,
+  ThrottlingException,
+  ValidationException,
+} from "@aws-sdk/client-bedrock-runtime";
+import type { TaggedException } from "@effect-aws/commons";
+import { SdkError as CommonSdkError } from "@effect-aws/commons";
+
+export const AllServiceErrors = [
+  "AccessDeniedException",
+  "ConflictException",
+  "InternalServerException",
+  "ModelErrorException",
+  "ModelNotReadyException",
+  "ModelStreamErrorException",
+  "ModelTimeoutException",
+  "ResourceNotFoundException",
+  "ServiceQuotaExceededException",
+  "ServiceUnavailableException",
+  "ThrottlingException",
+  "ValidationException",
+] as const;
+
+export type AccessDeniedError = TaggedException<AccessDeniedException>;
+export type ConflictError = TaggedException<ConflictException>;
+export type InternalServerError = TaggedException<InternalServerException>;
+export type ModelError = TaggedException<ModelErrorException>;
+export type ModelNotReadyError = TaggedException<ModelNotReadyException>;
+export type ModelStreamError = TaggedException<ModelStreamErrorException>;
+export type ModelTimeoutError = TaggedException<ModelTimeoutException>;
+export type ResourceNotFoundError = TaggedException<ResourceNotFoundException>;
+export type ServiceQuotaExceededError = TaggedException<ServiceQuotaExceededException>;
+export type ServiceUnavailableError = TaggedException<ServiceUnavailableException>;
+export type ThrottlingError = TaggedException<ThrottlingException>;
+export type ValidationError = TaggedException<ValidationException>;
+
+export type SdkError = CommonSdkError;
+export const SdkError = CommonSdkError;

package/src/index.ts

@@ -0,0 +1,44 @@
+/**
+ * @since 1.0.0
+ */
+import { BedrockRuntimeService } from "./BedrockRuntimeService.js";
+
+/**
+ * @since 1.0.0
+ */
+export * from "./Errors.js";
+
+/**
+ * @since 1.0.0
+ */
+export * as BedrockRuntimeClientInstance from "./BedrockRuntimeClientInstance.js";
+
+/**
+ * @since 1.0.0
+ */
+export * as BedrockRuntimeServiceConfig from "./BedrockRuntimeServiceConfig.js";
+
+/**
+ * @since 1.0.0
+ */
+export * from "./BedrockRuntimeService.js";
+
+/**
+ * @since 1.0.0
+ * @category exports
+ * @alias BedrockRuntimeService
+ */
+export declare namespace BedrockRuntime {
+  /**
+   * @since 1.0.0
+   * @alias BedrockRuntimeService.Config
+   */
+  export type Config = BedrockRuntimeService.Config;
+}
+
+/**
+ * @since 1.0.0
+ * @category exports
+ * @alias BedrockRuntimeService
+ */
+export const BedrockRuntime = BedrockRuntimeService;