Run black

Author: maximusunc

shepherd_utils/db.py

@@ -239,9 +239,7 @@ async def save_logs(
             new_logs.reverse()
             existing_logs.extend(new_logs)
         await client.set(
-            response_id,
-            orjson.dumps(existing_logs),
-            ex=settings.redis_ttl
+            response_id, orjson.dumps(existing_logs), ex=settings.redis_ttl
         )
         await client.aclose()
     except Exception as e:

shepherd_utils/shared.py

@@ -314,9 +314,9 @@ def merge_kgraph(og_message, new_message, source, logger: logging.Logger):
                     value["sources"],
                     [
                         {
-                        "resource_id": source,
-                        "resource_role": "aggregator_knowledge_source",
-                        "upstream_resource_ids": ["infores:retriever"]
+                            "resource_id": source,
+                            "resource_role": "aggregator_knowledge_source",
+                            "upstream_resource_ids": ["infores:retriever"],
                         }
                     ],
                     logger,

workers/merge_message/worker.py

@@ -101,7 +101,7 @@ def add_knowledge_edge(target, result_message, aux_graph_ids, answer):
             {
                 "resource_id": source,
                 "resource_role": "primary_knowledge_source",
-                "upstream_resource_ids": []
+                "upstream_resource_ids": [],
             }
         ],
     }
@@ -317,19 +317,38 @@ def merge_messages(
     result["message"]["knowledge_graph"] = pydantic_kgraph
     for result_message in [response, new_response]:
         if "auxiliary_graphs" in result_message["message"]:
-            for aux_id, aux_dict in result_message["message"]["auxiliary_graphs"].items():
+            for aux_id, aux_dict in result_message["message"][
+                "auxiliary_graphs"
+            ].items():
                 if aux_id in result["message"]["auxiliary_graphs"]:
                     for key, val in aux_dict.items():
                         if key in result["message"]["auxiliary_graphs"][aux_id]:
-                            if isinstance(result["message"]["auxiliary_graphs"][aux_id][key], list):
+                            if isinstance(
+                                result["message"]["auxiliary_graphs"][aux_id][key], list
+                            ):
                                 # combine both lists and then list/set it for uniqueness
-                                result["message"]["auxiliary_graphs"][aux_id][key] = list(set(result["message"]["auxiliary_graphs"][aux_id][key] + val))
+                                result["message"]["auxiliary_graphs"][aux_id][key] = (
+                                    list(
+                                        set(
+                                            result["message"]["auxiliary_graphs"][
+                                                aux_id
+                                            ][key]
+                                            + val
+                                        )
+                                    )
+                                )
                             else:
-                                logger.warning(f"Message had an invalid aux graph property: {key}")
+                                logger.warning(
+                                    f"Message had an invalid aux graph property: {key}"
+                                )
                         else:
-                            result["message"]["auxiliary_graphs"][aux_id][key] = copy.deepcopy(val)
+                            result["message"]["auxiliary_graphs"][aux_id][key] = (
+                                copy.deepcopy(val)
+                            )
                 else:
-                    result["message"]["auxiliary_graphs"][aux_id] = copy.deepcopy(aux_dict)
+                    result["message"]["auxiliary_graphs"][aux_id] = copy.deepcopy(
+                        aux_dict
+                    )
     # The result with the direct lookup needs to be handled specially.   It's the one with the lookup query graph
     lookup_results = []  # in case we don't have any
     lookup_results = (