Changes to make graphrag ui work

Signed-off-by: theresa <[email protected]>
opea-project · Nov 13, 2024 · e7efac2 · e7efac2
1 parent 3cd7675
commit e7efac2
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 5 deletions.
diff --git a/GraphRAG/graphrag.py b/GraphRAG/graphrag.py
@@ -70,7 +70,11 @@ def align_outputs(self, data, cur_node, inputs, runtime_graph, llm_parameters_di
         # handle template
         # if user provides template, then format the prompt with it
         # otherwise, use the default template
-        prompt = inputs.messages[0]["content"]
+        print("outputs before align:\n", inputs)
+        if isinstance(inputs.messages, str):
+            prompt = inputs.messages
+        else:
+            prompt = inputs.messages[0]["content"]
         chat_template = llm_parameters_dict["chat_template"]
         if chat_template:
             prompt_template = PromptTemplate.from_template(chat_template)

diff --git a/GraphRAG/tests/test_compose_on_gaudi.sh b/GraphRAG/tests/test_compose_on_gaudi.sh
@@ -2,7 +2,7 @@
 # Copyright (C) 2024 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-set -e
+set -xe
 IMAGE_REPO=${IMAGE_REPO:-"opea"}
 IMAGE_TAG=${IMAGE_TAG:-"latest"}
 echo "REGISTRY=IMAGE_REPO=${IMAGE_REPO}"
@@ -11,7 +11,6 @@ export REGISTRY=${IMAGE_REPO}
 export TAG=${IMAGE_TAG}
 
 WORKPATH=$(dirname "$PWD")
-WORKPATH=/home/rbrugaro/GenAIExamples/GraphRAG
 LOG_PATH="$WORKPATH/tests"
 ip_address=$(hostname -I | awk '{print $1}')
 

diff --git a/GraphRAG/ui/svelte/.env b/GraphRAG/ui/svelte/.env
@@ -1,4 +1,4 @@
-CHAT_BASE_URL = '/v1/chatqna'
+CHAT_BASE_URL = '/v1/graphrag'
 
 UPLOAD_FILE_BASE_URL = '/v1/dataprep'
 

diff --git a/GraphRAG/ui/svelte/src/routes/+page.svelte b/GraphRAG/ui/svelte/src/routes/+page.svelte
@@ -197,7 +197,7 @@
 >
 	<div class="mx-auto flex h-full w-full flex-col sm:mt-0 sm:w-[72%]">
 		<div class="flex justify-between p-2">
-			<p class="text-[1.7rem] font-bold tracking-tight">ChatQnA</p>
+			<p class="text-[1.7rem] font-bold tracking-tight">GraphRAG</p>
 			<UploadFile />
 		</div>
 		<div