Skip to content

Commit 2cf75d0

Browse files
committed
add emum query
1 parent b589e8b commit 2cf75d0

File tree

2 files changed

+2
-12
lines changed

2 files changed

+2
-12
lines changed

api/schema/dataset_schema.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@
4141
DatasetLicense,
4242
DatasetStatus,
4343
DatasetType,
44+
PromptFormat,
4445
PromptTaskType,
4546
UseCaseStatus,
4647
)
@@ -57,6 +58,7 @@
5758
DatasetLicenseENUM = strawberry.enum(DatasetLicense) # type: ignore
5859
DatasetTypeENUM = strawberry.enum(DatasetType) # type: ignore
5960
PromptTaskTypeENUM = strawberry.enum(PromptTaskType) # type: ignore
61+
PromptFormatENUM = strawberry.enum(PromptFormat) # type: ignore
6062

6163

6264
# Create permission classes dynamically with different operations

api/types/type_prompt_metadata.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -43,15 +43,3 @@ class TypePromptDataset(TypeDataset):
4343
domain: Optional[prompt_domain_enum]
4444
target_model_types: Optional[List[target_model_type_enum]]
4545
evaluation_criteria: Optional[strawberry.scalars.JSON]
46-
47-
48-
# Keep TypePromptMetadata as an alias for backward compatibility in nested queries
49-
@strawberry.type
50-
class TypePromptMetadata:
51-
"""Prompt-specific metadata fields (for embedding in TypeDataset)."""
52-
53-
task_type: Optional[prompt_task_type_enum]
54-
target_languages: Optional[List[target_language_enum]]
55-
domain: Optional[prompt_domain_enum]
56-
target_model_types: Optional[List[target_model_type_enum]]
57-
evaluation_criteria: Optional[strawberry.scalars.JSON]

0 commit comments

Comments
 (0)