Function test_model

py-src/data_formulator/routes/agents.py:238–269

()

@agent_bp.route('/test-model', methods=['GET', 'POST'])
def test_model():
    if not request.is_json:
        raise AppError(ErrorCode.INVALID_REQUEST, "Invalid request format")

    logger.info("# test-model request")
    content = request.get_json()

    logger.debug("content------------------------------")
    logger.debug(content)

    client = get_client(content['model'])

    try:
        response = client.get_completion(
            messages=[
                {"role": "system", "content": "You are a helpful assistant."},
                {"role": "user", "content": "Respond 'I can hear you.' if you can hear me. Do not say anything other than 'I can hear you.'"},
            ]
        )

        logger.debug(f"model: {content['model']}")
        logger.debug(f"welcome message: {response.choices[0].message.content}")

        if "I can hear you." in response.choices[0].message.content:
            return json_ok({"model": content['model'], "message": ""})
        else:
            raise AppError(ErrorCode.AGENT_ERROR, "Model responded but did not pass connectivity check")
    except AppError:
        raise
    except Exception as e:
        logger.exception(f"Error testing model {content['model'].get('id', '')}")
        raise classify_and_wrap_llm_error(e) from e

@agent_bp.route('/process-data-on-load', methods=['GET', 'POST'])
def process_data_on_load_request():