Provide embedding manager (#16)

* Provide the Embedding management UI

* Update Fastembed documentation

* Add validation when adding / updating embeddings

* Stop using the old ktem embeddings manager

* Set default local embedding models

* Move the local embeddings below in flowsettings

* Update flowsettings
This commit is contained in:
Duc Nguyen (john)
2024-04-10 15:11:44 +07:00
committed by GitHub
parent ed10020ea3
commit 7b3307e3c4
12 changed files with 608 additions and 30 deletions

View File

@@ -1,6 +1,7 @@
import gradio as gr
from ktem.app import BasePage
from ktem.db.models import User, engine
from ktem.embeddings.ui import EmbeddingManagement
from ktem.llms.ui import LLMManagement
from sqlmodel import Session, select
@@ -17,9 +18,12 @@ class AdminPage(BasePage):
with gr.Tab("User Management", visible=False) as self.user_management_tab:
self.user_management = UserManagement(self._app)
with gr.Tab("LLM Management") as self.llm_management_tab:
with gr.Tab("LLMs") as self.llm_management_tab:
self.llm_management = LLMManagement(self._app)
with gr.Tab("Embeddings") as self.llm_management_tab:
self.emb_management = EmbeddingManagement(self._app)
def on_subscribe_public_events(self):
if self._app.f_user_management:
self._app.subscribe_event(