too much

utils/__init__.py

@@ -0,0 +1,3 @@
+from .character_loader import load_characters
+
+__all__ = ['load_characters']

utils/character_loader.py

@@ -0,0 +1,77 @@
+import yaml
+from pathlib import Path
+from typing import List, Dict
+
+from living_agents import Character
+
+
+# def load_characters(characters_dir: str = "characters") -> List[Tuple[Character, List[str]]]:
+def load_characters(characters_dir: str = "characters") -> Dict[Character, List[str]]:
+    """
+    Load all character YAML files from the specified directory.
+
+    Args:
+        characters_dir (str): Path to the directory containing character YAML files
+
+    Returns:
+        List[Character]: List of Character objects loaded from YAML files
+
+    Raises:
+        FileNotFoundError: If the characters directory doesn't exist
+        yaml.YAMLError: If there's an error parsing a YAML file
+        ValueError: If a YAML file is missing required fields
+    """
+    characters = {}
+    characters_path = Path(characters_dir)
+
+    if not characters_path.exists():
+        raise FileNotFoundError(f"Characters directory '{characters_dir}' not found")
+
+    if not characters_path.is_dir():
+        raise ValueError(f"'{characters_dir}' is not a directory")
+
+    # Find all YAML files in the directory
+    yaml_files = list(characters_path.glob("*.yaml")) + list(characters_path.glob("*.yml"))
+
+    if not yaml_files:
+        print(f"No YAML files found in '{characters_dir}'")
+        return characters
+
+    for yaml_file in yaml_files:
+        try:
+            with open(yaml_file, 'r', encoding='utf-8') as file:
+                data = yaml.safe_load(file)
+
+            if data is None:
+                print(f"Warning: Empty YAML file '{yaml_file.name}', skipping")
+                continue
+
+            # Validate required fields
+            required_fields = ['name', 'age', 'personality', 'occupation', 'location']
+            missing_fields = [field for field in required_fields if field not in data]
+
+            if missing_fields:
+                raise ValueError(f"File '{yaml_file.name}' missing required fields: {missing_fields}")
+
+            # Create Character object
+            character = Character(
+                name=data['name'],
+                age=data['age'],
+                personality=data['personality'],
+                occupation=data['occupation'],
+                location=data['location'],
+                relationships=data.get('relationships', {}),
+                goals=data.get('goals', [])
+            )
+            initialize_memories = data.get('initialize_memories', [])
+
+            characters[character] = initialize_memories
+            # characters.append((character, initialize_memories))
+            print(f"Loaded character: {character.name} from {yaml_file.name}")
+
+        except yaml.YAMLError as e:
+            print(f"Error parsing YAML file '{yaml_file.name}': {e}")
+        except Exception as e:
+            print(f"Error loading character from '{yaml_file.name}': {e}")
+
+    return characters