Update agent.py

Author: DanielHashmi

src/agents/agent.py

@@ -256,7 +256,7 @@ def __post_init__(self):
 
  if (
  self.prompt is not None
- and not isinstance(self.prompt, (Prompt, DynamicPromptFunction))
+ and not isinstance(self.prompt, dict)
  and not callable(self.prompt)
  ):
  raise TypeError(
@@ -297,10 +297,11 @@ def __post_init__(self):
  f"got {type(self.output_type).__name__}"
  )
 
+ # Fixed hooks validation - use AgentHooksBase instead of generic AgentHooks
  if self.hooks is not None:
- from .lifecycle import AgentHooks
+ from .lifecycle import AgentHooksBase
 
- if not isinstance(self.hooks, AgentHooks):
+ if not isinstance(self.hooks, AgentHooksBase):
  raise TypeError(
  f"Agent hooks must be an AgentHooks instance or None, "
  f"got {type(self.hooks).__name__}"