Skip to content

Commit 31a2c41

Browse files
authored
refactor(evaluator): refine base_evaluator (#95)
* refactor(evaluator): refine base_evaluator * refactor(evaluator): fix adk evaluator --------- Co-authored-by: tangou <[email protected]>
1 parent 7cef667 commit 31a2c41

File tree

6 files changed

+211
-245
lines changed

6 files changed

+211
-245
lines changed

tests/test_evaluator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ def test_evaluator():
129129
with open(eval_set_file_path, "w") as f:
130130
json.dump(EVAL_SET_DATA, f)
131131

132-
base_evaluator.generate_eval_data(file_path=eval_set_file_path)
132+
base_evaluator.build_eval_set(file_path=eval_set_file_path)
133133

134134
assert len(base_evaluator.invocation_list) == 1
135135
assert len(base_evaluator.invocation_list[0].invocations) == 1
@@ -149,7 +149,7 @@ def test_tracing_file_to_evalset():
149149
with open(tracing_file_path, "w") as f:
150150
json.dump(TRACE_SET_DATA, f)
151151

152-
base_evaluator.generate_eval_data(file_path=tracing_file_path)
152+
base_evaluator.build_eval_set(file_path=tracing_file_path)
153153

154154
assert len(base_evaluator.invocation_list) == 1
155155
assert len(base_evaluator.invocation_list[0].invocations) == 1

veadk/evaluation/adk_evaluator/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,3 +11,7 @@
1111
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
14+
15+
from .adk_evaluator import ADKEvaluator
16+
17+
__all__ = ["ADKEvaluator"]

0 commit comments

Comments
 (0)