traceable-agent 2.0.0__tar.gz → 2.0.1__tar.gz

{traceable-agent-2.0.0 → traceable-agent-2.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: traceable-agent
-Version: 2.0.0
+Version: 2.0.1
 Summary: Traceable.ai Python Agent
 Home-page: https://traceable.ai
 Author: Traceable.ai

{traceable-agent-2.0.0 → traceable-agent-2.0.1}/setup.py

@@ -30,30 +30,30 @@ setup(
         )
     ],
     install_requires=[
-        "opentelemetry-api==1.26.0",
-        "opentelemetry-exporter-otlp==1.26.0",
-        "opentelemetry-instrumentation==0.47b0",
-        "opentelemetry-instrumentation-aiohttp-client==0.47b0",
-        "opentelemetry-instrumentation-boto==0.47b0",
-        "opentelemetry-instrumentation-botocore==0.47b0",
-        "opentelemetry-instrumentation-wsgi==0.47b0",
-        "opentelemetry-instrumentation-fastapi==0.47b0",
-        "opentelemetry-instrumentation-flask==0.47b0",
-        "opentelemetry-instrumentation-mysql==0.47b0",
-        "opentelemetry-instrumentation-psycopg2==0.47b0",
-        "opentelemetry-instrumentation-requests==0.47b0",
-        "opentelemetry-instrumentation-grpc==0.47b0",
-        "opentelemetry-instrumentation-django==0.47b0",
-        "opentelemetry-instrumentation-aws-lambda==0.47b0",
-        "opentelemetry-propagator-b3==1.26.0",
-        "opentelemetry-proto==1.26.0",
-        "opentelemetry-sdk==1.26.0",
-        "opentelemetry-util-http==0.47b0",
+        "opentelemetry-api==1.30.0",
+        "opentelemetry-exporter-otlp==1.30.0",
+        "opentelemetry-instrumentation==0.51b0",
+        "opentelemetry-instrumentation-aiohttp-client==0.51b0",
+        "opentelemetry-instrumentation-boto==0.51b0",
+        "opentelemetry-instrumentation-botocore==0.51b0",
+        "opentelemetry-instrumentation-wsgi==0.51b0",
+        "opentelemetry-instrumentation-fastapi==0.51b0",
+        "opentelemetry-instrumentation-flask==0.51b0",
+        "opentelemetry-instrumentation-mysql==0.51b0",
+        "opentelemetry-instrumentation-psycopg2==0.51b0",
+        "opentelemetry-instrumentation-requests==0.51b0",
+        "opentelemetry-instrumentation-grpc==0.51b0",
+        "opentelemetry-instrumentation-django==0.51b0",
+        "opentelemetry-instrumentation-aws-lambda==0.51b0",
+        "opentelemetry-propagator-b3==1.30.0",
+        "opentelemetry-proto==1.30.0",
+        "opentelemetry-sdk==1.30.0",
+        "opentelemetry-util-http==0.51b0",
         "google>=3.0.0",
         "pyyaml",
-        "protobuf<5",
+        "protobuf",
         "psutil",
-        "distro==1.6.0",
+        "distro",
         "cffi",
         "setuptools",
         "jaraco.text",

{traceable-agent-2.0.0 → traceable-agent-2.0.1}/src/traceable_agent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: traceable-agent
-Version: 2.0.0
+Version: 2.0.1
 Summary: Traceable.ai Python Agent
 Home-page: https://traceable.ai
 Author: Traceable.ai

traceable-agent-2.0.1/src/traceable_agent.egg-info/requires.txt

@@ -0,0 +1,28 @@
+opentelemetry-api==1.30.0
+opentelemetry-exporter-otlp==1.30.0
+opentelemetry-instrumentation==0.51b0
+opentelemetry-instrumentation-aiohttp-client==0.51b0
+opentelemetry-instrumentation-boto==0.51b0
+opentelemetry-instrumentation-botocore==0.51b0
+opentelemetry-instrumentation-wsgi==0.51b0
+opentelemetry-instrumentation-fastapi==0.51b0
+opentelemetry-instrumentation-flask==0.51b0
+opentelemetry-instrumentation-mysql==0.51b0
+opentelemetry-instrumentation-psycopg2==0.51b0
+opentelemetry-instrumentation-requests==0.51b0
+opentelemetry-instrumentation-grpc==0.51b0
+opentelemetry-instrumentation-django==0.51b0
+opentelemetry-instrumentation-aws-lambda==0.51b0
+opentelemetry-propagator-b3==1.30.0
+opentelemetry-proto==1.30.0
+opentelemetry-sdk==1.30.0
+opentelemetry-util-http==0.51b0
+google>=3.0.0
+pyyaml
+protobuf
+psutil
+distro
+cffi
+setuptools
+jaraco.text
+platformdirs

{traceable-agent-2.0.0 → traceable-agent-2.0.1}/src/traceableai/agent_init.py

@@ -110,19 +110,10 @@ class AgentInit:  # pylint: disable=R0902,R0903
         composite_propagators = CompositePropagator(propagator_list)
         set_global_textmap(composite_propagators)
 
-    def _set_wrapper_fields(self, wrapper):
-        data_cap = self._config.config.data_capture
-        wrapper.set_process_request_headers(data_cap.http_headers.request)
-        wrapper.set_process_request_body(data_cap.http_body.request)
-        wrapper.set_process_response_headers(data_cap.http_headers.response)
-        wrapper.set_process_response_body(data_cap.http_body.response)
-        wrapper.set_body_max_size(data_cap.body_max_size_bytes)
-
     def init_library_instrumentation(self, instrumentation_name, wrapper_instance):
         """used to configure instrumentation wrapper settings + apply instrumentation"""
         logger.debug("Attempting to initialize %s instrumentation", instrumentation_name)
         try:
-            self._set_wrapper_fields(wrapper_instance)
             wrapper_instance.instrument()
         except Exception as err: # pylint: disable=W0703
             logger.debug(constants.INST_WRAP_EXCEPTION_MSSG,

{traceable-agent-2.0.0 → traceable-agent-2.0.1}/src/traceableai/instrumentation/__init__.py

@@ -3,6 +3,7 @@ import sys
 import traceback
 from opentelemetry.trace.span import Span
 
+from traceableai.config.config import Config
 from traceableai.custom_logger import get_custom_logger
 logger = get_custom_logger(__name__)
 
@@ -24,46 +25,12 @@ class BaseInstrumentorWrapper:
     def __init__(self):
         '''constructor'''
         logger.debug('Entering BaseInstrumentorWrapper constructor.')
+        self._process_request_headers = Config()._instance.config.data_capture.http_headers.request.value
+        self._process_response_headers = Config()._instance.config.data_capture.http_headers.response.value
+        self._process_request_body = Config()._instance.config.data_capture.http_body.request.value
+        self._process_response_body = Config()._instance.config.data_capture.http_body.response.value
+        self._max_body_size = Config()._instance.config.data_capture.body_max_size_bytes.value
         super().__init__()
-        self._process_request_headers = False
-        self._process_response_headers = False
-        self._process_request_body = False
-        self._process_response_body = False
-        self._max_body_size = 128 * 1024
-
-    # Set whether request headers should be put in extended span, takes a BoolValue as input
-    def set_process_request_headers(self, process_request_headers) -> None:
-        '''Should it process request headers?'''
-        logger.debug('Setting self._process_request_headers to \'%s\'',
-                     process_request_headers.value)
-        self._process_request_headers = process_request_headers.value
-
-    # Set whether response headers should be put in extended span, takes a BoolValue as input
-    def set_process_response_headers(self, process_response_headers) -> None:
-        '''Should it process response headers?'''
-        logger.debug('Setting self._process_response_headers to \'%s\'',
-                     process_response_headers.value)
-        self._process_response_headers = process_response_headers.value
-
-    # Set whether request body should be put in extended span, takes a BoolValue as input
-    def set_process_request_body(self, process_request_body) -> None:
-        '''should it process request body?'''
-        logger.debug('Setting self._process_request_body to \'%s\'',
-                     process_request_body.value)
-        self._process_request_body = process_request_body.value
-
-    # Set whether response body should be put in extended span, takes a BoolValue as input
-    def set_process_response_body(self, process_response_body) -> None:
-        '''should it process response body?'''
-        logger.debug('Setting self._process_response_body to \'%s\'',
-                     process_response_body.value)
-        self._process_response_body = process_response_body.value
-
-    # Set max body size
-    def set_body_max_size(self, max_body_size) -> None:
-        '''Set the max body size for this instrumentor.'''
-        logger.debug('Setting self.body_max_size to %s.', max_body_size.value)
-        self._max_body_size = max_body_size.value
 
     # we need the headers lowercased multiple times
     # just do it once upfront

{traceable-agent-2.0.0 → traceable-agent-2.0.1}/src/traceableai/instrumentation/aws_lambda/__init__.py

@@ -29,7 +29,8 @@ class AwsLambdaInstrumentorWrapper(AwsLambdaInstrumentor, BaseInstrumentorWrappe
     def __init__(self):
         '''Constructor'''
         logger.debug('Entering AwsLambdaInstrumentorWrapper.__init__().')
-        super().__init__()
+        AwsLambdaInstrumentor.__init__(self)
+        BaseInstrumentorWrapper.__init__(self)
 
     def uninstrument(self, **kwargs):
         pass

{traceable-agent-2.0.0 → traceable-agent-2.0.1}/src/traceableai/instrumentation/flask/__init__.py

@@ -117,7 +117,8 @@ class FlaskInstrumentorWrapper(FlaskInstrumentor, BaseInstrumentorWrapper):
 
     def __init__(self):
         logger.debug('Entering FlaskInstrumentorWrapper constructor.')
-        super().__init__()
+        FlaskInstrumentor.__init__(self)
+        BaseInstrumentorWrapper.__init__(self)
         self._app = None
 
     def with_app(self, app=None):

{traceable-agent-2.0.0 → traceable-agent-2.0.1}/src/traceableai/instrumentation/grpc/__init__.py

@@ -29,7 +29,8 @@ class GrpcInstrumentorServerWrapper(GrpcInstrumentorServer, BaseInstrumentorWrap
     def __init__(self):
         '''constructor'''
         logger.debug('Entering GrpcInstrumentorServerWrapper constructor.')
-        super().__init__()
+        GrpcInstrumentorServer.__init__(self)
+        BaseInstrumentorWrapper.__init__(self)
         self._original_wrapper_func = None
 
     # Enable wrapper instrumentation

{traceable-agent-2.0.0 → traceable-agent-2.0.1}/src/traceableai/instrumentation/requests/__init__.py

@@ -11,7 +11,8 @@ class RequestsInstrumentorWrapper(RequestsInstrumentor, BaseInstrumentorWrapper)
     # Constructor
     def __init__(self):
         logger.debug('Entering RequestsInstrumentorWrapper.__init__().')
-        super().__init__()
+        RequestsInstrumentor.__init__(self)
+        BaseInstrumentorWrapper.__init__(self)
 
     def _instrument(self, **kwargs) -> None:
         '''internal enable instrumentation'''

{traceable-agent-2.0.0 → traceable-agent-2.0.1}/src/traceableai/version.py

@@ -1,4 +1,4 @@
 # Code generated by ./release.sh. DO NOT EDIT.
 
 """Package version"""
-__version__ = "2.0.0"
+__version__ = "2.0.1"

traceable-agent-2.0.0/src/traceable_agent.egg-info/requires.txt

@@ -1,28 +0,0 @@
-opentelemetry-api==1.26.0
-opentelemetry-exporter-otlp==1.26.0
-opentelemetry-instrumentation==0.47b0
-opentelemetry-instrumentation-aiohttp-client==0.47b0
-opentelemetry-instrumentation-boto==0.47b0
-opentelemetry-instrumentation-botocore==0.47b0
-opentelemetry-instrumentation-wsgi==0.47b0
-opentelemetry-instrumentation-fastapi==0.47b0
-opentelemetry-instrumentation-flask==0.47b0
-opentelemetry-instrumentation-mysql==0.47b0
-opentelemetry-instrumentation-psycopg2==0.47b0
-opentelemetry-instrumentation-requests==0.47b0
-opentelemetry-instrumentation-grpc==0.47b0
-opentelemetry-instrumentation-django==0.47b0
-opentelemetry-instrumentation-aws-lambda==0.47b0
-opentelemetry-propagator-b3==1.26.0
-opentelemetry-proto==1.26.0
-opentelemetry-sdk==1.26.0
-opentelemetry-util-http==0.47b0
-google>=3.0.0
-pyyaml
-protobuf<5
-psutil
-distro==1.6.0
-cffi
-setuptools
-jaraco.text
-platformdirs