linting fixes

Author: THardy98

activity_worker/activity_worker.py

@@ -1,13 +1,14 @@
 import asyncio
 import random
 import string
-from pathlib import Path
 
 from temporalio import activity
 from temporalio.client import Client
 from temporalio.envconfig import ClientConfig
 from temporalio.worker import Worker
 
+from util import get_temporal_config_path
+
 task_queue = "say-hello-task-queue"
 workflow_name = "say-hello-workflow"
 activity_name = "say-hello-activity"
@@ -20,10 +21,12 @@ async def say_hello_activity(name: str) -> str:
 
 async def main():
     # Create client to localhost on default namespace
-    # Get repo root - 1 level deep from root
-    repo_root = Path(__file__).resolve().parent.parent
-    config_file = repo_root / "temporal.toml"
-    config = ClientConfig.load_client_connect_config(config_file=str(config_file))
+    config = ClientConfig.load_client_connect_config(
+        config_file=str(get_temporal_config_path())
+    )
+    config = ClientConfig.load_client_connect_config(
+        config_file=str(get_temporal_config_path())
+    )
     config["target_host"] = "localhost:7233"
     client = await Client.connect(**config)
 

batch_sliding_window/starter.py

@@ -6,11 +6,13 @@
 import logging
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 
 from batch_sliding_window.batch_workflow import (
     ProcessBatchWorkflow,
     ProcessBatchWorkflowInput,
 )
+from util import get_temporal_config_path
 
 
 async def main():
@@ -19,10 +21,9 @@ async def main():
     logging.basicConfig(level=logging.INFO)
 
     # Create client
-        # Get repo root - 1 level deep from root
-        repo_root = Path(__file__).resolve().parent.parent
-        config_file = repo_root / "temporal.toml"
-    config = ClientConfig.load_client_connect_config(config_file=str(config_file))
+    config = ClientConfig.load_client_connect_config(
+        config_file=str(get_temporal_config_path())
+    )
     config["target_host"] = "localhost:7233"
     client = await Client.connect(**config)
 

batch_sliding_window/worker.py

@@ -6,11 +6,13 @@
 
 from temporalio import worker
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 
 from batch_sliding_window.batch_workflow import ProcessBatchWorkflow
 from batch_sliding_window.record_loader_activity import RecordLoader
 from batch_sliding_window.record_processor_workflow import RecordProcessorWorkflow
 from batch_sliding_window.sliding_window_workflow import SlidingWindowWorkflow
+from util import get_temporal_config_path
 
 
 async def main():
@@ -19,10 +21,9 @@ async def main():
     logging.basicConfig(level=logging.INFO)
 
     # Create client
-        # Get repo root - 1 level deep from root
-        repo_root = Path(__file__).resolve().parent.parent
-        config_file = repo_root / "temporal.toml"
-    config = ClientConfig.load_client_connect_config(config_file=str(config_file))
+    config = ClientConfig.load_client_connect_config(
+        config_file=str(get_temporal_config_path())
+    )
     config["target_host"] = "localhost:7233"
     client = await Client.connect(**config)
 

bedrock/basic/run_worker.py

@@ -3,18 +3,19 @@
 import logging
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from temporalio.worker import Worker
 from workflows import BasicBedrockWorkflow
 
 from bedrock.shared.activities import BedrockActivities
+from util import get_temporal_config_path
 
 
 async def main():
     # Create client connected to server at the given address
-        # Get repo root - 2 levels deep from root
-        repo_root = Path(__file__).resolve().parent.parent.parent
-        config_file = repo_root / "temporal.toml"
-    config = ClientConfig.load_client_connect_config(config_file=str(config_file))
+    config = ClientConfig.load_client_connect_config(
+        config_file=str(get_temporal_config_path())
+    )
     config["target_host"] = "localhost:7233"
     client = await Client.connect(**config)
     activities = BedrockActivities()

bedrock/basic/send_message.py

@@ -2,15 +2,17 @@
 import sys
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from workflows import BasicBedrockWorkflow
 
+from util import get_temporal_config_path
+
 
 async def main(prompt: str) -> str:
     # Create client connected to server at the given address
-        # Get repo root - 2 levels deep from root
-        repo_root = Path(__file__).resolve().parent.parent.parent
-        config_file = repo_root / "temporal.toml"
-    config = ClientConfig.load_client_connect_config(config_file=str(config_file))
+    config = ClientConfig.load_client_connect_config(
+        config_file=str(get_temporal_config_path())
+    )
     config["target_host"] = "localhost:7233"
     client = await Client.connect(**config)
 

bedrock/entity/end_chat.py

@@ -2,15 +2,17 @@
 import sys
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from workflows import EntityBedrockWorkflow
 
+from util import get_temporal_config_path
+
 
 async def main():
     # Create client connected to server at the given address
-        # Get repo root - 2 levels deep from root
-        repo_root = Path(__file__).resolve().parent.parent.parent
-        config_file = repo_root / "temporal.toml"
-    config = ClientConfig.load_client_connect_config(config_file=str(config_file))
+    config = ClientConfig.load_client_connect_config(
+        config_file=str(get_temporal_config_path())
+    )
     config["target_host"] = "localhost:7233"
     client = await Client.connect(**config)
 

bedrock/entity/get_history.py

@@ -1,15 +1,17 @@
 import asyncio
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from workflows import EntityBedrockWorkflow
 
+from util import get_temporal_config_path
+
 
 async def main():
     # Create client connected to server at the given address
-        # Get repo root - 2 levels deep from root
-        repo_root = Path(__file__).resolve().parent.parent.parent
-        config_file = repo_root / "temporal.toml"
-    config = ClientConfig.load_client_connect_config(config_file=str(config_file))
+    config = ClientConfig.load_client_connect_config(
+        config_file=str(get_temporal_config_path())
+    )
     config["target_host"] = "localhost:7233"
     client = await Client.connect(**config)
     workflow_id = "entity-bedrock-workflow"

bedrock/entity/run_worker.py

@@ -3,18 +3,19 @@
 import logging
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from temporalio.worker import Worker
 from workflows import EntityBedrockWorkflow
 
 from bedrock.shared.activities import BedrockActivities
+from util import get_temporal_config_path
 
 
 async def main():
     # Create client connected to server at the given address
-        # Get repo root - 2 levels deep from root
-        repo_root = Path(__file__).resolve().parent.parent.parent
-        config_file = repo_root / "temporal.toml"
-    config = ClientConfig.load_client_connect_config(config_file=str(config_file))
+    config = ClientConfig.load_client_connect_config(
+        config_file=str(get_temporal_config_path())
+    )
     config["target_host"] = "localhost:7233"
     client = await Client.connect(**config)
     activities = BedrockActivities()

bedrock/entity/send_message.py

@@ -2,15 +2,17 @@
 import sys
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from workflows import BedrockParams, EntityBedrockWorkflow
 
+from util import get_temporal_config_path
+
 
 async def main(prompt):
     # Create client connected to server at the given address
-        # Get repo root - 2 levels deep from root
-        repo_root = Path(__file__).resolve().parent.parent.parent
-        config_file = repo_root / "temporal.toml"
-    config = ClientConfig.load_client_connect_config(config_file=str(config_file))
+    config = ClientConfig.load_client_connect_config(
+        config_file=str(get_temporal_config_path())
+    )
     config["target_host"] = "localhost:7233"
     client = await Client.connect(**config)
 

bedrock/signals_and_queries/get_history.py

@@ -1,15 +1,17 @@
 import asyncio
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from workflows import SignalQueryBedrockWorkflow
 
+from util import get_temporal_config_path
+
 
 async def main():
     # Create client connected to server at the given address
-        # Get repo root - 2 levels deep from root
-        repo_root = Path(__file__).resolve().parent.parent.parent
-        config_file = repo_root / "temporal.toml"
-    config = ClientConfig.load_client_connect_config(config_file=str(config_file))
+    config = ClientConfig.load_client_connect_config(
+        config_file=str(get_temporal_config_path())
+    )
     config["target_host"] = "localhost:7233"
     client = await Client.connect(**config)
     workflow_id = "bedrock-workflow-with-signals"