File tree Expand file tree Collapse file tree 1 file changed +4
-11
lines changed Expand file tree Collapse file tree 1 file changed +4
-11
lines changed Original file line number Diff line number Diff line change @@ -61,18 +61,11 @@ def run(self, messages):
61
61
# Detect function support
62
62
if self .supports_functions != None :
63
63
supports_functions = self .supports_functions
64
+ elif litellm .supports_function_calling (self .model ):
65
+ supports_functions = True
64
66
else :
65
- # Guess whether or not it's a function calling LLM
66
- if not self .interpreter .offline and (
67
- self .interpreter .llm .model != "gpt-4-vision-preview"
68
- and self .model in litellm .open_ai_chat_completion_models
69
- or self .model .startswith ("azure/" )
70
- # Once Litellm supports it, add Anthropic models here
71
- ):
72
- supports_functions = True
73
- else :
74
- supports_functions = False
75
-
67
+ supports_functions = False
68
+
76
69
# Trim image messages if they're there
77
70
if self .supports_vision :
78
71
image_messages = [msg for msg in messages if msg ["type" ] == "image" ]
You can’t perform that action at this time.
0 commit comments