Skip to content

Commit f07f60e

Browse files
committed
[REL-12055] remove unnecessary check on typing
1 parent 8f30b0a commit f07f60e

1 file changed

Lines changed: 4 additions & 7 deletions

File tree

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

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,11 @@
11
"""Graph implementation for managing AI agent graphs."""
22

3-
from dataclasses import dataclass
4-
from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Set
3+
from typing import Any, Callable, Dict, List, Optional, Set
54

65
from ldclient import Context
76

87
from ldai.models import AIAgentConfig, AIAgentGraphConfig, Edge
9-
10-
if TYPE_CHECKING:
11-
from ldai.tracker import AIGraphTracker
8+
from ldai.tracker import AIGraphTracker
129

1310
DEFAULT_FALSE = AIAgentConfig(key="", enabled=False)
1411

@@ -57,15 +54,15 @@ def __init__(
5754
nodes: Dict[str, AgentGraphNode],
5855
context: Context,
5956
enabled: bool,
60-
tracker: Optional["AIGraphTracker"] = None,
57+
tracker: Optional[AIGraphTracker] = None,
6158
):
6259
self._agent_graph = agent_graph
6360
self._context = context
6461
self._nodes = nodes
6562
self.enabled = enabled
6663
self._tracker = tracker
6764

68-
def get_tracker(self) -> Optional["AIGraphTracker"]:
65+
def get_tracker(self) -> Optional[AIGraphTracker]:
6966
"""
7067
Get the graph tracker for this graph definition.
7168

0 commit comments

Comments
 (0)