feat: knowledge pipeline (#25360)
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: twwu <twwu@dify.ai> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com> Co-authored-by: jyong <718720800@qq.com> Co-authored-by: Wu Tianwei <30284043+WTW0313@users.noreply.github.com> Co-authored-by: QuantumGhost <obelisk.reg+git@gmail.com> Co-authored-by: lyzno1 <yuanyouhuilyz@gmail.com> Co-authored-by: quicksand <quicksandzn@gmail.com> Co-authored-by: Jyong <76649700+JohnJyong@users.noreply.github.com> Co-authored-by: lyzno1 <92089059+lyzno1@users.noreply.github.com> Co-authored-by: zxhlyh <jasonapring2015@outlook.com> Co-authored-by: Yongtao Huang <yongtaoh2022@gmail.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Co-authored-by: nite-knite <nkCoding@gmail.com> Co-authored-by: Hanqing Zhao <sherry9277@gmail.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Co-authored-by: Harry <xh001x@hotmail.com>
This commit is contained in:
62
api/core/schemas/schema_manager.py
Normal file
62
api/core/schemas/schema_manager.py
Normal file
@@ -0,0 +1,62 @@
|
||||
from collections.abc import Mapping
|
||||
from typing import Any
|
||||
|
||||
from core.schemas.registry import SchemaRegistry
|
||||
|
||||
|
||||
class SchemaManager:
|
||||
"""Schema manager provides high-level schema operations"""
|
||||
|
||||
def __init__(self, registry: SchemaRegistry | None = None):
|
||||
self.registry = registry or SchemaRegistry.default_registry()
|
||||
|
||||
def get_all_schema_definitions(self, version: str = "v1") -> list[Mapping[str, Any]]:
|
||||
"""
|
||||
Get all JSON Schema definitions for a specific version
|
||||
|
||||
Args:
|
||||
version: Schema version, defaults to v1
|
||||
|
||||
Returns:
|
||||
Array containing schema definitions, each element contains name and schema fields
|
||||
"""
|
||||
return self.registry.get_all_schemas_for_version(version)
|
||||
|
||||
def get_schema_by_name(self, schema_name: str, version: str = "v1") -> Mapping[str, Any] | None:
|
||||
"""
|
||||
Get a specific schema by name
|
||||
|
||||
Args:
|
||||
schema_name: Schema name
|
||||
version: Schema version, defaults to v1
|
||||
|
||||
Returns:
|
||||
Dictionary containing name and schema, returns None if not found
|
||||
"""
|
||||
uri = f"https://dify.ai/schemas/{version}/{schema_name}.json"
|
||||
schema = self.registry.get_schema(uri)
|
||||
|
||||
if schema:
|
||||
return {"name": schema_name, "schema": schema}
|
||||
return None
|
||||
|
||||
def list_available_schemas(self, version: str = "v1") -> list[str]:
|
||||
"""
|
||||
List all available schema names for a specific version
|
||||
|
||||
Args:
|
||||
version: Schema version, defaults to v1
|
||||
|
||||
Returns:
|
||||
List of schema names
|
||||
"""
|
||||
return self.registry.list_schemas(version)
|
||||
|
||||
def list_available_versions(self) -> list[str]:
|
||||
"""
|
||||
List all available schema versions
|
||||
|
||||
Returns:
|
||||
List of versions
|
||||
"""
|
||||
return self.registry.list_versions()
|
||||
Reference in New Issue
Block a user