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

dbgpt/agent/expand/resources/dbgpt_tool.py

@@ -0,0 +1,23 @@
+"""Some internal tools for the DB-GPT project."""
+
+from typing_extensions import Annotated, Doc
+
+from ...resource.tool.base import tool
+
+
+@tool(description="List the supported models in DB-GPT project.")
+def list_dbgpt_support_models(
+    model_type: Annotated[
+        str, Doc("The model type, LLM(Large Language Model) and EMBEDDING).")
+    ] = "LLM",
+) -> str:
+    """List the supported models in dbgpt."""
+    from dbgpt.configs.model_config import EMBEDDING_MODEL_CONFIG, LLM_MODEL_CONFIG
+
+    if model_type.lower() == "llm":
+        supports = list(LLM_MODEL_CONFIG.keys())
+    elif model_type.lower() == "embedding":
+        supports = list(EMBEDDING_MODEL_CONFIG.keys())
+    else:
+        raise ValueError(f"Unsupported model type: {model_type}")
+    return "\n\n".join(supports)