from semantic_kernel import Kernel
from semantic_kernel.connectors.ai.open_ai import OpenAIChatCompletion
from semantic_kernel.prompt_template import InputVariable, PromptTemplateConfig
kernel = Kernel()
kernel.add_service(
OpenAIChatCompletion(ai_model_id="gpt-4o",api_key="OPENAI_API_KEY"),
)
prompt = """
{{$input}}
Given the input above, answer the question to the best of your knowledge.
"""
prompt_template_config = PromptTemplateConfig(
template=prompt,
name="summarize",
template_format="semantic-kernel",
input_variables=[
InputVariable(name="input", description="user input", is_required=True),
]
)
summarize = kernel.add_function(
function_name="summarizeFunc",
plugin_name="summarizePlugin",
prompt_template_config=prompt_template_config,
)
input_text = "Summarize Arize AI platform in 50 words"
summary = await kernel.invoke(summarize, input=input_text)
print(summary)