writer_skill_fake.py

# Copyright (c) Microsoft. All rights reserved.

from semantic_kernel.skill_definition import sk_function, sk_function_context_parameter

# TODO: this fake skill is temporal usage.
# C# supports import skill from samples dir by using test helper and python should do the same
# `semantic-kernel/dotnet/src/IntegrationTests/TestHelpers.cs`


class WriterSkillFake:
    @sk_function(
        description="Translate",
        name="Translate",
    )
    def translate(self, language: str) -> str:
        return f"Translate: {language}"

    @sk_function(description="Write an outline for a novel", name="NovelOutline")
    @sk_function_context_parameter(
        name="endMarker",
        description="The marker to use to end each chapter.",
        default_value="<!--===ENDPART===-->",
    )
    def write_novel_outline(self, input: str) -> str:
        return f"Novel outline: {input}"