From 22c5ec0af4a65ca276a27d9529cf7daaa446f2d8 Mon Sep 17 00:00:00 2001 From: Mahathi Vatsal Date: Tue, 15 Apr 2025 08:52:06 -0700 Subject: [PATCH] Update AgentQnA and DocSum for Gaudi Compatibility Signed-off-by: Mahathi Vatsal --- AgentQnA/retrieval_tool/run_ingest_data.sh | 10 ++++++++++ DocSum/docker_compose/intel/hpu/gaudi/compose.yaml | 3 +++ 2 files changed, 13 insertions(+) diff --git a/AgentQnA/retrieval_tool/run_ingest_data.sh b/AgentQnA/retrieval_tool/run_ingest_data.sh index a55fd04457..f52249a7a3 100644 --- a/AgentQnA/retrieval_tool/run_ingest_data.sh +++ b/AgentQnA/retrieval_tool/run_ingest_data.sh @@ -6,4 +6,14 @@ port=6007 FILEDIR=${WORKDIR}/GenAIExamples/AgentQnA/example_data/ FILENAME=test_docs_music.jsonl +# Check if packages are installed +for package in "${packages[@]}"; do + if pip freeze | grep -q "$package="; then + echo "$package is installed" + else + echo "$package is not installed" + pip install --no-cache-dir "$package" + fi +done + python3 index_data.py --filedir ${FILEDIR} --filename ${FILENAME} --host_ip $host_ip --port $port diff --git a/DocSum/docker_compose/intel/hpu/gaudi/compose.yaml b/DocSum/docker_compose/intel/hpu/gaudi/compose.yaml index c5c70e438a..58f427c549 100644 --- a/DocSum/docker_compose/intel/hpu/gaudi/compose.yaml +++ b/DocSum/docker_compose/intel/hpu/gaudi/compose.yaml @@ -107,3 +107,6 @@ services: networks: default: driver: bridge + +volumes: + data: