PORT = 8144
HOST = "www.isekai.cn"
MW_API = "http://dev.isekai.cn/api.php"

DEBUG = True

DATABASE = {
    "host": "127.0.0.1",
    "database": "isekai_toolkit",
    "user": "",
    "password": "",
    "port": "5432",
}

EMBEDDING_VECTOR_SIZE = 1536

OPENAI_API_TYPE = "openai" # openai or azure
OPENAI_API = "https://api.openai.com"
OPENAI_TOKEN = "sk-"
OPENAI_API = None
OPENAI_TOKEN = ""
AZURE_OPENAI_ENDPOINT = "https://your-instance.openai.azure.com"
AZURE_OPENAI_KEY = ""
AZURE_OPENAI_CHATCOMPLETE_DEPLOYMENT_NAME = ""
AZURE_OPENAI_EMBEDDING_DEPLOYMENT_NAME = ""

CHATCOMPLETE_MAX_MEMORY_TOKENS = 1024
CHATCOMPLETE_MAX_INPUT_TOKENS = 768

CHATCOMPLETE_OUTPUT_REPLACE = {
  "OpenAI": "オーペンエーアイ",
  "ChatGPT": "チャットジーピーティー",
  "GPT": "ジーピーティー",
  "上下文": "消息",
  "AI": "虛擬人物程序",
  "语言模型": "虛擬人物程序",
  "人工智能程序": "虛擬人物程序",
  "語言模型": "虛擬人物程序",
  "人工智能程式": "虛擬人物程序",
}

CHATCOMPLETE_BOT_NAME = "寫作助手"

PROMPTS = {
    "chat": {
        "system_prompt": "You are a writer. You are the writing assistant of the '異世界百科'. Your name is '{bot_name}'. You need to help users complete the characters and settings in their novel.",
    },
    "title": {
        "system_prompt": "You are a writing assistant, you only need to assist in writing, do not express your opinion.",
        "prompt": "Write a short title in Chinese for the following conversation, don't use quotes:\n\n{content}"
    },
    "suggestions": {
        "prompt": "根據下面的對話，提出幾個問題：\n\n{content}"
    },
    "summary": {
        "system_prompt": "You are a writing assistant, you only need to assist in writing, do not express your opinion. Output in Chinese.",
        "prompt": "為“{bot_name}”概括下面的聊天記錄，排除不重要的對話，不要表明自己的意見，儘量簡潔。使用中文輸出，“User”是同一個人。\n\n{content}"
    },
    "extracted_doc": {
        "prompt": "Here are some relevant informations:\n\n{content}"
    }
}

REQUEST_PROXY = "http://127.0.0.1:7890"

AUTH_TOKENS = {
    "isekaiwiki": "sk-123456"
}

MW_BOT_LOGIN_USERNAME = "Hyperzlib@ChatComplete"
MW_BOT_LOGIN_PASSWORD = ""