File tree Expand file tree Collapse file tree 1 file changed +10
-3
lines changed
packages/ragbits-core/src/ragbits/core/llms Expand file tree Collapse file tree 1 file changed +10
-3
lines changed Original file line number Diff line number Diff line change @@ -106,9 +106,16 @@ def __init__(
106
106
else :
107
107
108
108
def download_and_register_model_cost () -> None :
109
- litellm .register_model (
110
- model_cost = "https://raw.githubusercontent.com/BerriAI/litellm/main/model_prices_and_context_window.json"
111
- )
109
+ # Suppress debug info to prevent "Provider List" messages during registration
110
+ original_suppress_debug = litellm .suppress_debug_info
111
+ litellm .suppress_debug_info = True
112
+ try :
113
+ litellm .register_model (
114
+ model_cost = "https://raw.githubusercontent.com/BerriAI/litellm/main/model_prices_and_context_window.json"
115
+ )
116
+ finally :
117
+ # Restore original setting
118
+ litellm .suppress_debug_info = original_suppress_debug
112
119
113
120
thread = threading .Thread (target = download_and_register_model_cost , daemon = True )
114
121
thread .start ()
You can’t perform that action at this time.
0 commit comments