Spaces:
Runtime error
Runtime error
| from __future__ import annotations | |
| from typing import Any, Dict | |
| from langchain.schema import BaseOutputParser | |
| class GuardrailsOutputParser(BaseOutputParser): | |
| guard: Any | |
| def _type(self) -> str: | |
| return "guardrails" | |
| def from_rail(cls, rail_file: str, num_reasks: int = 1) -> GuardrailsOutputParser: | |
| try: | |
| from guardrails import Guard | |
| except ImportError: | |
| raise ValueError( | |
| "guardrails-ai package not installed. " | |
| "Install it by running `pip install guardrails-ai`." | |
| ) | |
| return cls(guard=Guard.from_rail(rail_file, num_reasks=num_reasks)) | |
| def from_rail_string( | |
| cls, rail_str: str, num_reasks: int = 1 | |
| ) -> GuardrailsOutputParser: | |
| try: | |
| from guardrails import Guard | |
| except ImportError: | |
| raise ValueError( | |
| "guardrails-ai package not installed. " | |
| "Install it by running `pip install guardrails-ai`." | |
| ) | |
| return cls(guard=Guard.from_rail_string(rail_str, num_reasks=num_reasks)) | |
| def get_format_instructions(self) -> str: | |
| return self.guard.raw_prompt.format_instructions | |
| def parse(self, text: str) -> Dict: | |
| return self.guard.parse(text) | |