added custom handler v15

handler.py

@@ -109,12 +109,20 @@ class EndpointHandler:
                     logger.info(f"Function call: {function_call}")
                     logger.info(f"Text response: {text_response}")
                     # Return both text and tool message
-                    return [{"generations": [{"output": {"content": text_response}}]}]
+                    return [
+                        {
+                            "generated_text": text_response,
+                            "details": {
+                                "finish_reason": "stop",
+                                "generated_tokens": len(output_tokens[0])
+                            }
+                        }
+                    ]
             else:
                 response = output_text
                 
             logger.info(f"Generated response: {json.dumps(response)}")
-            return [{"generations": [{"output": {"content": response}}]}]
+            return [{"generated_text": response, "details": {"finish_reason": "stop", "generated_tokens": len(output_tokens[0])}}] 
             
         except Exception as e:
             logger.error(f"Error during generation: {str(e)}", exc_info=True)