Skip to content

Commit 7533bc9

Browse files
committed
fix: Fix various docstring issues
1 parent d45c13b commit 7533bc9

File tree

4 files changed

+22
-10
lines changed

4 files changed

+22
-10
lines changed

src/strands/hooks/registry.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -184,14 +184,12 @@ def invoke_callbacks(self, event: TInvokeEvent) -> TInvokeEvent:
184184
185185
This method finds all callbacks registered for the event's type and
186186
invokes them in the appropriate order. For events with should_reverse_callbacks=True,
187-
callbacks are invoked in reverse registration order.
187+
callbacks are invoked in reverse registration order. Any exceptions raised by callback
188+
functions will propagate to the caller.
188189
189190
Args:
190191
event: The event to dispatch to registered callbacks.
191192
192-
Raises:
193-
Any exceptions raised by callback functions will propagate to the caller.
194-
195193
Returns:
196194
The event dispatched to registered callbacks.
197195

src/strands/models/writer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -427,8 +427,8 @@ async def structured_output(
427427
"""Get structured output from the model.
428428
429429
Args:
430-
output_model(Type[BaseModel]): The output model to use for the agent.
431-
prompt(Messages): The prompt messages to use for the agent.
430+
output_model: The output model to use for the agent.
431+
prompt: The prompt messages to use for the agent.
432432
system_prompt: System prompt to provide context to the model.
433433
**kwargs: Additional keyword arguments for future extensibility.
434434
"""

src/strands/session/__init__.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
"""Session module.
2+
3+
This module provides session management functionality.
4+
"""
5+
6+
from .file_session_manager import FileSessionManager
7+
from .repository_session_manager import RepositorySessionManager
8+
from .s3_session_manager import S3SessionManager
9+
from .session_manager import SessionManager
10+
from .session_repository import SessionRepository
11+
12+
__all__ = [
13+
"FileSessionManager",
14+
"RepositorySessionManager",
15+
"S3SessionManager",
16+
"SessionManager",
17+
"SessionRepository",
18+
]

src/strands/telemetry/tracer.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -473,7 +473,6 @@ def end_agent_span(
473473
span: The span to end.
474474
response: The response from the agent.
475475
error: Any error that occurred.
476-
metrics: Metrics data to add to the span.
477476
"""
478477
attributes: Dict[str, AttributeValue] = {}
479478

@@ -541,9 +540,6 @@ def end_swarm_span(
541540
def get_tracer() -> Tracer:
542541
"""Get or create the global tracer.
543542
544-
Args:
545-
service_name: Name of the service for OpenTelemetry.
546-
547543
Returns:
548544
The global tracer instance.
549545
"""

0 commit comments

Comments
 (0)