AIJudge to Judge

Author: edwinokonkwo

ldai/__init__.py

@@ -5,7 +5,7 @@
 from ldai.chat import TrackedChat
 from ldai.client import LDAIClient
 # Export judge
-from ldai.judge import AIJudge
+from ldai.judge import Judge
 # Export models for convenience
 from ldai.models import (  # Deprecated aliases for backward compatibility
     AIAgentConfig, AIAgentConfigDefault, AIAgentConfigRequest, AIAgents,
@@ -25,7 +25,7 @@
     'AICompletionConfigDefault',
     'AIJudgeConfig',
     'AIJudgeConfigDefault',
-    'AIJudge',
+    'Judge',
     'TrackedChat',
     'EvalScore',
     'JudgeConfiguration',

ldai/chat/__init__.py

@@ -3,7 +3,7 @@
 import asyncio
 from typing import Any, Dict, List, Optional
 
-from ldai.judge import AIJudge
+from ldai.judge import Judge
 from ldai.models import AICompletionConfig, LDMessage
 from ldai.providers.ai_provider import AIProvider
 from ldai.providers.types import ChatResponse, JudgeResponse
@@ -24,7 +24,7 @@ def __init__(
         ai_config: AICompletionConfig,
         tracker: LDAIConfigTracker,
         provider: AIProvider,
-        judges: Optional[Dict[str, AIJudge]] = None,
+        judges: Optional[Dict[str, Judge]] = None,
         logger: Optional[Any] = None,
     ):
         """
@@ -151,7 +151,7 @@ def get_provider(self) -> AIProvider:
         """
         return self._provider
 
-    def get_judges(self) -> Dict[str, AIJudge]:
+    def get_judges(self) -> Dict[str, Judge]:
         """
         Get the judges associated with this TrackedChat.
 

ldai/client.py

@@ -6,7 +6,7 @@
 from ldclient.client import LDClient
 
 from ldai.chat import TrackedChat
-from ldai.judge import AIJudge
+from ldai.judge import Judge
 from ldai.models import (AIAgentConfig, AIAgentConfigDefault,
                          AIAgentConfigRequest, AIAgents, AICompletionConfig,
                          AICompletionConfigDefault, AIJudgeConfig,
@@ -121,7 +121,7 @@ async def create_judge(
         default_value: AIJudgeConfigDefault,
         variables: Optional[Dict[str, Any]] = None,
         default_ai_provider: Optional[SupportedAIProvider] = None,
-    ) -> Optional[AIJudge]:
+    ) -> Optional[Judge]:
         """
         Creates and returns a new Judge instance for AI evaluation.
 
@@ -182,7 +182,7 @@ async def create_judge(
             if not provider:
                 return None
 
-            return AIJudge(judge_config, judge_config.tracker, provider, self._logger)
+            return Judge(judge_config, judge_config.tracker, provider, self._logger)
         except Exception as error:
             # Would log error if logger available
             return None
@@ -193,7 +193,7 @@ async def _initialize_judges(
         context: Context,
         variables: Optional[Dict[str, Any]] = None,
         default_ai_provider: Optional[SupportedAIProvider] = None,
-    ) -> Dict[str, AIJudge]:
+    ) -> Dict[str, Judge]:
         """
         Initialize judges from judge configurations.
 
@@ -203,7 +203,7 @@ async def _initialize_judges(
         :param default_ai_provider: Optional default AI provider to use
         :return: Dictionary of judge instances keyed by their configuration keys
         """
-        judges: Dict[str, AIJudge] = {}
+        judges: Dict[str, Judge] = {}
 
         async def create_judge_for_config(judge_key: str):
             judge = await self.create_judge(

ldai/judge/__init__.py

@@ -13,7 +13,7 @@
 from ldai.tracker import LDAIConfigTracker
 
 
-class AIJudge:
+class Judge:
     """
     Judge implementation that handles evaluation functionality and conversation management.