add test /index.py

This commit is contained in:
Shiina
2023-10-21 21:29:16 +08:00
parent 6905749220
commit f71da6aefa

21
index.py Normal file
View File

@@ -0,0 +1,21 @@
import time
from jarvis.enviroment.base_env import BaseEnviroment
from jarvis.agent.openai_agent import OpenAIAgent
if __name__ == '__main__':
environment = BaseEnviroment()
agent = OpenAIAgent(
config_path="examples/config.json",
environment=environment
)
action = [
'turn_on_dark_mode',
'turn_on_light_mode'
]
# environment.init_env()
for a in action:
print(a)
command = agent.action_lib[a]
print(agent.env.step(command))
time.sleep(2)