11import warnings
22
3-
43warnings .filterwarnings ("ignore" )
54
65import json
1514from pycoingecko import CoinGeckoAPI
1615from web3 import Web3
1716
18-
1917# Configure _logger
2018_logger = setup_logger (__name__ )
2119
4644_aggregators_cache = None
4745_historical_data_cache = None
4846
47+ # Error list
48+ errors = []
4949
5050def get_coin_list ():
5151 global _coin_list_cache
@@ -56,7 +56,7 @@ def get_coin_list():
5656 response .raise_for_status ()
5757 _coin_list_cache = response .json ()
5858 except requests .RequestException as e :
59- _logger . error (f"Failed to fetch coin list: { e } )
59+ errors . append (f"Failed to fetch coin list: { e } )
6060 _coin_list_cache = []
6161 return _coin_list_cache
6262
@@ -72,7 +72,7 @@ def fetch_token_id(symbol):
7272 if coin ["symbol" ].lower () == symbol :
7373 return coin ["id" ]
7474
75- _logger . error (f"Failed to fetch id for coin with symbol: { symbol } )
75+ errors . append (f"Failed to fetch id for coin with symbol: { symbol } )
7676 return None
7777
7878
@@ -86,7 +86,8 @@ def fetch_historical_data(limit: int = 720):
8686 url = f"https://us-central1-stu-dashboard-a0ba2.cloudfunctions.net/getV2AggregatorHistoricalData?last_time={ one_month_ago_ms } { limit }
8787 response = requests .get (url )
8888 if response .status_code != 200 :
89- raise Exception ("Failed to fetch historical data from STURDY API." )
89+ errors .append ("Failed to fetch historical data from STURDY API." )
90+ return None
9091 _historical_data_cache = response .json ()
9192 return _historical_data_cache
9293
@@ -166,12 +167,12 @@ def fetch_aggregators() -> List[Dict[str, Any]]:
166167 response .raise_for_status ()
167168 result = response .json ()
168169 if "errors" in result :
169- _logger . error (f"REST API Errors: { result ['errors' ]} )
170+ errors . append (f"REST API Errors: { result ['errors' ]} )
170171 _aggregators_cache = []
171172 else :
172173 _aggregators_cache = result
173174 except requests .RequestException as e :
174- _logger . error (f"REST API request failed: { e } )
175+ errors . append (f"REST API request failed: { e } )
175176 _aggregators_cache = []
176177 return _aggregators_cache
177178
@@ -200,7 +201,7 @@ def filter_aggregators(
200201 filtered_aggregators .append (aggregator )
201202
202203 if not filtered_aggregators :
203- _logger . error ("No suitable aggregator found." )
204+ errors . append ("No suitable aggregator found." )
204205 return []
205206
206207 # If very few aggregators, return them directly
@@ -226,7 +227,7 @@ def filter_aggregators(
226227 scored_aggregators .append (aggregator )
227228
228229 if not scored_aggregators :
229- _logger . error ("No suitable aggregator found after scoring." )
230+ errors . append ("No suitable aggregator found after scoring." )
230231 return []
231232
232233 score_threshold = np .percentile (
@@ -239,7 +240,7 @@ def filter_aggregators(
239240 filtered_scored_aggregators .sort (key = lambda x : x ["score" ], reverse = True )
240241
241242 if not filtered_scored_aggregators :
242- _logger . error ("No suitable aggregator found after score threshold." )
243+ errors . append ("No suitable aggregator found after score threshold." )
243244 return []
244245
245246 # Limit to top 10 scored pools if more than 10
@@ -256,12 +257,16 @@ def calculate_il_risk_score_for_lending(
256257 time_period : int = 90 ,
257258) -> float :
258259 if not asset_token_1 or not asset_token_2 :
259- _logger . error (
260+ errors . append (
260261 "Tokens are required. Cannot calculate IL risk score without asset tokens"
261262 )
262- return float ( "nan" )
263+ return None
263264
264- cg = CoinGeckoAPI (demo_api_key = coingecko_api_key )
265+ is_pro = is_pro_api_key (coingecko_api_key )
266+ if is_pro :
267+ cg = CoinGeckoAPI (api_key = coingecko_api_key )
268+ else :
269+ cg = CoinGeckoAPI (demo_api_key = coingecko_api_key )
265270 to_timestamp = int (datetime .now ().timestamp ())
266271 from_timestamp = int ((datetime .now () - timedelta (days = time_period )).timestamp ())
267272
@@ -279,15 +284,15 @@ def calculate_il_risk_score_for_lending(
279284 to_timestamp = to_timestamp ,
280285 )
281286 except Exception as e :
282- _logger . error (f"Error fetching price data: { e } )
283- return float ( "nan" )
284-
287+ errors . append (f"Error fetching price data: Incorrect Coingecko API Key " )
288+ return None
289+
285290 prices_1_data = np .array ([x [1 ] for x in prices_1 ["prices" ]])
286291 prices_2_data = np .array ([x [1 ] for x in prices_2 ["prices" ]])
287292
288293 min_length = min (len (prices_1_data ), len (prices_2_data ))
289294 if min_length == 0 :
290- return float ( "nan" )
295+ return None
291296
292297 prices_1_data = prices_1_data [:min_length ]
293298 prices_2_data = prices_2_data [:min_length ]
@@ -319,7 +324,7 @@ def get_token_id(symbol):
319324
320325 token_0_id = get_token_id (token0_symbol )
321326 if not token_0_id :
322- return float ( "nan" )
327+ return None
323328
324329 for silo in silos :
325330 token_1_symbol = silo ["collateral" ].lower ()
@@ -329,12 +334,15 @@ def get_token_id(symbol):
329334 il_risk_score = calculate_il_risk_score_for_lending (
330335 token_0_id , token_1_id , coingecko_api_key
331336 )
337+ if not il_risk_score :
338+ return None
339+
332340 il_risk_scores .append (il_risk_score )
333341 else :
334- _logger . error (f"Failed to fetch token IDs for silo: { silo ['collateral' ]} )
342+ errors . append (f"Failed to fetch token IDs for silo: { silo ['collateral' ]} )
335343
336344 if not il_risk_scores :
337- return float ( "nan" )
345+ return None
338346
339347 return sum (il_risk_scores ) / len (il_risk_scores )
340348
@@ -355,6 +363,7 @@ def analyze_vault_liquidity(aggregator):
355363 break
356364
357365 if not tvl or not total_assets :
366+ errors .append ("Could not retrieve depth score and maximum position size." )
358367 return float ("nan" ), float ("nan" )
359368
360369 depth_score = (
@@ -389,15 +398,19 @@ def get_best_opportunities(
389398) -> List [Dict [str , Any ]]:
390399 data = fetch_aggregators ()
391400 if not data :
392- return []
401+ errors .append ("Failed to fetch aggregators." )
402+ return {"error" : errors }
393403
394404 filtered_aggregators = filter_aggregators (
395405 chains , data , lending_asset , current_positions
396406 )
397407 if not filtered_aggregators :
398- return []
408+ errors .append ("No suitable aggregators found." )
409+ return {"error" : errors }
399410
400411 historical_data = fetch_historical_data ()
412+ if historical_data is None :
413+ return {"error" : errors }
401414
402415 for aggregator in filtered_aggregators :
403416 silos = aggregator .get ("whitelistedSilos" , [])
@@ -427,6 +440,9 @@ def calculate_metrics(
427440 coingecko_api_key ,
428441 )
429442 historical_data = fetch_historical_data ()
443+ if historical_data is None :
444+ return {"error" : errors }
445+
430446 sharpe_ratio = get_sharpe_ratio_for_address (
431447 historical_data , position ["pool_address" ]
432448 )
@@ -439,10 +455,31 @@ def calculate_metrics(
439455 }
440456
441457
458+ def is_pro_api_key (coingecko_api_key : str ) -> bool :
459+ """
460+ Check if the provided CoinGecko API key is a pro key.
461+ """
462+ # Try using the key as a pro API key
463+ cg_pro = CoinGeckoAPI (api_key = coingecko_api_key )
464+ try :
465+ response = cg_pro .get_coin_market_chart_range_by_id (
466+ id = "bitcoin" ,
467+ vs_currency = "usd" ,
468+ from_timestamp = 0 ,
469+ to_timestamp = 0
470+ )
471+ if response :
472+ return True
473+ except Exception :
474+ return False
475+
476+ return False
477+
442478def run (* _args , ** kwargs ) -> Any :
443479 missing = check_missing_fields (kwargs )
444480 if missing :
445- return {"error" : f"Required kwargs { missing } }
481+ errors .append (f"Required kwargs { missing } )
482+ return {"error" : errors }
446483
447484 required_fields = list (REQUIRED_FIELDS )
448485 get_metrics = kwargs .get ("get_metrics" , False )
@@ -452,9 +489,14 @@ def run(*_args, **kwargs) -> Any:
452489 kwargs = remove_irrelevant_fields (kwargs , required_fields )
453490
454491 if get_metrics :
455- return calculate_metrics (** kwargs )
492+ metrics = calculate_metrics (** kwargs )
493+ if metrics is None :
494+ errors .append ("Failed to calculate metrics." )
495+ return {"error" : errors } if errors else metrics
456496 else :
457497 result = get_best_opportunities (** kwargs )
498+ if isinstance (result , dict ) and "error" in result :
499+ errors .append (result ["error" ])
458500 if not result :
459- return { "error" : " No suitable aggregators found"}
460- return result
501+ errors . append ( " No suitable aggregators found")
502+ return { "error" : errors } if errors else { " result" : result }
0 commit comments