chore: Update schema and service files for DuckDbMessageModel

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-26 16:11:29 -03:00
commit be0065fb2d
2 changed files with 6 additions and 7 deletions

View file

@ -1,6 +1,6 @@
import json
from datetime import datetime, timezone
from typing import Any, Optional
from typing import Any
from uuid import UUID
from pydantic import BaseModel, Field, field_serializer, field_validator
@ -81,8 +81,7 @@ class TransactionModelResponse(DefaultModel):
return v
class DeprecatedMessageModel(DefaultModel):
index: int | None = Field(default=None)
class DuckDbMessageModel(DefaultModel):
flow_id: str | None = Field(default=None, alias="flow_id")
timestamp: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
sender: str

View file

@ -1,6 +1,6 @@
from datetime import datetime
from pathlib import Path
from typing import TYPE_CHECKING, List, Optional, Union
from typing import TYPE_CHECKING, Union
import duckdb
from loguru import logger
@ -18,14 +18,14 @@ class MonitorService(Service):
name = "monitor_service"
def __init__(self, settings_service: "SettingsService"):
from langflow.services.monitor.schema import DeprecatedMessageModel, TransactionModel, VertexBuildModel
from langflow.services.monitor.schema import DuckDbMessageModel, TransactionModel, VertexBuildModel
self.settings_service = settings_service
self.base_cache_dir = Path(user_cache_dir("langflow"))
self.db_path = self.base_cache_dir / "monitor.duckdb"
self.table_map: dict[str, type[TransactionModel | DeprecatedMessageModel | VertexBuildModel]] = {
self.table_map: dict[str, type[TransactionModel | DuckDbMessageModel | VertexBuildModel]] = {
"transactions": TransactionModel,
"messages": DeprecatedMessageModel,
"messages": DuckDbMessageModel,
"vertex_builds": VertexBuildModel,
}