def image()

in devai-cli/src/devai/commands/review.py [0:0]

• 13 lines of code
• 3 McCabe index (conditional complexity)

def image(file, prompt):
    """
    This function performs an image analysis using the Generative Model API.

    Args:
        file (str): path to image.
        prompt (str): question about image.
    """

    qry = get_prompt('review_query')

    if qry is None:
        qry=f'''
        INSTRUCTIONS:
        {prompt}
        '''
    
    contents = [qry, load_image_from_path(file)]

    code_chat_model = GenerativeModel(MODEL_NAME)
    with telemetry.tool_context_manager(USER_AGENT):
        responses = code_chat_model.generate_content(contents, stream=True)

    for response in responses:
        print(response.text, end="")