const.py

import enum


class ClaudeModels(str, enum.Enum):
    Inatant1_1: str = "claude-instant-1"
    Instant1_2: str = "claude-instant-1.2"
    Instant1_3: str = "claude-instant-1.3"
    Claude2: str = "claude-2"


class ModelTokenLength(str, enum.Enum):
    ten: int = 10
    twenty_five: int = 25
    fifty: int = 50
    hundred: int = 100
    five_hundred: int = 500
    one_k: int = 1000
    five_k: int = 5000
    ten_k: int = 10000
    twenty_k: int = 20000
    fifty_k: int = 50000
    hundred_k: int = 100000


class Prompts(str, enum.Enum):
    general: str = "\n\nHuman: You're a AI bot who loves to gossip. Listin user's query and answer in markdown using Conversations. \n\nConversations: {memory} \n\nQuery: {question} \n\nAssistant:"