Files
semantic-kernel/python/samples/documentation_examples/plugin.py
T
Eduard van Valkenburg 6751f9d819 Python: Updated plugins (#5827)
### Motivation and Context

<!-- Thank you for your contribution to the semantic-kernel repo!
Please help reviewers and future users, providing the following
information:
  1. Why is this change required?
  2. What problem does it solve?
  3. What scenario does it contribute to?
  4. If it fixes an open issue, please link to the issue here.
-->

The loading of plugins and functions into Kernel was not fully
consistent and clear, this is now simplified.

Adding a plugin can now be done in two ways:
- multiple: `kernel.add_plugins[plugin1, plugin2]`
- single plugin: `kernel.add_plugin(plugin)`
- if the plugin here is a class, with methods that have the
kernel_function decorator then that is parsed into a plugin

Adding a function can now be done in three ways:
- multiple: `kernel.add_functions([func1, func2]` or
`kernel.add_function({'func1': func1})`
- single: `kernel.add_function('plugin_name', func1)`
- from a prompt: `kernel.add_function(function_name='func1',
prompt='test prompt', ...)`

In other words, all the different methods that were available have been
simplified, these methods also return the created plugin (or updated
plugin for the `add_functions` method. The `add_function` (singular)
method has a parameter `return_plugin` to control whether you get the
created or updated plugin, instead of the created function.

**An important callout:**

One big internal change that this introduces is that a function that
gets added to a plugin (new or existing) is automatically copied, the
metadata is deep-copied and the plugin_name in the metadata is updated,
so this sequence is valid now:

```python
@kernel_function(name='test')
def f(...):
   etc

func = KernelFunctionFromMethod(method=f)
func2 = kernel.add_function('plugin', func)
assert func != func2
```

Also closes: #5855

### Description

<!-- Describe your changes, the overall approach, the underlying design.
These notes will help understanding how your code works. Thanks! -->
Removed KernelPluginCollection
Updated KernelPlugin
   added from_... method to load from different sources
Updated KernelFunctionFromPrompt
   added from_yaml and from_directory methods.

Added and updated tests to 99% coverage of KernelPlugin and
KernelFunctionFromPrompt

### Contribution Checklist

<!-- Before submitting this PR, please make sure: -->

- [x] The code builds clean without any errors or warnings
- [x] The PR follows the [SK Contribution
Guidelines](https://github.com/microsoft/semantic-kernel/blob/main/CONTRIBUTING.md)
and the [pre-submission formatting
script](https://github.com/microsoft/semantic-kernel/blob/main/CONTRIBUTING.md#development-scripts)
raises no violations
- [x] All unit tests pass, and I have added new tests where possible
- [ ] I didn't break anyone 😄
2024-04-15 13:51:59 +00:00

73 lines
1.8 KiB
Python

# Copyright (c) Microsoft. All rights reserved.
import asyncio
import sys
from service_configurator import add_service
import semantic_kernel as sk
if sys.version_info >= (3, 9):
from typing import Annotated
else:
from typing_extensions import Annotated
from semantic_kernel.functions.kernel_function_decorator import kernel_function
# Let's define a light plugin
class LightPlugin:
is_on: bool = False
@kernel_function(
name="get_state",
description="Gets the state of the light.",
)
def get_state(
self,
) -> Annotated[str, "the output is a string"]:
"""Returns the state result of the light."""
return "On" if self.is_on else "Off"
@kernel_function(
name="change_state",
description="Changes the state of the light.",
)
def change_state(
self,
new_state: Annotated[bool, "the new state of the light"],
) -> Annotated[str, "the output is a string"]:
"""Changes the state of the light."""
self.is_on = new_state
state = self.get_state()
print(f"The light is now: {state}")
return state
async def main():
# Initialize the kernel
kernel = sk.Kernel()
# Add the service to the kernel
# use_chat: True to use chat completion, False to use text completion
kernel = add_service(kernel=kernel, use_chat=True)
light_plugin = kernel.add_plugin(
LightPlugin(),
plugin_name="LightPlugin",
)
result = await kernel.invoke(light_plugin["get_state"])
print(f"The light is: {result}")
print("Changing the light's state...")
# Kernel Arguments are passed in as kwargs.
result = await kernel.invoke(light_plugin["change_state"], new_state=True)
print(f"The light is: {result}")
# Run the main function
if __name__ == "__main__":
asyncio.run(main())