refactor(api): migrate snippet workspace endpoints to BaseModel (#37956)

Co-authored-by: Asuka Minato <i@asukaminato.eu.org>
This commit is contained in:
chariri
2026-07-09 04:21:27 +00:00
committed by GitHub
co-authored by Asuka Minato
parent 3cd8d850fa
commit 9bb3b1fa98
10 changed files with 28 additions and 99 deletions
+2 -4
View File
@@ -1,6 +1,6 @@
import json
import logging
from collections.abc import Iterator, Mapping, Sequence
from collections.abc import Generator, Mapping, Sequence
from contextlib import contextmanager
from datetime import UTC, datetime
from typing import Any
@@ -68,7 +68,7 @@ class SnippetService:
self._workflow_run_repo = DifyAPIRepositoryFactory.create_api_workflow_run_repository(session_maker)
@contextmanager
def _session_scope(self) -> Iterator[Session]:
def _session_scope(self) -> Generator[Session, None, None]:
current_session = getattr(self, "_session", None)
if current_session is not None:
yield current_session
@@ -625,8 +625,6 @@ class SnippetService:
conversation_variables=[],
rag_pipeline_variables=draft_workflow.rag_pipeline_variables,
kind=WorkflowKind.SNIPPET.value,
marked_name="",
marked_comment="",
)
session.add(workflow)