def get_live_prompt_id()

in pipeline/pipelineUtils/db.py [0:0]

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

def get_live_prompt_id():
    """
    Retrieve the live prompt ID from the configuration container.
    Assumes a document with id 'live_prompt_config' exists.
    """
    client = CosmosClient(COSMOS_DB_URI, credential=config.credential)
    database = client.get_database_client(COSMOS_DB_DATABASE)
    config_container = database.get_container_client(COSMOS_DB_CONFIG_CONTAINER)
    
    try:
        config_item = config_container.read_item(
            item="live_prompt_config",
            partition_key="live_prompt_config"
        )
        return config_item.get("prompt_id")
    except Exception as e:
        logging.error(f"Error retrieving live prompt config: {str(e)}")
        return None