Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion lib/ruby_llm/error.rb
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,8 @@ class << self
/token count exceeds/i,
/input[_\s-]?token/i,
/input or output tokens? must be reduced/i,
/reduce the length of messages/i
/reduce the length of messages/i,
/prompt is too long/i
].freeze

def parse_error(provider:, response:) # rubocop:disable Metrics/PerceivedComplexity
Expand Down
10 changes: 10 additions & 0 deletions spec/ruby_llm/error_middleware_spec.rb
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,16 @@
end.to raise_error(RubyLLM::ContextLengthExceededError)
end

it "maps Anthropic's 'prompt is too long' 400 error to ContextLengthExceededError" do
msg = 'prompt is too long: 209025 tokens > 200000 maximum'
response = Struct.new(:status, :body).new(400, %({"error":{"message":"#{msg}"}}))
provider = instance_double(RubyLLM::Provider, parse_error: msg)

expect do
described_class.parse_error(provider: provider, response: response)
end.to raise_error(RubyLLM::ContextLengthExceededError)
end

it 'keeps regular 400 errors as BadRequestError' do
response = Struct.new(:status, :body).new(400, '{"error":{"message":"Invalid model specified"}}')
provider = instance_double(RubyLLM::Provider, parse_error: 'Invalid model specified')
Expand Down
Loading