|
@@ -131,7 +131,10 @@ class LLM:
|
|
|
# litellm actually uses base Exception here for unknown model
|
|
# litellm actually uses base Exception here for unknown model
|
|
|
self.model_info = None
|
|
self.model_info = None
|
|
|
try:
|
|
try:
|
|
|
- self.model_info = litellm.get_model_info(self.model_name.split(':')[0])
|
|
|
|
|
|
|
+ if not self.model_name.startswith('openrouter'):
|
|
|
|
|
+ self.model_info = litellm.get_model_info(self.model_name.split(':')[0])
|
|
|
|
|
+ else:
|
|
|
|
|
+ self.model_info = litellm.get_model_info(self.model_name)
|
|
|
# noinspection PyBroadException
|
|
# noinspection PyBroadException
|
|
|
except Exception:
|
|
except Exception:
|
|
|
logger.warning(f'Could not get model info for {self.model_name}')
|
|
logger.warning(f'Could not get model info for {self.model_name}')
|