Decorators support

instrumentation-genai/opentelemetry-genai-sdk/.gitignore

@@ -0,0 +1,170 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# Mac files
+.DS_Store
+
+# Environment variables
+.env
+
+# sqlite database files
+*.db
+*.db-shm
+*.db-wal
+
+# PNG files
+*.png
+
+demo/
+
+.ruff_cache
+
+.vscode/
+
+output/
+
+.terraform.lock.hcl
+.terraform/
+foo.sh
+tfplan
+tfplan.txt
+tfplan.json
+terraform_output.json
+
+
+# IntelliJ / PyCharm
+.idea
+
+
+*.txt
+
+.dockerconfigjson
+
+app/src/bedrock_agent/deploy

instrumentation-genai/opentelemetry-genai-sdk/pyproject.toml

@@ -25,11 +25,11 @@ classifiers = [
   "Programming Language :: Python :: 3.13",
 ]
 dependencies = [
-  "opentelemetry-api ~= 1.30",
   "opentelemetry-instrumentation ~= 0.51b0",
   "opentelemetry-semantic-conventions ~= 0.51b0",
   "opentelemetry-api>=1.31.0",
   "opentelemetry-sdk>=1.31.0",
+  "pydantic-core>=2.33.2",
 ]
 
 [project.optional-dependencies]

instrumentation-genai/opentelemetry-genai-sdk/src/opentelemetry/genai/sdk/api.py

@@ -21,7 +21,7 @@
 from .exporters import SpanMetricEventExporter, SpanMetricExporter
 from .data import Message, ChatGeneration, Error
 
-from opentelemetry.instrumentation.langchain.version import __version__
+from .version import __version__
 from opentelemetry.metrics import get_meter
 from opentelemetry.trace import get_tracer
 from opentelemetry._events import get_event_logger

instrumentation-genai/opentelemetry-genai-sdk/src/opentelemetry/genai/sdk/data.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-
+from typing import Optional
 
 @dataclass
 class Message:
@@ -10,7 +10,7 @@ class Message:
 class ChatGeneration:
     content: str
     type: str
-    finish_reason: str = None
+    finish_reason: Optional[str] = None
 
 @dataclass
 class Error:

instrumentation-genai/opentelemetry-genai-sdk/src/opentelemetry/genai/sdk/decorators/__init__.py

@@ -0,0 +1,140 @@
+import inspect
+from typing import Optional, Union, TypeVar, Callable, Awaitable
+
+from typing_extensions import ParamSpec
+
+from opentelemetry.genai.sdk.decorators.base import (
+    entity_class,
+    entity_method,
+)
+from opentelemetry.genai.sdk.utils.const import (
+    ObserveSpanKindValues,
+)
+
+P = ParamSpec("P")
+R = TypeVar("R")
+F = TypeVar("F", bound=Callable[P, Union[R, Awaitable[R]]])
+
+
+def task(
+    name: Optional[str] = None,
+    description: Optional[str] = None,
+    version: Optional[int] = None,
+    protocol: Optional[str] = None,
+    method_name: Optional[str] = None,
+    tlp_span_kind: Optional[ObserveSpanKindValues] = ObserveSpanKindValues.TASK,
+) -> Callable[[F], F]:
+    def decorator(target):
+        # Check if target is a class
+        if inspect.isclass(target):
+            return entity_class(
+                name=name,
+                description=description,
+                version=version,
+                protocol=protocol,
+                method_name=method_name,
+                tlp_span_kind=tlp_span_kind,
+            )(target)
+        else:
+            # Target is a function/method
+            return entity_method(
+                name=name,
+                description=description,
+                version=version,
+                protocol=protocol,
+                tlp_span_kind=tlp_span_kind,
+            )(target)
+    return decorator
+
+
+def workflow(
+    name: Optional[str] = None,
+    description: Optional[str] = None,
+    version: Optional[int] = None,
+    protocol: Optional[str] = None,
+    method_name: Optional[str] = None,
+    tlp_span_kind: Optional[
+        Union[ObserveSpanKindValues, str]
+    ] = ObserveSpanKindValues.WORKFLOW,
+) -> Callable[[F], F]:
+    def decorator(target):
+        # Check if target is a class
+        if inspect.isclass(target):
+            return entity_class(
+                name=name,
+                description=description,
+                version=version,
+                protocol=protocol,
+                method_name=method_name,
+                tlp_span_kind=tlp_span_kind,
+            )(target)
+        else:
+            # Target is a function/method
+            return entity_method(
+                name=name,
+                description=description,
+                version=version,
+                protocol=protocol,
+                tlp_span_kind=tlp_span_kind,
+            )(target)
+
+    return decorator
+
+
+def agent(
+    name: Optional[str] = None,
+    description: Optional[str] = None,
+    version: Optional[int] = None,
+    protocol: Optional[str] = None,
+    method_name: Optional[str] = None,
+) -> Callable[[F], F]:
+    return workflow(
+        name=name,
+        description=description,
+        version=version,
+        protocol=protocol,
+        method_name=method_name,
+        tlp_span_kind=ObserveSpanKindValues.AGENT,
+    )
+
+
+def tool(
+    name: Optional[str] = None,
+    description: Optional[str] = None,
+    version: Optional[int] = None,
+    method_name: Optional[str] = None,
+) -> Callable[[F], F]:
+    return task(
+        name=name,
+        description=description,
+        version=version,
+        method_name=method_name,
+        tlp_span_kind=ObserveSpanKindValues.TOOL,
+    )
+
+
+def llm(
+    name: Optional[str] = None,
+    description: Optional[str] = None,
+    version: Optional[int] = None,
+    method_name: Optional[str] = None,
+) -> Callable[[F], F]:
+    def decorator(target):
+        # Check if target is a class
+        if inspect.isclass(target):
+            return entity_class(
+                name=name,
+                description=description,
+                version=version,
+                method_name=method_name,
+                tlp_span_kind=ObserveSpanKindValues.LLM,
+            )(target)
+        else:
+            # Target is a function/method
+            return entity_method(
+                name=name,
+                description=description,
+                version=version,
+                tlp_span_kind=ObserveSpanKindValues.LLM,
+            )(target)
+    return decorator