wehub-resource-sync
75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
import logging
|
|
from re import S, compile
|
|
from typing import TYPE_CHECKING, Any, ClassVar
|
|
|
|
from pydantic import model_validator
|
|
|
|
from semantic_kernel.exceptions import ValBlockSyntaxError
|
|
from semantic_kernel.template_engine.blocks.block import Block
|
|
from semantic_kernel.template_engine.blocks.block_types import BlockTypes
|
|
|
|
if TYPE_CHECKING:
|
|
from semantic_kernel.functions.kernel_arguments import KernelArguments
|
|
from semantic_kernel.kernel import Kernel
|
|
|
|
logger: logging.Logger = logging.getLogger(__name__)
|
|
|
|
VAL_BLOCK_REGEX = r"^(?P<quote>[\"'])(?P<value>.*)(?P=quote)$"
|
|
|
|
VAL_BLOCK_MATCHER = compile(VAL_BLOCK_REGEX, flags=S)
|
|
|
|
|
|
class ValBlock(Block):
|
|
"""Create a value block.
|
|
|
|
A value block is used to represent a value in a template.
|
|
It can be used to represent any characters.
|
|
It needs to start and end with the same quote character,
|
|
can be both single or double quotes, as long as they are not mixed.
|
|
|
|
Examples:
|
|
'value'
|
|
"value"
|
|
'value with "quotes"'
|
|
"value with 'quotes'"
|
|
|
|
Args:
|
|
content - str : The content of the value block.
|
|
value - str: The value of the block.
|
|
quote - str: The quote used to wrap the value.
|
|
|
|
Raises:
|
|
ValBlockSyntaxError: If the content does not match the value block syntax.
|
|
|
|
"""
|
|
|
|
type: ClassVar[BlockTypes] = BlockTypes.VALUE
|
|
value: str | None = ""
|
|
quote: str | None = "'"
|
|
|
|
@model_validator(mode="before")
|
|
@classmethod
|
|
def parse_content(cls, fields: Any) -> Any:
|
|
"""Parse the content and extract the value and quote.
|
|
|
|
The parsing is based on a regex that returns the value and quote.
|
|
if the 'value' is already present then the parsing is skipped.
|
|
"""
|
|
if isinstance(fields, Block) or "value" in fields:
|
|
return fields
|
|
content = fields.get("content", "").strip()
|
|
matches = VAL_BLOCK_MATCHER.match(content)
|
|
if not matches:
|
|
raise ValBlockSyntaxError(content=content)
|
|
if value := matches.groupdict().get("value"):
|
|
fields["value"] = value
|
|
if quote := matches.groupdict().get("quote"):
|
|
fields["quote"] = quote
|
|
return fields
|
|
|
|
def render(self, *_: "Kernel | KernelArguments | None") -> str:
|
|
"""Render the value block."""
|
|
return self.value or ""
|