From ad584db0d1bc28e7eb2de073ea88534935fb00e3 Mon Sep 17 00:00:00 2001
From: Mohammad Amin <dadgaramin96@gmail.com>
Date: Thu, 17 Jul 2025 14:43:58 +0330
Subject: [PATCH] feat: add retry policy to Hivemind query execution!

---
 tasks/hivemind/query_data_sources.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tasks/hivemind/query_data_sources.py b/tasks/hivemind/query_data_sources.py
index 5014395..3bd6a1b 100644
--- a/tasks/hivemind/query_data_sources.py
+++ b/tasks/hivemind/query_data_sources.py
@@ -7,6 +7,7 @@
 from langchain.tools import tool
 from tc_temporal_backend.client import TemporalClient
 from tc_temporal_backend.schema.hivemind import HivemindQueryPayload
+from temporalio.common import RetryPolicy
 
 nest_asyncio.apply()
 
@@ -46,6 +47,7 @@ async def query(self, query: str) -> str | None:
             payload,
             id=f"hivemind-query-{self.community_id}-{self.workflow_id}",
             task_queue=hivemind_queue,
+            retry_policy=RetryPolicy(maximum_attempts=3),
         )
 
         return result