Skip to content

Commit

Permalink
Merge pull request #21 from jujimeizuo/main
Browse files Browse the repository at this point in the history 
feat knowledge config path and fix func in role and action
  • Loading branch information
@sanbuphy
sanbuphy authored Jan 29, 2024
2 parents e94c23d + 62ef51d commit b5e872c
Show file tree
Hide file tree
Showing 20 changed files with 442 additions and 335 deletions.
147 changes: 97 additions & 50 deletions run/metagpt_webui.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,25 +8,25 @@
from streamlit_chat import message
import asyncio
from metagpt.logs import logger
from tianji.agents.metagpt_agents.wendao_agent import wendao
from tianji.agents.metagpt_agents.qianbianzhe_agent import qianbianzhe
from tianji.agents.metagpt_agents.ruyi_agent import ruyi
from tianji.agents.metagpt_agents.wendao import WenDao
from tianji.agents.metagpt_agents.qianbianzhe import QianBianZhe
from tianji.agents.metagpt_agents.ruyi import RuYi
from tianji.utils.json_from import SharedDataSingleton
from datetime import datetime

import uuid

# 初始化session_state变量
if 'user_id' not in st.session_state:
if "user_id" not in st.session_state:
# 为新用户会话生成一个唯一的UUID
logger.log(0,"add uuid")
st.session_state['user_id'] = str(uuid.uuid4())
logger.log(0, "add uuid")
st.session_state["user_id"] = str(uuid.uuid4())
# 设计思路 给定人设并导入参考聊天话术、历史聊天语料进行聊天。

# async def start(msg):
# role_wendao = wendao()
# logger.info(msg)
# result = await role_wendao.run(msg)
# result = await role_wendao.run(msg)

# sharedData = SharedDataSingleton.get_instance()
# json_from_data = sharedData.json_from_data
Expand All @@ -41,86 +41,133 @@
# logger.info(result)
# print("final ans :\n",qianjibian_ans,ruyi_ans)


def timestamp_str():
# 获取当前时间
now = datetime.now()
# 将当前时间转换为字符串格式
timestamp_str = now.strftime("%Y-%m-%d %H:%M:%S.%f")
# 从字符串中去除最后的微秒部分，并转换为时间戳形式的字符串
timestamp_str = timestamp_str[:-3].replace(" ", "_").replace(":", "").replace("-", "").replace(".", "")
timestamp_str = (
timestamp_str[:-3]
.replace(" ", "_")
.replace(":", "")
.replace("-", "")
.replace(".", "")
)
return timestamp_str


async def main():
first_status_user_history = ""
# 对话导入
# 对话导入
# msg = "test"
# msg = """元旦节下午，我和哥哥一起去图书馆学习。我像给哥哥一个祝福。我的哥哥，一位医学院的学生，正在为即将到来的考试做准备。他今年24岁，对医学充满热情。图书馆里非常安静，我们专心致志地学习。哥哥的爱好是玩篮球，他经常说运动是放松大脑的最佳方式。他总是希望我也能热爱学习，努力追求知识。"""
st.markdown("#### 人情世故大模型_祝福模块")
if 'generated' not in st.session_state:
st.session_state['generated'] = []
if 'past' not in st.session_state:
st.session_state['past'] = []
user_input=st.text_input("我是人情世故大模型团队开发的祝福agents。你可以在这里找到一个完整的祝福。我会告诉你怎么写，还会针对你的祝福给你生成专属的知识文档。\n 首先你需要完整的告诉我，你想在什么节日给谁送祝福？这个人是谁呢（是妈妈）？他会有什么愿望呢？你想在什么时候送给他？可以告诉我他的爱好、性格、年龄段、最近的状态。\n 就像这段：【元旦节下午，我和哥哥一起去图书馆学习。我想给哥哥一个祝福。我的哥哥，一位医学院的学生，正在为即将到来的考试做准备。他今年24岁，对医学充满热情。图书馆里非常安静，我们专心致志地学习。哥哥的爱好是玩篮球，他经常说运动是放松大脑的最佳方式。他总是希望我也能热爱学习，努力追求知识。】\n请输入你的问题：",key='input')
if "generated" not in st.session_state:
st.session_state["generated"] = []
if "past" not in st.session_state:
st.session_state["past"] = []
user_input = st.text_input(
"我是人情世故大模型团队开发的祝福agents。你可以在这里找到一个完整的祝福。我会告诉你怎么写，还会针对你的祝福给你生成专属的知识文档。\n 首先你需要完整的告诉我，你想在什么节日给谁送祝福？这个人是谁呢（是妈妈）？他会有什么愿望呢？你想在什么时候送给他？可以告诉我他的爱好、性格、年龄段、最近的状态。\n 就像这段：【元旦节下午，我和哥哥一起去图书馆学习。我想给哥哥一个祝福。我的哥哥，一位医学院的学生，正在为即将到来的考试做准备。他今年24岁，对医学充满热情。图书馆里非常安静，我们专心致志地学习。哥哥的爱好是玩篮球，他经常说运动是放松大脑的最佳方式。他总是希望我也能热爱学习，努力追求知识。】\n请输入你的问题：",
key="input",
)
task_status = 0
# 显示用户的UUID（仅供演示）
st.write(f"您的会话ID1是: {st.session_state['user_id']}")

if user_input:
# 显示用户的UUID（仅供演示）
st.write(f"您的会话ID2是: {st.session_state['user_id']}")
sharedData = SharedDataSingleton.get_instance()
for one_message in sharedData.first_status_message_list :
message(one_message['message'],is_user=one_message['is_user'],key=one_message['keyname'])

st.session_state['past'].append(user_input)
message(st.session_state['past'][-1], is_user=True,key="_user")
sharedData.first_status_message_list.append({"message":st.session_state['past'][-1],"is_user":True,"keyname":"_user"+str(timestamp_str)})
sharedData = SharedDataSingleton.get_instance()
for one_message in sharedData.first_status_message_list:
message(
one_message["message"],
is_user=one_message["is_user"],
key=one_message["keyname"],
)

st.session_state["past"].append(user_input)
message(st.session_state["past"][-1], is_user=True, key="_user")
sharedData.first_status_message_list.append(
{
"message": st.session_state["past"][-1],
"is_user": True,
"keyname": "_user" + str(timestamp_str),
}
)

# 第一阶段-搜集用户的需求

if task_status == 0:

sharedData.first_status_user_history = sharedData.first_status_user_history + "\n" + "user:" + str(user_input)
if task_status == 0:
sharedData.first_status_user_history = (
sharedData.first_status_user_history + "\n" + "user:" + str(user_input)
)

role_wendao = wendao()
result = await role_wendao.run(sharedData.first_status_user_history)
role_wendao = WenDao()
result = await role_wendao.run(sharedData.first_status_user_history)
first_status_result_list = result.content.split("|")
if first_status_result_list[0] == "NO":
sharedData.first_status_user_history = sharedData.first_status_user_history + "\n" + "assistant:" + str(first_status_result_list[1])
st.session_state['generated'].append(str(first_status_result_list[1]))
message(st.session_state["generated"][-1], key=str(len(st.session_state["generated"])))

sharedData.first_status_message_list.append({"message":st.session_state["generated"][-1],"is_user":False,"keyname":str(len(st.session_state["generated"]))+str(timestamp_str)})
sharedData.first_status_user_history = (
sharedData.first_status_user_history
+ "\n"
+ "assistant:"
+ str(first_status_result_list[1])
)
st.session_state["generated"].append(str(first_status_result_list[1]))
message(
st.session_state["generated"][-1],
key=str(len(st.session_state["generated"])),
)

sharedData.first_status_message_list.append(
{
"message": st.session_state["generated"][-1],
"is_user": False,
"keyname": str(len(st.session_state["generated"]))
+ str(timestamp_str),
}
)
st.text("请继续输入")
else:
task_status = 1
sharedData.ask_num +=1
print("ask_num:",sharedData.ask_num)
sharedData.ask_num += 1
print("ask_num:", sharedData.ask_num)

if task_status==1 or sharedData.ask_num>3:
sharedData = SharedDataSingleton.get_instance()
if task_status == 1 or sharedData.ask_num > 3:
sharedData = SharedDataSingleton.get_instance()
json_from_data = sharedData.json_from_data
knowledge_key= json_from_data["festival"] + json_from_data["requirement"]
st.session_state['generated'].append("知识库数据："+str(knowledge_key))
message(st.session_state["generated"][-1], key=str(len(st.session_state["generated"])))

role_qianbianzhe = qianbianzhe()
knowledge_key = json_from_data["festival"] + json_from_data["requirement"]
st.session_state["generated"].append("知识库数据：" + str(knowledge_key))
message(
st.session_state["generated"][-1],
key=str(len(st.session_state["generated"])),
)

role_qianbianzhe = QianBianZhe()
qianjibian_ans = await role_qianbianzhe.run(" ")
# st.session_state['past'].append("等待中！！！")

st.session_state['generated'].append("agent 千机变回答："+str(qianjibian_ans))
message(st.session_state["generated"][-1], key=str(len(st.session_state["generated"])))
role_ruyi = ruyi()
st.session_state["generated"].append("agent 千机变回答：" + str(qianjibian_ans))
message(
st.session_state["generated"][-1],
key=str(len(st.session_state["generated"])),
)
role_ruyi = RuYi()
ruyi_ans = await role_ruyi.run(" ")

st.session_state['generated'].append("agent 如意回答："+str(ruyi_ans))
message(st.session_state["generated"][-1], key=str(len(st.session_state["generated"])))

st.session_state["generated"].append("agent 如意回答：" + str(ruyi_ans))
message(
st.session_state["generated"][-1],
key=str(len(st.session_state["generated"])),
)

# if st.session_state['generated']:
# for i in range(len(st.session_state['generated'])-1, -1, -1):
# message(st.session_state["generated"][i], key=str(i))
# message(st.session_state['past'][i],
# is_user=True,
# message(st.session_state['past'][i],
# is_user=True,
# key=str(i)+'_user')



asyncio.run(main())
Loading

0 comments on commit b5e872c

Please sign in to comment.