Skip to content

Commit 44b02cf

Browse files
committed
fix: modify folder structure for types object
1 parent feeb8eb commit 44b02cf

File tree

3 files changed

+11
-8
lines changed

3 files changed

+11
-8
lines changed

src/mcpm/core/mcp/types.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
from enum import Enum
2+
3+
4+
class ResourceType(str, Enum):
5+
TOOL = "tool"
6+
PROMPT = "prompt"
7+
RESOURCE = "resource"
8+
RESOURCE_TEMPLATE = "resource_template"

src/mcpm/core/router/manager.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,8 @@
1717
from pydantic import AnyUrl
1818

1919
from mcpm.core.mcp.client_connection import ServerConnection
20-
from mcpm.core.schema import ResourceType, ServerConfig
20+
from mcpm.core.mcp.types import ResourceType
21+
from mcpm.core.schema import ServerConfig
2122
from mcpm.utils.config import PROMPT_SPLITOR, RESOURCE_SPLITOR, TOOL_SPLITOR
2223
from mcpm.utils.errlog_manager import ServerErrorLogManager
2324

src/mcpm/core/schema.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from typing import Any, Dict, List, Optional, Union
2-
from enum import Enum
2+
33
from pydantic import BaseModel
44

55

@@ -82,9 +82,3 @@ class Profile(BaseModel):
8282
name: str
8383
api_key: Optional[str]
8484
servers: list[ServerConfig]
85-
86-
class ResourceType(str, Enum):
87-
TOOL = "tool"
88-
PROMPT = "prompt"
89-
RESOURCE = "resource"
90-
RESOURCE_TEMPLATE = "resource_template"

0 commit comments

Comments
 (0)