docs: Add agents notebook

dbgpt/agent/__init__.py

@@ -7,6 +7,7 @@
     AgentGenerateContext,
     AgentMessage,
 )
+from .core.base_agent import ConversableAgent  # noqa: F401
 from .core.llm.llm import LLMConfig  # noqa: F401
 from .core.schema import PluginStorageType  # noqa: F401
 from .core.user_proxy_agent import UserProxyAgent  # noqa: F401
@@ -19,6 +20,7 @@
     "AgentContext",
     "AgentGenerateContext",
     "AgentMessage",
+    "ConversableAgent",
     "Action",
     "ActionOutput",
     "LLMConfig",

dbgpt/agent/core/agent_manage.py

@@ -98,10 +98,10 @@ def list_agents(self):
         return result
 
 
-agent_manage = AgentManager()
+agent_manager = AgentManager()
 
-agent_manage.register_agent(CodeAssistantAgent)
-agent_manage.register_agent(DashboardAssistantAgent)
-agent_manage.register_agent(DataScientistAgent)
-agent_manage.register_agent(SummaryAssistantAgent)
-agent_manage.register_agent(PluginAssistantAgent)
+agent_manager.register_agent(CodeAssistantAgent)
+agent_manager.register_agent(DashboardAssistantAgent)
+agent_manager.register_agent(DataScientistAgent)
+agent_manager.register_agent(SummaryAssistantAgent)
+agent_manager.register_agent(PluginAssistantAgent)

dbgpt/agent/core/llm/llm_client.py

@@ -171,7 +171,7 @@ async def _completions_create(self, llm_model, params) -> str:
             model_request = _build_model_request(payload)
             model_output = await self._llm_client.generate(model_request.copy())
             parsed_output = self._output_parser.parse_model_nostream_resp(
-                model_output, "###"
+                model_output, "#########################"
             )
             return parsed_output
         except Exception as e:

dbgpt/agent/plan/awel/agent_operator.py

@@ -17,7 +17,7 @@
 from dbgpt.model.operators.llm_operator import MixinLLMOperator
 
 from ...core.agent import Agent, AgentGenerateContext, AgentMessage
-from ...core.agent_manage import agent_manage
+from ...core.agent_manage import agent_manager
 from ...core.base_agent import ConversableAgent
 from ...core.llm.llm import LLMConfig
 from .agent_operator_resource import AWELAgent
@@ -244,7 +244,7 @@ async def get_agent(
     ) -> ConversableAgent:
         """Build the agent."""
         # agent build
-        agent_cls: Type[ConversableAgent] = agent_manage.get_by_name(  # type: ignore
+        agent_cls: Type[ConversableAgent] = agent_manager.get_by_name(  # type: ignore
             self.awel_agent.agent_profile
         )
         llm_config = self.awel_agent.llm_config

dbgpt/agent/plan/awel/agent_operator_resource.py

@@ -11,7 +11,7 @@
     register_resource,
 )
 
-from ...core.agent_manage import agent_manage
+from ...core.agent_manage import agent_manager
 from ...core.llm.llm import LLMConfig, LLMStrategyType
 from ...resource.resource_api import AgentResource, ResourceType
 
@@ -118,7 +118,7 @@ def pre_fill(cls, values: Dict[str, Any]) -> Dict[str, Any]:
 def _agent_resource_option_values() -> List[OptionValue]:
     return [
         OptionValue(label=item["name"], name=item["name"], value=item["name"])
-        for item in agent_manage.list_agents()
+        for item in agent_manager.list_agents()
     ]
 
 

dbgpt/serve/agent/agents/controller.py

@@ -10,7 +10,7 @@
 
 from dbgpt._private.config import Config
 from dbgpt.agent.core.agent import Agent, AgentContext
-from dbgpt.agent.core.agent_manage import agent_manage
+from dbgpt.agent.core.agent_manage import agent_manager
 from dbgpt.agent.core.base_agent import ConversableAgent
 from dbgpt.agent.core.llm.llm import LLMConfig, LLMStrategyType
 from dbgpt.agent.core.schema import Status
@@ -222,7 +222,7 @@ async def agent_team_chat_new(
         self.llm_provider = DefaultLLMClient(worker_manager, auto_convert_message=True)
 
         for record in gpts_app.details:
-            cls: Type[ConversableAgent] = agent_manage.get_by_name(record.agent_name)
+            cls: Type[ConversableAgent] = agent_manager.get_by_name(record.agent_name)
             llm_config = LLMConfig(
                 llm_client=self.llm_provider,
                 llm_strategy=LLMStrategyType(record.llm_strategy),
@@ -340,7 +340,7 @@ def gpts_conv_list(self, user_code: str = None, system_app: str = None):
 async def agents_list():
     logger.info("agents_list!")
     try:
-        agents = agent_manage.all_agents()
+        agents = agent_manager.all_agents()
         return Result.succ(agents)
     except Exception as e:
         return Result.failed(code="E30001", msg=str(e))

dbgpt/serve/agent/app/controller.py

@@ -3,7 +3,7 @@
 from fastapi import APIRouter
 
 from dbgpt._private.config import Config
-from dbgpt.agent.core.agent_manage import agent_manage
+from dbgpt.agent.core.agent_manage import agent_manager
 from dbgpt.agent.core.llm.llm import LLMStrategyType
 from dbgpt.agent.resource.resource_api import ResourceType
 from dbgpt.app.knowledge.api import knowledge_space_service
@@ -63,7 +63,7 @@ async def edit(gpts_app: GptsApp):
 @router.get("/v1/agents/list")
 async def all_agents():
     try:
-        return Result.succ(agent_manage.list_agents())
+        return Result.succ(agent_manager.list_agents())
     except Exception as ex:
         return Result.failed(code="E000X", msg=f"query agents error: {ex}")
 

examples/agents/awel_layout_agents_chat_examples.py

@@ -32,8 +32,6 @@
 from dbgpt.agent.resource import PluginFileLoadClient
 from dbgpt.configs.model_config import ROOT_PATH
 
-current_dir = os.getcwd()
-parent_dir = os.path.dirname(current_dir)
 test_plugin_dir = os.path.join(ROOT_PATH, "examples/test_files/plugins")