Working with Fireworks AI in Pixeltable

Pixeltable's Fireworks integration enables you to access LLMs hosted on the Fireworks platform.

Providers · Notebook · Updated

Colab runs it in your browser. If it calls a model provider, it asks for that provider's API key in a cell.

What it uses

Install
terminal
pip install pixeltable fireworks-ai
Environment variables it names
FIREWORKS_API_KEY
Pixeltable calls
fireworks.chat_completions()pxt.create_table()add_computed_column()insert()

A look at the code

working-with-fireworks.ipynb
from pixeltable.functions.fireworks import chat_completions
# Create a table in Pixeltable and pick a model hosted on Fireworks with some parameters
t = pxt.create_table('fireworks_demo/chat', {'input': pxt.String})
messages = [{'role': 'user', 'content': t.input}]
t.add_computed_column(
output=chat_completions(
messages=messages,
model='accounts/fireworks/models/gpt-oss-120b',
model_kwargs={
# Optional dict with parameters for the Fireworks API
'max_tokens': 300,
'top_k': 40,
'top_p': 0.9,
'temperature': 0.7,
},
)
)

One cell of the notebook, copied as it is. Open it in Colab for the rest.

Browse all 103