feat: set default None values for Pydantic models

langfuse/api/resources/commons/types/create_event_request.py

@@ -13,17 +13,17 @@
 
 
 class CreateEventRequest(pydantic.BaseModel):
-    id: typing.Optional[str]
-    trace_id: typing.Optional[str] = pydantic.Field(alias="traceId")
-    name: typing.Optional[str]
-    start_time: typing.Optional[dt.datetime] = pydantic.Field(alias="startTime")
-    metadata: typing.Optional[typing.Any]
-    input: typing.Optional[typing.Any]
-    output: typing.Optional[typing.Any]
-    level: typing.Optional[ObservationLevel]
-    status_message: typing.Optional[str] = pydantic.Field(alias="statusMessage")
-    parent_observation_id: typing.Optional[str] = pydantic.Field(alias="parentObservationId")
-    version: typing.Optional[str]
+    id: typing.Optional[str] = None
+    trace_id: typing.Optional[str] = pydantic.Field(alias="traceId", default=None)
+    name: typing.Optional[str] = None
+    start_time: typing.Optional[dt.datetime] = pydantic.Field(alias="startTime", default=None)
+    metadata: typing.Optional[typing.Any] = None
+    input: typing.Optional[typing.Any] = None
+    output: typing.Optional[typing.Any] = None
+    level: typing.Optional[ObservationLevel] = None
+    status_message: typing.Optional[str] = pydantic.Field(alias="statusMessage", default=None)
+    parent_observation_id: typing.Optional[str] = pydantic.Field(alias="parentObservationId", default=None)
+    version: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

langfuse/api/resources/commons/types/create_generation_request.py

@@ -15,12 +15,14 @@
 
 
 class CreateGenerationRequest(CreateSpanRequest):
-    completion_start_time: typing.Optional[dt.datetime] = pydantic.Field(alias="completionStartTime")
-    model: typing.Optional[str]
-    model_parameters: typing.Optional[typing.Dict[str, MapValue]] = pydantic.Field(alias="modelParameters")
-    prompt: typing.Optional[typing.Any]
-    completion: typing.Optional[typing.Any]
-    usage: typing.Optional[LlmUsage]
+    completion_start_time: typing.Optional[dt.datetime] = pydantic.Field(alias="completionStartTime", default=None)
+    model: typing.Optional[str] = None
+    model_parameters: typing.Optional[typing.Dict[str, MapValue]] = pydantic.Field(
+        alias="modelParameters", default=None
+    )
+    prompt: typing.Optional[typing.Any] = None
+    completion: typing.Optional[typing.Any] = None
+    usage: typing.Optional[LlmUsage] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

langfuse/api/resources/commons/types/create_span_request.py

@@ -13,7 +13,7 @@
 
 
 class CreateSpanRequest(CreateEventRequest):
-    end_time: typing.Optional[dt.datetime] = pydantic.Field(alias="endTime")
+    end_time: typing.Optional[dt.datetime] = pydantic.Field(alias="endTime", default=None)
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

langfuse/api/resources/commons/types/dataset_item.py

@@ -16,8 +16,8 @@ class DatasetItem(pydantic.BaseModel):
     id: str
     status: DatasetStatus
     input: typing.Any
-    expected_output: typing.Optional[typing.Any] = pydantic.Field(alias="expectedOutput")
-    source_observation_id: typing.Optional[str] = pydantic.Field(alias="sourceObservationId")
+    expected_output: typing.Optional[typing.Any] = pydantic.Field(alias="expectedOutput", default=None)
+    source_observation_id: typing.Optional[str] = pydantic.Field(alias="sourceObservationId", default=None)
     dataset_id: str = pydantic.Field(alias="datasetId")
     created_at: dt.datetime = pydantic.Field(alias="createdAt")
     updated_at: dt.datetime = pydantic.Field(alias="updatedAt")

langfuse/api/resources/commons/types/llm_usage.py

@@ -12,9 +12,9 @@
 
 
 class LlmUsage(pydantic.BaseModel):
-    prompt_tokens: typing.Optional[int] = pydantic.Field(alias="promptTokens")
-    completion_tokens: typing.Optional[int] = pydantic.Field(alias="completionTokens")
-    total_tokens: typing.Optional[int] = pydantic.Field(alias="totalTokens")
+    prompt_tokens: typing.Optional[int] = pydantic.Field(alias="promptTokens", default=None)
+    completion_tokens: typing.Optional[int] = pydantic.Field(alias="completionTokens", default=None)
+    total_tokens: typing.Optional[int] = pydantic.Field(alias="totalTokens", default=None)
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

langfuse/api/resources/commons/types/observation.py

@@ -15,24 +15,26 @@
 
 class Observation(pydantic.BaseModel):
     id: str
-    trace_id: typing.Optional[str] = pydantic.Field(alias="traceId")
+    trace_id: typing.Optional[str] = pydantic.Field(alias="traceId", default=None)
     type: str
-    name: typing.Optional[str]
+    name: typing.Optional[str] = None
     start_time: dt.datetime = pydantic.Field(alias="startTime")
-    end_time: typing.Optional[dt.datetime] = pydantic.Field(alias="endTime")
-    completion_start_time: typing.Optional[dt.datetime] = pydantic.Field(alias="completionStartTime")
-    model: typing.Optional[str]
-    model_parameters: typing.Optional[typing.Dict[str, MapValue]] = pydantic.Field(alias="modelParameters")
-    input: typing.Optional[typing.Any]
-    version: typing.Optional[str]
-    metadata: typing.Optional[typing.Any]
-    output: typing.Optional[typing.Any]
+    end_time: typing.Optional[dt.datetime] = pydantic.Field(alias="endTime", default=None)
+    completion_start_time: typing.Optional[dt.datetime] = pydantic.Field(alias="completionStartTime", default=None)
+    model: typing.Optional[str] = None
+    model_parameters: typing.Optional[typing.Dict[str, MapValue]] = pydantic.Field(
+        alias="modelParameters", default=None
+    )
+    input: typing.Optional[typing.Any] = None
+    version: typing.Optional[str] = None
+    metadata: typing.Optional[typing.Any] = None
+    output: typing.Optional[typing.Any] = None
     prompt_tokens: int = pydantic.Field(alias="promptTokens")
     completion_tokens: int = pydantic.Field(alias="completionTokens")
     total_tokens: int = pydantic.Field(alias="totalTokens")
     level: ObservationLevel
-    status_message: typing.Optional[str] = pydantic.Field(alias="statusMessage")
-    parent_observation_id: typing.Optional[str] = pydantic.Field(alias="parentObservationId")
+    status_message: typing.Optional[str] = pydantic.Field(alias="statusMessage", default=None)
+    parent_observation_id: typing.Optional[str] = pydantic.Field(alias="parentObservationId", default=None)
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

langfuse/api/resources/commons/types/score.py

@@ -16,9 +16,9 @@ class Score(pydantic.BaseModel):
     trace_id: str = pydantic.Field(alias="traceId")
     name: str
     value: float
-    observation_id: typing.Optional[str] = pydantic.Field(alias="observationId")
+    observation_id: typing.Optional[str] = pydantic.Field(alias="observationId", default=None)
     timestamp: dt.datetime
-    comment: typing.Optional[str]
+    comment: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

langfuse/api/resources/commons/types/trace.py

@@ -14,12 +14,14 @@
 class Trace(pydantic.BaseModel):
     id: str = pydantic.Field(description="The unique identifier of a trace")
     timestamp: dt.datetime
-    name: typing.Optional[str]
-    release: typing.Optional[str]
-    version: typing.Optional[str]
-    user_id: typing.Optional[str] = pydantic.Field(alias="userId")
-    metadata: typing.Optional[typing.Any]
-    public: typing.Optional[bool] = pydantic.Field(description="Public traces are accessible via url without login")
+    name: typing.Optional[str] = None
+    release: typing.Optional[str] = None
+    version: typing.Optional[str] = None
+    user_id: typing.Optional[str] = pydantic.Field(alias="userId", default=None)
+    metadata: typing.Optional[typing.Any] = None
+    public: typing.Optional[bool] = pydantic.Field(
+        default=None, description="Public traces are accessible via url without login"
+    )
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

langfuse/api/resources/dataset_items/types/create_dataset_item_request.py

@@ -14,8 +14,8 @@
 class CreateDatasetItemRequest(pydantic.BaseModel):
     dataset_name: str = pydantic.Field(alias="datasetName")
     input: typing.Any
-    expected_output: typing.Optional[typing.Any] = pydantic.Field(alias="expectedOutput")
-    id: typing.Optional[str]
+    expected_output: typing.Optional[typing.Any] = pydantic.Field(alias="expectedOutput", default=None)
+    id: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

langfuse/api/resources/generations/types/update_generation_request.py

@@ -16,20 +16,22 @@
 
 class UpdateGenerationRequest(pydantic.BaseModel):
     generation_id: str = pydantic.Field(alias="generationId")
-    trace_id: typing.Optional[str] = pydantic.Field(alias="traceId")
-    name: typing.Optional[str]
-    start_time: typing.Optional[dt.datetime] = pydantic.Field(alias="startTime")
-    end_time: typing.Optional[dt.datetime] = pydantic.Field(alias="endTime")
-    completion_start_time: typing.Optional[dt.datetime] = pydantic.Field(alias="completionStartTime")
-    model: typing.Optional[str]
-    model_parameters: typing.Optional[typing.Dict[str, MapValue]] = pydantic.Field(alias="modelParameters")
-    prompt: typing.Optional[typing.Any]
-    version: typing.Optional[str]
-    metadata: typing.Optional[typing.Any]
-    completion: typing.Optional[typing.Any]
-    usage: typing.Optional[LlmUsage]
-    level: typing.Optional[ObservationLevel]
-    status_message: typing.Optional[str] = pydantic.Field(alias="statusMessage")
+    trace_id: typing.Optional[str] = pydantic.Field(alias="traceId", default=None)
+    name: typing.Optional[str] = None
+    start_time: typing.Optional[dt.datetime] = pydantic.Field(alias="startTime", default=None)
+    end_time: typing.Optional[dt.datetime] = pydantic.Field(alias="endTime", default=None)
+    completion_start_time: typing.Optional[dt.datetime] = pydantic.Field(alias="completionStartTime", default=None)
+    model: typing.Optional[str] = None
+    model_parameters: typing.Optional[typing.Dict[str, MapValue]] = pydantic.Field(
+        alias="modelParameters", default=None
+    )
+    prompt: typing.Optional[typing.Any] = None
+    version: typing.Optional[str] = None
+    metadata: typing.Optional[typing.Any] = None
+    completion: typing.Optional[typing.Any] = None
+    usage: typing.Optional[LlmUsage] = None
+    level: typing.Optional[ObservationLevel] = None
+    status_message: typing.Optional[str] = pydantic.Field(alias="statusMessage", default=None)
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

langfuse/api/resources/score/types/create_score_request.py

@@ -12,12 +12,12 @@
 
 
 class CreateScoreRequest(pydantic.BaseModel):
-    id: typing.Optional[str]
+    id: typing.Optional[str] = None
     trace_id: str = pydantic.Field(alias="traceId")
     name: str
     value: float
-    observation_id: typing.Optional[str] = pydantic.Field(alias="observationId")
-    comment: typing.Optional[str]
+    observation_id: typing.Optional[str] = pydantic.Field(alias="observationId", default=None)
+    comment: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

langfuse/api/resources/span/types/update_span_request.py

@@ -14,16 +14,16 @@
 
 class UpdateSpanRequest(pydantic.BaseModel):
     span_id: str = pydantic.Field(alias="spanId")
-    trace_id: typing.Optional[str] = pydantic.Field(alias="traceId")
-    start_time: typing.Optional[dt.datetime] = pydantic.Field(alias="startTime")
-    end_time: typing.Optional[dt.datetime] = pydantic.Field(alias="endTime")
-    name: typing.Optional[str]
-    metadata: typing.Optional[typing.Any]
-    input: typing.Optional[typing.Any]
-    output: typing.Optional[typing.Any]
-    level: typing.Optional[ObservationLevel]
-    version: typing.Optional[str]
-    status_message: typing.Optional[str] = pydantic.Field(alias="statusMessage")
+    trace_id: typing.Optional[str] = pydantic.Field(alias="traceId", default=None)
+    start_time: typing.Optional[dt.datetime] = pydantic.Field(alias="startTime", default=None)
+    end_time: typing.Optional[dt.datetime] = pydantic.Field(alias="endTime", default=None)
+    name: typing.Optional[str] = None
+    metadata: typing.Optional[typing.Any] = None
+    input: typing.Optional[typing.Any] = None
+    output: typing.Optional[typing.Any] = None
+    level: typing.Optional[ObservationLevel] = None
+    version: typing.Optional[str] = None
+    status_message: typing.Optional[str] = pydantic.Field(alias="statusMessage", default=None)
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

langfuse/api/resources/trace/types/create_trace_request.py

@@ -12,14 +12,14 @@
 
 
 class CreateTraceRequest(pydantic.BaseModel):
-    id: typing.Optional[str]
-    name: typing.Optional[str]
-    user_id: typing.Optional[str] = pydantic.Field(alias="userId")
-    external_id: typing.Optional[str] = pydantic.Field(alias="externalId")
-    release: typing.Optional[str]
-    version: typing.Optional[str]
-    metadata: typing.Optional[typing.Any]
-    public: typing.Optional[bool] = pydantic.Field(description="Make trace publicly accessible via url")
+    id: typing.Optional[str] = None
+    name: typing.Optional[str] = None
+    user_id: typing.Optional[str] = pydantic.Field(alias="userId", default=None)
+    external_id: typing.Optional[str] = pydantic.Field(alias="externalId", default=None)
+    release: typing.Optional[str] = None
+    version: typing.Optional[str] = None
+    metadata: typing.Optional[typing.Any] = None
+    public: typing.Optional[bool] = pydantic.Field(default=None, description="Make trace publicly accessible via url")
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}