def __call__()

in 3-ai-native-e2e-sample/backend/agents/literature.py [0:0]

• 33 lines of code
• 9 McCabe index (conditional complexity)

    def __call__(self, run_status=None, run_step=None, message=None):
        try:
            if run_status:
                logger.info(f"Thread run status: {run_status}")
                
            if run_step:
                logger.info(f"Run step type: {run_step.type}, Status: {run_step.status}")
                
            if message:
                logger.info(f"Received thread message: {message.id}")
                if message.content:
                    try:
                        # Handle MessageTextDetails object properly
                        if hasattr(message.content, 'value'):
                            content = message.content.value
                        else:
                            content = str(message.content)
                            
                        # Return properly formatted JSON response
                        return None, None, json.dumps({
                            "type": "message",
                            "content": content
                        })
                    except Exception as e:
                        logger.error(f"Error processing message content: {str(e)}")
                        return None, None, json.dumps({
                            "type": "error",
                            "content": f"Failed to process message: {str(e)}"
                        })

            if run_status == RunStatus.COMPLETED:
                logger.info("Literature chat stream completed")
                
        except Exception as e:
            logger.error(f"Error in literature chat handler: {str(e)}")
            return None, None, json.dumps({
                "type": "error",
                "content": f"Handler error: {str(e)}"
            })
        
        return None, None, None