isekai-toolkit/api/route.py

38 lines
1.9 KiB
Python

from aiohttp import web
from api.controller.ChatComplete import ChatComplete
from api.controller.Hanzi import Hanzi
from api.controller.Index import Index
from api.controller.Kanji import Kanji
from api.controller.Hanja import Hanja
from api.controller.EmbeddingSearch import EmbeddingSearch
def init(app: web.Application):
app.router.add_routes([
web.route('*', '/hanzi/pinyin/', Hanzi.hanziToPinyin),
web.route('*', '/hanzi/split/', Hanzi.splitHanzi),
web.route('*', '/kanji/romaji/', Kanji.kanji2romaji),
web.route('*', '/hanja/romaja/', Hanja.hanja2roma),
web.route('*', '/title/info', Index.update_title_info),
web.route('*', '/title/search', Index.search_title),
web.route('*', '/user/info', Index.get_user_info),
web.route('*', '/conversation/list', Index.get_conversation_list),
web.route('*', '/conversation/info', Index.get_conversation_info),
web.route('POST', '/conversation/remove', Index.remove_conversation),
web.route('DELETE', '/conversation/remove', Index.remove_conversation),
web.route('POST', '/conversation/set_pinned', Index.set_conversation_pinned),
web.route('POST', '/conversation/set_title', Index.set_conversation_title),
web.route('*', '/embedding_search/index_page', EmbeddingSearch.index_page),
web.route('*', '/embedding_search/search', EmbeddingSearch.search),
web.route('*', '/chatcomplete/conversation_chunk/list', ChatComplete.get_conversation_chunk_list),
web.route('*', '/chatcomplete/conversation_chunk/info', ChatComplete.get_conversation_chunk),
web.route('POST', '/chatcomplete/message', ChatComplete.start_chat_complete),
web.route('GET', '/chatcomplete/message/stream', ChatComplete.chat_complete_stream),
web.route('POST', '/chatcomplete/get_point_cost', ChatComplete.get_point_cost),
])