d7b0fff954
Creates the API layer for Open Notebook Creates a services API gateway for the Streamlit front-end Migrates the SurrealDB SDK to the official one Change all database calls to async New podcast framework supporting multiple speaker configurations Implement the surreal-commands library for async processing Improve docker image and docker-compose configurations
39 lines
908 B
Python
39 lines
908 B
Python
from typing import Literal
|
|
|
|
import streamlit as st
|
|
|
|
from api.models_service import ModelsService
|
|
from open_notebook.domain.models import Model
|
|
|
|
# Initialize service instance
|
|
models_service = ModelsService()
|
|
|
|
|
|
def model_selector(
|
|
label,
|
|
key,
|
|
selected_id=None,
|
|
help=None,
|
|
model_type: Literal[
|
|
"language", "embedding", "speech_to_text", "text_to_speech"
|
|
] = "language",
|
|
) -> Model:
|
|
models = models_service.get_all_models(model_type=model_type)
|
|
models.sort(key=lambda x: (x.provider, x.name))
|
|
try:
|
|
index = (
|
|
next((i for i, m in enumerate(models) if m.id == selected_id), 0)
|
|
if selected_id
|
|
else 0
|
|
)
|
|
except Exception:
|
|
index = 0
|
|
|
|
return st.selectbox(
|
|
label,
|
|
models,
|
|
format_func=lambda x: f"{x.provider} - {x.name}",
|
|
help=help,
|
|
index=index,
|
|
key=key,
|
|
)
|