-
Notifications
You must be signed in to change notification settings - Fork 0
/
world_run.py
27 lines (20 loc) · 853 Bytes
/
world_run.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import os
import initialize
from bot.self_drive_bot import SelfDriveBot
from model.llm import get_openai_llm
from model.llm_broker import Character
from world.botbroker import SyncBotBroker
from world.world import BaseWorld
if __name__ == '__main__':
# 初始化,一些数据库session和日志等公共组件
initialize.initialize()
# print(f'your key is {os.environ["open_ai_key"]}')
# 获取llm实例,用于后面predict
llm = get_openai_llm(openai_api_key=os.environ['openai_api_key'])
# 设置聊天对象,name是唯一的,会根据对象去加载历史聊天记录
chr_bot = Character.get_by_name("hero")
bot_instance = SelfDriveBot(llm=llm, tools=[], character=chr_bot)
world = BaseWorld(steps_of_round=5, broker=SyncBotBroker())
world.join(bot_instance)
while True:
world.run()