- Replace CustomerContext to dataclass, - simplified main function by removing intermediate choice variable.

Author: SheikhMuhammadHamza-HS

examples/basic/dynamic_system_prompt.py

@@ -1,13 +1,12 @@
 import asyncio
 import random
 from typing import Literal
-
 from agents import Agent, RunContextWrapper, Runner
+from dataclasses import dataclass
 
-
+@dataclass
 class CustomContext:
-    def __init__(self, style: Literal["haiku", "pirate", "robot"]):
-        self.style = style
+    style: Literal["haiku", "pirate", "robot"] 
 
 
 def custom_instructions(
@@ -27,11 +26,9 @@ def custom_instructions(
     instructions=custom_instructions,
 )
 
-
 async def main():
-    choice: Literal["haiku", "pirate", "robot"] = random.choice(["haiku", "pirate", "robot"])
-    context = CustomContext(style=choice)
-    print(f"Using style: {choice}\n")
+    context = CustomContext(style=random.choice(["haiku", "pirate", "robot"]))
+    print(f"Using style: {context.style}\n")
 
     user_message = "Tell me a joke."
     print(f"User: {user_message}")
@@ -43,6 +40,7 @@ async def main():
 if __name__ == "__main__":
     asyncio.run(main())
 
+
 """
 $ python examples/basic/dynamic_system_prompt.py