junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-30 13:10:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

address final PR comments

Browse source
This commit is contained in:
Daniel Roth 2026-02-27 15:27:07 +00:00
parent b65d1e0f7e
commit 2f7823d6c0
3 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
backend/.env.test
View file

@ -19,4 +19,5 @@ PLAN_TRIGGER_BUCKET=test
DATA_BUCKET=test DATA_BUCKET=test
EPC_AUTH_TOKEN=test EPC_AUTH_TOKEN=test
ENGINE_SQS_URL=test ENGINE_SQS_URL=test
CATEGORISATION_SQS_URL=test
ENERGY_ASSESSMENTS_BUCKET=test ENERGY_ASSESSMENTS_BUCKET=test

2
backend/app/config.py
View file

@ -16,6 +16,7 @@ def resolve_env_file() -> Optional[str]:
if env == "local": if env == "local":
env_file = backend_dir / ".env" env_file = backend_dir / ".env"
print("USING ENV FILE:", env_file)
logger.debug("USING ENV FILE:", env_file) logger.debug("USING ENV FILE:", env_file)
return str(env_file) return str(env_file)
@ -36,6 +37,7 @@ class Settings(BaseSettings):
DATA_BUCKET: str = "changeme" DATA_BUCKET: str = "changeme"
PLAN_TRIGGER_BUCKET: str PLAN_TRIGGER_BUCKET: str
ENGINE_SQS_URL: str = "changeme" ENGINE_SQS_URL: str = "changeme"
CATEGORISATION_SQS_URL: str = "changeme"
# Third parties # Third parties
EPC_AUTH_TOKEN: str = "changeme" EPC_AUTH_TOKEN: str = "changeme"

4
backend/app/plan/router.py
View file

@ -70,7 +70,7 @@ async def trigger_categorisation(
# Create task # Create task
task_id, _ = TasksInterface.create_task( task_id, _ = TasksInterface.create_task(
task_source="backend/plan/router.py:trigger_categorisation", task_source="backend/plan/router.py:trigger_categorisation",
service="plan_engine", service="plan_categorisation",
inputs=payload.model_dump(), inputs=payload.model_dump(),
task_only=True, task_only=True,
) )
@ -102,7 +102,7 @@ async def trigger_categorisation(
batch_request.subtask_id = str(subtask_id) batch_request.subtask_id = str(subtask_id)
response = sqs_client.send_message( response = sqs_client.send_message(
QueueUrl="categorisation-queue-dev", QueueUrl=settings.CATEGORISATION_SQS_URL,
MessageBody=batch_request.model_dump_json(), MessageBody=batch_request.model_dump_json(),
) )