pixeltable 0.3.2__py3-none-any.whl → 0.3.3__py3-none-any.whl

pixeltable/functions/replicate.py

@@ -18,6 +18,7 @@ if TYPE_CHECKING:
 @register_client('replicate')
 def _(api_token: str) -> 'replicate.Client':
     import replicate
+
     return replicate.Client(api_token=api_token)
 
 
@@ -25,17 +26,17 @@ def _replicate_client() -> 'replicate.Client':
     return Env.get().get_client('replicate')
 
 
-@pxt.udf
-def run(
-    input: dict[str, Any],
-    *,
-    ref: str,
-) -> dict[str, Any]:
+@pxt.udf(resource_pool='request-rate:replicate')
+async def run(input: dict[str, Any], *, ref: str) -> dict[str, Any]:
     """
     Run a model on Replicate.
 
     For additional details, see: <https://replicate.com/docs/topics/models/run-a-model>
 
+    Request throttling:
+    Applies the rate limit set in the config (section `replicate`, key `rate_limit`). If no rate
+    limit is configured, uses a default of 600 RPM.
+
     __Requirements:__
 
     - `pip install replicate`
@@ -52,17 +53,17 @@ def run(
         to an existing Pixeltable column `tbl.prompt` of the table `tbl`:
 
         >>> input = {'system_prompt': 'You are a helpful assistant.', 'prompt': tbl.prompt}
-        ... tbl['response'] = run(input, ref='meta/meta-llama-3-8b-instruct')
+        ... tbl.add_computed_column(response=run(input, ref='meta/meta-llama-3-8b-instruct'))
 
         Add a computed column that uses the model `black-forest-labs/flux-schnell`
         to generate images from an existing Pixeltable column `tbl.prompt`:
 
         >>> input = {'prompt': tbl.prompt, 'go_fast': True, 'megapixels': '1'}
-        ... tbl['response'] = run(input, ref='black-forest-labs/flux-schnell')
-        ... tbl['image'] = tbl.response.output[0].astype(pxt.Image)
+        ... tbl.add_computed_column(response=run(input, ref='black-forest-labs/flux-schnell'))
+        ... tbl.add_computed_column(image=tbl.response.output[0].astype(pxt.Image))
     """
     Env.get().require_package('replicate')
-    return _replicate_client().run(ref, input, use_file_output=False)
+    return await _replicate_client().async_run(ref, input, use_file_output=False)
 
 
 __all__ = local_public_names(__name__)

pixeltable/functions/string.py

@@ -28,6 +28,7 @@ def capitalize(self: str) -> str:
     """
     return self.capitalize()
 
+
 @pxt.udf(is_method=True)
 def casefold(self: str) -> str:
     """
@@ -37,6 +38,7 @@ def casefold(self: str) -> str:
     """
     return self.casefold()
 
+
 @pxt.udf(is_method=True)
 def center(self: str, width: int, fillchar: str = ' ') -> str:
     """
@@ -50,6 +52,7 @@ def center(self: str, width: int, fillchar: str = ' ') -> str:
     """
     return self.center(width, fillchar)
 
+
 @pxt.udf(is_method=True)
 def contains(self: str, pattern: str, case: bool = True, flags: int = 0, regex: bool = True) -> bool:
     """
@@ -63,6 +66,7 @@ def contains(self: str, pattern: str, case: bool = True, flags: int = 0, regex:
     """
     if regex:
         import re
+
         if not case:
             flags |= re.IGNORECASE
         return bool(re.search(pattern, self, flags))
@@ -72,6 +76,7 @@ def contains(self: str, pattern: str, case: bool = True, flags: int = 0, regex:
         else:
             return pattern.lower() in self.lower()
 
+
 @pxt.udf(is_method=True)
 def count(self: str, pattern: str, flags: int = 0) -> int:
     """
@@ -83,8 +88,10 @@ def count(self: str, pattern: str, flags: int = 0) -> int:
     """
     import re
     from builtins import len
+
     return len(re.findall(pattern, self, flags))
 
+
 @pxt.udf(is_method=True)
 def endswith(self: str, pattern: str) -> bool:
     """
@@ -97,6 +104,7 @@ def endswith(self: str, pattern: str) -> bool:
     """
     return self.endswith(pattern)
 
+
 @pxt.udf(is_method=True)
 def fill(self: str, width: int, **kwargs: Any) -> str:
     """
@@ -109,8 +117,10 @@ def fill(self: str, width: int, **kwargs: Any) -> str:
         kwargs: Additional keyword arguments to pass to `textwrap.fill()`.
     """
     import textwrap
+
     return textwrap.fill(self, width, **kwargs)
 
+
 @pxt.udf(is_method=True)
 def find(self: str, substr: str, start: Optional[int] = 0, end: Optional[int] = None) -> int:
     """
@@ -125,6 +135,7 @@ def find(self: str, substr: str, start: Optional[int] = 0, end: Optional[int] =
     """
     return self.find(substr, start, end)
 
+
 @pxt.udf(is_method=True)
 def findall(self: str, pattern: str, flags: int = 0) -> list:
     """
@@ -137,8 +148,10 @@ def findall(self: str, pattern: str, flags: int = 0) -> list:
         flags: [flags](https://docs.python.org/3/library/re.html#flags) for the `re` module
     """
     import re
+
     return re.findall(pattern, self, flags)
 
+
 @pxt.udf(is_method=True)
 def format(self: str, *args: Any, **kwargs: Any) -> str:
     """
@@ -148,6 +161,7 @@ def format(self: str, *args: Any, **kwargs: Any) -> str:
     """
     return self.format(*args, **kwargs)
 
+
 @pxt.udf(is_method=True)
 def fullmatch(self: str, pattern: str, case: bool = True, flags: int = 0) -> bool:
     """
@@ -161,11 +175,13 @@ def fullmatch(self: str, pattern: str, case: bool = True, flags: int = 0) -> boo
         flags: [flags](https://docs.python.org/3/library/re.html#flags) for the `re` module
     """
     import re
+
     if not case:
         flags |= re.IGNORECASE
     _ = bool(re.fullmatch(pattern, self, flags))
     return bool(re.fullmatch(pattern, self, flags))
 
+
 @pxt.udf(is_method=True)
 def index(self: str, substr: str, start: Optional[int] = 0, end: Optional[int] = None) -> int:
     """
@@ -181,6 +197,7 @@ def index(self: str, substr: str, start: Optional[int] = 0, end: Optional[int] =
     """
     return self.index(substr, start, end)
 
+
 @pxt.udf(is_method=True)
 def isalnum(self: str) -> bool:
     """
@@ -191,6 +208,7 @@ def isalnum(self: str) -> bool:
     """
     return self.isalnum()
 
+
 @pxt.udf(is_method=True)
 def isalpha(self: str) -> bool:
     """
@@ -200,6 +218,7 @@ def isalpha(self: str) -> bool:
     """
     return self.isalpha()
 
+
 @pxt.udf(is_method=True)
 def isascii(self: str) -> bool:
     """
@@ -209,6 +228,7 @@ def isascii(self: str) -> bool:
     """
     return self.isascii()
 
+
 @pxt.udf(is_method=True)
 def isdecimal(self: str) -> bool:
     """
@@ -219,6 +239,7 @@ def isdecimal(self: str) -> bool:
     """
     return self.isdecimal()
 
+
 @pxt.udf(is_method=True)
 def isdigit(self: str) -> bool:
     """
@@ -228,6 +249,7 @@ def isdigit(self: str) -> bool:
     """
     return self.isdigit()
 
+
 @pxt.udf(is_method=True)
 def isidentifier(self: str) -> bool:
     """
@@ -247,6 +269,7 @@ def islower(self: str) -> bool:
     """
     return self.islower()
 
+
 @pxt.udf(is_method=True)
 def isnumeric(self: str) -> bool:
     """
@@ -256,6 +279,7 @@ def isnumeric(self: str) -> bool:
     """
     return self.isnumeric()
 
+
 @pxt.udf(is_method=True)
 def isupper(self: str) -> bool:
     """
@@ -265,6 +289,7 @@ def isupper(self: str) -> bool:
     """
     return self.isupper()
 
+
 @pxt.udf(is_method=True)
 def istitle(self: str) -> bool:
     """
@@ -274,6 +299,7 @@ def istitle(self: str) -> bool:
     """
     return self.istitle()
 
+
 @pxt.udf(is_method=True)
 def isspace(self: str) -> bool:
     """
@@ -283,6 +309,7 @@ def isspace(self: str) -> bool:
     """
     return self.isspace()
 
+
 @pxt.udf
 def join(sep: str, elements: list) -> str:
     """
@@ -292,6 +319,7 @@ def join(sep: str, elements: list) -> str:
     """
     return sep.join(elements)
 
+
 @pxt.udf(is_method=True)
 def len(self: str) -> int:
     """
@@ -301,6 +329,7 @@ def len(self: str) -> int:
     """
     return self.__len__()
 
+
 @pxt.udf(is_method=True)
 def ljust(self: str, width: int, fillchar: str = ' ') -> str:
     """
@@ -314,6 +343,7 @@ def ljust(self: str, width: int, fillchar: str = ' ') -> str:
     """
     return self.ljust(width, fillchar)
 
+
 @pxt.udf(is_method=True)
 def lower(self: str) -> str:
     """
@@ -323,6 +353,7 @@ def lower(self: str) -> str:
     """
     return self.lower()
 
+
 @pxt.udf(is_method=True)
 def lstrip(self: str, chars: Optional[str] = None) -> str:
     """
@@ -336,6 +367,7 @@ def lstrip(self: str, chars: Optional[str] = None) -> str:
     """
     return self.lstrip(chars)
 
+
 @pxt.udf(is_method=True)
 def match(self: str, pattern: str, case: bool = True, flags: int = 0) -> bool:
     """
@@ -347,10 +379,12 @@ def match(self: str, pattern: str, case: bool = True, flags: int = 0) -> bool:
         flags: [flags](https://docs.python.org/3/library/re.html#flags) for the `re` module
     """
     import re
+
     if not case:
         flags |= re.IGNORECASE
     return bool(re.match(pattern, self, flags))
 
+
 @pxt.udf(is_method=True)
 def normalize(self: str, form: str) -> str:
     """
@@ -362,8 +396,10 @@ def normalize(self: str, form: str) -> str:
         form: Unicode normal form (`‘NFC’`, `‘NFKC’`, `‘NFD’`, `‘NFKD’`)
     """
     import unicodedata
+
     return unicodedata.normalize(form, self)  # type: ignore[arg-type]
 
+
 @pxt.udf(is_method=True)
 def pad(self: str, width: int, side: str = 'left', fillchar: str = ' ') -> str:
     """
@@ -381,7 +417,8 @@ def pad(self: str, width: int, side: str = 'left', fillchar: str = ' ') -> str:
     elif side == 'both':
         return self.center(width, fillchar)
     else:
-        raise ValueError(f"Invalid side: {side}")
+        raise ValueError(f'Invalid side: {side}')
+
 
 @pxt.udf(is_method=True)
 def partition(self: str, sep: str = ' ') -> list:
@@ -394,7 +431,9 @@ def partition(self: str, sep: str = ' ') -> list:
     if idx == -1:
         return [self, '', '']
     from builtins import len
-    return [self[:idx], sep, self[idx + len(sep):]]
+
+    return [self[:idx], sep, self[idx + len(sep) :]]
+
 
 @pxt.udf(is_method=True)
 def removeprefix(self: str, prefix: str) -> str:
@@ -404,9 +443,11 @@ def removeprefix(self: str, prefix: str) -> str:
     if self.startswith(prefix):
         # we need to avoid referring to our symbol 'len'
         from builtins import len
-        return self[len(prefix):]
+
+        return self[len(prefix) :]
     return self
 
+
 @pxt.udf(is_method=True)
 def removesuffix(self: str, suffix: str) -> str:
     """
@@ -415,9 +456,11 @@ def removesuffix(self: str, suffix: str) -> str:
     if self.endswith(suffix):
         # we need to avoid referring to our symbol 'len'
         from builtins import len
-        return self[:-len(suffix)]
+
+        return self[: -len(suffix)]
     return self
 
+
 @pxt.udf(is_method=True)
 def repeat(self: str, n: int) -> str:
     """
@@ -425,9 +468,10 @@ def repeat(self: str, n: int) -> str:
     """
     return self * n
 
+
 @pxt.udf(is_method=True)
 def replace(
-        self: str, pattern: str, repl: str, n: int = -1, case: bool = True, flags: int = 0, regex: bool = False
+    self: str, pattern: str, repl: str, n: int = -1, case: bool = True, flags: int = 0, regex: bool = False
 ) -> str:
     """
     Replace occurrences of `pattern` with `repl`.
@@ -445,12 +489,14 @@ def replace(
     """
     if regex:
         import re
+
         if not case:
             flags |= re.IGNORECASE
         return re.sub(pattern, repl, self, 0 if n == -1 else n, flags)
     else:
         return self.replace(pattern, repl, n)
 
+
 @pxt.udf(is_method=True)
 def rfind(self: str, substr: str, start: Optional[int] = 0, end: Optional[int] = None) -> int:
     """
@@ -465,6 +511,7 @@ def rfind(self: str, substr: str, start: Optional[int] = 0, end: Optional[int] =
     """
     return self.rfind(substr, start, end)
 
+
 @pxt.udf(is_method=True)
 def rindex(self: str, substr: str, start: Optional[int] = 0, end: Optional[int] = None) -> int:
     """
@@ -475,6 +522,7 @@ def rindex(self: str, substr: str, start: Optional[int] = 0, end: Optional[int]
     """
     return self.rindex(substr, start, end)
 
+
 @pxt.udf(is_method=True)
 def rjust(self: str, width: int, fillchar: str = ' ') -> str:
     """
@@ -488,6 +536,7 @@ def rjust(self: str, width: int, fillchar: str = ' ') -> str:
     """
     return self.rjust(width, fillchar)
 
+
 @pxt.udf(is_method=True)
 def rpartition(self: str, sep: str = ' ') -> list:
     """
@@ -498,7 +547,9 @@ def rpartition(self: str, sep: str = ' ') -> list:
     if idx == -1:
         return [self, '', '']
     from builtins import len
-    return [self[:idx], sep, self[idx + len(sep):]]
+
+    return [self[:idx], sep, self[idx + len(sep) :]]
+
 
 @pxt.udf(is_method=True)
 def rstrip(self: str, chars: Optional[str] = None) -> str:
@@ -512,6 +563,7 @@ def rstrip(self: str, chars: Optional[str] = None) -> str:
     """
     return self.rstrip(chars)
 
+
 @pxt.udf(is_method=True)
 def slice(self: str, start: Optional[int] = None, stop: Optional[int] = None, step: Optional[int] = None) -> str:
     """
@@ -524,8 +576,11 @@ def slice(self: str, start: Optional[int] = None, stop: Optional[int] = None, st
     """
     return self[start:stop:step]
 
+
 @pxt.udf(is_method=True)
-def slice_replace(self: str, start: Optional[int] = None, stop: Optional[int] = None, repl: Optional[str] = None) -> str:
+def slice_replace(
+    self: str, start: Optional[int] = None, stop: Optional[int] = None, repl: Optional[str] = None
+) -> str:
     """
     Replace a positional slice with another value.
 
@@ -536,6 +591,7 @@ def slice_replace(self: str, start: Optional[int] = None, stop: Optional[int] =
     """
     return self[:start] + repl + self[stop:]
 
+
 @pxt.udf(is_method=True)
 def startswith(self: str, pattern: str) -> int:
     """
@@ -548,6 +604,7 @@ def startswith(self: str, pattern: str) -> int:
     """
     return self.startswith(pattern)
 
+
 @pxt.udf(is_method=True)
 def strip(self: str, chars: Optional[str] = None) -> str:
     """
@@ -560,6 +617,7 @@ def strip(self: str, chars: Optional[str] = None) -> str:
     """
     return self.strip(chars)
 
+
 @pxt.udf(is_method=True)
 def swapcase(self: str) -> str:
     """
@@ -569,6 +627,7 @@ def swapcase(self: str) -> str:
     """
     return self.swapcase()
 
+
 @pxt.udf(is_method=True)
 def title(self: str) -> str:
     """
@@ -579,6 +638,7 @@ def title(self: str) -> str:
     """
     return self.title()
 
+
 @pxt.udf(is_method=True)
 def upper(self: str) -> str:
     """
@@ -588,6 +648,7 @@ def upper(self: str) -> str:
     """
     return self.upper()
 
+
 @pxt.udf(is_method=True)
 def wrap(self: str, width: int, **kwargs: Any) -> list[str]:
     """
@@ -601,8 +662,10 @@ def wrap(self: str, width: int, **kwargs: Any) -> list[str]:
         kwargs: Additional keyword arguments to pass to `textwrap.fill()`.
     """
     import textwrap
+
     return textwrap.wrap(self, width, **kwargs)
 
+
 @pxt.udf(is_method=True)
 def zfill(self: str, width: int) -> str:
     """

pixeltable/functions/timestamp.py

@@ -135,6 +135,7 @@ def astimezone(self: datetime, tz: str) -> datetime:
         tz: The time zone to convert to. Must be a valid time zone name from the IANA Time Zone Database.
     """
     from zoneinfo import ZoneInfo
+
     tzinfo = ZoneInfo(tz)
     return self.astimezone(tzinfo)
 
@@ -210,7 +211,7 @@ def strftime(self: datetime, format: str) -> str:
 
 @pxt.udf(is_method=True)
 def make_timestamp(
-        year: int, month: int, day: int, hour: int = 0, minute: int = 0, second: int = 0, microsecond: int = 0
+    year: int, month: int, day: int, hour: int = 0, minute: int = 0, second: int = 0, microsecond: int = 0
 ) -> datetime:
     """
     Create a timestamp.
@@ -222,9 +223,13 @@ def make_timestamp(
 
 @make_timestamp.to_sql
 def _(
-        year: sql.ColumnElement, month: sql.ColumnElement, day: sql.ColumnElement,
-        hour: sql.ColumnElement = sql.literal(0), minute: sql.ColumnElement = sql.literal(0),
-        second: sql.ColumnElement = sql.literal(0), microsecond: sql.ColumnElement = sql.literal(0)
+    year: sql.ColumnElement,
+    month: sql.ColumnElement,
+    day: sql.ColumnElement,
+    hour: sql.ColumnElement = sql.literal(0),
+    minute: sql.ColumnElement = sql.literal(0),
+    second: sql.ColumnElement = sql.literal(0),
+    microsecond: sql.ColumnElement = sql.literal(0),
 ) -> sql.ColumnElement:
     return sql.func.make_timestamptz(
         sql.cast(year, sql.Integer),
@@ -232,7 +237,9 @@ def _(
         sql.cast(day, sql.Integer),
         sql.cast(hour, sql.Integer),
         sql.cast(minute, sql.Integer),
-        sql.cast(second + microsecond / 1000000.0, sql.Float))
+        sql.cast(second + microsecond / 1000000.0, sql.Float),
+    )
+
 
 # @pxt.udf
 # def date(self: datetime) -> datetime:
@@ -258,9 +265,15 @@ def _(
 
 @pxt.udf(is_method=True)
 def replace(
-        self: datetime, year: Optional[int] = None, month: Optional[int] = None, day: Optional[int] = None,
-        hour: Optional[int] = None, minute: Optional[int] = None, second: Optional[int] = None,
-        microsecond: Optional[int] = None) -> datetime:
+    self: datetime,
+    year: Optional[int] = None,
+    month: Optional[int] = None,
+    day: Optional[int] = None,
+    hour: Optional[int] = None,
+    minute: Optional[int] = None,
+    second: Optional[int] = None,
+    microsecond: Optional[int] = None,
+) -> datetime:
     """
     Return a datetime with the same attributes, except for those attributes given new values by whichever keyword
     arguments are specified.