update samples to use environment configuration for client

activity_worker/activity_worker.py

@@ -4,6 +4,7 @@
 
 from temporalio import activity
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from temporalio.worker import Worker
 
 task_queue = "say-hello-task-queue"
@@ -18,7 +19,9 @@ async def say_hello_activity(name: str) -> str:
 
 async def main():
     # Create client to localhost on default namespace
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
 
     # Run activity worker
     async with Worker(client, task_queue=task_queue, activities=[say_hello_activity]):

batch_sliding_window/starter.py

@@ -6,6 +6,7 @@
 import logging
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 
 from batch_sliding_window.batch_workflow import (
     ProcessBatchWorkflow,
@@ -19,7 +20,9 @@ async def main():
     logging.basicConfig(level=logging.INFO)
 
     # Create client
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
 
     # Create unique workflow ID with timestamp
     timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")

batch_sliding_window/worker.py

@@ -6,6 +6,7 @@
 
 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
@@ -19,7 +20,9 @@ async def main():
     logging.basicConfig(level=logging.INFO)
 
     # Create client
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
 
     # Create RecordLoader activity with sample data
     record_loader = RecordLoader(record_count=90)

bedrock/basic/run_worker.py

@@ -3,6 +3,7 @@
 import logging
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from temporalio.worker import Worker
 from workflows import BasicBedrockWorkflow
 
@@ -11,7 +12,10 @@
 
 async def main():
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
+
     activities = BedrockActivities()
 
     # Run the worker

bedrock/basic/send_message.py

@@ -2,12 +2,15 @@
 import sys
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from workflows import BasicBedrockWorkflow
 
 
 async def main(prompt: str) -> str:
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
 
     # Start the workflow
     workflow_id = "basic-bedrock-workflow"

bedrock/entity/end_chat.py

@@ -2,12 +2,15 @@
 import sys
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from workflows import EntityBedrockWorkflow
 
 
 async def main():
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
 
     workflow_id = "entity-bedrock-workflow"
 

bedrock/entity/get_history.py

@@ -1,12 +1,16 @@
 import asyncio
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from workflows import EntityBedrockWorkflow
 
 
 async def main():
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
+
     workflow_id = "entity-bedrock-workflow"
 
     handle = client.get_workflow_handle(workflow_id)

bedrock/entity/run_worker.py

@@ -3,6 +3,7 @@
 import logging
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from temporalio.worker import Worker
 from workflows import EntityBedrockWorkflow
 
@@ -11,7 +12,10 @@
 
 async def main():
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
+
     activities = BedrockActivities()
 
     # Run the worker

bedrock/entity/send_message.py

@@ -2,12 +2,15 @@
 import sys
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from workflows import BedrockParams, EntityBedrockWorkflow
 
 
 async def main(prompt):
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
 
     workflow_id = "entity-bedrock-workflow"
 

bedrock/signals_and_queries/get_history.py

@@ -1,12 +1,16 @@
 import asyncio
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from workflows import SignalQueryBedrockWorkflow
 
 
 async def main():
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
+
     workflow_id = "bedrock-workflow-with-signals"
 
     handle = client.get_workflow_handle(workflow_id)

bedrock/signals_and_queries/run_worker.py

@@ -3,6 +3,7 @@
 import logging
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from temporalio.worker import Worker
 from workflows import SignalQueryBedrockWorkflow
 
@@ -11,7 +12,10 @@
 
 async def main():
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
+
     activities = BedrockActivities()
 
     # Run the worker

bedrock/signals_and_queries/send_message.py

@@ -2,12 +2,15 @@
 import sys
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from workflows import SignalQueryBedrockWorkflow
 
 
 async def main(prompt):
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
 
     workflow_id = "bedrock-workflow-with-signals"
     inactivity_timeout_minutes = 1

cloud_export_to_parquet/create_schedule.py

@@ -10,6 +10,7 @@
     ScheduleSpec,
     WorkflowFailureError,
 )
+from temporalio.envconfig import ClientConfig
 
 from cloud_export_to_parquet.workflows import (
     ProtoToParquet,
@@ -20,7 +21,10 @@
 async def main() -> None:
     """Main function to run temporal workflow."""
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
+
     # TODO: update s3_bucket and namespace to the actual usecase
     wf_input = ProtoToParquetWorkflowInput(
         num_delay_hour=2,

cloud_export_to_parquet/run_worker.py

@@ -2,6 +2,7 @@
 from concurrent.futures import ThreadPoolExecutor
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from temporalio.worker import Worker
 from temporalio.worker.workflow_sandbox import (
     SandboxedWorkflowRunner,
@@ -18,7 +19,9 @@
 async def main() -> None:
     """Main worker function."""
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
 
     # Run the worker
     worker: Worker = Worker(

context_propagation/starter.py

@@ -2,6 +2,7 @@
 import logging
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 
 from context_propagation import interceptor, shared, workflows
 
@@ -12,9 +13,12 @@ async def main():
     # Set the user ID
     shared.user_id.set("some-user")
 
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+
     # Connect client
     client = await Client.connect(
-        "localhost:7233",
+        **config,
         # Use our interceptor
         interceptors=[interceptor.ContextPropagationInterceptor()],
     )

context_propagation/worker.py

@@ -2,6 +2,7 @@
 import logging
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from temporalio.worker import Worker
 
 from context_propagation import activities, interceptor, workflows
@@ -12,9 +13,12 @@
 async def main():
     logging.basicConfig(level=logging.INFO)
 
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+
     # Connect client
     client = await Client.connect(
-        "localhost:7233",
+        **config,
         # Use our interceptor
         interceptors=[interceptor.ContextPropagationInterceptor()],
     )

custom_converter/starter.py

@@ -1,6 +1,7 @@
 import asyncio
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 
 from custom_converter.shared import (
     GreetingInput,
@@ -11,9 +12,12 @@
 
 
 async def main():
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+
     # Connect client
     client = await Client.connect(
-        "localhost:7233",
+        **config,
         # Without this we get:
         #   TypeError: Object of type GreetingInput is not JSON serializable
         data_converter=greeting_data_converter,

custom_converter/worker.py

@@ -1,6 +1,7 @@
 import asyncio
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 from temporalio.worker import Worker
 
 from custom_converter.shared import greeting_data_converter
@@ -10,9 +11,12 @@
 
 
 async def main():
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+
     # Connect client
     client = await Client.connect(
-        "localhost:7233",
+        **config,
         # Without this, when trying to run a workflow, we get:
         #   KeyError: 'Unknown payload encoding my-greeting-encoding
         data_converter=greeting_data_converter,

custom_decorator/starter.py

@@ -1,13 +1,16 @@
 import asyncio
 
 from temporalio.client import Client
+from temporalio.envconfig import ClientConfig
 
 from custom_decorator.worker import WaitForCancelWorkflow
 
 
 async def main():
     # Connect client
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
 
     # Start the workflow
     handle = await client.start_workflow(

custom_decorator/worker.py

@@ -4,6 +4,7 @@
 from temporalio import activity, workflow
 from temporalio.client import Client
 from temporalio.common import RetryPolicy
+from temporalio.envconfig import ClientConfig
 from temporalio.worker import Worker
 
 from custom_decorator.activity_utils import auto_heartbeater
@@ -51,7 +52,9 @@ def cancel_activity(self) -> None:
 
 async def main():
     # Connect client
-    client = await Client.connect("localhost:7233")
+    config = ClientConfig.load_client_connect_config()
+    config.setdefault("target_host", "localhost:7233")
+    client = await Client.connect(**config)
 
     # Run a worker for the workflow
     async with Worker(