darabos commited on
Commit
6fa5290
·
1 Parent(s): 1d0e9e3

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(*, prompt_role: MessageRole, prompt_content: ops.LongStr):
 
 
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