From 343e5c73aed737d5729d5ba49e46c235fded3f51 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E0=AE=AE=E0=AE=A9=E0=AF=8B=E0=AE=9C=E0=AF=8D=E0=AE=95?= =?UTF-8?q?=E0=AF=81=E0=AE=AE=E0=AE=BE=E0=AE=B0=E0=AF=8D=20=E0=AE=AA?= =?UTF-8?q?=E0=AE=B4=E0=AE=A9=E0=AE=BF=E0=AE=9A=E0=AF=8D=E0=AE=9A=E0=AE=BE?= =?UTF-8?q?=E0=AE=AE=E0=AE=BF?= Date: Wed, 29 May 2024 14:24:27 +0530 Subject: [PATCH] Parsed model_name for model_info (#2122) --- opendevin/llm/llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/opendevin/llm/llm.py b/opendevin/llm/llm.py index 18103dab48..bc039f59b4 100644 --- a/opendevin/llm/llm.py +++ b/opendevin/llm/llm.py @@ -131,7 +131,7 @@ 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) + self.model_info = litellm.get_model_info(self.model_name.split(':')[0]) # noinspection PyBroadException except Exception: logger.warning(f'Could not get model info for {self.model_name}')