Skip to content

Commit 4545203

Browse files
committed
chore: fix agent init typing
1 parent 27b248e commit 4545203

2 files changed

Lines changed: 4 additions & 1 deletion

File tree

packages/optimization/src/ldai_optimizer/client.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -320,6 +320,7 @@ def _build_agent_config_for_context(
320320
key=self._agent_key,
321321
enabled=True,
322322
create_tracker=self._agent_config.create_tracker,
323+
evaluator=self._agent_config.evaluator,
323324
model=ModelConfig(
324325
name=ctx.current_model or "",
325326
parameters=ctx.current_parameters,

packages/optimization/tests/test_client.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,8 @@
66
from unittest.mock import AsyncMock, MagicMock, patch
77

88
import pytest
9-
from ldai import AIAgentConfig, AIJudgeConfig, AIJudgeConfigDefault, LDAIClient
9+
from ldai import AIAgentConfig, AIJudgeConfig, LDAIClient
10+
from ldai.client import Evaluator
1011
from ldai.models import LDMessage, ModelConfig
1112
from ldai.tracker import TokenUsage
1213
from ldclient import Context
@@ -74,6 +75,7 @@ def _make_agent_config(
7475
key="test-agent",
7576
enabled=True,
7677
create_tracker=MagicMock,
78+
evaluator=Evaluator.noop(),
7779
model=ModelConfig(name=model_name, parameters=parameters or {}),
7880
instructions=instructions,
7981
)

0 commit comments

Comments
 (0)