block level script run (#3710)
This commit is contained in:
@@ -4152,6 +4152,7 @@ class AgentDB:
|
||||
organization_id: str,
|
||||
script_block_label: str,
|
||||
script_file_id: str | None = None,
|
||||
run_signature: str | None = None,
|
||||
) -> ScriptBlock:
|
||||
"""Create a script block."""
|
||||
async with self.Session() as session:
|
||||
@@ -4161,6 +4162,7 @@ class AgentDB:
|
||||
organization_id=organization_id,
|
||||
script_block_label=script_block_label,
|
||||
script_file_id=script_file_id,
|
||||
run_signature=run_signature,
|
||||
)
|
||||
session.add(script_block)
|
||||
await session.commit()
|
||||
@@ -4172,6 +4174,7 @@ class AgentDB:
|
||||
script_block_id: str,
|
||||
organization_id: str,
|
||||
script_file_id: str | None = None,
|
||||
run_signature: str | None = None,
|
||||
) -> ScriptBlock:
|
||||
async with self.Session() as session:
|
||||
script_block = (
|
||||
@@ -4182,8 +4185,10 @@ class AgentDB:
|
||||
)
|
||||
).first()
|
||||
if script_block:
|
||||
if script_file_id:
|
||||
if script_file_id is not None:
|
||||
script_block.script_file_id = script_file_id
|
||||
if run_signature is not None:
|
||||
script_block.run_signature = run_signature
|
||||
await session.commit()
|
||||
await session.refresh(script_block)
|
||||
return convert_to_script_block(script_block)
|
||||
|
||||
Reference in New Issue
Block a user