change model type

app.py

@@ -33,7 +33,7 @@ async def onClick(vacationType,familyType,duration,purpose,interests):
     "Visual Arts: Imagery from H.R. Giger (Alien) or Moebius (The Incal).",
     "Graphic Novels: Saga, Transmetropolitan."
     ),
-    result_type = customModel.Planet
+    result_type = customModel.TravelAgency
     )
     result_1 = agent.run_sync(user_prompt="for space travel which planet is the best")
     result = result_1

planet_model.py

@@ -4,7 +4,7 @@ from dataclasses import dataclass, field
 from typing import List, Dict
 
 @dataclass
-class Planet(BaseModel):
+class Planet:
     name: str
     description: str
     accommodation: str
@@ -12,7 +12,7 @@ class Planet(BaseModel):
     photography: str
 
 @dataclass
-class Category(BaseModel):
+class Category:
     budget_range: str
     planets: List[Planet] = field(default_factory=list)