diff --git a/python/packages/main/agent_framework/guard_rails.py b/python/packages/main/agent_framework/guard_rails.py deleted file mode 100644 index 46f9d44352..0000000000 --- a/python/packages/main/agent_framework/guard_rails.py +++ /dev/null @@ -1,27 +0,0 @@ -# Copyright (c) Microsoft. All rights reserved. - - -from typing import Generic, Protocol, TypeVar, runtime_checkable - -TInput = TypeVar("TInput") -TResponse = TypeVar("TResponse") - -__all__ = ["InputGuardrail", "OutputGuardrail"] - - -@runtime_checkable -class InputGuardrail(Protocol, Generic[TInput]): - """A protocol for input guardrails that can validate and transform input messages.""" - - def __call__(self, message: TInput) -> TInput: - """Validate and possibly transform the input message.""" - ... - - -@runtime_checkable -class OutputGuardrail(Protocol, Generic[TResponse]): - """A protocol for output guardrails that can validate and transform output messages.""" - - def __call__(self, message: TResponse) -> TResponse: - """Validate and possibly transform the output message.""" - ...