vrr/agent-zero
2
0
Fork 0
mirror of https://github.com/agent0ai/agent-zero.git synced 2026-04-29 04:00:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

whisper remote preload

Browse source
This commit is contained in:
frdel 2024-12-03 19:41:29 +01:00
parent e0b0b6f636
commit 357909c16a
4 changed files with 68 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

9
python/helpers/settings.py
View file

@ -1,3 +1,4 @@
import asyncio
import json
import os
import re
@ -5,12 +6,13 @@ import subprocess
from typing import Any, Literal, TypedDict
import models
from python.helpers import runtime, whisper
from python.helpers import runtime, whisper, defer
from . import files, dotenv
from models import get_model, ModelProvider, ModelType
from langchain_core.language_models.chat_models import BaseChatModel
from langchain_core.embeddings import Embeddings
class Settings(TypedDict):
chat_model_provider: str
chat_model_name: str
@ -743,8 +745,8 @@ def _apply_settings():
agent.config = ctx.config
agent = agent.get_data(agent.DATA_NAME_SUBORDINATE)
# reload whisper model if necessary
whisper.preload()
# reload whisper model if necessary
task = defer.DeferredTask(whisper.preload, _settings["stt_model_size"])
def _env_to_dict(data: str):
@ -776,7 +778,6 @@ def set_root_password(password: str):
raise Exception("root password can only be set in dockerized environments")
subprocess.run(f"echo 'root:{password}' | chpasswd", shell=True, check=True)
dotenv.save_dotenv_value(dotenv.KEY_ROOT_PASSWORD, password)
def get_runtime_config(set: Settings):