Skip to content

Commit a92c84f

Browse files
committed
linting fixes
1 parent ead750d commit a92c84f

File tree

107 files changed

+699
-783
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

107 files changed

+699
-783
lines changed

activity_worker/activity_worker.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,14 @@
11
import asyncio
22
import random
33
import string
4-
from pathlib import Path
54

65
from temporalio import activity
76
from temporalio.client import Client
87
from temporalio.envconfig import ClientConfig
98
from temporalio.worker import Worker
109

10+
from util import get_temporal_config_path
11+
1112
task_queue = "say-hello-task-queue"
1213
workflow_name = "say-hello-workflow"
1314
activity_name = "say-hello-activity"
@@ -20,11 +21,10 @@ async def say_hello_activity(name: str) -> str:
2021

2122
async def main():
2223
# Create client to localhost on default namespace
23-
# Get repo root - 1 level deep from root
24-
repo_root = Path(__file__).resolve().parent.parent
25-
config_file = repo_root / "temporal.toml"
26-
config = ClientConfig.load_client_connect_config(config_file=str(config_file))
27-
config["target_host"] = "localhost:7233"
24+
config = ClientConfig.load_client_connect_config(
25+
config_file=str(get_temporal_config_path())
26+
)
27+
2828
client = await Client.connect(**config)
2929

3030
# Run activity worker

batch_sliding_window/starter.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,13 @@
66
import logging
77

88
from temporalio.client import Client
9+
from temporalio.envconfig import ClientConfig
910

1011
from batch_sliding_window.batch_workflow import (
1112
ProcessBatchWorkflow,
1213
ProcessBatchWorkflowInput,
1314
)
15+
from util import get_temporal_config_path
1416

1517

1618
async def main():
@@ -19,11 +21,10 @@ async def main():
1921
logging.basicConfig(level=logging.INFO)
2022

2123
# Create client
22-
# Get repo root - 1 level deep from root
23-
repo_root = Path(__file__).resolve().parent.parent
24-
config_file = repo_root / "temporal.toml"
25-
config = ClientConfig.load_client_connect_config(config_file=str(config_file))
26-
config["target_host"] = "localhost:7233"
24+
config = ClientConfig.load_client_connect_config(
25+
config_file=str(get_temporal_config_path())
26+
)
27+
2728
client = await Client.connect(**config)
2829

2930
# Create unique workflow ID with timestamp

batch_sliding_window/worker.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,13 @@
66

77
from temporalio import worker
88
from temporalio.client import Client
9+
from temporalio.envconfig import ClientConfig
910

1011
from batch_sliding_window.batch_workflow import ProcessBatchWorkflow
1112
from batch_sliding_window.record_loader_activity import RecordLoader
1213
from batch_sliding_window.record_processor_workflow import RecordProcessorWorkflow
1314
from batch_sliding_window.sliding_window_workflow import SlidingWindowWorkflow
15+
from util import get_temporal_config_path
1416

1517

1618
async def main():
@@ -19,11 +21,10 @@ async def main():
1921
logging.basicConfig(level=logging.INFO)
2022

2123
# Create client
22-
# Get repo root - 1 level deep from root
23-
repo_root = Path(__file__).resolve().parent.parent
24-
config_file = repo_root / "temporal.toml"
25-
config = ClientConfig.load_client_connect_config(config_file=str(config_file))
26-
config["target_host"] = "localhost:7233"
24+
config = ClientConfig.load_client_connect_config(
25+
config_file=str(get_temporal_config_path())
26+
)
27+
2728
client = await Client.connect(**config)
2829

2930
# Create RecordLoader activity with sample data

bedrock/basic/run_worker.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,19 +3,20 @@
33
import logging
44

55
from temporalio.client import Client
6+
from temporalio.envconfig import ClientConfig
67
from temporalio.worker import Worker
78
from workflows import BasicBedrockWorkflow
89

910
from bedrock.shared.activities import BedrockActivities
11+
from util import get_temporal_config_path
1012

1113

1214
async def main():
1315
# Create client connected to server at the given address
14-
# Get repo root - 2 levels deep from root
15-
repo_root = Path(__file__).resolve().parent.parent.parent
16-
config_file = repo_root / "temporal.toml"
17-
config = ClientConfig.load_client_connect_config(config_file=str(config_file))
18-
config["target_host"] = "localhost:7233"
16+
config = ClientConfig.load_client_connect_config(
17+
config_file=str(get_temporal_config_path())
18+
)
19+
1920
client = await Client.connect(**config)
2021
activities = BedrockActivities()
2122

bedrock/basic/send_message.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,16 +2,18 @@
22
import sys
33

44
from temporalio.client import Client
5+
from temporalio.envconfig import ClientConfig
56
from workflows import BasicBedrockWorkflow
67

8+
from util import get_temporal_config_path
9+
710

811
async def main(prompt: str) -> str:
912
# Create client connected to server at the given address
10-
# Get repo root - 2 levels deep from root
11-
repo_root = Path(__file__).resolve().parent.parent.parent
12-
config_file = repo_root / "temporal.toml"
13-
config = ClientConfig.load_client_connect_config(config_file=str(config_file))
14-
config["target_host"] = "localhost:7233"
13+
config = ClientConfig.load_client_connect_config(
14+
config_file=str(get_temporal_config_path())
15+
)
16+
1517
client = await Client.connect(**config)
1618

1719
# Start the workflow

bedrock/entity/end_chat.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,16 +2,18 @@
22
import sys
33

44
from temporalio.client import Client
5+
from temporalio.envconfig import ClientConfig
56
from workflows import EntityBedrockWorkflow
67

8+
from util import get_temporal_config_path
9+
710

811
async def main():
912
# Create client connected to server at the given address
10-
# Get repo root - 2 levels deep from root
11-
repo_root = Path(__file__).resolve().parent.parent.parent
12-
config_file = repo_root / "temporal.toml"
13-
config = ClientConfig.load_client_connect_config(config_file=str(config_file))
14-
config["target_host"] = "localhost:7233"
13+
config = ClientConfig.load_client_connect_config(
14+
config_file=str(get_temporal_config_path())
15+
)
16+
1517
client = await Client.connect(**config)
1618

1719
workflow_id = "entity-bedrock-workflow"

bedrock/entity/get_history.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,18 @@
11
import asyncio
22

33
from temporalio.client import Client
4+
from temporalio.envconfig import ClientConfig
45
from workflows import EntityBedrockWorkflow
56

7+
from util import get_temporal_config_path
8+
69

710
async def main():
811
# Create client connected to server at the given address
9-
# Get repo root - 2 levels deep from root
10-
repo_root = Path(__file__).resolve().parent.parent.parent
11-
config_file = repo_root / "temporal.toml"
12-
config = ClientConfig.load_client_connect_config(config_file=str(config_file))
13-
config["target_host"] = "localhost:7233"
12+
config = ClientConfig.load_client_connect_config(
13+
config_file=str(get_temporal_config_path())
14+
)
15+
1416
client = await Client.connect(**config)
1517
workflow_id = "entity-bedrock-workflow"
1618

bedrock/entity/run_worker.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,19 +3,20 @@
33
import logging
44

55
from temporalio.client import Client
6+
from temporalio.envconfig import ClientConfig
67
from temporalio.worker import Worker
78
from workflows import EntityBedrockWorkflow
89

910
from bedrock.shared.activities import BedrockActivities
11+
from util import get_temporal_config_path
1012

1113

1214
async def main():
1315
# Create client connected to server at the given address
14-
# Get repo root - 2 levels deep from root
15-
repo_root = Path(__file__).resolve().parent.parent.parent
16-
config_file = repo_root / "temporal.toml"
17-
config = ClientConfig.load_client_connect_config(config_file=str(config_file))
18-
config["target_host"] = "localhost:7233"
16+
config = ClientConfig.load_client_connect_config(
17+
config_file=str(get_temporal_config_path())
18+
)
19+
1920
client = await Client.connect(**config)
2021
activities = BedrockActivities()
2122

bedrock/entity/send_message.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,16 +2,18 @@
22
import sys
33

44
from temporalio.client import Client
5+
from temporalio.envconfig import ClientConfig
56
from workflows import BedrockParams, EntityBedrockWorkflow
67

8+
from util import get_temporal_config_path
9+
710

811
async def main(prompt):
912
# Create client connected to server at the given address
10-
# Get repo root - 2 levels deep from root
11-
repo_root = Path(__file__).resolve().parent.parent.parent
12-
config_file = repo_root / "temporal.toml"
13-
config = ClientConfig.load_client_connect_config(config_file=str(config_file))
14-
config["target_host"] = "localhost:7233"
13+
config = ClientConfig.load_client_connect_config(
14+
config_file=str(get_temporal_config_path())
15+
)
16+
1517
client = await Client.connect(**config)
1618

1719
workflow_id = "entity-bedrock-workflow"

bedrock/signals_and_queries/get_history.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,18 @@
11
import asyncio
22

33
from temporalio.client import Client
4+
from temporalio.envconfig import ClientConfig
45
from workflows import SignalQueryBedrockWorkflow
56

7+
from util import get_temporal_config_path
8+
69

710
async def main():
811
# Create client connected to server at the given address
9-
# Get repo root - 2 levels deep from root
10-
repo_root = Path(__file__).resolve().parent.parent.parent
11-
config_file = repo_root / "temporal.toml"
12-
config = ClientConfig.load_client_connect_config(config_file=str(config_file))
13-
config["target_host"] = "localhost:7233"
12+
config = ClientConfig.load_client_connect_config(
13+
config_file=str(get_temporal_config_path())
14+
)
15+
1416
client = await Client.connect(**config)
1517
workflow_id = "bedrock-workflow-with-signals"
1618

0 commit comments

Comments
 (0)