diff --git a/docs/examples/planning-tasks.md b/docs/examples/planning-tasks.md
index 6960c30c5..65d6e71c0 100644
--- a/docs/examples/planning-tasks.md
+++ b/docs/examples/planning-tasks.md
@@ -20,7 +20,8 @@ With the OpenAI Function Call model, you can customize the planning process and
 Let's define the necessary Pydantic models to represent the query plan and the queries.
 
 ```python
-from typing import List
+import enum
+from typing import List, Literal
 from pydantic import Field, BaseModel