fix: lint + missed variable rename

Author: andrewklatzke

packages/optimization/src/ldai_optimization/client.py

@@ -215,7 +215,7 @@ def _serialize_scores(
         """
         return {key: result.to_json() for key, result in judge_results.items()}
 
-    def _extract_agent_tools(self, parameters: Dict[str, Any]) -> List[StructuredOutputTool]:
+    def _extract_agent_tools(self, parameters: Dict[str, Any]) -> List[ToolDefinition]:
         """
         Extract and normalise the tools list from agent parameters.
 

packages/optimization/src/ldai_optimization/util.py

@@ -80,10 +80,9 @@ async def await_if_needed(
     :param result: Either a string or an awaitable that returns a string
     :return: The string result
     """
-    if inspect.iscoroutine(result):
-        return await result
-    else:
+    if isinstance(result, str):
         return result
+    return await result
 
 
 def create_evaluation_tool() -> ToolDefinition: