Skip to content

Commit 3f6fb83

Browse files
committed
Move Provider Concerns into folder
1 parent d11b103 commit 3f6fb83

14 files changed

+23
-31
lines changed

lib/active_agent/generation_provider/_base_provider.rb

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
1-
# lib/active_agent/generation_provider/base.rb
2-
3-
require_relative "error_handling"
4-
require_relative "parameter_builder"
1+
require_relative "concerns/error_handling"
2+
require_relative "concerns/parameter_builder"
53

64
module ActiveAgent
75
module GenerationProvider

lib/active_agent/generation_provider/anthropic_provider.rb

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,9 @@
1010
require "active_agent/action_prompt/action"
1111
require_relative "_base_provider"
1212
require_relative "response"
13-
require_relative "stream_processing"
14-
require_relative "message_formatting"
15-
require_relative "tool_management"
13+
require_relative "concerns/stream_processing"
14+
require_relative "concerns/message_formatting"
15+
require_relative "concerns/tool_management"
1616

1717
module ActiveAgent
1818
module GenerationProvider

lib/active_agent/generation_provider/error_handling.rb renamed to lib/active_agent/generation_provider/concerns/error_handling.rb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def retry_delay(attempt)
4747
def handle_generation_error(error)
4848
error_message = format_error_message(error)
4949
# Create new error with original backtrace preserved
50-
new_error = ActiveAgent::GenerationProvider::Base::GenerationProviderError.new(error_message)
50+
new_error = ActiveAgent::GenerationProvider::BaseProvider::GenerationProviderError.new(error_message)
5151
new_error.set_backtrace(error.backtrace) if error.respond_to?(:backtrace)
5252

5353
# Log detailed error if verbose mode is enabled

lib/active_agent/generation_provider/open_ai_provider.rb

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,9 @@
99
require_relative "_base_provider"
1010
require_relative "response"
1111
require_relative "responses_adapter"
12-
require_relative "stream_processing"
13-
require_relative "message_formatting"
14-
require_relative "tool_management"
12+
require_relative "concerns/stream_processing"
13+
require_relative "concerns/message_formatting"
14+
require_relative "concerns/tool_management"
1515

1616
module ActiveAgent
1717
module GenerationProvider

test/generation_provider/error_handling_test.rb renamed to test/generation_provider/concerns/error_handling_test.rb

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,7 @@
22

33
require "test_helper"
44
require "active_support/rescuable"
5-
require "active_agent/generation_provider/base"
6-
require "active_agent/generation_provider/error_handling"
5+
require "active_agent/generation_provider/_base_provider"
76

87
class ErrorHandlingTest < ActiveSupport::TestCase
98
class TestError < StandardError; end
@@ -42,7 +41,7 @@ def operation_that_succeeds_on_retry
4241

4342
test "with_error_handling uses rescue_with_handler" do
4443
# The default rescue_from StandardError handler will convert to GenerationProviderError
45-
error = assert_raises(ActiveAgent::GenerationProvider::Base::GenerationProviderError) do
44+
error = assert_raises(ActiveAgent::GenerationProvider::BaseProvider::GenerationProviderError) do
4645
@provider.with_error_handling do
4746
@provider.operation_that_fails
4847
end
@@ -53,7 +52,7 @@ def operation_that_succeeds_on_retry
5352

5453
test "preserves original backtrace" do
5554
original_line = nil
56-
error = assert_raises(ActiveAgent::GenerationProvider::Base::GenerationProviderError) do
55+
error = assert_raises(ActiveAgent::GenerationProvider::BaseProvider::GenerationProviderError) do
5756
@provider.with_error_handling do
5857
original_line = __LINE__ + 1
5958
raise TestError, "With backtrace"
@@ -79,7 +78,7 @@ def operation_that_succeeds_on_retry
7978
@provider.class.retry_on_errors = [ TestError ]
8079
@provider.class.max_retries = 2
8180

82-
assert_raises(ActiveAgent::GenerationProvider::Base::GenerationProviderError) do
81+
assert_raises(ActiveAgent::GenerationProvider::BaseProvider::GenerationProviderError) do
8382
@provider.with_error_handling do
8483
@provider.operation_that_fails
8584
end
@@ -243,7 +242,7 @@ class VerboseProvider < TestProvider
243242
@provider.class.retry_on_errors = []
244243
@provider.class.max_retries = 3
245244

246-
assert_raises(ActiveAgent::GenerationProvider::Base::GenerationProviderError) do
245+
assert_raises(ActiveAgent::GenerationProvider::BaseProvider::GenerationProviderError) do
247246
@provider.with_error_handling do
248247
@provider.operation_that_fails
249248
end
@@ -293,7 +292,7 @@ class VerboseProvider < TestProvider
293292
@provider.config = { "logger" => logger }
294293

295294
error = TestError.new("Original error")
296-
assert_raises(ActiveAgent::GenerationProvider::Base::GenerationProviderError) do
295+
assert_raises(ActiveAgent::GenerationProvider::BaseProvider::GenerationProviderError) do
297296
@provider.send(:handle_generation_error, error)
298297
end
299298

@@ -304,7 +303,7 @@ class VerboseProvider < TestProvider
304303
# This test would require mocking ActiveSupport::Notifications
305304
# For now, just ensure the method doesn't error
306305
error = TestError.new("Test")
307-
wrapped = ActiveAgent::GenerationProvider::Base::GenerationProviderError.new("Wrapped")
306+
wrapped = ActiveAgent::GenerationProvider::BaseProvider::GenerationProviderError.new("Wrapped")
308307

309308
assert_nothing_raised do
310309
@provider.send(:instrument_error, error, wrapped)

test/generation_provider/message_formatting_test.rb renamed to test/generation_provider/concerns/message_formatting_test.rb

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,7 @@
22

33
require "test_helper"
44
require "active_agent/action_prompt/message"
5-
require "active_agent/action_prompt/action"
6-
require "active_agent/generation_provider/message_formatting"
5+
require "active_agent/generation_provider/_base_provider"
76

87
class MessageFormattingTest < ActiveSupport::TestCase
98
class TestProvider

0 commit comments

Comments
 (0)