observagent 0.1.0__py3-none-any.whl

observagent/__init__.py

@@ -0,0 +1,4 @@
+# observagent_sdk/src/observagent/__init__.py
+from .core import Observagent
+
+__all__ = ["Observagent"]

observagent/core.py

@@ -0,0 +1,92 @@
+# observagent_sdk/src/observagent/client.py
+import os
+from contextlib import contextmanager
+from opentelemetry import trace, metrics
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+
+# Added imports for Metrics
+from opentelemetry.sdk.metrics import MeterProvider
+from opentelemetry.sdk.metrics.export import PeriodicExportingMetricReader
+from opentelemetry.exporter.otlp.proto.http.metric_exporter import OTLPMetricExporter
+
+from opentelemetry.sdk.resources import Resource
+from opentelemetry.instrumentation.google_genai import GoogleGenAiSdkInstrumentor
+
+class Observagent:
+    _trace_provider = None
+    _tracer = None
+    _metric_provider = None
+    _meter = None
+
+    @staticmethod
+    def init(service_name: str, endpoint: str = "http://34.186.209.74:30080"):
+        print(f"👁️  Initializing Observagent for service: '{service_name}'...")
+        
+        os.environ["OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT"] = "true"
+        
+        resource = Resource.create({"service.name": service_name})
+        
+        # ------------------------------------------------------------
+        # 1. Tracing Setup
+        # ------------------------------------------------------------
+        Observagent._trace_provider = TracerProvider(resource=resource)
+        trace_exporter = OTLPSpanExporter(f"{endpoint}/v1/traces")
+        span_processor = BatchSpanProcessor(trace_exporter)
+        Observagent._trace_provider.add_span_processor(span_processor)
+        trace.set_tracer_provider(Observagent._trace_provider)
+        Observagent._tracer = trace.get_tracer("observagent.core")
+
+        # ------------------------------------------------------------
+        # 2. Metrics Setup
+        # ------------------------------------------------------------
+        metric_exporter = OTLPMetricExporter(endpoint=f"{endpoint}/v1/metrics")
+        # PeriodicExportingMetricReader batches and sends metrics at regular intervals
+        reader = PeriodicExportingMetricReader(metric_exporter)
+        
+        Observagent._metric_provider = MeterProvider(
+            resource=resource, 
+            metric_readers=[reader]
+        )
+        metrics.set_meter_provider(Observagent._metric_provider)
+        Observagent._meter = metrics.get_meter("observagent.core")
+        
+        # ------------------------------------------------------------
+        # 3. Instrumentation
+        # ------------------------------------------------------------
+        # This instrumentor will now automatically pick up both the tracer and meter providers
+        GoogleGenAiSdkInstrumentor().instrument()
+        
+        print("🚀 Observagent is active and monitoring AI interactions (Traces & Metrics).")
+
+    @staticmethod
+    @contextmanager
+    def track_session(conversation_id: str, user_id: str = "anonymous"):
+        """
+        A context manager wrapper that links a conversation flow together
+        and attaches specific user metadata context to the execution.
+        """
+        if not Observagent._tracer:
+            yield
+            return
+
+        # Start an explicit orchestration span
+        with Observagent._tracer.start_as_current_span("observagent.chat_session") as span:
+            # Inject official OpenTelemetry GenAI Semantic Convention tags
+            span.set_attribute("gen_ai.conversation.id", conversation_id)
+            span.set_attribute("enduser.id", user_id)
+            try:
+                yield span
+            except Exception as e:
+                span.record_exception(e)
+                span.set_status(trace.StatusCode.ERROR, str(e))
+                raise
+
+    @staticmethod
+    def shutdown():
+        # Clean shutdown for both providers to ensure remaining data is flushed
+        if Observagent._trace_provider:
+            Observagent._trace_provider.shutdown()
+        if Observagent._metric_provider:
+            Observagent._metric_provider.shutdown()

observagent-0.1.0.dist-info/METADATA

@@ -0,0 +1,29 @@
+Metadata-Version: 2.4
+Name: observagent
+Version: 0.1.0
+Summary: An OpenTelemetry-based SDK for monitoring GenAI applications.
+Author-email: Andre <your.email@example.com>
+License: MIT
+Project-URL: Homepage, https://github.com/yourusername/observagent
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: opentelemetry-api
+Requires-Dist: opentelemetry-sdk
+Requires-Dist: opentelemetry-exporter-otlp-proto-http
+Requires-Dist: opentelemetry-instrumentation-google-genai
+Dynamic: license-file
+
+# Observagent SDK
+
+An OpenTelemetry-based observability SDK for automatically capturing Traces and Metrics from Agents 
+
+## Installation
+
+Install the SDK directly from PyPI
+
+```bash
+pip install observagent

observagent-0.1.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+observagent/__init__.py,sha256=WVb7RVzQ64mv-4hqrMK4mQz-qcPKH8C_T_8jVkyClLI,102
+observagent/core.py,sha256=X7gKFfBif7IMm41VJ_50B3HG9HnjlDFr_J6RykgH0UQ,4050
+observagent-0.1.0.dist-info/licenses/LICENSE,sha256=zK1or5JeP0aAEBQU-Mcb1MfQtkcZGHLol1L71FJswMo,1061
+observagent-0.1.0.dist-info/METADATA,sha256=GVahGw4WX11kCjkxtIRHDVkiCLy1RK4bhy_FmvawEHg,889
+observagent-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+observagent-0.1.0.dist-info/top_level.txt,sha256=LYpvNra-bWofIxK0f6pV0cPOASeXcvTaIo2YQ5_iH40,12
+observagent-0.1.0.dist-info/RECORD,,

observagent-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

observagent-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Andre
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

observagent-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+observagent