pixeltable 0.2.16__py3-none-any.whl → 0.2.17__py3-none-any.whl

pixeltable/__version__.py

@@ -1,3 +1,3 @@
 # These version placeholders will be replaced during build.
-__version__ = "0.2.16"
-__version_tuple__ = (0, 2, 16)
+__version__ = "0.2.17"
+__version_tuple__ = (0, 2, 17)

pixeltable/env.py

@@ -444,6 +444,7 @@ class Env:
             self._spacy_nlp = spacy.load('en_core_web_sm')
         check('tiktoken')
         check('openai')
+        check('anthropic')
         check('together')
         check('fireworks')
         check('label_studio_sdk')

pixeltable/functions/__init__.py

@@ -1,4 +1,4 @@
-from . import audio, fireworks, huggingface, image, json, openai, string, timestamp, together, video, vision
+from . import anthropic, audio, fireworks, huggingface, image, json, openai, string, timestamp, together, video, vision
 from .globals import *
 from pixeltable.utils.code import local_public_names
 

pixeltable/functions/anthropic.py

@@ -0,0 +1,97 @@
+"""
+Pixeltable [UDFs](https://pixeltable.readme.io/docs/user-defined-functions-udfs)
+that wrap various endpoints from the Anthropic API. In order to use them, you must
+first `pip install anthropic` and configure your Anthropic credentials, as described in
+the [Working with Anthropic](https://pixeltable.readme.io/docs/working-with-anthropic) tutorial.
+"""
+
+from typing import TYPE_CHECKING, Any, Optional, TypeVar, Union
+
+import pixeltable as pxt
+from pixeltable import env
+from pixeltable.utils.code import local_public_names
+
+if TYPE_CHECKING:
+    import anthropic
+
+
+@env.register_client('anthropic')
+def _(api_key: str) -> 'anthropic.Anthropic':
+    import anthropic
+
+    return anthropic.Anthropic(api_key=api_key)
+
+
+def _anthropic_client() -> 'anthropic.Anthropic':
+    return env.Env.get().get_client('anthropic')
+
+
+@pxt.udf
+def messages(
+    messages: list[dict[str, str]],
+    *,
+    model: str,
+    max_tokens: int = 1024,
+    metadata: Optional[dict[str, Any]] = None,
+    stop_sequences: Optional[list[str]] = None,
+    system: Optional[str] = None,
+    temperature: Optional[float] = None,
+    tool_choice: Optional[list[dict]] = None,
+    tools: Optional[dict] = None,
+    top_k: Optional[int] = None,
+    top_p: Optional[float] = None,
+) -> dict:
+    """
+    Create a Message.
+
+    Equivalent to the Anthropic `messages` API endpoint.
+    For additional details, see: <https://docs.anthropic.com/en/api/messages>
+
+    __Requirements:__
+
+    - `pip install anthropic`
+
+    Args:
+        messages: Input messages.
+        model: The model that will complete your prompt.
+
+    For details on the other parameters, see: <https://docs.anthropic.com/en/api/messages>
+
+    Returns:
+        A dictionary containing the response and other metadata.
+
+    Examples:
+        Add a computed column that applies the model `claude-3-haiku-20240307`
+        to an existing Pixeltable column `tbl.prompt` of the table `tbl`:
+
+        >>> msgs = [{'role': 'user', 'content': tbl.prompt}]
+        ... tbl['response'] = messages(msgs, model='claude-3-haiku-20240307')
+    """
+    return _anthropic_client().messages.create(
+        messages=messages,
+        model=model,
+        max_tokens=max_tokens,
+        metadata=_opt(metadata),
+        stop_sequences=_opt(stop_sequences),
+        system=_opt(system),
+        temperature=_opt(temperature),
+        tool_choice=_opt(tool_choice),
+        tools=_opt(tools),
+        top_k=_opt(top_k),
+        top_p=_opt(top_p),
+    ).dict()
+
+
+_T = TypeVar('_T')
+
+
+def _opt(arg: _T) -> Union[_T, 'anthropic.NotGiven']:
+    from anthropic import NOT_GIVEN
+    return arg if arg is not None else NOT_GIVEN
+
+
+__all__ = local_public_names(__name__)
+
+
+def __dir__():
+    return __all__

pixeltable/functions/fireworks.py

@@ -60,7 +60,7 @@ def chat_completions(
         to an existing Pixeltable column `tbl.prompt` of the table `tbl`:
 
         >>> messages = [{'role': 'user', 'content': tbl.prompt}]
-        ... tbl['response'] = chat_completions(tbl.prompt, model='accounts/fireworks/models/mixtral-8x22b-instruct')
+        ... tbl['response'] = chat_completions(messages, model='accounts/fireworks/models/mixtral-8x22b-instruct')
     """
     kwargs = {'max_tokens': max_tokens, 'top_k': top_k, 'top_p': top_p, 'temperature': temperature}
     kwargs_not_none = {k: v for k, v in kwargs.items() if v is not None}

pixeltable/functions/together.py

@@ -136,7 +136,7 @@ def chat_completions(
         of the table `tbl`:
 
         >>> messages = [{'role': 'user', 'content': tbl.prompt}]
-        ... tbl['response'] = chat_completions(tbl.prompt, model='mistralai/Mixtral-8x7B-v0.1')
+        ... tbl['response'] = chat_completions(messages, model='mistralai/Mixtral-8x7B-v0.1')
     """
     return (
         _together_client()

{pixeltable-0.2.16.dist-info → pixeltable-0.2.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pixeltable
-Version: 0.2.16
+Version: 0.2.17
 Summary: Pixeltable: The Multimodal AI Data Plane
 Author: Pixeltable, Inc.
 Author-email: contact@pixeltable.com

{pixeltable-0.2.16.dist-info → pixeltable-0.2.17.dist-info}/RECORD

@@ -1,5 +1,5 @@
 pixeltable/__init__.py,sha256=ISVoeqNCvSJonD0EXJ32Mer84m29YQDyLhhUydyCV_o,1272
-pixeltable/__version__.py,sha256=rP4-HoR0fqIhlpjT4eZqvrq6OqDcSAHSNTMYUoFT4xo,114
+pixeltable/__version__.py,sha256=trDVbtw9jdTT0ICXbanwd2shP4jvU1zFX_mKVo08BXQ,114
 pixeltable/catalog/__init__.py,sha256=E41bxaPeQIcgRYzTWc2vkDOboQhRymrJf4IcHQO7o_8,453
 pixeltable/catalog/catalog.py,sha256=8gsFWm6o9Qg4_BEO6oejdpmP4MAOlhmuKRaJP0o2UPU,7906
 pixeltable/catalog/column.py,sha256=_LnuEsHpopccIgh_Uq0BEHUB5Bk_WsvYaiYscmZZVHk,9628
@@ -15,7 +15,7 @@ pixeltable/catalog/table_version.py,sha256=T78HuBg-gph1WauElDzvqenu7ipeK3DaViObW
 pixeltable/catalog/table_version_path.py,sha256=eyQ07VQYQEbl2_AzaLPVLVe9hQ6c1njiIpDjzWhW5Ic,6398
 pixeltable/catalog/view.py,sha256=ZsEblhWsAotLOoPg1Q_Ts3UNnn4I-SVOTh5bJMDnY20,10599
 pixeltable/dataframe.py,sha256=rmKgmAeJrOmD3ELCnbh-jS3VYwt5rMMUpF5pTWvSSaw,34383
-pixeltable/env.py,sha256=uMMOHJoCMkEslOkTSe43BaXPWD1glO5m0BZJ7zvuDIA,21791
+pixeltable/env.py,sha256=7TVo0WpFjtALyct7SEHwKqZf0nuyQ3-smgyJUQ8JbgE,21818
 pixeltable/exceptions.py,sha256=MSP9zeL0AmXT93XqjdvgGN4rzno1_KRrGriq6hpemnw,376
 pixeltable/exec/__init__.py,sha256=VRENEONsAv3PPoBV0r7h-7nAB7SWM4Uglmu1FVQE5uQ,507
 pixeltable/exec/aggregation_node.py,sha256=NtN71QrMmzqnAaWcdTuhovr6FCxSuTpvI3S5a0ZZZYA,3341
@@ -69,9 +69,10 @@ pixeltable/func/globals.py,sha256=sEwn6lGgHMp6VQORb_P5qRd_-Q2_bUSqvqM9-XPN_ec,14
 pixeltable/func/query_template_function.py,sha256=BUU0KZYkqVPjIFg6g6msU4PzVGf2fkEKveDaEMGXhzI,3680
 pixeltable/func/signature.py,sha256=6Lwf32auSnmhYGrN4NkHk07BmG2a73TNICbWnelkH1s,8638
 pixeltable/func/udf.py,sha256=xrfCuYFF7Qqx-I_vE11WT9AoY6UbbzIIZ2DBGzYGzDE,7772
-pixeltable/functions/__init__.py,sha256=N8BRODx5IhozS7SoluCmuufoMpqvDgMSrucT0Kr4uZs,321
+pixeltable/functions/__init__.py,sha256=An9rxP0uG38UYF5kF-wEt9RVioGAjbR9he24rqp7ehc,332
+pixeltable/functions/anthropic.py,sha256=vXiZVIdF3GIKPYLow7_N7m5WC199ZBtAw3q9BUkXnJs,2716
 pixeltable/functions/audio.py,sha256=Z94EB2iJgrXxWYMOGjxJ2KWSDhcEO61Jgs4YoFLYH78,797
-pixeltable/functions/fireworks.py,sha256=69EoYE8aS9s9SV6iTJGeUb7mNEHdodHh96JsOa2KL7Y,2575
+pixeltable/functions/fireworks.py,sha256=2D4TR4RCUHpDcmHEvidMU4gNyLUa8l9wyduaYnekJ9s,2573
 pixeltable/functions/globals.py,sha256=9yz5XzhGhh4kuvHJ8eiFvy9SCPYaZrF5ltwscMH9xEg,2692
 pixeltable/functions/huggingface.py,sha256=KW2OrRZyzqzpN4PekG5wMCWQsPOA61qvZmJzKURUqlk,12790
 pixeltable/functions/image.py,sha256=DQ95DyddxeMAIryBJb-U-y1ck7kvoNYyo-0PLDBaoyk,13313
@@ -79,7 +80,7 @@ pixeltable/functions/json.py,sha256=mJ8oEJMJ2aqt5jCrPSN7vf7oSeQOsMN8Wn1H3PSLoKs,
 pixeltable/functions/openai.py,sha256=4FyL60mQUPcqb19DkFMBbBHf6utV3xsOs6SPWZrfXp8,15981
 pixeltable/functions/string.py,sha256=JEng3jwYZUSYSfjWjto7tLxlIvQLW1JvQ7OmncIgd5M,20085
 pixeltable/functions/timestamp.py,sha256=pfP28wErdjWOMLgJaxqXPMMScf4_LGUFja7oYb_cNBg,6620
-pixeltable/functions/together.py,sha256=Gvy7ontUQmTx1n-eFvGaDEfl48v95eClNy0FmoT3Ivw,8587
+pixeltable/functions/together.py,sha256=M_eIUPiZE5nxFVoUPGTPvkYuBZMgoKlcuIp2ywC1Tus,8585
 pixeltable/functions/util.py,sha256=F2iiIL7UfhYdCVzdCa3efYqWbaeLKFrbycKnuPkG57M,650
 pixeltable/functions/video.py,sha256=7_PF4QBlsH0IvOb_JGxARlyOqhtesBykTRcOUIOyqhI,7148
 pixeltable/functions/vision.py,sha256=ONOpkj426ytWCVtuJn8lmSuPJlqXC38ugcUebHVEFL4,13690
@@ -136,8 +137,8 @@ pixeltable/utils/pytorch.py,sha256=VWczSB_FT_aOU5Xqv4T5ONTsnQN6KDlZmMkuoBuji08,3
 pixeltable/utils/s3.py,sha256=DBfXp0SYubhiKckdAD7PsiVBX_YfVP8Rcu6DCG_3SaQ,433
 pixeltable/utils/sql.py,sha256=5n5_OmXAGtqFdL6z5XvgnU-vlx6Ba6f1WJrO1ZwUle8,765
 pixeltable/utils/transactional_directory.py,sha256=UGzCrGtLR3hEEf8sYGuWBzLVFAEQml3vdIavigWeTBM,1349
-pixeltable-0.2.16.dist-info/LICENSE,sha256=0UNMmwuqWPC0xDY1NWMm4uNJ2_MyA1pnTNRgQTvuBiQ,746
-pixeltable-0.2.16.dist-info/METADATA,sha256=xE2LDpSg2emsk0Gd87nc1-5zF8wxfNxIO8kHB3W2KaM,11252
-pixeltable-0.2.16.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-pixeltable-0.2.16.dist-info/entry_points.txt,sha256=TNI1Gb5vPwFrTdw6TimSYjO8FeK8c_HuPr28vcf7o_I,108
-pixeltable-0.2.16.dist-info/RECORD,,
+pixeltable-0.2.17.dist-info/LICENSE,sha256=0UNMmwuqWPC0xDY1NWMm4uNJ2_MyA1pnTNRgQTvuBiQ,746
+pixeltable-0.2.17.dist-info/METADATA,sha256=s_D-zWxRad72KJyyn9lLatWdaoiWVkMEp72pcN-Vft8,11252
+pixeltable-0.2.17.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+pixeltable-0.2.17.dist-info/entry_points.txt,sha256=TNI1Gb5vPwFrTdw6TimSYjO8FeK8c_HuPr28vcf7o_I,108
+pixeltable-0.2.17.dist-info/RECORD,,