@@ -47,7 +47,7 @@ class Seedream(SimpleService):
     def predict_price(cls, content: str, file_exists: bool, info: dict[str, Any]) -> Decimal | None:
         version = info.get('version', 'seedream-boosted')
         size = info.get('size', '2K')
-        price = cls.TOKEN_COST[version][size]
+        price = cls.TOKEN_COST[version].get(size, Decimal('0'))
 
         return price.quantize(Decimal('0.1'), rounding='ROUND_UP')