chore(internal): add internal helpers & improve build scripts (#73)

anthropics · Jan 22, 2024 · d12b655 · d12b655
1 parent d4db6b7
commit d12b655
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 5 deletions.
diff --git a/scripts/fix-index-exports.cjs b/scripts/fix-index-exports.cjs
@@ -1,7 +1,11 @@
 const fs = require('fs');
 const path = require('path');
 
-const indexJs = path.resolve(__dirname, '..', 'dist', 'index.js');
+const indexJs =
+  process.env['DIST_PATH'] ?
+    path.resolve(process.env['DIST_PATH'], 'index.js')
+  : path.resolve(__dirname, '..', 'dist', 'index.js');
+
 let before = fs.readFileSync(indexJs, 'utf8');
 let after = before.replace(
   /^\s*exports\.default\s*=\s*(\w+)/m,

diff --git a/scripts/make-dist-package-json.cjs b/scripts/make-dist-package-json.cjs
@@ -1,4 +1,4 @@
-const pkgJson = require('../package.json');
+const pkgJson = require(process.env['PKG_JSON_PATH'] || '../package.json');
 
 function processExportMap(m) {
   for (const key in m) {

diff --git a/scripts/postprocess-files.cjs b/scripts/postprocess-files.cjs
@@ -2,7 +2,12 @@ const fs = require('fs');
 const path = require('path');
 const { parse } = require('@typescript-eslint/parser');
 
-const distDir = path.resolve(__dirname, '..', 'dist');
+const pkgImportPath = process.env['PKG_IMPORT_PATH'] ?? '@anthropic-ai/bedrock-sdk/'
+
+const distDir =
+  process.env['DIST_PATH'] ?
+    path.resolve(process.env['DIST_PATH'])
+  : path.resolve(__dirname, '..', 'dist');
 const distSrcDir = path.join(distDir, 'src');
 
 /**
@@ -105,11 +110,11 @@ async function postprocess() {
 
     let transformed = mapModulePaths(code, (importPath) => {
       if (file.startsWith(distSrcDir)) {
-        if (importPath.startsWith('@anthropic-ai/bedrock-sdk/')) {
+        if (importPath.startsWith(pkgImportPath)) {
           // convert self-references in dist/src to relative paths
           let relativePath = path.relative(
             path.dirname(file),
-            path.join(distSrcDir, importPath.substring('@anthropic-ai/bedrock-sdk/'.length)),
+            path.join(distSrcDir, importPath.substring(pkgImportPath.length)),
           );
           if (!relativePath.startsWith('.')) relativePath = `./${relativePath}`;
           return relativePath;

diff --git a/src/core.ts b/src/core.ts
@@ -342,6 +342,11 @@ export abstract class APIClient {
     return reqHeaders;
   }
 
+  /**
+   * Used as a callback for mutating the given `FinalRequestOptions` object.
+   */
+  protected async prepareOptions(options: FinalRequestOptions): Promise<void> {}
+
   /**
    * Used as a callback for mutating the given `RequestInit` object.
    *
@@ -387,6 +392,8 @@ export abstract class APIClient {
       retriesRemaining = options.maxRetries ?? this.maxRetries;
     }
 
+    await this.prepareOptions(options);
+
     const { req, url, timeout } = this.buildRequest(options);
 
     await this.prepareRequest(req, { url, options });
@@ -1140,3 +1147,7 @@ export const toBase64 = (str: string | null | undefined): string => {
 
   throw new AnthropicBedrockError('Cannot generate b64 string; Expected `Buffer` or `btoa` to be defined');
 };
+
+export function isObj(obj: unknown): obj is Record<string, unknown> {
+  return obj != null && typeof obj === 'object' && !Array.isArray(obj);
+}