From b1e7bc9afa7965ded729845075ac286897130631 Mon Sep 17 00:00:00 2001
From: Max Wang <max@covar.com>
Date: Wed, 17 May 2023 14:48:40 -0400
Subject: [PATCH] Remove message merging of original query

---
 app/server.py | 13 +------------
 1 file changed, 1 insertion(+), 12 deletions(-)

diff --git a/app/server.py b/app/server.py
index cee7c42..a779fa5 100644
--- a/app/server.py
+++ b/app/server.py
@@ -99,12 +99,6 @@ async def run_workflow(
     log_level = request_dict.get("log_level", "ERROR")
     logger.setLevel(logging._nameToLevel[log_level])
     qgraph = message["query_graph"]
-    kgraph = {"nodes": {}, "edges": {}}
-    results = message.get("results", [])
-    aux_graphs = message.get("auxiliary_graphs", {})
-    if "knowledge_graph" in message.keys():
-        if "nodes" in message["knowledge_graph"].keys():
-            kgraph = message["knowledge_graph"]
     completed_workflow = []
 
     async with httpx.AsyncClient(verify=False, timeout=60.0) as client:
@@ -185,12 +179,7 @@ async def run_workflow(
                     break
             
             logger.debug(f"Merging {len(service_operation_responses)} responses for '{operation}'...")
-            m = Message(
-                query_graph=QueryGraph.parse_obj(qgraph),
-                knowledge_graph=KnowledgeGraph.parse_obj(kgraph),
-                results=Results.parse_obj(results),
-                auxiliary_graphs=AuxiliaryGraphs.parse_obj(aux_graphs),
-                )
+            m = Message(query_graph=QueryGraph.parse_obj(qgraph))
             for response in service_operation_responses:
                 response["message"]["query_graph"] = qgraph
                 m.update(Message.parse_obj(response["message"]))