first

ai_bot/states/MidjourneyStates.py

@@ -0,0 +1,24 @@
+from bot.States import States
+
+MIDJOURNEY_SET_PROMPT = 41
+
+def set_states():
+    states_arr = {
+        "set_prompt": MIDJOURNEY_SET_PROMPT
+    }
+
+    return states_arr
+
+
+class MidjourneyStates(States):
+
+    def __init__(self):
+        pass
+
+    @staticmethod
+    def get_state_by_key(key: str):
+        states = set_states()
+        if key in states:
+            return states[key]
+
+        return None