Class ComplexNestedXMLTool

tests/main/test_xml_tool_message.py:339–395

class ComplexNestedXMLTool(XMLToolMessage):
    request: str = "complex_nested_tool"
    purpose: str = "To present a complex nested structure"

    person: Person
    hobbies: List[str]
    phones: Dict[str, int]
    friends: List[Person] | None = None

    @classmethod
    def examples(cls) -> List[XMLToolMessage | Tuple[str, XMLToolMessage]]:
        return [
            (
                "I want to present a person named John Doe, aged 30, "
                "living at 123 Main St, Anytown, USA, with hobbies of "
                "reading and cycling, "
                " phone numbers: home (1234567890) and work (9876543210)"
                " and two friends: "
                "   Jane Doe, aged 28, living at 456 Elm St, Somewhere, Canada, "
                "   Jack Doe, aged 32, living at 789 Oak St, Anywhere, UK",
                cls(
                    person=Person(
                        name="John Doe",
                        age=30,
                        address=Address(
                            street="123 Main St", city="Anytown", country="USA"
                        ),
                    ),
                    hobbies=["reading", "cycling"],
                    phones={"home": 1234567890, "work": 9876543210},
                    friends=[
                        Person(
                            name="Jane Doe",
                            age=28,
                            address=Address(
                                street="456 Elm St", city="Somewhere", country="Canada"
                            ),
                        ),
                        Person(
                            name="Jack Doe",
                            age=32,
                            address=Address(
                                street="789 Oak St", city="Anywhere", country="UK"
                            ),
                        ),
                    ],
                ),
            )
        ]

    def handle(self) -> ResultTool:
        return ResultTool(
            person=self.person,
            hobbies=self.hobbies,
            phones=self.phones,
            friends=self.friends,
        )