diff --git a/app/primitives/index.ts b/app/primitives/index.ts
index 66b8a8e..a3239c7 100644
--- a/app/primitives/index.ts
+++ b/app/primitives/index.ts
@@ -13,6 +13,7 @@ export enum BittePrimitiveName {
   SWAP_ON_AFTERMATH = 'aftermath-sui-swap',
   GET_SUI_BALANCES = 'get-sui-balances',
   SUI_LIQUID_STAKING = 'sui-lst',
+  DATA_RETRIEVAL = 'data-retrieval',
 }
 
 export const isBittePrimitiveName = (
diff --git a/lib/constants.ts b/lib/constants.ts
index e3a0b94..7cd797a 100644
--- a/lib/constants.ts
+++ b/lib/constants.ts
@@ -24,6 +24,7 @@ export const BittePrimitiveNames = [
   "sign-message",
   "submit-query",
   "transfer-ft",
+  "data-retrieval",
 ];
 
 export const DEFAULT_MODEL = "gpt-4o";