Merge branch 'main' into jb/aic-1628/make-defaults-optional

Author: jsonbailey

.release-please-manifest.json

@@ -1,5 +1,5 @@
 {
-  "packages/sdk/server-ai": "0.14.2",
+  "packages/sdk/server-ai": "0.15.0",
   "packages/ai-providers/server-ai-langchain": "0.3.1",
   "packages/ai-providers/server-ai-openai": "0.1.1"
 }

packages/sdk/server-ai/CHANGELOG.md

@@ -2,6 +2,13 @@
 
 All notable changes to the LaunchDarkly Python AI package will be documented in this file. This project adheres to [Semantic Versioning](http://semver.org).
 
+## [0.15.0](https://github.com/launchdarkly/python-server-sdk-ai/compare/launchdarkly-server-sdk-ai-0.14.2...launchdarkly-server-sdk-ai-0.15.0) (2026-03-02)
+
+
+### Features
+
+* add agent graph tracker ([#89](https://github.com/launchdarkly/python-server-sdk-ai/issues/89)) ([9d371fa](https://github.com/launchdarkly/python-server-sdk-ai/commit/9d371faa43c3028062e32f44f8693ede685cb7c6))
+
 ## [0.14.2](https://github.com/launchdarkly/python-server-sdk-ai/compare/launchdarkly-server-sdk-ai-0.14.1...launchdarkly-server-sdk-ai-0.14.2) (2026-02-25)
 
 

packages/sdk/server-ai/PROVENANCE.md

@@ -10,7 +10,7 @@ To verify SLSA provenance attestations, we recommend using [slsa-verifier](https
 
 ```
 # Set the version of the library to verify
-VERSION=0.14.2
+VERSION=0.15.0
 ```
 
 <!-- x-release-please-end -->

packages/sdk/server-ai/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "launchdarkly-server-sdk-ai"
-version = "0.14.2"
+version = "0.15.0"
 description = "LaunchDarkly SDK for AI"
 authors = ["LaunchDarkly <dev@launchdarkly.com>"]
 license = "Apache-2.0"

packages/sdk/server-ai/src/ldai/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.14.2"  # x-release-please-version
+__version__ = "0.15.0"  # x-release-please-version
 
 from ldclient import log
 
@@ -13,6 +13,7 @@
     Edge, JudgeConfiguration, LDAIAgent, LDAIAgentConfig, LDAIAgentDefaults,
     LDMessage, ModelConfig, ProviderConfig)
 from ldai.providers.types import EvalScore, JudgeResponse
+from ldai.tracker import AIGraphTracker
 
 __all__ = [
     'LDAIClient',
@@ -21,6 +22,7 @@
     'AIAgentConfigRequest',
     'AIAgents',
     'AIAgentGraphConfig',
+    'AIGraphTracker',
     'Edge',
     'AICompletionConfig',
     'AICompletionConfigDefault',

packages/sdk/server-ai/src/ldai/agent_graph/__init__.py

@@ -1,11 +1,11 @@
 """Graph implementation for managing AI agent graphs."""
 
-from dataclasses import dataclass
 from typing import Any, Callable, Dict, List, Optional, Set
 
 from ldclient import Context
 
 from ldai.models import AIAgentConfig, AIAgentGraphConfig, Edge
+from ldai.tracker import AIGraphTracker
 
 DEFAULT_FALSE = AIAgentConfig(key="", enabled=False)
 
@@ -54,11 +54,21 @@ def __init__(
         nodes: Dict[str, AgentGraphNode],
         context: Context,
         enabled: bool,
+        tracker: Optional[AIGraphTracker] = None,
     ):
         self._agent_graph = agent_graph
         self._context = context
         self._nodes = nodes
         self.enabled = enabled
+        self._tracker = tracker
+
+    def get_tracker(self) -> Optional[AIGraphTracker]:
+        """
+        Get the graph tracker for this graph definition.
+
+        :return: The AIGraphTracker instance, or None if not available.
+        """
+        return self._tracker
 
     def is_enabled(self) -> bool:
         """Check if the graph is enabled."""

packages/sdk/server-ai/src/ldai/client.py

@@ -16,7 +16,7 @@
                          ProviderConfig)
 from ldai.providers.ai_provider_factory import AIProviderFactory
 from ldai.sdk_info import AI_SDK_LANGUAGE, AI_SDK_NAME, AI_SDK_VERSION
-from ldai.tracker import LDAIConfigTracker
+from ldai.tracker import AIGraphTracker, LDAIConfigTracker
 
 _TRACK_SDK_INFO = '$ld:ai:sdk:info'
 _TRACK_USAGE_COMPLETION_CONFIG = '$ld:ai:usage:completion-config'
@@ -485,6 +485,19 @@ def agent_graph(
         """
         variation = self._client.variation(key, context, {})
 
+        # Extract variation metadata for tracker
+        variation_key = variation.get("_ldMeta", {}).get("variationKey", "")
+        version = int(variation.get("_ldMeta", {}).get("version", 1))
+
+        # Create graph tracker
+        tracker = AIGraphTracker(
+            self._client,
+            variation_key,
+            key,
+            version,
+            context,
+        )
+
         if not variation.get("root"):
             log.debug(f"Agent graph {key} is disabled, no root config key found")
             return AgentGraphDefinition(
@@ -497,6 +510,7 @@ def agent_graph(
                 nodes={},
                 context=context,
                 enabled=False,
+                tracker=tracker,
             )
 
         edge_keys = list[str](variation.get("edges", {}).keys())
@@ -524,6 +538,7 @@ def agent_graph(
                 nodes={},
                 context=context,
                 enabled=False,
+                tracker=tracker,
             )
 
         try:
@@ -554,6 +569,7 @@ def agent_graph(
                 nodes={},
                 context=context,
                 enabled=False,
+                tracker=tracker,
             )
 
         nodes = AgentGraphDefinition.build_nodes(
@@ -566,6 +582,7 @@ def agent_graph(
             nodes=nodes,
             context=context,
             enabled=agent_graph_config.enabled,
+            tracker=tracker,
         )
 
     def agents(