# constants to use

# OpenAI constants
SYSTEM = "system"
ROLE = "role"
USER = "user"
CONTENT = "content"
GPT_4o = "gpt-4o"
GPT_4o_MINI = "gpt-4o-mini"
DEEPSEEK_V3 = "deepseek-ai/DeepSeek-V3"

LLAMA_START_HEADER = "<|start_header_id|>"
LLAMA_END_HEADER = "<|end_header_id|>"
BODY = "body"
MESSAGE = "message"
MESSAGES = "messages"
TEMPERATURE = "temperature"
TOP_P = "top_p"
MODEL = "model"
MAX_COMPLETION_TOKENS = "max_completion_tokens"
CUSTOM_ID = "custom_id"
INPUT_FILE = "input_file"
RESPONSE = "response"
CHOICES = "choices"
RESULT_FILE_ID = "result_file_id"
PROMPTS = "prompts"
PROMPT = "prompt"
TEXT = "text"
RESULT = "result"
PARAMS = "params"
JOB_NAME = "job_name"
API_CALL = "api_call"
TARGET = "target"
BATCH_ID = "batch_id"
TYPE = "type"
MAX_TOKENS = "max_tokens"
BATCH = "batch"
BLOCK_NONE = "BLOCK_NONE"
MODELINPUT = "modelInput"
RECORDID = "recordId"
TIMESTAMP = "timestamp"
CREATED = "created"
DISPLAY_NAME = "display_name"
KEY = "key"
CANDIDATES = "candidates"
PARTS = "parts"
EXAMPLES = "examples"


CLAUDE = "claude"
GPT = "gpt"
MISTRAL = "mistral"
ANTHROPIC = "anthropic"
OPENAI = "openai"
BEDROCK = "bedrock"
GEMINI = "gemini"
DEEPSEEK = "deepseek"



PX = "p(x)"
ENTROPY = "entropy"
COUNT = "count"


S3URI = "s3Uri"
JOB_ARN = "jobArn"

S3_OUTPUT_DATA_CONFIG = "s3OutputDataConfig"
S3_INPUT_DATA_CONFIG  = "s3InputDataConfig"
MODEL_OUTPUT = "modelOutput"
GENERATION = "generation"

OUTPUT_DATASET_ID = "outputDatasetId"
INPUT_DATASET_ID = "inputDatasetId"
FIREWORKS_AI = "FireworksAI"
USAGE = "usage"
OBJECT = "object"
ID = "id"

