hindsight-client 0.1.15__py3-none-any.whl → 0.2.0__py3-none-any.whl

hindsight_client/hindsight_client.py

@@ -112,6 +112,7 @@ class Hindsight:
         context: Optional[str] = None,
         document_id: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
+        entities: Optional[List[Dict[str, str]]] = None,
     ) -> RetainResponse:
         """
         Store a single memory (simplified interface).
@@ -123,13 +124,14 @@ class Hindsight:
             context: Optional context description
             document_id: Optional document ID for grouping
             metadata: Optional user-defined metadata
+            entities: Optional list of entities [{"text": "...", "type": "..."}]
 
         Returns:
             RetainResponse with success status
         """
         return self.retain_batch(
             bank_id=bank_id,
-            items=[{"content": content, "timestamp": timestamp, "context": context, "metadata": metadata}],
+            items=[{"content": content, "timestamp": timestamp, "context": context, "metadata": metadata, "entities": entities}],
             document_id=document_id,
         )
 
@@ -145,24 +147,34 @@ class Hindsight:
 
         Args:
             bank_id: The memory bank ID
-            items: List of memory items with 'content' and optional 'timestamp', 'context', 'metadata', 'document_id'
+            items: List of memory items with 'content' and optional 'timestamp', 'context', 'metadata', 'document_id', 'entities'
             document_id: Optional document ID for grouping memories (applied to items that don't have their own)
             retain_async: If True, process asynchronously in background (default: False)
 
         Returns:
             RetainResponse with success status and item count
         """
-        memory_items = [
-            memory_item.MemoryItem(
-                content=item["content"],
-                timestamp=item.get("timestamp"),
-                context=item.get("context"),
-                metadata=item.get("metadata"),
-                # Use item's document_id if provided, otherwise fall back to batch-level document_id
-                document_id=item.get("document_id") or document_id,
+        from hindsight_client_api.models.entity_input import EntityInput
+
+        memory_items = []
+        for item in items:
+            entities = None
+            if item.get("entities"):
+                entities = [
+                    EntityInput(text=e["text"], type=e.get("type"))
+                    for e in item["entities"]
+                ]
+            memory_items.append(
+                memory_item.MemoryItem(
+                    content=item["content"],
+                    timestamp=item.get("timestamp"),
+                    context=item.get("context"),
+                    metadata=item.get("metadata"),
+                    # Use item's document_id if provided, otherwise fall back to batch-level document_id
+                    document_id=item.get("document_id") or document_id,
+                    entities=entities,
+                )
             )
-            for item in items
-        ]
 
         request_obj = retain_request.RetainRequest(
             items=memory_items,
@@ -229,6 +241,8 @@ class Hindsight:
         query: str,
         budget: str = "low",
         context: Optional[str] = None,
+        max_tokens: Optional[int] = None,
+        response_schema: Optional[Dict[str, Any]] = None,
     ) -> ReflectResponse:
         """
         Generate a contextual answer based on bank identity and memories.
@@ -238,14 +252,21 @@ class Hindsight:
             query: The question or prompt
             budget: Budget level for reflection - "low", "mid", or "high" (default: "low")
             context: Optional additional context
+            max_tokens: Maximum tokens for the response (server default: 4096)
+            response_schema: Optional JSON Schema for structured output. When provided,
+                the response will include a 'structured_output' field with the LLM
+                response parsed according to this schema.
 
         Returns:
-            ReflectResponse with answer text and optionally facts used
+            ReflectResponse with answer text, optionally facts used, and optionally
+            structured_output if response_schema was provided
         """
         request_obj = reflect_request.ReflectRequest(
             query=query,
             budget=budget,
             context=context,
+            max_tokens=max_tokens,
+            response_schema=response_schema,
         )
 
         return _run_async(self._memory_api.reflect(bank_id, request_obj))
@@ -303,24 +324,34 @@ class Hindsight:
 
         Args:
             bank_id: The memory bank ID
-            items: List of memory items with 'content' and optional 'timestamp', 'context', 'metadata', 'document_id'
+            items: List of memory items with 'content' and optional 'timestamp', 'context', 'metadata', 'document_id', 'entities'
             document_id: Optional document ID for grouping memories (applied to items that don't have their own)
             retain_async: If True, process asynchronously in background (default: False)
 
         Returns:
             RetainResponse with success status and item count
         """
-        memory_items = [
-            memory_item.MemoryItem(
-                content=item["content"],
-                timestamp=item.get("timestamp"),
-                context=item.get("context"),
-                metadata=item.get("metadata"),
-                # Use item's document_id if provided, otherwise fall back to batch-level document_id
-                document_id=item.get("document_id") or document_id,
+        from hindsight_client_api.models.entity_input import EntityInput
+
+        memory_items = []
+        for item in items:
+            entities = None
+            if item.get("entities"):
+                entities = [
+                    EntityInput(text=e["text"], type=e.get("type"))
+                    for e in item["entities"]
+                ]
+            memory_items.append(
+                memory_item.MemoryItem(
+                    content=item["content"],
+                    timestamp=item.get("timestamp"),
+                    context=item.get("context"),
+                    metadata=item.get("metadata"),
+                    # Use item's document_id if provided, otherwise fall back to batch-level document_id
+                    document_id=item.get("document_id") or document_id,
+                    entities=entities,
+                )
             )
-            for item in items
-        ]
 
         request_obj = retain_request.RetainRequest(
             items=memory_items,
@@ -337,6 +368,7 @@ class Hindsight:
         context: Optional[str] = None,
         document_id: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
+        entities: Optional[List[Dict[str, str]]] = None,
     ) -> RetainResponse:
         """
         Store a single memory (async).
@@ -348,13 +380,14 @@ class Hindsight:
             context: Optional context description
             document_id: Optional document ID for grouping
             metadata: Optional user-defined metadata
+            entities: Optional list of entities [{"text": "...", "type": "..."}]
 
         Returns:
             RetainResponse with success status
         """
         return await self.aretain_batch(
             bank_id=bank_id,
-            items=[{"content": content, "timestamp": timestamp, "context": context, "metadata": metadata}],
+            items=[{"content": content, "timestamp": timestamp, "context": context, "metadata": metadata, "entities": entities}],
             document_id=document_id,
         )
 

{hindsight_client-0.1.15.dist-info → hindsight_client-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hindsight-client
-Version: 0.1.15
+Version: 0.2.0
 Summary: Python client for Hindsight - Semantic memory system with personality-driven thinking
 Author: Hindsight Team
 Requires-Python: >=3.10

{hindsight_client-0.1.15.dist-info → hindsight_client-0.2.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hindsight_client/__init__.py,sha256=PyDJ4UVKmtRN5OeBs0-rl-tUtqS8OoX53qvejKGC3JU,3114
-hindsight_client/hindsight_client.py,sha256=sOyI7okjIuq6U6Qf30dXsTR9ikDMC6FVtX71aZ_Ribw,14453
+hindsight_client/hindsight_client.py,sha256=FzlKL3HsoOvyYRipe4sKnZUJ-FWHrNe86fkAavPPSZA,16172
 hindsight_client_api/__init__.py,sha256=5PKjxuRIBdYzBm_35NF_B583xiI5TAlbxmTTKkF60j0,7004
 hindsight_client_api/api_client.py,sha256=6Q0Hdx1h4U549-Bfk9ckwSlIdF27arfLMMQsGeoyR98,27860
 hindsight_client_api/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
@@ -54,14 +54,14 @@ hindsight_client_api/docs/RecallResponse.md,sha256=VbABZlZwMTaSuqaqAAOOAjkuqaDB4
 hindsight_client_api/docs/RecallResult.md,sha256=1QTIqPN86Gq--tw-LwFVBmO5QD4dj8DbzJqe0ywUrew,1297
 hindsight_client_api/docs/ReflectFact.md,sha256=f9wmQde692is1dF6sw6S5_zIDI9Z_AHSbCvD7ALluLY,1075
 hindsight_client_api/docs/ReflectIncludeOptions.md,sha256=tH-pQeSv0QH9hoLmQhsg0PO5R3jVolDK6JFUG5CHi50,1112
-hindsight_client_api/docs/ReflectRequest.md,sha256=QIkRNJJsZS47yy20ZsoDFQhOFx8kYmB2wd0az85OelY,1158
-hindsight_client_api/docs/ReflectResponse.md,sha256=ziZtjLkXfX7RieaBtL5XC1JxgyG5_1rda-pZHHrQAaA,1019
+hindsight_client_api/docs/ReflectRequest.md,sha256=MxOgwBhzXZCbBWBhDDs89aWtOWZQ9bjQSGmsrirNkMo,1309
+hindsight_client_api/docs/ReflectResponse.md,sha256=oA6GmB264k9N8_j7tQYU4Ubj_S-mWXe0_quJTJ57k6w,1082
 hindsight_client_api/docs/RetainRequest.md,sha256=u-IhxRJo2iFWtskDsRpBJJjOdm7mfLGE6j1W6iNTcKY,1090
 hindsight_client_api/docs/RetainResponse.md,sha256=2a3Wz7ilrnQ_BFf6YTqWDmA94gohs8kU9sCR3mJYrj8,1065
 hindsight_client_api/docs/UpdateDispositionRequest.md,sha256=C3Q9UcVts2wPjYQgy1JH2ch1jy0tP2xswGe8sUzkZ_8,1112
 hindsight_client_api/docs/ValidationError.md,sha256=moV1dyA1z8n2DSD18ri7x7GDJujHc6--fbwGr4u8cl4,1001
 hindsight_client_api/docs/ValidationErrorLocInner.md,sha256=WP7Zm2LNt2HZb6XHo0cLvLCrht-EGFKp0i4zD3jPMto,988
-hindsight_client_api/models/__init__.py,sha256=a-vD1DQG9S6RMncyRj29QJZnkzCNuF17VVAwWMYrH58,3537
+hindsight_client_api/models/__init__.py,sha256=7d7fYdP_wGCEkXoXA3iA7aMZNJR1BZn49rUxiZeLt6I,3602
 hindsight_client_api/models/add_background_request.py,sha256=3rld3CjxNSPbxX9CoDSr5WEuhEoo-gRrCZ709Vvy9rI,2867
 hindsight_client_api/models/background_response.py,sha256=SUFt-k6ipl2SXAvHb3IU3IJ5LuXIGEoK8_pVbZO3Dgw,3144
 hindsight_client_api/models/bank_list_item.py,sha256=grJYCFBinjxEy1M8ZxPiy8asUVKj58b2KnScE2tmU-Q,4137
@@ -80,6 +80,7 @@ hindsight_client_api/models/disposition_traits.py,sha256=I61B7L3BkGQlAxBONDILwE-
 hindsight_client_api/models/document_response.py,sha256=hvtlBhm0EEGldM5Gh82cwnFYCL2ckwI2upDySdhbFZ0,3256
 hindsight_client_api/models/entity_detail_response.py,sha256=51Pv6lQEVGjUZ2KimRP0xFCowwOgt0ShcNITeHLF0Ro,4312
 hindsight_client_api/models/entity_include_options.py,sha256=j6o0SPWslW7y461xahbw4l1VXMIU1B5vxR3MX8cJ2FY,2635
+hindsight_client_api/models/entity_input.py,sha256=-ZGF71tAznuuTOHK1gIL-wNAfiAD2zvsIRldtgOmrsU,2431
 hindsight_client_api/models/entity_list_item.py,sha256=ZtdnkJ-xz7djdls1RaYXl7BEF_q1YIcan61MrHH0x6g,3602
 hindsight_client_api/models/entity_list_response.py,sha256=J3EYrjHKJk8AjcUQH2uEI3s1Hw11mIyQDgKrvPRhd4w,2913
 hindsight_client_api/models/entity_observation_response.py,sha256=Sbz_m0MssD-gYyV4sAeD56HGrnmlWcdO0bBs73K65YU,2804
@@ -89,7 +90,7 @@ hindsight_client_api/models/http_validation_error.py,sha256=KKvY67lajAI7hAys0IZH
 hindsight_client_api/models/include_options.py,sha256=FlB8QT_bqA141_tSv51QvZRbqKZdcaR2estkHg6eBGc,3645
 hindsight_client_api/models/list_documents_response.py,sha256=8lXIfprjJHpWCgiKkw4-M8d2y4NOTcK76HoF1otMekI,2675
 hindsight_client_api/models/list_memory_units_response.py,sha256=gEC-opRav_3cWdKvoO7Hure3dnVyFiFC0gtDP0TE7H8,2684
-hindsight_client_api/models/memory_item.py,sha256=2DisYT6Hnfpi9IbVkQfkOhlQQk_E9AqAmGn4wo_WcIA,3741
+hindsight_client_api/models/memory_item.py,sha256=xDMmKYS3ZixyiNATBX4E4_OLxYMfPxtXjs_Jhv792A0,4330
 hindsight_client_api/models/operation_response.py,sha256=ach9QI35J3FtPKLzZDqb54Xfov7ouiEhto4nztCKc8I,3467
 hindsight_client_api/models/operations_list_response.py,sha256=3XOfvTBBkv7VmYo-PZX0wVK_RtMCZkQdaPkXXhGIiRU,3088
 hindsight_client_api/models/recall_request.py,sha256=IPYeipQO7qxUepB01iCkz2ACNWQg5jXLIPzSAxsTQts,4053
@@ -97,13 +98,13 @@ hindsight_client_api/models/recall_response.py,sha256=t_qYAUMHXFBnXX7kCF8EeLfqey
 hindsight_client_api/models/recall_result.py,sha256=fI-TCBDFGuSIytFsI7FADtJkmvsTOX-G3bFFRxLJbMY,5402
 hindsight_client_api/models/reflect_fact.py,sha256=sYW00LmDpY7VptK0E4I0gUWHkCRaCTeypuHSdciknw8,3973
 hindsight_client_api/models/reflect_include_options.py,sha256=q3TqMu-zrJw8LIHOl5pb0baDEQlOlFkBI5gE9MEmKQQ,2584
-hindsight_client_api/models/reflect_request.py,sha256=_Rt9QSGz040309-WTN8YXWkfQ6KzGWWQAmGsaGgw0Iw,3407
-hindsight_client_api/models/reflect_response.py,sha256=z4rzUMGlFzk4DQKINQhaLHWmwe6kocgglSFePyaKLE0,3024
+hindsight_client_api/models/reflect_request.py,sha256=OOfOiLh0pmrIVo5ZFBFL09H69v8UGzWWlyp5I-CPCpE,4007
+hindsight_client_api/models/reflect_response.py,sha256=jqh3gJvqlz2MKbk4H9sgHe3EfprmpIoMZgxHMuxQlVw,3414
 hindsight_client_api/models/retain_request.py,sha256=LABq0vwi3ZYnb1RB6UCLw6PoFT4Q4qap8LNWLquh7jE,3178
 hindsight_client_api/models/retain_response.py,sha256=b3t5H7_ixi7jZw_kvb2XYh0Q6IDTfHvWRR2KaKqf7ls,2796
 hindsight_client_api/models/update_disposition_request.py,sha256=6D3qceEJByBWcZrfSbNXrdlrDtnj0_QXh2zosc49-5E,2817
 hindsight_client_api/models/validation_error.py,sha256=XnK71WeEUbZyXPbzv1uKaNAFEYxfsiS7G0cvjTgCxiM,3029
 hindsight_client_api/models/validation_error_loc_inner.py,sha256=q51Yi64xsxhsy3_AAD5DlcfI2-g_81hQoN6Olh3I8ag,4812
-hindsight_client-0.1.15.dist-info/METADATA,sha256=292P4UtcOi9P0ZYK_Alw_YJPlzLC_Jypo7fRawFu0n0,648
-hindsight_client-0.1.15.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-hindsight_client-0.1.15.dist-info/RECORD,,
+hindsight_client-0.2.0.dist-info/METADATA,sha256=cD7W3A8iLYPqI9llPg0kChvZda9gaWkJOh60pC10Zzs,647
+hindsight_client-0.2.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+hindsight_client-0.2.0.dist-info/RECORD,,

hindsight_client_api/docs/ReflectRequest.md

@@ -9,7 +9,9 @@ Name | Type | Description | Notes
 **query** | **str** |  | 
 **budget** | [**Budget**](Budget.md) |  | [optional] 
 **context** | **str** |  | [optional] 
+**max_tokens** | **int** | Maximum tokens for the response | [optional] [default to 4096]
 **include** | [**ReflectIncludeOptions**](ReflectIncludeOptions.md) | Options for including additional data (disabled by default) | [optional] 
+**response_schema** | **Dict[str, object]** |  | [optional] 
 
 ## Example
 

hindsight_client_api/docs/ReflectResponse.md

@@ -8,6 +8,7 @@ Name | Type | Description | Notes
 ------------ | ------------- | ------------- | -------------
 **text** | **str** |  | 
 **based_on** | [**List[ReflectFact]**](ReflectFact.md) |  | [optional] [default to []]
+**structured_output** | **Dict[str, object]** |  | [optional] 
 
 ## Example
 

hindsight_client_api/models/__init__.py

@@ -31,6 +31,7 @@ from hindsight_client_api.models.disposition_traits import DispositionTraits
 from hindsight_client_api.models.document_response import DocumentResponse
 from hindsight_client_api.models.entity_detail_response import EntityDetailResponse
 from hindsight_client_api.models.entity_include_options import EntityIncludeOptions
+from hindsight_client_api.models.entity_input import EntityInput
 from hindsight_client_api.models.entity_list_item import EntityListItem
 from hindsight_client_api.models.entity_list_response import EntityListResponse
 from hindsight_client_api.models.entity_observation_response import EntityObservationResponse

hindsight_client_api/models/entity_input.py

@@ -0,0 +1,82 @@
+# coding: utf-8
+
+"""
+    Hindsight HTTP API
+
+    HTTP API for Hindsight
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+
+class EntityInput(BaseModel):
+    """
+    Entity to associate with retained content.
+    """  # noqa: E501
+
+    text: StrictStr = Field(description="The entity name/text")
+    type: Optional[StrictStr] = Field(
+        default=None, description="Optional entity type (e.g., 'PERSON', 'ORG', 'CONCEPT')"
+    )
+    __properties: ClassVar[List[str]] = ["text", "type"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of EntityInput from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias."""
+        excluded_fields: Set[str] = set([])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # set to None if type (nullable) is None
+        # and model_fields_set contains the field
+        if self.type is None and "type" in self.model_fields_set:
+            _dict["type"] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of EntityInput from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({"text": obj.get("text"), "type": obj.get("type")})
+        return _obj

hindsight_client_api/models/memory_item.py

@@ -18,10 +18,11 @@ import re  # noqa: F401
 import json
 
 from datetime import datetime
-from pydantic import BaseModel, ConfigDict, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
+from hindsight_client_api.models.entity_input import EntityInput
 
 class MemoryItem(BaseModel):
     """
@@ -32,7 +33,11 @@ class MemoryItem(BaseModel):
     context: Optional[StrictStr] = None
     metadata: Optional[Dict[str, StrictStr]] = None
     document_id: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["content", "timestamp", "context", "metadata", "document_id"]
+    entities: Optional[List[EntityInput]] = Field(
+        default=None,
+        description="Optional entities to combine with auto-extracted entities."
+    )
+    __properties: ClassVar[List[str]] = ["content", "timestamp", "context", "metadata", "document_id", "entities"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -93,6 +98,11 @@ class MemoryItem(BaseModel):
         if self.document_id is None and "document_id" in self.model_fields_set:
             _dict['document_id'] = None
 
+        # set to None if entities (nullable) is None
+        # and model_fields_set contains the field
+        if self.entities is None and "entities" in self.model_fields_set:
+            _dict['entities'] = None
+
         return _dict
 
     @classmethod
@@ -109,7 +119,8 @@ class MemoryItem(BaseModel):
             "timestamp": obj.get("timestamp"),
             "context": obj.get("context"),
             "metadata": obj.get("metadata"),
-            "document_id": obj.get("document_id")
+            "document_id": obj.get("document_id"),
+            "entities": [EntityInput.from_dict(_item) for _item in obj["entities"]] if obj.get("entities") is not None else None
         })
         return _obj
 

hindsight_client_api/models/reflect_request.py

@@ -17,7 +17,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from hindsight_client_api.models.budget import Budget
 from hindsight_client_api.models.reflect_include_options import ReflectIncludeOptions
@@ -31,8 +31,10 @@ class ReflectRequest(BaseModel):
     query: StrictStr
     budget: Optional[Budget] = None
     context: Optional[StrictStr] = None
+    max_tokens: Optional[StrictInt] = Field(default=4096, description="Maximum tokens for the response")
     include: Optional[ReflectIncludeOptions] = Field(default=None, description="Options for including additional data (disabled by default)")
-    __properties: ClassVar[List[str]] = ["query", "budget", "context", "include"]
+    response_schema: Optional[Dict[str, Any]] = None
+    __properties: ClassVar[List[str]] = ["query", "budget", "context", "max_tokens", "include", "response_schema"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -81,6 +83,11 @@ class ReflectRequest(BaseModel):
         if self.context is None and "context" in self.model_fields_set:
             _dict['context'] = None
 
+        # set to None if response_schema (nullable) is None
+        # and model_fields_set contains the field
+        if self.response_schema is None and "response_schema" in self.model_fields_set:
+            _dict['response_schema'] = None
+
         return _dict
 
     @classmethod
@@ -96,7 +103,9 @@ class ReflectRequest(BaseModel):
             "query": obj.get("query"),
             "budget": obj.get("budget"),
             "context": obj.get("context"),
-            "include": ReflectIncludeOptions.from_dict(obj["include"]) if obj.get("include") is not None else None
+            "max_tokens": obj.get("max_tokens") if obj.get("max_tokens") is not None else 4096,
+            "include": ReflectIncludeOptions.from_dict(obj["include"]) if obj.get("include") is not None else None,
+            "response_schema": obj.get("response_schema")
         })
         return _obj
 

hindsight_client_api/models/reflect_response.py

@@ -29,7 +29,8 @@ class ReflectResponse(BaseModel):
     """ # noqa: E501
     text: StrictStr
     based_on: Optional[List[ReflectFact]] = None
-    __properties: ClassVar[List[str]] = ["text", "based_on"]
+    structured_output: Optional[Dict[str, Any]] = None
+    __properties: ClassVar[List[str]] = ["text", "based_on", "structured_output"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -77,6 +78,11 @@ class ReflectResponse(BaseModel):
                 if _item_based_on:
                     _items.append(_item_based_on.to_dict())
             _dict['based_on'] = _items
+        # set to None if structured_output (nullable) is None
+        # and model_fields_set contains the field
+        if self.structured_output is None and "structured_output" in self.model_fields_set:
+            _dict['structured_output'] = None
+
         return _dict
 
     @classmethod
@@ -90,7 +96,8 @@ class ReflectResponse(BaseModel):
 
         _obj = cls.model_validate({
             "text": obj.get("text"),
-            "based_on": [ReflectFact.from_dict(_item) for _item in obj["based_on"]] if obj.get("based_on") is not None else None
+            "based_on": [ReflectFact.from_dict(_item) for _item in obj["based_on"]] if obj.get("based_on") is not None else None,
+            "structured_output": obj.get("structured_output")
         })
         return _obj