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)
