Przeglądaj źródła

fix: messages updated_at (#23869)

tags/1.8.0
Will 2 miesięcy temu
rodzic
commit
5aaa47e25d
No account linked to committer's email address

+ 2
- 0
api/core/app/apps/advanced_chat/generate_task_pipeline.py Wyświetl plik

from core.workflow.workflow_cycle_manager import CycleManagerWorkflowInfo, WorkflowCycleManager from core.workflow.workflow_cycle_manager import CycleManagerWorkflowInfo, WorkflowCycleManager
from events.message_event import message_was_created from events.message_event import message_was_created
from extensions.ext_database import db from extensions.ext_database import db
from libs.datetime_utils import naive_utc_now
from models import Conversation, EndUser, Message, MessageFile from models import Conversation, EndUser, Message, MessageFile
from models.account import Account from models.account import Account
from models.enums import CreatorUserRole from models.enums import CreatorUserRole
def _save_message(self, *, session: Session, graph_runtime_state: Optional[GraphRuntimeState] = None) -> None: def _save_message(self, *, session: Session, graph_runtime_state: Optional[GraphRuntimeState] = None) -> None:
message = self._get_message(session=session) message = self._get_message(session=session)
message.answer = self._task_state.answer message.answer = self._task_state.answer
message.updated_at = naive_utc_now()
message.provider_response_latency = time.perf_counter() - self._base_task_pipeline._start_at message.provider_response_latency = time.perf_counter() - self._base_task_pipeline._start_at
message.message_metadata = self._task_state.metadata.model_dump_json() message.message_metadata = self._task_state.metadata.model_dump_json()
message_files = [ message_files = [

+ 2
- 0
api/core/app/task_pipeline/easy_ui_based_generate_task_pipeline.py Wyświetl plik

from core.prompt.utils.prompt_template_parser import PromptTemplateParser from core.prompt.utils.prompt_template_parser import PromptTemplateParser
from events.message_event import message_was_created from events.message_event import message_was_created
from extensions.ext_database import db from extensions.ext_database import db
from libs.datetime_utils import naive_utc_now
from models.model import AppMode, Conversation, Message, MessageAgentThought from models.model import AppMode, Conversation, Message, MessageAgentThought


logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
if llm_result.message.content if llm_result.message.content
else "" else ""
) )
message.updated_at = naive_utc_now()
message.answer_tokens = usage.completion_tokens message.answer_tokens = usage.completion_tokens
message.answer_unit_price = usage.completion_unit_price message.answer_unit_price = usage.completion_unit_price
message.answer_price_unit = usage.completion_price_unit message.answer_price_unit = usage.completion_price_unit

Ładowanie…
Anuluj
Zapisz