mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-10-23 01:49:58 +00:00
76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
import logging
|
||
import os
|
||
from typing import Optional
|
||
|
||
from dbgpt.agent.plugin.commands.command_manage import execute_command
|
||
from dbgpt.agent.plugin.generator import PluginPromptGenerator
|
||
from dbgpt.agent.plugin.plugins_util import scan_plugin_file, scan_plugins
|
||
from dbgpt.agent.resource.resource_api import AgentResource
|
||
|
||
from .resource_api import ResourceClient, ResourceType
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class ResourcePluginClient(ResourceClient):
|
||
@property
|
||
def type(self):
|
||
return ResourceType.Plugin
|
||
|
||
def get_data_type(self, resource: AgentResource) -> str:
|
||
return "Tools"
|
||
|
||
async def get_data_introduce(
|
||
self, resource: AgentResource, question: Optional[str] = None
|
||
) -> str:
|
||
return await self.a_plugins_prompt(resource.value)
|
||
|
||
async def a_load_plugin(
|
||
self,
|
||
value: str,
|
||
plugin_generator: Optional[PluginPromptGenerator] = None,
|
||
) -> PluginPromptGenerator:
|
||
raise NotImplementedError("The run method should be implemented in a subclass.")
|
||
|
||
async def a_plugins_prompt(
|
||
self, value: str, plugin_generator: Optional[PluginPromptGenerator] = None
|
||
) -> str:
|
||
plugin_generator = await self.a_load_plugin(value)
|
||
return plugin_generator.generate_commands_string()
|
||
|
||
async def a_execute_command(
|
||
self,
|
||
command_name: str,
|
||
arguments: Optional[dict],
|
||
plugin_generator: Optional[PluginPromptGenerator],
|
||
):
|
||
if plugin_generator is None:
|
||
raise ValueError("No plugin commands loaded into the executable！")
|
||
return execute_command(command_name, arguments, plugin_generator)
|
||
|
||
|
||
class PluginFileLoadClient(ResourcePluginClient):
|
||
async def a_load_plugin(
|
||
self, value: str, plugin_generator: Optional[PluginPromptGenerator] = None
|
||
) -> PluginPromptGenerator:
|
||
logger.info(f"PluginFileLoadClient load plugin:{value}")
|
||
if plugin_generator is None:
|
||
plugin_generator = PluginPromptGenerator()
|
||
plugins = []
|
||
if os.path.isabs(value):
|
||
if not os.path.exists(value):
|
||
raise ValueError(f"Wrong plugin file path configured {value}!")
|
||
if os.path.isfile(value):
|
||
plugins.extend(scan_plugin_file(value))
|
||
else:
|
||
plugins.extend(scan_plugins(value))
|
||
else:
|
||
raise ValueError(
|
||
f"The current mode cannot support plug-in loading with relative paths.{value}"
|
||
)
|
||
for plugin in plugins:
|
||
if not plugin.can_handle_post_prompt():
|
||
continue
|
||
plugin_generator = plugin.post_prompt(plugin_generator)
|
||
return plugin_generator
|