Skip to main content

Run a single prompt and stream the result

Executes a single prompt using specified variables, with a streaming response.

Examples

import os

from freeplay import Freeplay
from freeplay.provider_config import ProviderConfig, OpenAIConfig

FREEPLAY_API_KEY = os.environ["FREEPLAY_API_KEY"]
OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
FREEPLAY_CUSTOMER_NAME = os.environ["FREEPLAY_CUSTOMER_NAME"]
FREEPLAY_PROJECT_ID = os.environ["FREEPLAY_PROJECT_ID"]

freeplay_chat = Freeplay(
    provider_config=ProviderConfig(openai=OpenAIConfig(OPENAI_API_KEY)),
    freeplay_api_key=FREEPLAY_API_KEY,
    api_base=f'https://{FREEPLAY_CUSTOMER_NAME}.freeplay.ai/api')

completion_stream = freeplay_chat.get_completion_stream(project_id=FREEPLAY_PROJECT_ID,
                                          template_name="album_bot",
                                          variables={"pop_star": "Bruno Mars"})

for chunk in completion_stream:
    print("Chunk: %s" % chunk.text.strip())