Skip to content

Commit cddc9ee

Browse files
authoredJan 3, 2025
Merge pull request elizaOS#1745 from sinecose/console-to-elizaLogger
feat(plugin-near): replace console.log to eliza logger
2 parents 7cb04dc + 48fbcab commit cddc9ee

File tree

3 files changed

+26
-18
lines changed

3 files changed

+26
-18
lines changed
 

‎packages/plugin-near/src/actions/swap.ts

+9-8
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import {
22
ActionExample,
33
HandlerCallback,
4+
elizaLogger,
45
IAgentRuntime,
56
Memory,
67
ModelClass,
@@ -34,7 +35,7 @@ async function checkStorageBalance(
3435
});
3536
return balance !== null && balance.total !== "0";
3637
} catch (error) {
37-
console.log(`Error checking storage balance: ${error}`);
38+
elizaLogger.log(`Error checking storage balance: ${error}`);
3839
return false;
3940
}
4041
}
@@ -142,7 +143,7 @@ async function swapToken(
142143

143144
return transactions;
144145
} catch (error) {
145-
console.error("Error in swapToken:", error);
146+
elizaLogger.error("Error in swapToken:", error);
146147
throw error;
147148
}
148149
}
@@ -186,8 +187,8 @@ export const executeSwap: Action = {
186187
"TRADE_TOKENS_NEAR",
187188
"EXCHANGE_TOKENS_NEAR",
188189
],
189-
validate: async (runtime: IAgentRuntime, message: Memory) => {
190-
console.log("Message:", message);
190+
validate: async (_runtime: IAgentRuntime, message: Memory) => {
191+
elizaLogger.log("Message:", message);
191192
return true;
192193
},
193194
description: "Perform a token swap using Ref Finance.",
@@ -221,14 +222,14 @@ export const executeSwap: Action = {
221222
modelClass: ModelClass.LARGE,
222223
});
223224

224-
console.log("Response:", response);
225+
elizaLogger.log("Response:", response);
225226

226227
if (
227228
!response.inputTokenId ||
228229
!response.outputTokenId ||
229230
!response.amount
230231
) {
231-
console.log("Missing required parameters, skipping swap");
232+
elizaLogger.log("Missing required parameters, skipping swap");
232233
const responseMsg = {
233234
text: "I need the input token ID, output token ID, and amount to perform the swap",
234235
};
@@ -290,7 +291,7 @@ export const executeSwap: Action = {
290291
}
291292
}
292293

293-
console.log("Swap completed successfully!");
294+
elizaLogger.log("Swap completed successfully!");
294295
const txHashes = results.map((r) => r.transaction.hash).join(", ");
295296

296297
const responseMsg = {
@@ -300,7 +301,7 @@ export const executeSwap: Action = {
300301
callback?.(responseMsg);
301302
return true;
302303
} catch (error) {
303-
console.error("Error during token swap:", error);
304+
elizaLogger.error("Error during token swap:", error);
304305
const responseMsg = {
305306
text: `Error during swap: ${error instanceof Error ? error.message : String(error)}`,
306307
};

‎packages/plugin-near/src/actions/transfer.ts

+3-2
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ import {
66
Memory,
77
ModelClass,
88
State,
9+
elizaLogger,
910
type Action,
1011
composeContext,
1112
generateObject,
@@ -132,7 +133,7 @@ export const executeTransfer: Action = {
132133

133134
// Validate transfer content
134135
if (!isTransferContent(runtime, content)) {
135-
console.error("Invalid content for TRANSFER_NEAR action.");
136+
elizaLogger.error("Invalid content for TRANSFER_NEAR action.");
136137
if (callback) {
137138
callback({
138139
text: "Unable to process transfer request. Invalid content provided.",
@@ -163,7 +164,7 @@ export const executeTransfer: Action = {
163164

164165
return true;
165166
} catch (error) {
166-
console.error("Error during NEAR transfer:", error);
167+
elizaLogger.error("Error during NEAR transfer:", error);
167168
if (callback) {
168169
callback({
169170
text: `Error transferring NEAR: ${error}`,

‎packages/plugin-near/src/providers/wallet.ts

+14-8
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,10 @@
1-
import { IAgentRuntime, Memory, Provider, State } from "@elizaos/core";
1+
import {
2+
IAgentRuntime,
3+
Memory,
4+
Provider,
5+
State,
6+
elizaLogger,
7+
} from "@elizaos/core";
28
import { KeyPair, keyStores, connect, Account, utils } from "near-api-js";
39
import BigNumber from "bignumber.js";
410
import { KeyPairString } from "near-api-js/lib/utils";
@@ -51,7 +57,7 @@ export class WalletProvider implements Provider {
5157
try {
5258
return await this.getFormattedPortfolio(runtime);
5359
} catch (error) {
54-
console.error("Error in wallet provider:", error);
60+
elizaLogger.error("Error in wallet provider:", error);
5561
return null;
5662
}
5763
}
@@ -102,7 +108,7 @@ export class WalletProvider implements Provider {
102108
}
103109
return await response.json();
104110
} catch (error) {
105-
console.error(`Attempt ${i + 1} failed:`, error);
111+
elizaLogger.error(`Attempt ${i + 1} failed:`, error);
106112
lastError = error as Error;
107113
if (i < PROVIDER_CONFIG.MAX_RETRIES - 1) {
108114
await new Promise((resolve) =>
@@ -125,7 +131,7 @@ export class WalletProvider implements Provider {
125131
const cachedValue = this.cache.get<WalletPortfolio>(cacheKey);
126132

127133
if (cachedValue) {
128-
console.log("Cache hit for fetchPortfolioValue");
134+
elizaLogger.log("Cache hit for fetchPortfolioValue");
129135
return cachedValue;
130136
}
131137

@@ -160,7 +166,7 @@ export class WalletProvider implements Provider {
160166
this.cache.set(cacheKey, portfolio);
161167
return portfolio;
162168
} catch (error) {
163-
console.error("Error fetching portfolio:", error);
169+
elizaLogger.error("Error fetching portfolio:", error);
164170
throw error;
165171
}
166172
}
@@ -181,7 +187,7 @@ export class WalletProvider implements Provider {
181187
this.cache.set(cacheKey, price);
182188
return price;
183189
} catch (error) {
184-
console.error("Error fetching NEAR price:", error);
190+
elizaLogger.error("Error fetching NEAR price:", error);
185191
return 0;
186192
}
187193
}
@@ -214,7 +220,7 @@ export class WalletProvider implements Provider {
214220
const portfolio = await this.fetchPortfolioValue(runtime);
215221
return this.formatPortfolio(runtime, portfolio);
216222
} catch (error) {
217-
console.error("Error generating portfolio report:", error);
223+
elizaLogger.error("Error generating portfolio report:", error);
218224
return "Unable to fetch wallet information. Please try again later.";
219225
}
220226
}
@@ -234,7 +240,7 @@ const walletProvider: Provider = {
234240
const provider = new WalletProvider(accountId);
235241
return await provider.getFormattedPortfolio(runtime);
236242
} catch (error) {
237-
console.error("Error in wallet provider:", error);
243+
elizaLogger.error("Error in wallet provider:", error);
238244
return null;
239245
}
240246
},

0 commit comments

Comments
 (0)
Please sign in to comment.