# Code description generation task
# ———— 5-shot Prompt ————
TEMPLATE_5SHOT = """
Context: The Hubble Space Telescope was launched into low Earth orbit in 1990 aboard the Space Shuttle Discovery. It has since captured landmark images such as the Hubble Ultra-Deep Field, revealing thousands of distant galaxies. In 2009, the final servicing mission upgraded its cameras and sensors.
Question: Which space telescope captured the Ultra-Deep Field image?
Answer: Hubble Space Telescope

Context: ‘The Lord of the Rings’ is a high-fantasy trilogy originally published in three volumes between 1954 and 1955. Written by J.R.R. Tolkien, it follows the quest of Frodo Baggins to destroy the One Ring and defeat the Dark Lord Sauron.
Question: Who is the author of ‘The Lord of the Rings’?
Answer: J.R.R. Tolkien

Context: Ratatouille is a classic vegetable stew from southern France, typically including eggplant, zucchini, bell peppers, tomatoes, onions, and garlic, flavored with herbs de Provence. It is named after a city on the Côte d’Azur where it originated.
Question: Which French city is ratatouille traditionally associated with?
Answer: Nice

Question: {question}{instruction}
Let's think step by step
Answer: 
"""



#


#
# Context: At the 63rd Annual Grammy Awards (held in March 2021), ‘Everything I Wanted’ by Billie Eilish won Record of the Year, ‘Savage’ by Megan Thee Stallion won Best Rap Song, and Taylor Swift’s album ‘Folklore’ won Album of the Year.
# Question: Which album won Album of the Year at the 63rd Grammy Awards?
# Answer: Folklore
#
# Context: The Komodo dragon is the world’s largest lizard species, reaching up to 3 m in length. It is endemic to a few Indonesian islands in the Lesser Sunda chain, most notably the island that shares its name.
# Question: On which Indonesian island is the Komodo dragon predominantly found?
# Answer: Komodo

# ———— Chain-of-Thought Prompt ————
TEMPLATE_COT = """Question: {question}
Let's think step by step
Answer: """
# TEMPLATE = """{question}{instruction}\nEnglish language description: """
cot = False
shots = 3
if cot is True:
    TEMPLATE = TEMPLATE_COT
else:
    TEMPLATE = TEMPLATE_5SHOT
print("TEMPLATE:", TEMPLATE)
FOLLOW_TEMPLATE = "So the final answer is: "

# GPU_ID = "4"
# GPU_ID = "5"
GPU_ID = "6"
# GPU_ID = "7"


MODEL_DIR = '/data/'
# RUNNING_MODEL = "Meta-Llama-3.1-8B"
RUNNING_MODEL = "Gemma-7B"
# RUNNING_MODEL = "Qwen2.5-32B"
# RUNNING_MODEL = "Qwen2.5-7B"


acceptable_tokens = {
    "Mistral-7B" :[13,5380],
    "Gemma-7B":[108, 109],
    "Meta-Llama-3.1-8B":[271,198,627],
    "Qwen2.5-3B":[624,5267],
    "Qwen2.5-7B":[624,5267],
    "Qwen2.5-14B":[624,5267],
    "Qwen2.5-32B":[624,5267]
}
# "gap"
# "gap_ratio"
# "p1p2"
# "p1_over_rest"
# "topk_coverage"
# "entropy"

config_base = {
# 'measure_methods' : ["logits"],
'measure_methods' : ["gap"],
'sample_methods' :["fibonacci"],
'branch_methods':["local_min"],
'lowest_probs':[0.2],
'filter_token_gen':250,
'filter_token_follow':50,
'k_values' : [10]
}

config_new = {
'measure_methods' : ["gap"],
'sample_methods' : ["fibonacci"],
'branch_methods' : ["local_min"],
'lowest_probs' : [0.2],
'filter_token_gen':250,
'filter_token_follow':50,
'k_values' : [10]
}

task_instructions = {
    "option": "",
    "bool": "",
    "yes_or_no": "",
    # "yes_or_no": "",
    "math": "",
    "dyck": "",
    "sorting": "",
    "open":""
}

# RUNNING_TASK = ['auto_categorization']
# RUNNING_TASK = ['causal_judgement']
RUNNING_TASK = ['processed_squaddev']
# RUNNING_TASK = ['processed_bridging_anaphora_resolution_barqa']
# processed_code_line_description
# processed_bridging_anaphora_resolution_barqa
# processed_disfl_qa
# processed_freetext
# processed_squaddev_v1.1
# RUNNING_TASK = ['web_of_lies']
