import sys import json from utils import create_client from mindieclient.python.common import Log logger = Log(__name__).getlog() if __name__ == "__main__": # get argument and create client try: mindie_client = create_client() except Exception as e: logger.exception("Client Creation falied!") sys.exit(1) model_name = "llama_65b" if mindie_client.is_server_live(): logger.info("The server is alive!") else: logger.error("The server is not alive!") sys.exit(1) if mindie_client.is_server_ready(): logger.info("The server is ready!") else: logger.error("The server is not ready!") sys.exit(1) if mindie_client.is_model_ready(model_name): logger.info("The model is ready!") else: logger.error("The model is not ready!") sys.exit(1) server_metadata_dict = mindie_client.get_server_metadata() logger.info("get_server_metadata: %s", json.dumps(server_metadata_dict)) model_metadata_dict = mindie_client.get_model_metadata(model_name) logger.info("get_model_metadata: %s", json.dumps(model_metadata_dict)) model_config_dict = mindie_client.get_model_config(model_name) logger.info("get_model_config: %s", json.dumps(model_config_dict)) # get slots result = mindie_client.get_slot_count(model_name) logger.info("gets_slot_count: %s", result.get_response())