vikram-fresche commited on
Commit
1e856c7
·
verified ·
1 Parent(s): 0e056a8

handler v4 (#4)

Browse files

- added custom handler v5 (ff143df46516300d8540ca22cb0f91b1926d067f)

Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -55,7 +55,7 @@ class EndpointHandler:
55
  logger.info("Processing new request")
56
  logger.info(f"Input data: {data}")
57
 
58
- messages = data.get("messages", [])
59
  if not messages:
60
  logger.warning("No input messages provided")
61
  return [{"role": "assistant", "content": "No input messages provided"}]
 
55
  logger.info("Processing new request")
56
  logger.info(f"Input data: {data}")
57
 
58
+ messages = data.get("inputs", [])
59
  if not messages:
60
  logger.warning("No input messages provided")
61
  return [{"role": "assistant", "content": "No input messages provided"}]