SyngyeonTak commited on
Commit
8afc022
·
1 Parent(s): e29b52a

cluster_predictor updates

Browse files
Files changed (1) hide show
  1. cluster_predictor.py +2 -2
cluster_predictor.py CHANGED
@@ -52,12 +52,12 @@ def query_llm_for_variables(user_query, use_prompt=True, use_fewshot=True):
52
  prompt_parts = []
53
 
54
  if use_prompt:
55
- with open("PROMPT_PATH", "r", encoding="utf-8") as f:
56
  custom_prompt = f.read()
57
  prompt_parts.append(custom_prompt)
58
 
59
  if use_fewshot:
60
- with open("FEWSHOT_PATH", "r", encoding="utf-8") as f:
61
  few_shot_examples = f.read()
62
  prompt_parts.append(few_shot_examples)
63
 
 
52
  prompt_parts = []
53
 
54
  if use_prompt:
55
+ with open(PROMPT_PATH, "r", encoding="utf-8") as f:
56
  custom_prompt = f.read()
57
  prompt_parts.append(custom_prompt)
58
 
59
  if use_fewshot:
60
+ with open(FEWSHOT_PATH, "r", encoding="utf-8") as f:
61
  few_shot_examples = f.read()
62
  prompt_parts.append(few_shot_examples)
63