Fireworks is a platform for providing endpoints for Large Language models.

Authentication

Set your FIREWORKS_API_KEY environment variable. Get your key from here.

export FIREWORKS_API_KEY=***

Example

Use Fireworks with your Agent:

from phi.agent import Agent, RunResponse
from phi.model.fireworks import Fireworks

agent = Agent(
    model=Fireworks(id="accounts/fireworks/models/firefunction-v2"),
    markdown=True
)

# Get the response in a variable
# run: RunResponse = agent.run("Share a 2 sentence horror story.")
# print(run.content)

# Print the response in the terminal
agent.print_response("Share a 2 sentence horror story.")

Params

ParameterTypeDefaultDescription
idstr"accounts/fireworks/models/firefunction-v2"The specific model ID used for generating responses.
namestr"Fireworks: {id}"The name identifier for the agent. Defaults to "Fireworks: " followed by the model ID.
providerstr"Fireworks"The provider of the model.
api_keyOptional[str]-The API key for authenticating requests to the service. Retrieved from the environment variable FIREWORKS_API_KEY.
base_urlstr"https://api.fireworks.ai/inference/v1"The base URL for making API requests to the Fireworks service.

Fireworks also supports the params of OpenAI.