Add missing type hints to DatasetCase

[aviralgarg05]

Summary

Add type hints to the DatasetCase dataclass fields that currently lack them.

Tasks

• Review src/agentunit/datasets/base.py for missing or incomplete type hints.
• Add str, list[str], dict[str, Any], etc. where appropriate.
• Ensure from __future__ import annotations is present.

Acceptance criteria

• All fields in DatasetCase have explicit type hints.
• poetry run ruff check . passes.
• Tests pass.

[evanjhoward11]

Hello, I'd like to work on this issue. Can you please assign it to me?

[aviralgarg05]

Assigned! Go through the codebase, if any issues let me know here

[evanjhoward11]

It looks like base.py (and everything within its folder) already has complete and accurate type hints, and passes the formatting and linting tests. However, when running mypy on src/agentunit, issues for the other folders/files come up. Besides missing type hints, there are import-related errors, such as [import-not-found] and [import-untyped].
Can you confirm that there are in fact missing dependencies and it's not just an issue with my machine?
Should I work on the other files that mypy shows are missing type hints?

TIA!

[aviralgarg05]

You're right - base.py already has proper type hints. The mypy errors you're seeing are real issues, not your machine.

Import errors (import-not-found, import-untyped) - These happen because some dependencies might not be installed or don't have type stubs. This is normal for third-party packages.
Other files needing type hints - Yes, if mypy shows other files with missing type hints, those are real issues that could be fixed.

[evanjhoward11]

Hi @aviralgarg05, I submitted PR #45 for this issue. I does not resolve it, but I hope a made a small dent in all the mypy errors. Let me know if my changes are appropriate and didn't break anything.