Working with Together AI in Pixeltable

%pip install -qU pixeltable together

import getpass
import os

if 'TOGETHER_API_KEY' not in os.environ:
    os.environ['TOGETHER_API_KEY'] = getpass.getpass('Together API Key: ')

import pixeltable as pxt

# Remove the 'together_demo' directory and its contents, if it exists
pxt.drop_dir('together_demo', force=True)
pxt.create_dir('together_demo')

from pixeltable.functions import together

chat_t = pxt.create_table('together_demo/chat', {'input': pxt.String})

messages = [{'role': 'user', 'content': chat_t.input}]

chat_t.add_computed_column(
    output=together.chat_completions(
        messages=messages,
        model='Qwen/Qwen3.5-9B',
        model_kwargs={
            # Optional dict with parameters for the Together API
            'max_tokens': 300,
            'stop': ['\n'],
            'temperature': 0.7,
            'top_p': 0.9,
        },
    )
)
chat_t.add_computed_column(
    response=chat_t.output.choices[0].message.content
)

# Start a conversation
chat_t.insert(
    [
        {'input': 'How many species of felids have been classified?'},
        {'input': 'Can you make me a coffee?'},
    ]
)
chat_t.select(chat_t.input, chat_t.response).head()

emb_t = pxt.create_table(
    'together_demo/embeddings', {'input': pxt.String}
)
emb_t.add_computed_column(
    embedding=together.embeddings(
        input=emb_t.input, model='intfloat/multilingual-e5-large-instruct'
    )
)

emb_t.insert(
    [{'input': 'Together AI provides a variety of embeddings models.'}]
)

emb_t.head()

image_t = pxt.create_table('together_demo/images', {'input': pxt.String})
image_t.add_computed_column(
    img=together.image_generations(
        image_t.input,
        model='black-forest-labs/FLUX.1-schnell',
        model_kwargs={'steps': 5},
    )
)

image_t.insert(
    [{'input': 'A friendly dinosaur playing tennis in a cornfield'}]
)

image_t

image_t.head()