Spaces:
Running
Running
Better error message for wrong enum value.
Browse files
lynxkite-core/src/lynxkite/core/ops.py
CHANGED
@@ -144,6 +144,7 @@ def _param_to_type(name, value, type):
|
|
144 |
assert value != "", f"{name} is unset."
|
145 |
return float(value)
|
146 |
if isinstance(type, enum.EnumMeta):
|
|
|
147 |
return type[value]
|
148 |
if isinstance(type, types.UnionType):
|
149 |
match type.__args__:
|
|
|
144 |
assert value != "", f"{name} is unset."
|
145 |
return float(value)
|
146 |
if isinstance(type, enum.EnumMeta):
|
147 |
+
assert value in type.__members__, f"{value} is not an option for {name}."
|
148 |
return type[value]
|
149 |
if isinstance(type, types.UnionType):
|
150 |
match type.__args__:
|
lynxkite-lynxscribe/src/lynxkite_lynxscribe/lynxscribe_ops.py
CHANGED
@@ -679,7 +679,9 @@ def chat_processor(processor, *, _ctx: one_by_one.Context):
|
|
679 |
|
680 |
@output_on_top
|
681 |
@op("LynxScribe Message")
|
682 |
-
def lynxscribe_message(
|
|
|
|
|
683 |
return_message = Message(role=prompt_role.value, content=prompt_content.strip())
|
684 |
return {"prompt_message": return_message}
|
685 |
|
|
|
679 |
|
680 |
@output_on_top
|
681 |
@op("LynxScribe Message")
|
682 |
+
def lynxscribe_message(
|
683 |
+
*, prompt_role: MessageRole = MessageRole.SYSTEM, prompt_content: ops.LongStr
|
684 |
+
):
|
685 |
return_message = Message(role=prompt_role.value, content=prompt_content.strip())
|
686 |
return {"prompt_message": return_message}
|
687 |
|