refactor(agent): Refactor resource of agents (#1518)

dbgpt/agent/resource/pack.py

@@ -0,0 +1,115 @@
+"""Resource pack module.
+
+Resource pack is a collection of resources(also, it is a resource) that can be executed
+together.
+"""
+
+import dataclasses
+from typing import Any, Dict, List, Optional
+
+from .base import Resource, ResourceParameters, ResourceType
+
+
+@dataclasses.dataclass
+class PackResourceParameters(ResourceParameters):
+    """Resource pack parameters class."""
+
+    pass
+
+
+class ResourcePack(Resource[PackResourceParameters]):
+    """Resource pack class."""
+
+    def __init__(
+        self,
+        resources: List[Resource],
+        name: str = "Resource Pack",
+        prompt_separator: str = "\n",
+    ):
+        """Initialize the resource pack."""
+        self._resources: Dict[str, Resource] = {
+            resource.name: resource for resource in resources
+        }
+        self._name = name
+        self._prompt_separator = prompt_separator
+
+    @classmethod
+    def type(cls) -> ResourceType:
+        """Return the resource type."""
+        return ResourceType.Pack
+
+    @property
+    def name(self) -> str:
+        """Return the resource name."""
+        return self._name
+
+    def _get_resource_by_name(self, name: str) -> Optional[Resource]:
+        """Get the resource by name."""
+        return self._resources.get(name, None)
+
+    async def get_prompt(
+        self,
+        *,
+        lang: str = "en",
+        prompt_type: str = "default",
+        question: Optional[str] = None,
+        resource_name: Optional[str] = None,
+        **kwargs,
+    ) -> str:
+        """Get the prompt."""
+        prompt_list = []
+        for name, resource in self._resources.items():
+            prompt = await resource.get_prompt(
+                lang=lang,
+                prompt_type=prompt_type,
+                question=question,
+                resource_name=resource_name,
+                **kwargs,
+            )
+            prompt_list.append(prompt)
+        return self._prompt_separator.join(prompt_list)
+
+    def append(self, resource: Resource, overwrite: bool = False):
+        """Append a resource to the pack."""
+        name = resource.name
+        if name in self._resources and not overwrite:
+            raise ValueError(f"Resource {name} already exists in the pack.")
+        self._resources[name] = resource
+
+    def execute(
+        self,
+        *args,
+        resource_name: Optional[str] = None,
+        **kwargs,
+    ) -> Any:
+        """Execute the resource."""
+        if not resource_name:
+            raise ValueError("No resource name provided, will not execute.")
+        resource = self._resources.get(resource_name)
+        if resource:
+            return resource.execute(*args, **kwargs)
+        raise ValueError("No resource parameters provided, will not execute.")
+
+    async def async_execute(
+        self,
+        *args,
+        resource_name: Optional[str] = None,
+        **kwargs,
+    ) -> Any:
+        """Execute the resource asynchronously."""
+        if not resource_name:
+            raise ValueError("No resource name provided, will not execute.")
+        resource = self._resources.get(resource_name)
+        if resource:
+            return await resource.async_execute(*args, **kwargs)
+        raise ValueError("No resource parameters provided, will not execute.")
+
+    @property
+    def is_pack(self) -> bool:
+        """Return whether the resource is a pack."""
+        return True
+
+    @property
+    def sub_resources(self) -> List[Resource]:
+        """Return the resources."""
+        return list(self._resources.values())