Function generate

web/pgadmin/tools/sqleditor/__init__.py:2983–3099

Generator for SSE events.

()

        )

    def generate():
        """Generator for SSE events."""
        import secrets as py_secrets
        from pgadmin.llm.models import Message, Role

        try:
            # Send thinking status
            yield _nlq_sse_event({
                'type': 'thinking',
                'message': gettext('Analyzing your request...')
            })

            # Deserialize conversation history if provided
            conversation_history = None
            if history_data:
                conversation_history = []
                for item in (history_data or []):
                    if not isinstance(item, dict):
                        continue
                    role_str = item.get('role', '')
                    content = item.get('content', '')
                    try:
                        role = Role(role_str)
                    except ValueError:
                        continue
                    conversation_history.append(
                        Message(role=role, content=content)
                    )

            # Stream the LLM response with database tools
            response_text = ''
            updated_messages = []
            for item in chat_with_database_stream(
                user_message=user_message,
                sid=trans_obj.sid,
                did=trans_obj.did,
                system_prompt=NLQ_SYSTEM_PROMPT,
                conversation_history=conversation_history
            ):
                if isinstance(item, str):
                    # Text chunk from streaming LLM response
                    yield _nlq_sse_event({
                        'type': 'text_delta',
                        'content': item
                    })
                elif isinstance(item, tuple) and \
                        item[0] == 'tool_use':
                    # Tool execution in progress - reset streaming
                    yield _nlq_sse_event({
                        'type': 'thinking',
                        'message': gettext(
                            'Querying the database...'
                        )
                    })
                elif isinstance(item, tuple) and \
                        item[0] == 'complete':
                    # Final result: ('complete', response_text, messages)
                    response_text = item[1]