vikram-fresche commited on
Commit
c950f55
·
verified ·
1 Parent(s): 407d7fe

handler_v14 (#14)

Browse files

- added custom handler v14 (8780767e17ff7d0b241f01b5f981c595d8b524d5)

Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -109,12 +109,12 @@ class EndpointHandler:
109
  logger.info(f"Function call: {function_call}")
110
  logger.info(f"Text response: {text_response}")
111
  # Return both text and tool message
112
- return {"generations": [{"output": {"content": text_response}}]}
113
  else:
114
  response = output_text
115
 
116
  logger.info(f"Generated response: {json.dumps(response)}")
117
- return {"generations": [{"output": {"content": response}}]}
118
 
119
  except Exception as e:
120
  logger.error(f"Error during generation: {str(e)}", exc_info=True)
 
109
  logger.info(f"Function call: {function_call}")
110
  logger.info(f"Text response: {text_response}")
111
  # Return both text and tool message
112
+ return [{"generations": [{"output": {"content": text_response}}]}]
113
  else:
114
  response = output_text
115
 
116
  logger.info(f"Generated response: {json.dumps(response)}")
117
+ return [{"generations": [{"output": {"content": response}}]}]
118
 
119
  except Exception as e:
120
  logger.error(f"Error during generation: {str(e)}", exc_info=True)