Add HTTPSCRAMAuth to authed fetch

edgedb · Sep 6, 2024 · 1be3174 · 1be3174
1 parent fd82d48
commit 1be3174
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/packages/ai/src/core.ts b/packages/ai/src/core.ts
@@ -15,11 +15,15 @@ import type {
   RAGRequest,
   StreamingMessage,
 } from "./types.js";
+import { getHTTPSCRAMAuth } from "edgedb/dist/httpScram.js";
+import browserCryptoUtils from "edgedb/dist/browserCrypto.js";
 
 export function createAI(client: Client, options: AIOptions) {
   return new EdgeDBAI(client, options);
 }
 
+const httpSCRAMAuth = getHTTPSCRAMAuth(browserCryptoUtils.default);
+
 export class EdgeDBAI {
   /** @internal */
   private readonly authenticatedFetch: Promise<AuthenticatedFetch>;
@@ -46,7 +50,7 @@ export class EdgeDBAI {
       await (client as any).pool._getNormalizedConnectConfig()
     ).connectionParams;
 
-    return getAuthenticatedFetch(connectConfig, "ext/ai/");
+    return getAuthenticatedFetch(connectConfig, httpSCRAMAuth, "ext/ai/");
   }
 
   withConfig(options: Partial<AIOptions>) {