Fix/telemetry

README.md

@@ -137,9 +137,9 @@ puts result.text  # Generated text response
 The gem supports OpenTelemetry integration for monitoring and observability. You can configure telemetry settings to control what data is recorded and add metadata for better tracing:
 
 **Telemetry Options:**
-- `is_enabled`: Enable/disable telemetry (default: false)
-- `record_inputs`: Record input messages (default: true, disable for sensitive data)
-- `record_outputs`: Record output responses (default: true, disable for sensitive data)
+- `enabled`: Enable/disable telemetry (default: true)
+- `record_inputs`: Record input messages (default: false, disable for sensitive data)
+- `record_outputs`: Record output responses (default: false, disable for sensitive data)
 - `function_id`: Identifier for grouping telemetry data by function
 - `metadata`: Additional metadata for OpenTelemetry traces (agent identification, service info, etc.)
 
@@ -157,7 +157,8 @@ telemetry_settings = Ai::TelemetrySettings.new(
     'service.namespace' => 'customer-service',
     'cx.application.name' => 'ai-tracing',
     'cx.subsystem.name' => 'mastra-agents'
-  }
+  },
+  tracer:nil
 )
 
 # Use with text generation

lib/ai/clients/mastra.rb

@@ -8,6 +8,10 @@ module Ai
   module Clients
     class Mastra < Ai::Client
       extend T::Sig
+      RUBY_TO_API_KEY_MAPPING = T.let({
+        'enabled' => 'is_enabled'
+      }.freeze, 
+T::Hash[String, String])
 
       sig { params(endpoint: String).void }
       def initialize(endpoint)
@@ -45,6 +49,11 @@ def generate(agent_name, messages:, options: {})
 
       private
 
+      sig { params(key: String).returns(String) }
+      def map_ruby_key_to_api(key)
+        RUBY_TO_API_KEY_MAPPING.fetch(key, key)
+      end
+
       sig do
         params(
           url: URI::Generic,
@@ -61,7 +70,11 @@ def response(url:, messages:, options:)
         request['Origin'] = Ai.config.origin
 
         # convert to camelCase and unpacking for API compatibility
-        camelized_options = options.deep_transform_keys { |key| key.to_s.camelize(:lower).to_sym }
+        camelized_options = options.deep_transform_keys do |key|
+          mapped_key = map_ruby_key_to_api(key.to_s)
+          mapped_key.camelize(:lower).to_sym
+        end
+
         request.body = { messages: messages, **camelized_options }.to_json
 
         response = http.request(request)

lib/ai/types/telemetry_settings.rb

@@ -3,7 +3,7 @@
 module Ai
   class TelemetrySettings < T::Struct
     # Enable or disable telemetry. Enabled by default.
-    const :is_enabled, T::Boolean, default: true
+    const :enabled, T::Boolean, default: true
 
     # Enable or disable input recording. You might want to disable this to avoid 
     # recording sensitive information, reduce data transfers, or increase performance.