Defillama additional integration

eval/eval_widgets.txt

@@ -9,6 +9,7 @@ fetch_nft_collection_traits
 fetch_nft_collection_trait_values
 fetch_nft_asset_traits
 fetch_yields
+fetch_tvl
 fetch_price
 ens_from_address
 address_from_ens

integrations/defillama.py

@@ -22,6 +22,15 @@ def container_name(self) -> str:
     def container_params(self) -> Dict:
         return dataclass_to_container_params(self)
 
+def fetch_tvl(protocol) -> str:
+    url = f"https://api.llama.fi/tvl/{protocol}"
+    response = requests.get(url)
+    try:
+        response.raise_for_status()
+    except requests.exceptions.HTTPError:
+        return "No TVL data available for this protocol"
+    obj = response.json()
+    return obj
 
 def fetch_yields(token, network, count) -> List[Yield]:
     normalized_network_name = _network_name_normalizer(network)

knowledge_base/widgets.yaml

@@ -614,6 +614,17 @@
     - value
     type: object
   return_value_description: ''
+- _name_: fetch_tvl
+  description: Get the tvl for crypto protocol.
+  parameters:
+    properties:
+      protocol:
+        description: Name of the protocol
+        type: string
+    required:
+    - protocol
+    type: object
+  return_value_description: The TVl of the protocol
 - _name_: display_yield_protocol_lend
   description: use the yield protocol to lend tokens at a fixed rate
   parameters:
@@ -677,4 +688,4 @@
     required:
       - borrowToken
     type: object
-  return_value_description: ""
+  return_value_description: ""

tools/index_widget.py

@@ -243,6 +243,8 @@ def replace_match(m: re.Match) -> Union[str, Generator, Callable]:
         return str(fetch_gas(*params))
     elif command == 'fetch-yields':
         return str(fetch_yields(*params))
+    elif command == 'fetch-tvl':
+        return str(fetch_tvl(*params))
     elif command == 'fetch-app-info':
         return fetch_app_info(*params)
     elif command == 'fetch-scraped-sites':
@@ -537,6 +539,9 @@ def fetch_nft_buy(network: str, address: str, token_id: str) -> str:
     ret = opensea.fetch_nft_buy(network, address, token_id)
     return ret
 
+@error_wrap
+def fetch_tvl(protocol: str) -> str:
+    return defillama.fetch_tvl(protocol)
 
 @error_wrap
 def fetch_yields(token, network, count) -> str:

utils/constants.py

@@ -50,7 +50,7 @@
 WIDGET_INFO_TOKEN_LIMIT = 4000
 
 # Widget Index
-WIDGET_INDEX_NAME = "WidgetV16"
+WIDGET_INDEX_NAME = "WidgetV19"
 
 def get_widget_index_name():
     if env.is_local():