examples: use type hinting generics in standard collections for struc…

…tured outputs (#389) --------- Signed-off-by: Tomoya Fujita <[email protected]> Co-authored-by: ParthSareen <[email protected]>
ollama · Dec 20, 2024 · 7d1e002 · 7d1e002
1 parent 4daf4af
commit 7d1e002
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/examples/structured-outputs-image.py b/examples/structured-outputs-image.py
@@ -1,6 +1,6 @@
 from pathlib import Path
 from pydantic import BaseModel
-from typing import List, Optional, Literal
+from typing import Literal
 from ollama import chat
 
 
@@ -13,12 +13,12 @@ class Object(BaseModel):
 
 class ImageDescription(BaseModel):
   summary: str
-  objects: List[Object]
+  objects: list[Object]
   scene: str
-  colors: List[str]
+  colors: list[str]
   time_of_day: Literal['Morning', 'Afternoon', 'Evening', 'Night']
   setting: Literal['Indoor', 'Outdoor', 'Unknown']
-  text_content: Optional[str] = None
+  text_content: str | None = None
 
 
 # Get path from user input