* feat: support timestamp="unset" to retain content without a date When callers retain timeless content (e.g. fictional documents, static reference material), passing timestamp="unset" now skips the utcnow() default so mentioned_at is stored as NULL instead of an artificial date. - HTTP: validate_timestamp recognises "unset" sentinel and threads it through api_retain as event_date=None (key present, value None), which the orchestrator distinguishes from key-absent (still defaults to now) - Orchestrator: new branching logic separates "key absent" → utcnow() from "key present but None" → no date - types.py: RetainContent.event_date and ProcessedFact.mentioned_at are now datetime | None; removed the unused _now_utc factory - fact_extraction.py: all event_date params accept datetime | None; _build_user_message emits "Event Date: Unknown" when None; removed mentioned_at from the Fact LLM response model (LLM never sets it) - embedding_processing: skip date suffix when fact_date is None - entity_resolver: COALESCE(event_date, now()) for first_seen/last_seen so entities table NOT NULL constraint is preserved - link_utils: skip temporal linking for units without event_date - Migration aa2b3c4d5e6f: DROP NOT NULL on memory_units.event_date - Tests: test_retain_no_timestamp and test_retain_omit_timestamp_defaults_to_now - Docs + OpenAPI + TypeScript client updated * refactor: replace _TIMESTAMP_UNKNOWN sentinel with plain string comparison The sentinel object() was only needed to distinguish "unset" from None at the boundary — but since the field type is datetime | str | None, "unset" can pass through the validator unchanged and be compared directly. * chore: regenerate OpenAPI spec and clients after timestamp type change timestamp field is now datetime | str | None to accept the "unset" sentinel value.
145 lines
5.1 KiB
Python
145 lines
5.1 KiB
Python
# coding: utf-8
|
|
|
|
"""
|
|
Hindsight HTTP API
|
|
|
|
HTTP API for Hindsight
|
|
|
|
The version of the OpenAPI document: 0.4.14
|
|
Generated by OpenAPI Generator (https://openapi-generator.tech)
|
|
|
|
Do not edit the class manually.
|
|
""" # noqa: E501
|
|
|
|
|
|
from __future__ import annotations
|
|
from inspect import getfullargspec
|
|
import json
|
|
import pprint
|
|
import re # noqa: F401
|
|
from datetime import datetime
|
|
from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
|
|
from typing import Optional
|
|
from typing import Union, Any, List, Set, TYPE_CHECKING, Optional, Dict
|
|
from typing_extensions import Literal, Self
|
|
from pydantic import Field
|
|
|
|
TIMESTAMP_ANY_OF_SCHEMAS = ["datetime", "str"]
|
|
|
|
class Timestamp(BaseModel):
|
|
"""
|
|
When the content occurred. Accepts an ISO 8601 datetime string (e.g. '2024-01-15T10:30:00Z'), null/omitted (defaults to now), or the special string 'unset' to explicitly store without any timestamp (use this for timeless content such as fictional documents or static reference material).
|
|
"""
|
|
|
|
# data type: datetime
|
|
anyof_schema_1_validator: Optional[datetime] = None
|
|
# data type: str
|
|
anyof_schema_2_validator: Optional[StrictStr] = None
|
|
if TYPE_CHECKING:
|
|
actual_instance: Optional[Union[datetime, str]] = None
|
|
else:
|
|
actual_instance: Any = None
|
|
any_of_schemas: Set[str] = { "datetime", "str" }
|
|
|
|
model_config = {
|
|
"validate_assignment": True,
|
|
"protected_namespaces": (),
|
|
}
|
|
|
|
def __init__(self, *args, **kwargs) -> None:
|
|
if args:
|
|
if len(args) > 1:
|
|
raise ValueError("If a position argument is used, only 1 is allowed to set `actual_instance`")
|
|
if kwargs:
|
|
raise ValueError("If a position argument is used, keyword arguments cannot be used.")
|
|
super().__init__(actual_instance=args[0])
|
|
else:
|
|
super().__init__(**kwargs)
|
|
|
|
@field_validator('actual_instance')
|
|
def actual_instance_must_validate_anyof(cls, v):
|
|
if v is None:
|
|
return v
|
|
|
|
instance = Timestamp.model_construct()
|
|
error_messages = []
|
|
# validate data type: datetime
|
|
try:
|
|
instance.anyof_schema_1_validator = v
|
|
return v
|
|
except (ValidationError, ValueError) as e:
|
|
error_messages.append(str(e))
|
|
# validate data type: str
|
|
try:
|
|
instance.anyof_schema_2_validator = v
|
|
return v
|
|
except (ValidationError, ValueError) as e:
|
|
error_messages.append(str(e))
|
|
if error_messages:
|
|
# no match
|
|
raise ValueError("No match found when setting the actual_instance in Timestamp with anyOf schemas: datetime, str. Details: " + ", ".join(error_messages))
|
|
else:
|
|
return v
|
|
|
|
@classmethod
|
|
def from_dict(cls, obj: Dict[str, Any]) -> Self:
|
|
return cls.from_json(json.dumps(obj))
|
|
|
|
@classmethod
|
|
def from_json(cls, json_str: str) -> Self:
|
|
"""Returns the object represented by the json string"""
|
|
instance = cls.model_construct()
|
|
if json_str is None:
|
|
return instance
|
|
|
|
error_messages = []
|
|
# deserialize data into datetime
|
|
try:
|
|
# validation
|
|
instance.anyof_schema_1_validator = json.loads(json_str)
|
|
# assign value to actual_instance
|
|
instance.actual_instance = instance.anyof_schema_1_validator
|
|
return instance
|
|
except (ValidationError, ValueError) as e:
|
|
error_messages.append(str(e))
|
|
# deserialize data into str
|
|
try:
|
|
# validation
|
|
instance.anyof_schema_2_validator = json.loads(json_str)
|
|
# assign value to actual_instance
|
|
instance.actual_instance = instance.anyof_schema_2_validator
|
|
return instance
|
|
except (ValidationError, ValueError) as e:
|
|
error_messages.append(str(e))
|
|
|
|
if error_messages:
|
|
# no match
|
|
raise ValueError("No match found when deserializing the JSON string into Timestamp with anyOf schemas: datetime, str. Details: " + ", ".join(error_messages))
|
|
else:
|
|
return instance
|
|
|
|
def to_json(self) -> str:
|
|
"""Returns the JSON representation of the actual instance"""
|
|
if self.actual_instance is None:
|
|
return "null"
|
|
|
|
if hasattr(self.actual_instance, "to_json") and callable(self.actual_instance.to_json):
|
|
return self.actual_instance.to_json()
|
|
else:
|
|
return json.dumps(self.actual_instance)
|
|
|
|
def to_dict(self) -> Optional[Union[Dict[str, Any], datetime, str]]:
|
|
"""Returns the dict representation of the actual instance"""
|
|
if self.actual_instance is None:
|
|
return None
|
|
|
|
if hasattr(self.actual_instance, "to_dict") and callable(self.actual_instance.to_dict):
|
|
return self.actual_instance.to_dict()
|
|
else:
|
|
return self.actual_instance
|
|
|
|
def to_str(self) -> str:
|
|
"""Returns the string representation of the actual instance"""
|
|
return pprint.pformat(self.model_dump())
|
|
|
|
|