From 32dbcc8b52215e640a523fbd5d9e0e1500be66e8 Mon Sep 17 00:00:00 2001
From: Benjamin Smith <bh2smith@users.noreply.github.com>
Date: Wed, 22 Jan 2025 19:04:43 +0100
Subject: [PATCH] Possibly undefined TokenDetails

---
 packages/agent-sdk/src/evm/tokens.ts       | 4 ++--
 packages/agent-sdk/tests/evm/token.spec.ts | 5 +++++
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/packages/agent-sdk/src/evm/tokens.ts b/packages/agent-sdk/src/evm/tokens.ts
index 5bd53bb..301c5de 100644
--- a/packages/agent-sdk/src/evm/tokens.ts
+++ b/packages/agent-sdk/src/evm/tokens.ts
@@ -2,7 +2,7 @@ import { isAddress } from "viem";
 import { getTokenInfo } from "./erc20";
 import { type TokenInfo } from "./types";
 
-type SymbolMapping = Record<string, TokenInfo>;
+type SymbolMapping = Record<string, TokenInfo | undefined>;
 type ChainId = number;
 export type BlockchainMapping = Record<ChainId, SymbolMapping>;
 
@@ -36,7 +36,7 @@ export async function getTokenDetails(
   chainId: number,
   symbolOrAddress: string,
   tokenMap?: BlockchainMapping,
-): Promise<TokenInfo> {
+): Promise<TokenInfo | undefined> {
   if (isAddress(symbolOrAddress, { strict: false })) {
     return getTokenInfo(chainId, symbolOrAddress);
   }
diff --git a/packages/agent-sdk/tests/evm/token.spec.ts b/packages/agent-sdk/tests/evm/token.spec.ts
index 9ed5564..0bfd0d7 100644
--- a/packages/agent-sdk/tests/evm/token.spec.ts
+++ b/packages/agent-sdk/tests/evm/token.spec.ts
@@ -12,4 +12,9 @@ describe("getTokenDetails", () => {
     );
     expect(tokenDetails).toBeDefined();
   });
+
+  it("should return the token details for a given symbol", async () => {
+    const tokenDetails = await getTokenDetails(43114, "UNI");
+    expect(tokenDetails).toBeUndefined();
+  });
 });