use simplified Chat name

Author: jsonbailey

ldai/__init__.py

@@ -2,7 +2,7 @@
 
 # Export main client
 # Export chat
-from ldai.chat import TrackedChat
+from ldai.chat import Chat
 from ldai.client import LDAIClient
 # Export judge
 from ldai.judge import Judge
@@ -26,7 +26,7 @@
     'AIJudgeConfig',
     'AIJudgeConfigDefault',
     'Judge',
-    'TrackedChat',
+    'Chat',
     'EvalScore',
     'JudgeConfiguration',
     'JudgeResponse',

ldai/chat/__init__.py

@@ -1,4 +1,4 @@
-"""TrackedChat implementation for managing AI chat conversations."""
+"""Chat implementation for managing AI chat conversations."""
 
 import asyncio
 from typing import Any, Dict, List, Optional
@@ -10,9 +10,9 @@
 from ldai.tracker import LDAIConfigTracker
 
 
-class TrackedChat:
+class Chat:
     """
-    Concrete implementation of TrackedChat that provides chat functionality
+    Concrete implementation of Chat that provides chat functionality
     by delegating to an AIProvider implementation.
 
     This class handles conversation management and tracking, while delegating
@@ -28,7 +28,7 @@ def __init__(
         logger: Optional[Any] = None,
     ):
         """
-        Initialize the TrackedChat.
+        Initialize the Chat.
 
         :param ai_config: The completion AI configuration
         :param tracker: The tracker for the completion configuration
@@ -126,15 +126,15 @@ async def evaluate_judge(judge_config):
 
     def get_config(self) -> AICompletionConfig:
         """
-        Get the underlying AI configuration used to initialize this TrackedChat.
+        Get the underlying AI configuration used to initialize this Chat.
 
         :return: The AI completion configuration
         """
         return self._ai_config
 
     def get_tracker(self) -> LDAIConfigTracker:
         """
-        Get the underlying AI configuration tracker used to initialize this TrackedChat.
+        Get the underlying AI configuration tracker used to initialize this Chat.
 
         :return: The tracker instance
         """
@@ -152,7 +152,7 @@ def get_provider(self) -> AIProvider:
 
     def get_judges(self) -> Dict[str, Judge]:
         """
-        Get the judges associated with this TrackedChat.
+        Get the judges associated with this Chat.
 
         Returns a dictionary of judge instances keyed by their configuration keys.
 

ldai/client.py

@@ -5,7 +5,7 @@
 from ldclient import Context
 from ldclient.client import LDClient
 
-from ldai.chat import TrackedChat
+from ldai.chat import Chat
 from ldai.judge import Judge
 from ldai.models import (AIAgentConfig, AIAgentConfigDefault,
                          AIAgentConfigRequest, AIAgents, AICompletionConfig,
@@ -241,16 +241,16 @@ async def create_chat(
         default_value: AICompletionConfigDefault,
         variables: Optional[Dict[str, Any]] = None,
         default_ai_provider: Optional[SupportedAIProvider] = None,
-    ) -> Optional[TrackedChat]:
+    ) -> Optional[Chat]:
         """
-        Creates and returns a new TrackedChat instance for AI chat conversations.
+        Creates and returns a new Chat instance for AI conversations.
 
         :param key: The key identifying the AI completion configuration to use
         :param context: Standard Context used when evaluating flags
         :param default_value: A default value representing a standard AI config result
         :param variables: Dictionary of values for instruction interpolation
         :param default_ai_provider: Optional default AI provider to use
-        :return: TrackedChat instance or None if disabled/unsupported
+        :return: Chat instance or None if disabled/unsupported
 
         Example::
 
@@ -296,7 +296,7 @@ async def create_chat(
                 default_ai_provider,
             )
 
-        return TrackedChat(config, config.tracker, provider, judges, self._logger)
+        return Chat(config, config.tracker, provider, judges, self._logger)
 
     def agent_config(
         self,