Spaces:
Sleeping
Sleeping
| """ | |
| Example: MCP Server for Text Adventures | |
| A complete MCP server that exposes text adventure games via tools. | |
| This demonstrates a full-featured server with memory, mapping, and inventory. | |
| """ | |
| import sys | |
| import os | |
| # Add parent directory to path to import games module | |
| sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) | |
| from fastmcp import FastMCP | |
| from games.zork_env import TextAdventureEnv, list_available_games | |
| # Get game from environment variable (default: zork1) | |
| INITIAL_GAME = os.environ.get("GAME", "zork1") | |
| # Create the MCP server | |
| mcp = FastMCP("Text Adventure Server") | |
| class GameState: | |
| """Manages the text adventure game state and exploration data.""" | |
| def __init__(self, game: str = "zork1"): | |
| self.game_name = game | |
| self.env = TextAdventureEnv(game) | |
| self.state = self.env.reset() | |
| self.history: list[tuple[str, str]] = [] | |
| self.location_moves : dict = {} | |
| def take_action(self, action: str) -> str: | |
| """Execute a game action and return the result.""" | |
| self.state = self.env.step(action) | |
| result = self.state.observation | |
| # Track history | |
| self.history.append((action, result)) | |
| if len(self.history) > 50: | |
| self.history = self.history[-50:] | |
| return result | |
| def get_map(self) -> str: | |
| """Get a map of explored locations.""" | |
| if not self.explored_locations: | |
| return "Map: No locations explored yet. Try moving around!" | |
| lines = ["Explored Locations and Exits:"] | |
| for loc, exits in sorted(self.explored_locations.items()): | |
| lines.append(f"\n* {loc}") | |
| for exit_info in sorted(exits): | |
| lines.append(f" -> {exit_info}") | |
| lines.append(f"\n[Current] {self.current_location}") | |
| return "\n".join(lines) | |
| def get_inventory(self) -> str: | |
| """Get current inventory.""" | |
| items = self.state.inventory if hasattr(self.state, 'inventory') and self.state.inventory else [] | |
| if not items: | |
| return "Inventory: You are empty-handed." | |
| item_names = [] | |
| for item in items: | |
| item_str = str(item) | |
| item_lower = item_str.lower() | |
| if "parent" in item_lower: | |
| idx = item_lower.index("parent") | |
| name = item_str[:idx].strip() | |
| if ":" in name: | |
| name = name.split(":", 1)[1].strip() | |
| item_names.append(name) | |
| elif ":" in item_str: | |
| name = item_str.split(":")[1].strip() | |
| item_names.append(name) | |
| else: | |
| item_names.append(item_str) | |
| return f"Inventory: {', '.join(item_names)}" | |
| # Global game state | |
| _game_state: GameState | None = None | |
| def get_game() -> GameState: | |
| """Get or initialize the game state.""" | |
| global _game_state | |
| if _game_state is None: | |
| _game_state = GameState(INITIAL_GAME) | |
| return _game_state | |
| # ============================================================================= | |
| # MCP Tools | |
| # ============================================================================= | |
| def play_action(action: str) -> str: | |
| """ | |
| Execute a game action in the text adventure. | |
| Args: | |
| action: The command to execute (e.g., 'north', 'take lamp', 'open mailbox') | |
| Returns: | |
| The game's response to your action | |
| """ | |
| game = get_game() | |
| result = game.take_action(action) | |
| if action == "listen" or action == "look": | |
| result = game.take_action(action) | |
| # Add score info | |
| # score_info = f"\n\n[Score: {game.state.score} | Moves: {game.state.moves}]" | |
| # if game.state.reward > 0: | |
| # score_info = f"\n\n+{game.state.reward} points! (Total: {game.state.score})" | |
| # done_info = "" | |
| # if game.state.done: | |
| # done_info = "\n\nGAME OVER" | |
| return result | |
| def get_score() -> int: | |
| """ | |
| Return score | |
| """ | |
| return get_game().state.score | |
| def inventory() -> str: | |
| """ | |
| Check what items you are currently carrying. | |
| """ | |
| return get_game().get_inventory() | |
| # ============================================================================= | |
| # Main | |
| # ============================================================================= | |
| if __name__ == "__main__": | |
| mcp.run() | |