diff --git a/src/core/config.py b/src/core/config.py index 1646ff3..d4658e0 100644 --- a/src/core/config.py +++ b/src/core/config.py @@ -1,8 +1,7 @@ +from httpx import Client from pydantic import BaseModel from pydantic_settings import BaseSettings -from httpx import Client - class Settings(BaseSettings): VAULT_HOST: str @@ -35,6 +34,8 @@ class Config(BaseModel): OPENAI_API_KEY: str + TEMPOLAR_URL: str = "temporal:7233" + def get_config() -> Config: settings = Settings() # type: ignore @@ -45,7 +46,7 @@ def get_config() -> Config: headers={ "X-Vault-Token": settings.VAULT_TOKEN, "Content-Type": "application/json", - } + }, ) response.raise_for_status() diff --git a/src/core/temporal.py b/src/core/temporal.py index b58ea17..05bbe75 100644 --- a/src/core/temporal.py +++ b/src/core/temporal.py @@ -1,10 +1,10 @@ from temporalio.client import Client from temporalio.contrib.pydantic import pydantic_data_converter +from core.config import config + async def get_client() -> Client: return await Client.connect( - "temporal:7233", - namespace="default", - data_converter=pydantic_data_converter + config.TEMPOLAR_URL, namespace="default", data_converter=pydantic_data_converter )