Преглед на файлове

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

tobitege преди 1 година
родител
ревизия
4b76f98b26
променени са 1 файла, в които са добавени 4 реда и са изтрити 1 реда
  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}')