From c994011144341bb8e0d3114ed736140348a20be1 Mon Sep 17 00:00:00 2001
From: Dillon Laird <dillonalaird@gmail.com>
Date: Tue, 20 Feb 2024 12:19:16 -0800
Subject: [PATCH] fix type issue

---
 lmm_tools/lmm/lmm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lmm_tools/lmm/lmm.py b/lmm_tools/lmm/lmm.py
index e6f954a4..c5615114 100644
--- a/lmm_tools/lmm/lmm.py
+++ b/lmm_tools/lmm/lmm.py
@@ -1,7 +1,7 @@
 import base64
 from abc import ABC, abstractmethod
 from pathlib import Path
-from typing import Any, Dict, Optional, Union, cast
+from typing import Any, Dict, List, Optional, Union, cast
 
 
 def encode_image(image: Union[str, Path]) -> str:
@@ -36,7 +36,7 @@ def __init__(self, name: str):
         self.client = OpenAI()
 
     def generate(self, prompt: str, image: Optional[Union[str, Path]]) -> str:
-        message: list[Dict[str, Any]] = [
+        message: List[Dict[str, Any]] = [
             {
                 "role": "user",
                 "content": [