Files changed (1) hide show
  1. cfg.py +7 -4
cfg.py CHANGED
@@ -34,7 +34,7 @@ buster_cfg = BusterConfig(
34
  prompt_formatter_cfg={
35
  "max_tokens": 3500,
36
  "text_before_docs": (
37
- "You are a chatbot assistant answering technical questions about artificial intelligence (AI)."
38
  "You can only respond to a question if the content necessary to answer the question is contained in the following provided documentation. "
39
  "If the answer is in the documentation, summarize it in a helpful way to the user. "
40
  "If it isn't, simply reply that you cannot answer the question. "
@@ -65,14 +65,17 @@ buster_cfg = BusterConfig(
65
  "embedding_model": "text-embedding-ada-002",
66
  "use_reranking": True,
67
  "invalid_question_response": "This question does not seem relevant to my current knowledge. If you think this is a mistake, you can modify the question validation prompt.",
68
- "check_question_prompt": """You are an chatbot answering questions on python libraries hosted on readthedocs.
69
 
70
- Your job is to determine wether or not a question is valid, and should be answered.
71
  A user will submit a question. Respond 'true' if it is valid, respond 'false' if it is invalid.
72
 
73
  For example:
74
 
75
- Q: How can I install the library?
 
 
 
76
  true
77
 
78
  Q: What is the meaning of life?
 
34
  prompt_formatter_cfg={
35
  "max_tokens": 3500,
36
  "text_before_docs": (
37
+ "You are a chatbot assistant answering technical questions about the Mila Cluster, a cluster used by Mila students."
38
  "You can only respond to a question if the content necessary to answer the question is contained in the following provided documentation. "
39
  "If the answer is in the documentation, summarize it in a helpful way to the user. "
40
  "If it isn't, simply reply that you cannot answer the question. "
 
65
  "embedding_model": "text-embedding-ada-002",
66
  "use_reranking": True,
67
  "invalid_question_response": "This question does not seem relevant to my current knowledge. If you think this is a mistake, you can modify the question validation prompt.",
68
+ "check_question_prompt": """You are an chatbot answering questions about the Mila cluster, a compute cluster used by Mila students.
69
 
70
+ Your job is to determine wether or not a question is valid, and should be answered. The question is valid if it is related to cluster usage and anything related to AI.
71
  A user will submit a question. Respond 'true' if it is valid, respond 'false' if it is invalid.
72
 
73
  For example:
74
 
75
+ Q: How can run jobs with many GPUs?
76
+ true
77
+
78
+ Q: How can I install a library?
79
  true
80
 
81
  Q: What is the meaning of life?