diff --git a/src/arbiter/agents/__init__.py b/src/arbiter/agents/__init__.py index ad2e995..a1a40c3 100644 --- a/src/arbiter/agents/__init__.py +++ b/src/arbiter/agents/__init__.py @@ -1,6 +1,6 @@ """Review agents for code analysis.""" -from arbiter.agents.base import Agent, ReviewContext +from arbiter.agents.base import Agent, ExplainContext, ExplainResult, ReviewContext from arbiter.agents.complexity import ComplexityAgent from arbiter.agents.security import SecurityAgent from arbiter.agents.style import StyleAgent @@ -8,6 +8,8 @@ from arbiter.agents.style import StyleAgent __all__ = [ "Agent", "ComplexityAgent", + "ExplainContext", + "ExplainResult", "ReviewContext", "SecurityAgent", "StyleAgent",