Skip to content

Commit

Permalink
Add module level docstrings (#4652)
Browse files Browse the repository at this point in the history
* Improve init docs

* Add docstring for messages

* Add docstring for agents

* Add docstring for teams

* Add doc string for conditions

* Add docs for ui

* Update module docstring in __init__.py

* Clarify BaseChatAgent description in __init__.py

* Fix formatting
  • Loading branch information
gagb authored Dec 11, 2024
1 parent 0aeb781 commit 34b9977
Show file tree
Hide file tree
Showing 6 changed files with 30 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
"""
This module provides the main entry point for the autogen_agentchat package.
It includes logger names for trace and event logs, and retrieves the package version.
"""

import importlib.metadata

TRACE_LOGGER_NAME = "autogen_agentchat"
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
"""
This module initializes various pre-defined agents provided by the package.
BaseChatAgent is the base class for all agents in AgentChat.
"""

from ._assistant_agent import AssistantAgent, Handoff # type: ignore
from ._base_chat_agent import BaseChatAgent
from ._code_executor_agent import CodeExecutorAgent
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
"""
This module provides various termination conditions for controlling the behavior of
multi-agent teams.
"""

from ._terminations import (
ExternalTermination,
HandoffTermination,
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,9 @@
"""
This module defines various message types used for agent-to-agent communication.
Each message type inherits from the BaseMessage class and includes specific fields
relevant to the type of message being sent.
"""

from typing import List, Literal

from autogen_core import FunctionCall, Image
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
"""
This module provides implementation of various pre-defined multi-agent teams.
Each team inherits from the BaseGroupChat class.
"""

from ._group_chat._base_group_chat import BaseGroupChat
from ._group_chat._magentic_one import MagenticOneGroupChat
from ._group_chat._round_robin_group_chat import RoundRobinGroupChat
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,7 @@
"""
This module implements utility classes for formatting/printing agent messages.
"""

from ._console import Console

__all__ = ["Console"]

0 comments on commit 34b9977

Please sign in to comment.