https://github.com/openai/openai-cookbook在TuGraph上的实现
- 环境要求python>=3.12
pip install -r requirements.txt安装依赖python all.py运行完整代码.env中配置api_key,默认使用百炼平台api,如果需要使用别的api,需要修改config.py中BASE_URLTuGraph用户名、密码、数据库设置在tugraph_api.py中修改,db_interface.py只是接口- 数据加载路径在
utils.py中修改load_transcripts_from_pickle修改directory_path参数 prompts.py中只设计了zero-shot版本的提示词,如果需要few-shot,需要在提示词中添加合适的示例