Skip to content

qbxlvnf11/multi-turn-rag-conversation-compression

Repository files navigation

Multi-Turn Conversation RAG

- Conversation compression method for multi-turn RAG

Docker Environment

- Docker Build

docker build -t multi_turn_test_env .

- Docker Run

docker run -it --gpus all --name multi_turn_test_env --shm-size=64G -p {port}:{port} -e GRANT_SUDO=yes --user root -v {root_folder}:/workspace/multi_turn_conversation -w /workspace/multi_turn_conversation multi_turn_test_env bash

- Docker Exec

docker exec -it multi_turn_test_env bash

Multi-turn Conversation RAG Test

- LLM Summarization Strategy

python multi_turn_conversation_test_summarization.py --config_path {config_path} --conversation_history_file {conversation_history_file_path} --demonstration_file {demonstration_file_path}

- Simple Rewriting Strategy

  • When using the GPT API, enter the API KEY in the '.env' file created after executing the initialization command

  • Step 1. Extracting keywords

  • Step 2(3). Combine keywords and generate final questions

python multi_turn_conversation_test_simple_rewriting.py --config_path {config_path}

Author

About

Conversation compression method for multi-turn RAG

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages