jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

chore: Update schema and service files for DuckDbMessageModel

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-26 16:11:29 -03:00
commit bb7f365ccd
2 changed files with 6 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/services/monitor/schema.py
View file

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

8
src/backend/base/langflow/services/monitor/service.py
View file

@ -1,6 +1,6 @@
from datetime import datetime from datetime import datetime
from pathlib import Path from pathlib import Path
from typing import TYPE_CHECKING, List, Optional, Union from typing import TYPE_CHECKING, Union
import duckdb import duckdb
from loguru import logger from loguru import logger
@ -18,14 +18,14 @@ class MonitorService(Service):
name = "monitor_service" name = "monitor_service"
def __init__(self, settings_service: "SettingsService"): 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.settings_service = settings_service
self.base_cache_dir = Path(user_cache_dir("langflow")) self.base_cache_dir = Path(user_cache_dir("langflow"))
self.db_path = self.base_cache_dir / "monitor.duckdb" 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, "transactions": TransactionModel,
"messages": DeprecatedMessageModel, "messages": DuckDbMessageModel,
"vertex_builds": VertexBuildModel, "vertex_builds": VertexBuildModel,
} }