Fix issue with workflows schema endpoint

roboflow · Nov 28, 2024 · c3d3800 · c3d3800
1 parent a011334
commit c3d3800
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/inference/core/entities/responses/workflows.py b/inference/core/entities/responses/workflows.py
@@ -1,6 +1,6 @@
 from typing import Any, Dict, List, Optional, Union
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, ConfigDict, Field
 
 from inference.core.workflows.core_steps.common.query_language.entities.introspection import (
     OperationDescription,
@@ -160,9 +160,8 @@ class ExecutionEngineVersions(BaseModel):
 
 
 class WorkflowsBlocksSchemaDescription(BaseModel):
-    blocks_schema: dict = Field(
-        description="Schema for validating block definitions", alias="schema"
-    )
+    model_config = ConfigDict(protected_namespaces=())
+    schema: dict = Field(description="Schema for validating block definitions")
 
 
 class DescribeInterfaceResponse(BaseModel):

diff --git a/inference/core/workflows/execution_engine/v1/compiler/syntactic_parser.py b/inference/core/workflows/execution_engine/v1/compiler/syntactic_parser.py
@@ -97,4 +97,4 @@ def get_workflow_schema_description() -> WorkflowsBlocksSchemaDescription:
         available_blocks=available_blocks
     )
     schema = workflow_definition_class.model_json_schema()
-    return WorkflowsBlocksSchemaDescription(blocks_schema=schema)
+    return WorkflowsBlocksSchemaDescription(schema=schema)