Compare commits

...

6 Commits

Author SHA1 Message Date
Bently
e0c4fb7871 Merge branch 'dev' into aarushikansal/remove-title-input-blocks 2024-12-18 16:00:39 -05:00
Aarushi
f9d02ad340 fix tests 2024-12-05 13:33:42 +00:00
Aarushi
e08921b5af remove title 2024-12-05 13:19:44 +00:00
Aarushi
a7b9ee6ab9 fix tests 2024-12-04 18:07:15 +00:00
Aarushi
5cdb382ea4 Merge branch 'dev' into aarushikansal/remove-title-input-blocks 2024-12-04 17:52:42 +00:00
Aarushi
0a4229b15b remove title from input and output blocks 2024-12-03 12:43:28 +00:00
3 changed files with 3 additions and 15 deletions

View File

@@ -146,9 +146,6 @@ class AgentInputBlock(Block):
description="The value to be passed as input.",
default=None,
)
title: str | None = SchemaField(
description="The title of the input.", default=None, advanced=True
)
description: str | None = SchemaField(
description="The description of the input.",
default=None,
@@ -230,11 +227,6 @@ class AgentOutputBlock(Block):
advanced=False,
)
name: str = SchemaField(description="The name of the output.")
title: str | None = SchemaField(
description="The title of the output.",
default=None,
advanced=True,
)
description: str | None = SchemaField(
description="The description of the output.",
default=None,

View File

@@ -194,7 +194,6 @@ class Graph(BaseDbModel):
p.name: {
"secret": p.secret,
"advanced": p.advanced,
"title": p.title or p.name,
**({"description": p.description} if p.description else {}),
**({"default": p.value} if p.value is not None else {}),
}

View File

@@ -90,7 +90,7 @@ async def test_get_input_schema(server: SpinTestServer):
Node(
id="node_0_a",
block_id=input_block,
input_default={"name": "in_key_a", "title": "Key A", "value": "A"},
input_default={"name": "in_key_a", "value": "A"},
metadata={"id": "node_0_a"},
),
Node(
@@ -138,20 +138,17 @@ async def test_get_input_schema(server: SpinTestServer):
)
class ExpectedInputSchema(BlockSchema):
in_key_a: Any = SchemaField(title="Key A", default="A", advanced=False)
in_key_b: Any = SchemaField(title="in_key_b", advanced=True)
in_key_a: Any = SchemaField(default="A", advanced=False)
in_key_b: Any = SchemaField(advanced=True)
class ExpectedOutputSchema(BlockSchema):
out_key: Any = SchemaField(
description="This is an output key",
title="out_key",
advanced=False,
)
input_schema = created_graph.input_schema
input_schema["title"] = "ExpectedInputSchema"
assert input_schema == ExpectedInputSchema.jsonschema()
output_schema = created_graph.output_schema
output_schema["title"] = "ExpectedOutputSchema"
assert output_schema == ExpectedOutputSchema.jsonschema()