def setUpClass(cls): cls.client = craft_ai.Client(settings.CRAFT_CFG) cls.n_generators = 5 cls.generators_id = [ generate_entity_id("list_generators") for i in range(cls.n_generators) ] cls.agent_id = generate_entity_id("list_generators_agent")
def test_get_decision_tree_w_smallish_timeout(self): other_client_cfg = settings.CRAFT_CFG.copy() other_client_cfg["decisionTreeRetrievalTimeout"] = 500 other_client = craft_ai.Client(other_client_cfg) last_operation = VALID_L_OPERATIONS[-1][-1] self.assertRaises( craft_ai.errors.CraftAiLongRequestTimeOutError, other_client.get_agent_decision_tree, self.agent_id, last_operation["timestamp"], )
def setUp(self): self.client = craft_ai.Client(settings.CRAFT_CFG)
def setUpClass(cls): cls.client = craft_ai.Client(settings.CRAFT_CFG)
def setUpClass(cls): cls.client = craft_ai.Client(settings.CRAFT_CFG) cls.agent_id = generate_entity_id("test_delete_agent_with_unknown_id")
def setUpClass(cls): cls.client = craft_ai.Client(settings.CRAFT_CFG) cls.agent_id = generate_entity_id("get_operations")
import craft_ai import json import random import os ## 1) Connect to craftai api client = craft_ai.Client({ "token": os.environ.get("CRAFT_TOKEN") }) JOB_ID = str(os.environ.get("JOB_ID")) if "JOB_ID" in os.environ else "" ## 2) Create the agent agent_id = "Wellness" + JOB_ID configuration = { "context": { "day_off": { # day is off or not "type": "enum" }, "next_day_off": { # next day is off or not "type": "enum" }, "sleep_at_home": { # sleep at home or not "type" : "enum" }, "sleep_start": { # hour at which he started to sleep "type": "time_of_day", "is_generated" : False, }, "timezone": { # timezone for sleep_start hour "type" : "timezone"
def setUpClass(cls): cls.client = craft_ai.Client(settings.CRAFT_CFG) cls.generator_id = generate_entity_id("test_delete_generator")
def setUpClass(cls): cls.client = craft_ai.Client(settings.CRAFT_CFG) cls.generator_id = generate_entity_id("get_generator_operations") cls.agent_id_1 = generate_entity_id("get_generator_operations") cls.agent_id_2 = generate_entity_id("get_generator_operations") cls.filter = [cls.agent_id_1, cls.agent_id_2]
def setUpClass(cls): cls.client = craft_ai.Client(settings.CRAFT_CFG) cls.agent_id = generate_entity_id("test_get_decision_tree")
import craft_ai import os from dotenv import load_dotenv, find_dotenv load_dotenv(find_dotenv()) CRAFT_CFG = {"token": os.environ.get("CRAFT_TOKEN")} JOB_ID = os.environ.get("TRAVIS_JOB_ID", "loc") print("Cleaning dangling agents and generators from test session.") craft_client = craft_ai.Client(CRAFT_CFG) agents_list = craft_client.list_agents() agents_to_delete = [ agent_id for agent_id in agents_list if agent_id[-3:] == JOB_ID[-3:] ] for agent_id in agents_to_delete: craft_client.delete_agent(agent_id) generators_list = craft_client.list_generators() generators_to_delete = [ generator_id for generator_id in generators_list if generator_id[-3:] == JOB_ID[-3:] ] for generator_id in generators_to_delete: craft_client.delete_generator(generator_id)