[Eigenwise]

Problem

When you subclass AtomicAgent with a custom constructor, the generic type parameters get lost. The input_schema and output_schema properties end up returning BasicChatInputSchema and BasicChatOutputSchema instead of the types you actually specified.

Example that was broken:

class MyAgent(AtomicAgent[MyInputSchema, MyOutputSchema]):
    def __init__(self, extra_param):
        self.extra_param = extra_param
        super().__init__(config)

agent = MyAgent("test")
# These would incorrectly return Basic* schemas instead of My* schemas
print(agent.input_schema)
print(agent.output_schema)

The issue is that __orig_class__ only gets set for direct instantiation, not when subclassing.

Solution

Added __init_subclass__ to capture the generic parameters at class creation time. This is the same approach already used by BasePrompt, BaseTool, and BaseResource, so it's consistent with the codebase.

The schema properties now check in this order:

1. Class attributes (set by __init_subclass__) - handles subclassing
2. Instance __orig_class__ - handles direct instantiation
3. Default schemas - handles untyped usage

Testing

• Added test_subclass_with_custom_constructor() to verify the fix
• All existing tests still pass (35/35)
• Code formatted and linted

Notes

This is backward compatible - doesn't break any existing usage patterns. Direct instantiation still works the same way it did before.

[Eigenwise]

Fixes #185