23 lines
937 B
Python
23 lines
937 B
Python
|
|
import sys
|
|
import os
|
|
|
|
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/../')
|
|
from agent import ChatglmWithSharedMemoryOpenaiLLM
|
|
|
|
if __name__ == "__main__":
|
|
# 创建 ChatglmWithSharedMemoryOpenaiLLM 类的实例
|
|
chatglm_instance = ChatglmWithSharedMemoryOpenaiLLM()
|
|
|
|
# 使用代理链运行一些示例输入
|
|
chatglm_instance.agent_chain.run(input="我跟露露聊了什么?")
|
|
chatglm_instance.agent_chain.run(input="她开心吗?")
|
|
chatglm_instance.agent_chain.run(input="她有表达意见吗?")
|
|
chatglm_instance.agent_chain.run(input="根据历史对话总结下?")
|
|
# chatglm_instance.agent_chain.run(input="""可以拓展下吗?，比如写个小作文。
|
|
# 大纲：游戏的美好回忆，触不可及的距离，不在乎得失
|
|
# 主题：露露的陪伴无比珍贵
|
|
# 背景：游戏，通话，当下
|
|
# 开篇需要以游戏相识你挑逗的话语讲起
|
|
# """)
|