Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 14 additions & 0 deletions langfuse/api/resources/commons/types/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,20 @@ class Dataset(pydantic_v1.BaseModel):
name: str
description: typing.Optional[str] = None
metadata: typing.Optional[typing.Any] = None
input_schema: typing.Optional[typing.Any] = pydantic_v1.Field(
alias="inputSchema", default=None
)
"""
JSON Schema for validating dataset item inputs
"""

expected_output_schema: typing.Optional[typing.Any] = pydantic_v1.Field(
alias="expectedOutputSchema", default=None
)
"""
JSON Schema for validating dataset item expected outputs
"""

project_id: str = pydantic_v1.Field(alias="projectId")
created_at: dt.datetime = pydantic_v1.Field(alias="createdAt")
updated_at: dt.datetime = pydantic_v1.Field(alias="updatedAt")
Expand Down
15 changes: 15 additions & 0 deletions langfuse/api/resources/datasets/types/create_dataset_request.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,19 @@ class CreateDatasetRequest(pydantic_v1.BaseModel):
name: str
description: typing.Optional[str] = None
metadata: typing.Optional[typing.Any] = None
input_schema: typing.Optional[typing.Any] = pydantic_v1.Field(
alias="inputSchema", default=None
)
"""
JSON Schema for validating dataset item inputs. When set, all new and existing dataset items will be validated against this schema.
"""

expected_output_schema: typing.Optional[typing.Any] = pydantic_v1.Field(
alias="expectedOutputSchema", default=None
)
"""
JSON Schema for validating dataset item expected outputs. When set, all new and existing dataset items will be validated against this schema.
"""

def json(self, **kwargs: typing.Any) -> str:
kwargs_with_defaults: typing.Any = {
Expand Down Expand Up @@ -40,5 +53,7 @@ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
class Config:
frozen = True
smart_union = True
allow_population_by_field_name = True
populate_by_name = True
extra = pydantic_v1.Extra.allow
json_encoders = {dt.datetime: serialize_datetime}
2 changes: 1 addition & 1 deletion langfuse/api/resources/score/types/create_score_request.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ class CreateScoreRequest(pydantic_v1.BaseModel):
"""

comment: typing.Optional[str] = None
metadata: typing.Optional[typing.Any] = None
metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
environment: typing.Optional[str] = pydantic_v1.Field(default=None)
"""
The environment of the score. Can be any lowercase alphanumeric string with hyphens and underscores that does not start with 'langfuse'.
Expand Down