test_chatglm_with_shared_memory_openai_llm.py.py 937 B

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