fix type issue

landing-ai · Feb 20, 2024 · c994011 · c994011
1 parent 220e83c
commit c994011
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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": [