import osfrom scale_gp import SGPClientfrom scale_gp.types.completion_create_params import ModelParametersfrom scale_gp.types.chat_completion_create_params import ( ModelParameters as ChatModelParameters, MessageEgpAPIBackendServerAPIModelsEgpModelsUserMessage,)client = SGPClient(api_key=api_key)model_parameters = ModelParameters(max_tokens=200, temperature=0.5, top_k=1, top_p=1)completion = client.completions.create( model="gpt-4", prompt="Why is the sky blue?", account_id=account_id, model_parameters=model_parameters,)print(completion)chat_model_parameters = ChatModelParameters(max_tokens=200, temperature=0.5, top_k=1, top_p=1)message = MessageEgpAPIBackendServerAPIModelsEgpModelsUserMessage(role="user", content="What is the capital of Canada?")chat_completion = client.chat_completions.create( model="gpt-4", messages=[ message ], # messages is a list of historical messages in the conversation, with roles usually alternating between user, assistant and system account_id=account_id, model_parameters=model_parameters, instructions="Answer the question like an elementary school teacher.",)print(chat_completion)
CompletionsResponse( completion=Completion(text="The sky appears blue because of the way Earth's atmosphere scatters sunlight in all directions and blue light is scattered more than other colors because it travels in smaller, shorter waves. This is known as Rayleigh scattering.", finish_reason='stop', response_metadata=None), token_usage=TokenUsage(total=56, completion=43, prompt=13))
Inference
Create Completion and Chat Completions
Use this recipe to use the Scale GenAI Platform SDK to perform completion and chat completions
Use this file to discover all available pages before exploring further.
1. Instantiate Client
Follow the instructions in the Quickstart Guide to setup the SGP Client
from scale_gp import SGPClientclient = SGPClient(api_key=api_key)
2. Perform a completion
Define model name and parameters to perform a completion.
model_parameters = ModelParameters(max_tokens=200, temperature=0.5, top_k=1, top_p=1)completion = client.completions.create( model="gpt-4", prompt="Why is the sky blue?", account_id=account_id, model_parameters=model_parameters,)
3. Perform a chat completion
Call a chat model to perform a completion using a message and the parameters defined above. Chat completion allows for a series of messages to be sent to the model,
serving as a conversation between the user and the model
chat_model_parameters = ChatModelParameters(max_tokens=200, temperature=0.5, top_k=1, top_p=1)message = MessageEgpAPIBackendServerAPIModelsEgpModelsUserMessage(role="user", content="What is the capital of Canada?")chat_completion = client.chat_completions.create( model="gpt-4", messages=[ message ], # messages is a list of historical messages in the conversation, with roles usually alternating between user, assistant and system account_id=account_id, model_parameters=model_parameters, instructions="Answer the question like an elementary school teacher.",)
import osfrom scale_gp import SGPClientfrom scale_gp.types.completion_create_params import ModelParametersfrom scale_gp.types.chat_completion_create_params import ( ModelParameters as ChatModelParameters, MessageEgpAPIBackendServerAPIModelsEgpModelsUserMessage,)client = SGPClient(api_key=api_key)model_parameters = ModelParameters(max_tokens=200, temperature=0.5, top_k=1, top_p=1)completion = client.completions.create( model="gpt-4", prompt="Why is the sky blue?", account_id=account_id, model_parameters=model_parameters,)print(completion)chat_model_parameters = ChatModelParameters(max_tokens=200, temperature=0.5, top_k=1, top_p=1)message = MessageEgpAPIBackendServerAPIModelsEgpModelsUserMessage(role="user", content="What is the capital of Canada?")chat_completion = client.chat_completions.create( model="gpt-4", messages=[ message ], # messages is a list of historical messages in the conversation, with roles usually alternating between user, assistant and system account_id=account_id, model_parameters=model_parameters, instructions="Answer the question like an elementary school teacher.",)print(chat_completion)
CompletionsResponse( completion=Completion(text="The sky appears blue because of the way Earth's atmosphere scatters sunlight in all directions and blue light is scattered more than other colors because it travels in smaller, shorter waves. This is known as Rayleigh scattering.", finish_reason='stop', response_metadata=None), token_usage=TokenUsage(total=56, completion=43, prompt=13))