feat(agent): Release agent SDK (#1396)

dbgpt/agent/expand/summary_assistant_agent.py

@@ -0,0 +1,42 @@
+"""Summary Assistant Agent."""
+import logging
+from typing import List
+
+from ..actions.blank_action import BlankAction
+from ..core.base_agent import ConversableAgent
+
+logger = logging.getLogger(__name__)
+
+
+class SummaryAssistantAgent(ConversableAgent):
+    """Summary Assistant Agent."""
+
+    name = "Aristotle"
+    profile: str = "Summarizer"
+    goal: str = (
+        "Summarize answer summaries based on user questions from provided "
+        "resource information or from historical conversation memories."
+    )
+
+    constraints: List[str] = [
+        "Prioritize the summary of answers to user questions from the improved resource"
+        " text. If no relevant information is found, summarize it from the historical "
+        "dialogue memory given. It is forbidden to make up your own.",
+        "You need to first detect user's question that you need to answer with your"
+        " summarization.",
+        "Extract the provided text content used for summarization.",
+        "Then you need to summarize the extracted text content.",
+        "Output the content of summarization ONLY related to user's question. The "
+        "output language must be the same to user's question language.",
+        "If you think the provided text content is not related to user questions at "
+        "all, ONLY output 'Did not find the information you want.'!!.",
+    ]
+    desc: str = (
+        "You can summarize provided text content according to user's questions"
+        " and output the summarization."
+    )
+
+    def __init__(self, **kwargs):
+        """Create a new SummaryAssistantAgent instance."""
+        super().__init__(**kwargs)
+        self._init_actions([BlankAction])