feat(agent): Add retrieve summary agent for both local files and URLs (…

…#1063) Co-authored-by: fangfcg <[email protected]>
eosphoros-ai · Jan 16, 2024 · fb2d18a · fb2d18a
1 parent 40c8535
commit fb2d18a
Show file tree

Hide file tree

Showing 6 changed files with 727 additions and 0 deletions.
diff --git a/dbgpt/agent/agents/agents_mange.py b/dbgpt/agent/agents/agents_mange.py
@@ -10,6 +10,7 @@
 from .expand.plugin_assistant_agent import PluginAssistantAgent
 from .expand.sql_assistant_agent import SQLAssistantAgent
 from .expand.summary_assistant_agent import SummaryAssistantAgent
+from .expand.retrieve_summary_assistant_agent import RetrieveSummaryAssistantAgent
 
 logger = logging.getLogger(__name__)
 
@@ -88,3 +89,4 @@ def all_agents(self):
 agent_mange.register_agent(SQLAssistantAgent)
 agent_mange.register_agent(SummaryAssistantAgent)
 agent_mange.register_agent(PluginAssistantAgent)
+agent_mange.register_agent(RetrieveSummaryAssistantAgent)