Bladeren bron

fix: keep colon part in model name for OpenRouter (#2223)

tobitege 1 jaar geleden
bovenliggende
commit
4b76f98b26
1 gewijzigde bestanden met toevoegingen van 4 en 1 verwijderingen
  1. 4 1
      opendevin/llm/llm.py

+ 4 - 1
opendevin/llm/llm.py

@@ -131,7 +131,10 @@ class LLM:
         # litellm actually uses base Exception here for unknown model
         self.model_info = None
         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
         except Exception:
             logger.warning(f'Could not get model info for {self.model_name}')