Add partition key to state operations

Signed-off-by: Albert Callarisa <albert@diagrid.io>
This commit is contained in:
Albert Callarisa 2025-08-05 15:44:05 +02:00
parent d993f9090b
commit d99e7f3036
No known key found for this signature in database
GPG Key ID: 450F38BCAA761685
1 changed files with 8 additions and 2 deletions

View File

@ -288,7 +288,10 @@ class AgenticWorkflow(
store_name=store_name,
key=store_key,
value=json.dumps({}),
state_metadata={"contentType": "application/json"},
state_metadata={
"contentType": "application/json",
"partitionKey": store_key,
},
options=StateOptions(
concurrency=Concurrency.first_write,
consistency=Consistency.strong,
@ -315,7 +318,10 @@ class AgenticWorkflow(
operation_type=TransactionOperationType.upsert,
)
],
transactional_metadata={"contentType": "application/json"},
transactional_metadata={
"contentType": "application/json",
"partitionKey": store_key,
},
)
except Exception as e:
raise e