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

View File

@@ -194,7 +194,6 @@ class Graph(BaseDbModel):
p.name: { p.name: {
"secret": p.secret, "secret": p.secret,
"advanced": p.advanced, "advanced": p.advanced,
"title": p.title or p.name,
**({"description": p.description} if p.description else {}), **({"description": p.description} if p.description else {}),
**({"default": p.value} if p.value is not None 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( Node(
id="node_0_a", id="node_0_a",
block_id=input_block, 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"}, metadata={"id": "node_0_a"},
), ),
Node( Node(
@@ -138,20 +138,17 @@ async def test_get_input_schema(server: SpinTestServer):
) )
class ExpectedInputSchema(BlockSchema): class ExpectedInputSchema(BlockSchema):
in_key_a: Any = SchemaField(title="Key A", default="A", advanced=False) in_key_a: Any = SchemaField(default="A", advanced=False)
in_key_b: Any = SchemaField(title="in_key_b", advanced=True) in_key_b: Any = SchemaField(advanced=True)
class ExpectedOutputSchema(BlockSchema): class ExpectedOutputSchema(BlockSchema):
out_key: Any = SchemaField( out_key: Any = SchemaField(
description="This is an output key", description="This is an output key",
title="out_key",
advanced=False, advanced=False,
) )
input_schema = created_graph.input_schema input_schema = created_graph.input_schema
input_schema["title"] = "ExpectedInputSchema"
assert input_schema == ExpectedInputSchema.jsonschema() assert input_schema == ExpectedInputSchema.jsonschema()
output_schema = created_graph.output_schema output_schema = created_graph.output_schema
output_schema["title"] = "ExpectedOutputSchema"
assert output_schema == ExpectedOutputSchema.jsonschema() assert output_schema == ExpectedOutputSchema.jsonschema()