From 2b152639a5d6a1fe77eba979823c72d9b545c261 Mon Sep 17 00:00:00 2001
From: Spacebean <wndgud00@gmail.com>
Date: Mon, 9 Sep 2024 11:37:55 -0600
Subject: [PATCH] update wells-sdk provider implementation

---
 projects/sdk-wells/src/lib/WellsSDK.ts | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/projects/sdk-wells/src/lib/WellsSDK.ts b/projects/sdk-wells/src/lib/WellsSDK.ts
index 9fe2bcb948..d5e61af8ee 100644
--- a/projects/sdk-wells/src/lib/WellsSDK.ts
+++ b/projects/sdk-wells/src/lib/WellsSDK.ts
@@ -1,4 +1,4 @@
-import { ChainId } from "@beanstalk/sdk-core";
+import { Address, ChainId } from "@beanstalk/sdk-core";
 import { ethers } from "ethers";
 import { addresses } from "src/constants/addresses";
 import { enumFromValue } from "src/utils";
@@ -69,7 +69,7 @@ export class WellsSDK {
 
   handleConfig(config: SDKConfig = {}) {
     if (config.rpcUrl) {
-      config.provider = this.getProviderFromUrl(config.rpcUrl);
+      config.provider = this.getProviderFromUrl(config.rpcUrl, config.provider);
     }
 
     this.signer = config.signer;
@@ -84,12 +84,13 @@ export class WellsSDK {
     this.DEBUG = config.DEBUG ?? false;
   }
 
-  private getProviderFromUrl(url: string): Provider {
+  private getProviderFromUrl(url: string, _provider: SDKConfig["provider"]): Provider {
+    const networkish = _provider?.network || _provider?._network || Address.defaultChainId;
     if (url.startsWith("ws")) {
-      return new ethers.providers.WebSocketProvider(url);
+      return new ethers.providers.WebSocketProvider(url, networkish);
     }
     if (url.startsWith("http")) {
-      return new ethers.providers.JsonRpcProvider(url);
+      return new ethers.providers.JsonRpcProvider(url, networkish);
     }
 
     throw new Error("Invalid rpcUrl");