File size: 930 Bytes
01f199c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | from promptings.CoT import CoTStrategy
from promptings.Direct import DirectStrategy
from promptings.Analogical import AnalogicalStrategy
from promptings.SelfPlanning import SelfPlanningStrategy
from promptings.DebateCoder import DebateCoder
from promptings.MapCoder import MapCoder as MapCoder
class PromptingFactory:
@staticmethod
def get_prompting_class(prompting_name):
if prompting_name == "CoT":
return CoTStrategy
elif prompting_name == "MapCoder":
return MapCoder
elif prompting_name == "Direct":
return DirectStrategy
elif prompting_name == "Analogical":
return AnalogicalStrategy
elif prompting_name == "SelfPlanning":
return SelfPlanningStrategy
elif prompting_name == "DebateCoder":
return DebateCoder
else:
raise Exception(f"Unknown prompting name {prompting_name}")
|