pixeltable 0.2.16__tar.gz → 0.2.17__tar.gz

{pixeltable-0.2.16 → pixeltable-0.2.17}/PKG-INFO

@@ -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 → pixeltable-0.2.17}/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-0.2.16 → pixeltable-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-0.2.16 → pixeltable-0.2.17}/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-0.2.17/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-0.2.16 → pixeltable-0.2.17}/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-0.2.16 → pixeltable-0.2.17}/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 → pixeltable-0.2.17}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry_dynamic_versioning.backend"
 
 [tool.poetry]
 name = "pixeltable"
-version = "0.2.16"
+version = "0.2.17"
 description = "Pixeltable: The Multimodal AI Data Plane"
 authors = ["Pixeltable, Inc. <contact@pixeltable.com>"]
 readme = "README.md"
@@ -83,6 +83,7 @@ torch = "^2.2"
 torchvision = "^0.17"
 pyarrow = ">=13.0.0"
 openai = "^1.10.0"
+anthropic = ">=0.34.2"
 together = "^1.1"
 fireworks-ai = "^0.13.0"
 boto3 = "==1.35.5"  # Locking a specific version of boto3 dramatically improves `poetry lock` runtimes