Skip to content

Commit d8f04f1

Browse files
committed
How about just claude-3 as the model name since OCR is the default now
1 parent 359a46c commit d8f04f1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

operate/models/apis.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ async def get_next_action(model, messages, objective, session_id):
5555
if model == "llava":
5656
operation = call_ollama_llava(messages)
5757
return operation, None
58-
if model == "claude-3-with-ocr":
58+
if model == "claude-3":
5959
operation = await call_claude_3_with_ocr(messages, objective, model)
6060
return operation, None
6161
raise ModelNotRecognizedException(model)
@@ -716,6 +716,7 @@ async def call_claude_3_with_ocr(messages, objective, model):
716716

717717
return gpt_4_fallback(gpt4_messages, objective, model)
718718

719+
719720
def get_last_assistant_message(messages):
720721
"""
721722
Retrieve the last message from the assistant in the messages array.

0 commit comments

Comments
 (0)