Add OpenTelemetry instrumentation for LiteLLM

.github/workflows/loongsuite_lint_0.yml

@@ -146,3 +146,22 @@ jobs:
       - name: Run tests
         run: tox -c tox-loongsuite.ini -e lint-util-genai
 
+  lint-loongsuite-instrumentation-litellm:
+    name: LoongSuite loongsuite-instrumentation-litellm
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
+    steps:
+      - name: Checkout repo @ SHA - ${{ github.sha }}
+        uses: actions/checkout@v4
+
+      - name: Set up Python 3.13
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.13"
+
+      - name: Install tox
+        run: pip install tox-uv
+
+      - name: Run tests
+        run: tox -c tox-loongsuite.ini -e lint-loongsuite-instrumentation-litellm
+

.github/workflows/loongsuite_test_0.yml

@@ -1001,3 +1001,79 @@ jobs:
       - name: Run tests
         run: tox -c tox-loongsuite.ini -e pypy3-test-util-genai -- -ra
 
+  py310-test-loongsuite-instrumentation-litellm_ubuntu-latest:
+    name: LoongSuite loongsuite-instrumentation-litellm 3.10 Ubuntu
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
+    steps:
+      - name: Checkout repo @ SHA - ${{ github.sha }}
+        uses: actions/checkout@v4
+
+      - name: Set up Python 3.10
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.10"
+
+      - name: Install tox
+        run: pip install tox-uv
+
+      - name: Run tests
+        run: tox -c tox-loongsuite.ini -e py310-test-loongsuite-instrumentation-litellm -- -ra
+
+  py311-test-loongsuite-instrumentation-litellm_ubuntu-latest:
+    name: LoongSuite loongsuite-instrumentation-litellm 3.11 Ubuntu
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
+    steps:
+      - name: Checkout repo @ SHA - ${{ github.sha }}
+        uses: actions/checkout@v4
+
+      - name: Set up Python 3.11
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+
+      - name: Install tox
+        run: pip install tox-uv
+
+      - name: Run tests
+        run: tox -c tox-loongsuite.ini -e py311-test-loongsuite-instrumentation-litellm -- -ra
+
+  py312-test-loongsuite-instrumentation-litellm_ubuntu-latest:
+    name: LoongSuite loongsuite-instrumentation-litellm 3.12 Ubuntu
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
+    steps:
+      - name: Checkout repo @ SHA - ${{ github.sha }}
+        uses: actions/checkout@v4
+
+      - name: Set up Python 3.12
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+
+      - name: Install tox
+        run: pip install tox-uv
+
+      - name: Run tests
+        run: tox -c tox-loongsuite.ini -e py312-test-loongsuite-instrumentation-litellm -- -ra
+
+  py313-test-loongsuite-instrumentation-litellm_ubuntu-latest:
+    name: LoongSuite loongsuite-instrumentation-litellm 3.13 Ubuntu
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
+    steps:
+      - name: Checkout repo @ SHA - ${{ github.sha }}
+        uses: actions/checkout@v4
+
+      - name: Set up Python 3.13
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.13"
+
+      - name: Install tox
+        run: pip install tox-uv
+
+      - name: Run tests
+        run: tox -c tox-loongsuite.ini -e py313-test-loongsuite-instrumentation-litellm -- -ra
+

instrumentation-loongsuite/loongsuite-instrumentation-litellm/CHANGELOG.md

@@ -0,0 +1,13 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## Unreleased
+
+### Added
+
+- Initialize the instrumentation for Litellm
+  ([#88](https://github.com/alibaba/loongsuite-python-agent/pull/88))

instrumentation-loongsuite/loongsuite-instrumentation-litellm/README.rst

@@ -0,0 +1,68 @@
+LoongSuite LiteLLM Instrumentation
+======================================
+
+|pypi|
+
+.. |pypi| image:: https://badge.fury.io/py/opentelemetry-instrumentation-litellm.svg
+   :target: https://pypi.org/project/opentelemetry-instrumentation-litellm/
+
+This library provides automatic instrumentation for the
+`LiteLLM <https://github.com/BerriAI/litellm>`_ library, which provides
+a unified interface to 100+ LLM providers.
+
+Installation
+------------
+
+::
+
+     git clone https://github.com/alibaba/loongsuite-python-agent.git
+     cd loongsuite-python-agent
+     pip install ./instrumentation-loongsuite/loongsuite-instrumentation-litellm
+
+Configuration
+-------------
+
+The instrumentation can be enabled/disabled using environment variables:
+
+* ``ENABLE_LITELLM_INSTRUMENTOR``: Enable/disable instrumentation (default: true)
+
+Usage
+-----
+
+.. code:: python
+
+    from opentelemetry.instrumentation.litellm import LiteLLMInstrumentor
+    import litellm
+
+    # Instrument LiteLLM
+    LiteLLMInstrumentor().instrument()
+
+    # Use LiteLLM as normal
+    response = litellm.completion(
+        model="gpt-3.5-turbo",
+        messages=[{"role": "user", "content": "Hello!"}]
+    )
+
+Features
+--------
+
+This instrumentation automatically captures:
+
+* LLM completion calls (sync and async)
+* Streaming completions
+* Embedding calls
+* Retry mechanisms
+* Tool/function calls
+* Request and response metadata
+* Token usage
+* Model information
+
+The instrumentation follows OpenTelemetry semantic conventions for GenAI operations.
+
+References
+----------
+
+* `OpenTelemetry LiteLLM Instrumentation <https://opentelemetry-python-contrib.readthedocs.io/en/latest/instrumentation/litellm/litellm.html>`_
+* `OpenTelemetry Project <https://opentelemetry.io/>`_
+* `LiteLLM Documentation <https://docs.litellm.ai/>`_
+

instrumentation-loongsuite/loongsuite-instrumentation-litellm/pyproject.toml

@@ -0,0 +1,57 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "loongsuite-instrumentation-litellm"
+dynamic = ["version"]
+description = "LoongSuite LiteLLM instrumentation"
+readme = "README.rst"
+license = "Apache-2.0"
+requires-python = ">=3.10"
+authors = [
+  { name = "OpenTelemetry Authors", email = "cncf-opentelemetry-contributors@lists.cncf.io" },
+]
+classifiers = [
+  "Development Status :: 4 - Beta",
+  "Intended Audience :: Developers",
+  "License :: OSI Approved :: Apache Software License",
+  "Programming Language :: Python",
+  "Programming Language :: Python :: 3",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+]
+dependencies = [
+  "opentelemetry-api >= 1.37.0",
+  "opentelemetry-instrumentation >= 0.58b0",
+  "opentelemetry-semantic-conventions >= 0.60b0",
+  "wrapt >= 1.0.0, < 2.0.0",
+  "opentelemetry-util-genai",
+]
+
+[project.optional-dependencies]
+instruments = [
+  "litellm >= 1.0.0",
+]
+
+[project.entry-points.opentelemetry_instrumentor]
+litellm = "opentelemetry.instrumentation.litellm:LiteLLMInstrumentor"
+
+[project.urls]
+Homepage = "https://github.com/open-telemetry/opentelemetry-python-contrib/tree/main/instrumentation/opentelemetry-instrumentation-litellm"
+Repository = "https://github.com/open-telemetry/opentelemetry-python-contrib"
+
+[tool.hatch.version]
+path = "src/opentelemetry/instrumentation/litellm/version.py"
+
+[tool.hatch.build.targets.sdist]
+include = [
+  "/src",
+  "/tests",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/opentelemetry"]
+