feat: add tools to completion configs with test coverage

Co-Authored-By: Paul Loeb <ploeb@launchdarkly.com>

Author: devin-ai-integration[bot]

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

@@ -52,15 +52,31 @@ def _completion_config(
         default: AICompletionConfigDefault,
         variables: Optional[Dict[str, Any]] = None,
     ) -> AICompletionConfig:
-        model, provider, messages, instructions, tracker, enabled, judge_configuration, _ = self.__evaluate(
+        model, provider, messages, instructions, tracker, enabled, judge_configuration, variation = self.__evaluate(
             key, context, default.to_dict(), variables
         )
 
+        # Parse tools from variation data
+        tools = None
+        if 'tools' in variation and isinstance(variation['tools'], list):
+            tools = [
+                AITool(
+                    key=tool['key'],
+                    version=tool.get('version', 0),
+                    instructions=tool.get('instructions'),
+                    examples=tool.get('examples'),
+                    custom_parameters=tool.get('customParameters'),
+                )
+                for tool in variation['tools']
+                if isinstance(tool, dict) and 'key' in tool
+            ] or None
+
         config = AICompletionConfig(
             key=key,
             enabled=bool(enabled),
             model=model,
             messages=messages,
+            tools=tools if tools is not None else (default.tools if default.tools else None),
             provider=provider,
             tracker=tracker,
             judge_configuration=judge_configuration,

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

@@ -239,6 +239,7 @@ class AICompletionConfigDefault(AIConfigDefault):
     Default Completion AI Config (default mode).
     """
     messages: Optional[List[LDMessage]] = None
+    tools: Optional[List[AITool]] = None
     judge_configuration: Optional[JudgeConfiguration] = None
 
     def to_dict(self) -> dict:
@@ -247,6 +248,8 @@ def to_dict(self) -> dict:
         """
         result = self._base_to_dict()
         result['messages'] = [message.to_dict() for message in self.messages] if self.messages else None
+        if self.tools is not None:
+            result['tools'] = [tool.to_dict() for tool in self.tools]
         if self.judge_configuration is not None:
             result['judgeConfiguration'] = self.judge_configuration.to_dict()
         return result
@@ -258,6 +261,7 @@ class AICompletionConfig(AIConfig):
     Completion AI Config (default mode).
     """
     messages: Optional[List[LDMessage]] = None
+    tools: Optional[List[AITool]] = None
     judge_configuration: Optional[JudgeConfiguration] = None
 
     def to_dict(self) -> dict:
@@ -266,6 +270,8 @@ def to_dict(self) -> dict:
         """
         result = self._base_to_dict()
         result['messages'] = [message.to_dict() for message in self.messages] if self.messages else None
+        if self.tools is not None:
+            result['tools'] = [tool.to_dict() for tool in self.tools]
         if self.judge_configuration is not None:
             result['judgeConfiguration'] = self.judge_configuration.to_dict()
         return result