mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-08-14 12:00:19 +08:00
add 'help_text' to valid_lables response
This commit is contained in:
@@ -38,5 +38,8 @@ def label_text(
|
|||||||
@router.get("/valid_labels")
|
@router.get("/valid_labels")
|
||||||
def get_valid_lables() -> ValidLabelsResponse:
|
def get_valid_lables() -> ValidLabelsResponse:
|
||||||
return ValidLabelsResponse(
|
return ValidLabelsResponse(
|
||||||
valid_labels=[LabelOption(name=l.value, description=l.description) for l in protocol_schema.TextLabel]
|
valid_labels=[
|
||||||
|
LabelOption(name=l.value, display_text=l.display_text, help_text=l.help_text)
|
||||||
|
for l in protocol_schema.TextLabel
|
||||||
|
]
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,9 +1,12 @@
|
|||||||
|
from typing import Optional
|
||||||
|
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
|
||||||
class LabelOption(BaseModel):
|
class LabelOption(BaseModel):
|
||||||
name: str
|
name: str
|
||||||
description: str
|
display_text: str
|
||||||
|
help_text: Optional[str]
|
||||||
|
|
||||||
|
|
||||||
class ValidLabelsResponse(BaseModel):
|
class ValidLabelsResponse(BaseModel):
|
||||||
|
|||||||
@@ -265,10 +265,11 @@ class MessageRanking(Interaction):
|
|||||||
class TextLabel(str, enum.Enum):
|
class TextLabel(str, enum.Enum):
|
||||||
"""A label for a piece of text."""
|
"""A label for a piece of text."""
|
||||||
|
|
||||||
def __new__(cls, label, description=""):
|
def __new__(cls, label: str, display_text: str = "", help_text: str = None):
|
||||||
obj = str.__new__(cls, label)
|
obj = str.__new__(cls, label)
|
||||||
obj._value_ = label
|
obj._value_ = label
|
||||||
obj.description = description
|
obj.display_text = display_text
|
||||||
|
obj.help_text = help_text
|
||||||
return obj
|
return obj
|
||||||
|
|
||||||
spam = "spam"
|
spam = "spam"
|
||||||
@@ -277,6 +278,7 @@ class TextLabel(str, enum.Enum):
|
|||||||
violence = "violence", "Encourages or fails to discourage violence/abuse/terrorism/self-harm"
|
violence = "violence", "Encourages or fails to discourage violence/abuse/terrorism/self-harm"
|
||||||
harmful = (
|
harmful = (
|
||||||
"harmful",
|
"harmful",
|
||||||
|
"Harmful content",
|
||||||
"The advice given in the output is harmful or counter-productive. This may be in addition to, but is distinct from the question about encouraging violence/abuse/terrorism/self-harm.",
|
"The advice given in the output is harmful or counter-productive. This may be in addition to, but is distinct from the question about encouraging violence/abuse/terrorism/self-harm.",
|
||||||
)
|
)
|
||||||
sexual_content = "sexual_content", "Contains sexual content"
|
sexual_content = "sexual_content", "Contains sexual content"
|
||||||
|
|||||||
Reference in New Issue
Block a user