only support str or list[str] for product data

app/schemas/requests.py

@@ -18,7 +18,7 @@ class ExtractionRequest(BaseModel):
     img_urls: Optional[List[str]] = None
     product_taxonomy: str
     request_meta: Optional[Dict[str, str]] = None
-    product_data: Optional[Dict[str, Union[str, List]]] = None
+    product_data: Optional[Dict[str, Union[str, List[str]]]] = None
     ai_model: str = settings.DEFAULT_MODEL  # type: ignore
     max_attempts: int = settings.DEFAULT_MAX_ATTEMPTS  # type: ignore
 

app/services/base.py

@@ -34,7 +34,7 @@ class BaseAttributionService(ABC):
         ai_model: str,
         img_urls: List[str],
         product_taxonomy: str,
-        product_data: Dict[str, Union[str, List]],
+        product_data: Dict[str, Union[str, List[str]]],
         pil_images: List[Any] = None,
         img_paths: List[str] = None,
     ) -> Dict[str, Any]:

app/services/service_anthropic.py

@@ -41,7 +41,7 @@ class AnthropicService(BaseAttributionService):
         ai_model: str,
         img_urls: List[str],
         product_taxonomy: str,
-        product_data: Dict[str, Union[str, List]],
+        product_data: Dict[str, Union[str, List[str]]],
         pil_images: List[Any] = None,  # do not remove, this is for weave
         img_paths: List[str] = None,
     ) -> Dict[str, Any]:

app/services/service_openai.py

@@ -58,7 +58,7 @@ class OpenAIService(BaseAttributionService):
         ai_model: str,
         img_urls: List[str],
         product_taxonomy: str,
-        product_data: Dict[str, Union[str, List]],
+        product_data: Dict[str, Union[str, List[str]]],
         pil_images: List[Any] = None,  # do not remove, this is for weave
         img_paths: List[str] = None,
     ) -> Dict[str, Any]:

app/utils/converter.py

@@ -1,4 +1,7 @@
-def product_data_to_str(product_data: dict[str, any]) -> str:
+from typing import Dict, List, Union
+
+
+def product_data_to_str(product_data: Dict[str, Union[str, List[str]]]) -> str:
     """
     Convert product data to a string.