Fork me on GitHub

llm大语言模型用于电影问答

项目:

本项目利用chatglm6b实现一个中文版的简单电影问答功能,目的是探索llm agent与图数据库neo4j进行交互的能力,主要根据用户的意图与neo4j进行交互。

Tools

agent使用几个tools与neo4j进行交互:

  1. Person_Tool:

根据人物(演员)person(actor)检索电影。

2.Genre_Tool:

根据类型(genre)检索电影 。

chatglm server

部署 chatglm service服务, 见chatglm_server/chatglm_flask_server.py

neo4j server

部署neo4j service 服务and 导入数据, 见 insert_data_to_neo4j.py.

data

为了简单起见,我在这里只处理和使用了少量的电影数据,参见data/。

test

我想看剧情类的电影

李丽珍参演的电影

 

posted @ 2024-07-22 20:34  石头木  阅读(106)  评论(0)    收藏  举报