dapr
/
dapr-agents
mirror of https://github.com/dapr/dapr-agents.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
dapr_agents/workflow/agentic.py
View File

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