fixes and clean up !

pull/944/head
harshalmore31 2 days ago
parent 7ec57fe39e
commit 81a6040527

@ -429,9 +429,7 @@ class LiteLLM:
if out is False: if out is False:
raise ValueError( raise ValueError(
f"Model {self.model_name} does not support vision. " f"Model {self.model_name} does not support vision"
f"Use a vision-capable model like gpt-4-vision-preview, "
f"claude-3-sonnet, or gemini-pro-vision."
) )
def run( def run(
@ -570,7 +568,7 @@ class LiteLLM:
try: try:
messages = self._prepare_messages(task=task, img=img) messages = self._prepare_messages(task=task, img=img)
# Base completion parameters # Prepare common completion parameters
completion_params = { completion_params = {
"model": self.model_name, "model": self.model_name,
"messages": messages, "messages": messages,

Loading…
Cancel
Save