From 19b034e4eadc348d629a46989a39c75388568670 Mon Sep 17 00:00:00 2001 From: dimitrievgs Date: Thu, 30 Apr 2026 19:30:51 +0300 Subject: [PATCH] Add small fixes due to def voice llm model name --- app/heartbeat_monitor.py | 7 +++++-- app/nodes.py | 2 -- app/title_generator.py | 9 ++++++--- 3 files changed, 11 insertions(+), 7 deletions(-) diff --git a/app/heartbeat_monitor.py b/app/heartbeat_monitor.py index 89204c1..7b8e417 100644 --- a/app/heartbeat_monitor.py +++ b/app/heartbeat_monitor.py @@ -191,9 +191,10 @@ class VoiceHeartbeatMonitor(threading.Thread): print(f"🎀 New Voice Command detected: {cmd}") # ΠžΠ’ΠŸΠ ΠΠ’Π›Π―Π•Πœ Π’ LLM Ρ‡Π΅Ρ€Π΅Π· TitleGenerator + prompt = f"{(s.get('systemPrompt') or '')}\n\n{(s.get('prompt_commands') or '')}" self.generator.add_voice_task_to_queue( cmd, - s['prompt_commands'] # Π­Ρ‚ΠΎ ΠΏΡƒΡ‚ΡŒ ΠΊ .md Ρ„Π°ΠΉΠ»Ρƒ ΠΈΠ· настроСк + prompt ) self._processed_hashes.add(cmd_hash) @@ -204,5 +205,7 @@ class VoiceHeartbeatMonitor(threading.Thread): raw_text = self.vs.get_last_chars(s['voice_context_limit']) if not raw_text.strip(): return + prompt = f"{(s.get('systemPrompt') or '')}\n\n{(s.get('prompt_regular') or '')}" + # Π”ΠΎΠ±Π°Π²ΠΈΠΌ Π½ΠΎΠ²Ρ‹ΠΉ Ρ‚ΠΈΠΏ 'voice_regular' Π²TitleGenerator Π°Π½Π°Π»ΠΎΠ³ΠΈΡ‡Π½ΠΎ 'voice_command' - self.generator.add_voice_regular_task(raw_text, s['prompt_regular']) \ No newline at end of file + self.generator.add_voice_regular_task(raw_text, prompt) \ No newline at end of file diff --git a/app/nodes.py b/app/nodes.py index 22cd8dc..a744514 100644 --- a/app/nodes.py +++ b/app/nodes.py @@ -143,8 +143,6 @@ def execute_command_node(state: AgentState) -> AgentState: DEFAULT_LLM_NAME = "gemini-2.0-flash" #"gemini-2.5-flash" main_llm = get_llm(DEFAULT_LLM_NAME) -DEFAULT_SUMMARIZATION_LLM_NAME = "mistral-small-latest" # "mistral-small-latest" #"gemini-2.0-flash-r" - def call_llm_node(state: AgentState) -> AgentState: """ diff --git a/app/title_generator.py b/app/title_generator.py index 81f8656..b612a1e 100644 --- a/app/title_generator.py +++ b/app/title_generator.py @@ -7,11 +7,13 @@ import sqlite3 import threading import time from typing import Any, Dict, Optional -from nodes import DEFAULT_SUMMARIZATION_LLM_NAME from llm_client import get_llm from langchain_core.messages import SystemMessage, HumanMessage import os +DEFAULT_SUMMARIZATION_LLM_NAME = "mistral-small-latest" # "mistral-small-latest" #"gemini-2.0-flash-r" +DEFAULT_VOICE_LLM_NAME = "gemini-3.0-flash-openrouter" + MAX_TITLE_GENERATION_CONTENT_LENGTH = 5000 # МаксимальноС количСство символов для Π³Π΅Π½Π΅Ρ€Π°Ρ†ΠΈΠΈ Π·Π°Π³ΠΎΠ»ΠΎΠ²ΠΊΠΎΠ² class TitleGenerator: @@ -22,6 +24,7 @@ class TitleGenerator: self._create_table() self.history_manager = history_manager self.llm = get_llm(DEFAULT_SUMMARIZATION_LLM_NAME) + self.voice_llm = get_llm(DEFAULT_VOICE_LLM_NAME) self.running = False self.thread: Optional[threading.Thread] = None @@ -376,7 +379,7 @@ class TitleGenerator: SystemMessage(content=prompt), HumanMessage(content=f"ПослСдний Ρ„Ρ€Π°Π³ΠΌΠ΅Π½Ρ‚ Π΄ΠΈΠ°Π»ΠΎΠ³Π° для Π°Π½Π°Π»ΠΈΠ·Π°: {text}") ] - response = self.llm.invoke(llm_messages) + response = self.voice_llm.invoke(llm_messages) import app.api as api api.voice_inst.responses["regular"] = response # ЗаписываСм Π² Response 1 @@ -392,7 +395,7 @@ class TitleGenerator: SystemMessage(content=prompt), HumanMessage(content=f"Π’Ρ‹ΠΏΠΎΠ»Π½ΠΈ ΠΊΠΎΠΌΠ°Π½Π΄Ρƒ ΠΈΠ· транскрибации: {text}") ] - response = self.llm.invoke(llm_messages) + response = self.voice_llm.invoke(llm_messages) import app.api as api # Π€ΠΎΡ€ΠΌΠ°Ρ‚ΠΈΡ€ΡƒΠ΅ΠΌ для Π²ΠΊΠ»Π°Π΄ΠΊΠΈ Response 2