neurograph-core 1.202509220142__py3-none-any.whl → 1.202509242050__py3-none-any.whl

neurograph/v1/models/{client_client_url_row.py → client_url_row.py}

@@ -23,9 +23,9 @@ from neurograph.v1.models.client_probe import ClientProbe
 from typing import Optional, Set
 from typing_extensions import Self
 
-class ClientClientUrlRow(BaseModel):
+class ClientUrlRow(BaseModel):
     """
-    ClientClientUrlRow
+    ClientUrlRow
     """ # noqa: E501
     client_id: Optional[StrictStr] = None
     customer_environment: Optional[StrictStr] = None
@@ -54,7 +54,7 @@ class ClientClientUrlRow(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of ClientClientUrlRow from a JSON string"""
+        """Create an instance of ClientUrlRow from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -86,7 +86,7 @@ class ClientClientUrlRow(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of ClientClientUrlRow from a dict"""
+        """Create an instance of ClientUrlRow from a dict"""
         if obj is None:
             return None
 

neurograph/v1/models/{client_client_url_snapshot_response.py → client_url_snapshot_response.py}

@@ -19,15 +19,15 @@ import json
 
 from pydantic import BaseModel, ConfigDict, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
-from neurograph.v1.models.client_client_url_snapshot_row import ClientClientURLSnapshotRow
+from neurograph.v1.models.client_url_snapshot_row import ClientUrlSnapshotRow
 from typing import Optional, Set
 from typing_extensions import Self
 
-class ClientClientURLSnapshotResponse(BaseModel):
+class ClientURLSnapshotResponse(BaseModel):
     """
-    ClientClientURLSnapshotResponse
+    ClientURLSnapshotResponse
     """ # noqa: E501
-    data: Optional[ClientClientURLSnapshotRow] = None
+    data: Optional[ClientUrlSnapshotRow] = None
     error: Optional[StrictStr] = None
     query_probe_id: Optional[StrictStr] = None
     __properties: ClassVar[List[str]] = ["data", "error", "query_probe_id"]
@@ -50,7 +50,7 @@ class ClientClientURLSnapshotResponse(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of ClientClientURLSnapshotResponse from a JSON string"""
+        """Create an instance of ClientURLSnapshotResponse from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -78,7 +78,7 @@ class ClientClientURLSnapshotResponse(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of ClientClientURLSnapshotResponse from a dict"""
+        """Create an instance of ClientURLSnapshotResponse from a dict"""
         if obj is None:
             return None
 
@@ -86,7 +86,7 @@ class ClientClientURLSnapshotResponse(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "data": ClientClientURLSnapshotRow.from_dict(obj["data"]) if obj.get("data") is not None else None,
+            "data": ClientUrlSnapshotRow.from_dict(obj["data"]) if obj.get("data") is not None else None,
             "error": obj.get("error"),
             "query_probe_id": obj.get("query_probe_id")
         })

neurograph/v1/models/{client_client_url_snapshot_row.py → client_url_snapshot_row.py}

@@ -22,9 +22,9 @@ from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
-class ClientClientURLSnapshotRow(BaseModel):
+class ClientUrlSnapshotRow(BaseModel):
     """
-    ClientClientURLSnapshotRow
+    ClientUrlSnapshotRow
     """ # noqa: E501
     client_url_id: Optional[StrictInt] = None
     html_raw: Optional[StrictStr] = Field(default=None, description="full HTML; can be large")
@@ -53,7 +53,7 @@ class ClientClientURLSnapshotRow(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of ClientClientURLSnapshotRow from a JSON string"""
+        """Create an instance of ClientUrlSnapshotRow from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -78,7 +78,7 @@ class ClientClientURLSnapshotRow(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of ClientClientURLSnapshotRow from a dict"""
+        """Create an instance of ClientUrlSnapshotRow from a dict"""
         if obj is None:
             return None
 

neurograph/v1/models/{client_client_url_upsert_request.py → client_url_upsert_request.py}

@@ -22,9 +22,9 @@ from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
-class ClientClientUrlUpsertRequest(BaseModel):
+class ClientUrlUpsertRequest(BaseModel):
     """
-    ClientClientUrlUpsertRequest
+    ClientUrlUpsertRequest
     """ # noqa: E501
     client_id: Optional[StrictStr] = None
     customer_environment: Optional[StrictStr] = None
@@ -51,7 +51,7 @@ class ClientClientUrlUpsertRequest(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of ClientClientUrlUpsertRequest from a JSON string"""
+        """Create an instance of ClientUrlUpsertRequest from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -76,7 +76,7 @@ class ClientClientUrlUpsertRequest(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of ClientClientUrlUpsertRequest from a dict"""
+        """Create an instance of ClientUrlUpsertRequest from a dict"""
         if obj is None:
             return None
 

neurograph/v1/models/{client_client_url_upsert_response.py → client_url_upsert_response.py}

@@ -23,9 +23,9 @@ from neurograph.v1.models.pgtype_timestamp import PgtypeTimestamp
 from typing import Optional, Set
 from typing_extensions import Self
 
-class ClientClientUrlUpsertResponse(BaseModel):
+class ClientUrlUpsertResponse(BaseModel):
     """
-    ClientClientUrlUpsertResponse
+    ClientUrlUpsertResponse
     """ # noqa: E501
     client_id: Optional[StrictStr] = None
     customer_environment: Optional[StrictStr] = None
@@ -55,7 +55,7 @@ class ClientClientUrlUpsertResponse(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of ClientClientUrlUpsertResponse from a JSON string"""
+        """Create an instance of ClientUrlUpsertResponse from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -86,7 +86,7 @@ class ClientClientUrlUpsertResponse(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of ClientClientUrlUpsertResponse from a dict"""
+        """Create an instance of ClientUrlUpsertResponse from a dict"""
         if obj is None:
             return None
 

neurograph/v1/models/{client_client_urls_response.py → client_urls_response.py}

@@ -19,15 +19,15 @@ import json
 
 from pydantic import BaseModel, ConfigDict, StrictInt, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
-from neurograph.v1.models.client_client_url_row import ClientClientUrlRow
+from neurograph.v1.models.client_url_row import ClientUrlRow
 from typing import Optional, Set
 from typing_extensions import Self
 
-class ClientClientUrlsResponse(BaseModel):
+class ClientUrlsResponse(BaseModel):
     """
-    ClientClientUrlsResponse
+    ClientUrlsResponse
     """ # noqa: E501
-    data: Optional[List[ClientClientUrlRow]] = None
+    data: Optional[List[ClientUrlRow]] = None
     error: Optional[StrictStr] = None
     query_client_id: Optional[StrictStr] = None
     total_rows: Optional[StrictInt] = None
@@ -51,7 +51,7 @@ class ClientClientUrlsResponse(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of ClientClientUrlsResponse from a JSON string"""
+        """Create an instance of ClientUrlsResponse from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -83,7 +83,7 @@ class ClientClientUrlsResponse(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of ClientClientUrlsResponse from a dict"""
+        """Create an instance of ClientUrlsResponse from a dict"""
         if obj is None:
             return None
 
@@ -91,7 +91,7 @@ class ClientClientUrlsResponse(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "data": [ClientClientUrlRow.from_dict(_item) for _item in obj["data"]] if obj.get("data") is not None else None,
+            "data": [ClientUrlRow.from_dict(_item) for _item in obj["data"]] if obj.get("data") is not None else None,
             "error": obj.get("error"),
             "query_client_id": obj.get("query_client_id"),
             "total_rows": obj.get("total_rows")

neurograph/v1/models/{dagster_dagster_info.py → dagster_info.py}

@@ -22,9 +22,9 @@ from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
-class DagsterDagsterInfo(BaseModel):
+class DagsterInfo(BaseModel):
     """
-    DagsterDagsterInfo
+    DagsterInfo
     """ # noqa: E501
     id: Optional[StrictStr] = None
     job: Optional[StrictStr] = None
@@ -51,7 +51,7 @@ class DagsterDagsterInfo(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of DagsterDagsterInfo from a JSON string"""
+        """Create an instance of DagsterInfo from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -76,7 +76,7 @@ class DagsterDagsterInfo(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of DagsterDagsterInfo from a dict"""
+        """Create an instance of DagsterInfo from a dict"""
         if obj is None:
             return None
 

neurograph/v1/models/{dagster_dagster_log_create_request.py → dagster_log_create_request.py}

@@ -22,9 +22,9 @@ from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
-class DagsterDagsterLogCreateRequest(BaseModel):
+class DagsterLogCreateRequest(BaseModel):
     """
-    DagsterDagsterLogCreateRequest
+    DagsterLogCreateRequest
     """ # noqa: E501
     core_id: Optional[StrictStr] = None
     job: Optional[StrictStr] = None
@@ -50,7 +50,7 @@ class DagsterDagsterLogCreateRequest(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of DagsterDagsterLogCreateRequest from a JSON string"""
+        """Create an instance of DagsterLogCreateRequest from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -75,7 +75,7 @@ class DagsterDagsterLogCreateRequest(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of DagsterDagsterLogCreateRequest from a dict"""
+        """Create an instance of DagsterLogCreateRequest from a dict"""
         if obj is None:
             return None
 

neurograph/v1/models/{dagster_dagster_log_create_response.py → dagster_log_create_response.py}

@@ -22,9 +22,9 @@ from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
-class DagsterDagsterLogCreateResponse(BaseModel):
+class DagsterLogCreateResponse(BaseModel):
     """
-    DagsterDagsterLogCreateResponse
+    DagsterLogCreateResponse
     """ # noqa: E501
     error: Optional[StrictStr] = None
     id: Optional[StrictStr] = None
@@ -49,7 +49,7 @@ class DagsterDagsterLogCreateResponse(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of DagsterDagsterLogCreateResponse from a JSON string"""
+        """Create an instance of DagsterLogCreateResponse from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -74,7 +74,7 @@ class DagsterDagsterLogCreateResponse(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of DagsterDagsterLogCreateResponse from a dict"""
+        """Create an instance of DagsterLogCreateResponse from a dict"""
         if obj is None:
             return None
 

neurograph/v1/models/{personas_persona_kpi_create_response.py → dagster_log_get_response.py}

@@ -19,15 +19,15 @@ import json
 
 from pydantic import BaseModel, ConfigDict, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
-from neurograph.v1.models.personas_persona_kpi_resp import PersonasPersonaKpiResp
+from neurograph.v1.models.dagster_info import DagsterInfo
 from typing import Optional, Set
 from typing_extensions import Self
 
-class PersonasPersonaKpiCreateResponse(BaseModel):
+class DagsterLogGetResponse(BaseModel):
     """
-    PersonasPersonaKpiCreateResponse
+    DagsterLogGetResponse
     """ # noqa: E501
-    data: Optional[List[PersonasPersonaKpiResp]] = None
+    data: Optional[List[DagsterInfo]] = None
     error: Optional[StrictStr] = None
     __properties: ClassVar[List[str]] = ["data", "error"]
 
@@ -49,7 +49,7 @@ class PersonasPersonaKpiCreateResponse(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of PersonasPersonaKpiCreateResponse from a JSON string"""
+        """Create an instance of DagsterLogGetResponse from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -81,7 +81,7 @@ class PersonasPersonaKpiCreateResponse(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of PersonasPersonaKpiCreateResponse from a dict"""
+        """Create an instance of DagsterLogGetResponse from a dict"""
         if obj is None:
             return None
 
@@ -89,7 +89,7 @@ class PersonasPersonaKpiCreateResponse(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "data": [PersonasPersonaKpiResp.from_dict(_item) for _item in obj["data"]] if obj.get("data") is not None else None,
+            "data": [DagsterInfo.from_dict(_item) for _item in obj["data"]] if obj.get("data") is not None else None,
             "error": obj.get("error")
         })
         return _obj

neurograph/v1/models/knowledge_entity.py

@@ -37,9 +37,10 @@ class KnowledgeEntity(BaseModel):
     name: Optional[StrictStr] = None
     relations: Optional[KnowledgeEntityRelations] = None
     var_schema: Optional[KnowledgeEntitySchema] = Field(default=None, alias="schema")
+    source_id: Optional[StrictStr] = None
     source_updated_at: Optional[StrictInt] = None
     ts_updated: Optional[StrictInt] = None
-    __properties: ClassVar[List[str]] = ["client_id", "data", "ext", "extra", "kind", "name", "relations", "schema", "source_updated_at", "ts_updated"]
+    __properties: ClassVar[List[str]] = ["client_id", "data", "ext", "extra", "kind", "name", "relations", "schema", "source_id", "source_updated_at", "ts_updated"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -109,6 +110,7 @@ class KnowledgeEntity(BaseModel):
             "name": obj.get("name"),
             "relations": KnowledgeEntityRelations.from_dict(obj["relations"]) if obj.get("relations") is not None else None,
             "schema": KnowledgeEntitySchema.from_dict(obj["schema"]) if obj.get("schema") is not None else None,
+            "source_id": obj.get("source_id"),
             "source_updated_at": obj.get("source_updated_at"),
             "ts_updated": obj.get("ts_updated")
         })

neurograph/v1/models/knowledge_entity_create_request.py

@@ -17,7 +17,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from neurograph.v1.models.knowledge_entity import KnowledgeEntity
 from typing import Optional, Set
@@ -27,8 +27,9 @@ class KnowledgeEntityCreateRequest(BaseModel):
     """
     KnowledgeEntityCreateRequest
     """ # noqa: E501
+    batch_id: Optional[StrictStr] = None
     data: Optional[List[KnowledgeEntity]] = None
-    __properties: ClassVar[List[str]] = ["data"]
+    __properties: ClassVar[List[str]] = ["batch_id", "data"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -88,6 +89,7 @@ class KnowledgeEntityCreateRequest(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
+            "batch_id": obj.get("batch_id"),
             "data": [KnowledgeEntity.from_dict(_item) for _item in obj["data"]] if obj.get("data") is not None else None
         })
         return _obj

neurograph/v1/models/knowledge_entity_create_response.py

@@ -19,7 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, StrictInt, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
-from neurograph.v1.models.knowledge_err_entity_row import KnowledgeErrEntityRow
+from neurograph.v1.models.knowledge_entity_err_row import KnowledgeEntityErrRow
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -28,7 +28,7 @@ class KnowledgeEntityCreateResponse(BaseModel):
     KnowledgeEntityCreateResponse
     """ # noqa: E501
     error: Optional[StrictStr] = None
-    error_rows: Optional[List[KnowledgeErrEntityRow]] = None
+    error_rows: Optional[List[KnowledgeEntityErrRow]] = None
     rows_created: Optional[StrictInt] = None
     __properties: ClassVar[List[str]] = ["error", "error_rows", "rows_created"]
 
@@ -91,7 +91,7 @@ class KnowledgeEntityCreateResponse(BaseModel):
 
         _obj = cls.model_validate({
             "error": obj.get("error"),
-            "error_rows": [KnowledgeErrEntityRow.from_dict(_item) for _item in obj["error_rows"]] if obj.get("error_rows") is not None else None,
+            "error_rows": [KnowledgeEntityErrRow.from_dict(_item) for _item in obj["error_rows"]] if obj.get("error_rows") is not None else None,
             "rows_created": obj.get("rows_created")
         })
         return _obj

neurograph/v1/models/{knowledge_err_entity_row.py → knowledge_entity_err_row.py}

@@ -22,9 +22,9 @@ from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
-class KnowledgeErrEntityRow(BaseModel):
+class KnowledgeEntityErrRow(BaseModel):
     """
-    KnowledgeErrEntityRow
+    KnowledgeEntityErrRow
     """ # noqa: E501
     error: Optional[StrictStr] = None
     index: Optional[StrictInt] = None
@@ -48,7 +48,7 @@ class KnowledgeErrEntityRow(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of KnowledgeErrEntityRow from a JSON string"""
+        """Create an instance of KnowledgeEntityErrRow from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -73,7 +73,7 @@ class KnowledgeErrEntityRow(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of KnowledgeErrEntityRow from a dict"""
+        """Create an instance of KnowledgeEntityErrRow from a dict"""
         if obj is None:
             return None
 

neurograph/v1/models/knowledge_entity_in_db.py

@@ -29,6 +29,7 @@ class KnowledgeEntityInDb(BaseModel):
     """
     KnowledgeEntityInDb
     """ # noqa: E501
+    batch_id: Optional[StrictStr] = None
     client_id: Optional[StrictStr] = Field(default=None, description="nullable.")
     data: Optional[Dict[str, Any]] = None
     ext: Optional[KnowledgeEntityExtra] = None
@@ -38,10 +39,11 @@ class KnowledgeEntityInDb(BaseModel):
     name: Optional[StrictStr] = None
     relations: Optional[KnowledgeEntityRelations] = None
     var_schema: Optional[KnowledgeEntitySchema] = Field(default=None, alias="schema")
+    source_id: Optional[StrictStr] = None
     source_updated_at: Optional[StrictInt] = None
     ts_created: Optional[StrictInt] = None
     ts_updated: Optional[StrictInt] = None
-    __properties: ClassVar[List[str]] = ["client_id", "data", "ext", "extra", "id", "kind", "name", "relations", "schema", "source_updated_at", "ts_created", "ts_updated"]
+    __properties: ClassVar[List[str]] = ["batch_id", "client_id", "data", "ext", "extra", "id", "kind", "name", "relations", "schema", "source_id", "source_updated_at", "ts_created", "ts_updated"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -103,6 +105,7 @@ class KnowledgeEntityInDb(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
+            "batch_id": obj.get("batch_id"),
             "client_id": obj.get("client_id"),
             "data": obj.get("data"),
             "ext": KnowledgeEntityExtra.from_dict(obj["ext"]) if obj.get("ext") is not None else None,
@@ -112,6 +115,7 @@ class KnowledgeEntityInDb(BaseModel):
             "name": obj.get("name"),
             "relations": KnowledgeEntityRelations.from_dict(obj["relations"]) if obj.get("relations") is not None else None,
             "schema": KnowledgeEntitySchema.from_dict(obj["schema"]) if obj.get("schema") is not None else None,
+            "source_id": obj.get("source_id"),
             "source_updated_at": obj.get("source_updated_at"),
             "ts_created": obj.get("ts_created"),
             "ts_updated": obj.get("ts_updated")

neurograph/v1/models/{knowledge_type_response.py → knowledge_entity_type_response.py}

@@ -23,9 +23,9 @@ from neurograph.v1.models.knowledge_entity_type_row import KnowledgeEntityTypeRo
 from typing import Optional, Set
 from typing_extensions import Self
 
-class KnowledgeTypeResponse(BaseModel):
+class KnowledgeEntityTypeResponse(BaseModel):
     """
-    KnowledgeTypeResponse
+    KnowledgeEntityTypeResponse
     """ # noqa: E501
     error: Optional[StrictStr] = None
     types: Optional[List[KnowledgeEntityTypeRow]] = None
@@ -49,7 +49,7 @@ class KnowledgeTypeResponse(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of KnowledgeTypeResponse from a JSON string"""
+        """Create an instance of KnowledgeEntityTypeResponse from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -81,7 +81,7 @@ class KnowledgeTypeResponse(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of KnowledgeTypeResponse from a dict"""
+        """Create an instance of KnowledgeEntityTypeResponse from a dict"""
         if obj is None:
             return None
 

neurograph/v1/models/{personas_persona_trait_create_request.py → knowledge_entity_type_upsert_request.py}

@@ -19,15 +19,15 @@ import json
 
 from pydantic import BaseModel, ConfigDict
 from typing import Any, ClassVar, Dict, List, Optional
-from neurograph.v1.models.personas_personality_trait_in import PersonasPersonalityTraitIn
+from neurograph.v1.models.knowledge_entity_type_upsert_row import KnowledgeEntityTypeUpsertRow
 from typing import Optional, Set
 from typing_extensions import Self
 
-class PersonasPersonaTraitCreateRequest(BaseModel):
+class KnowledgeEntityTypeUpsertRequest(BaseModel):
     """
-    PersonasPersonaTraitCreateRequest
+    KnowledgeEntityTypeUpsertRequest
     """ # noqa: E501
-    data: Optional[List[PersonasPersonalityTraitIn]] = None
+    data: Optional[List[KnowledgeEntityTypeUpsertRow]] = None
     __properties: ClassVar[List[str]] = ["data"]
 
     model_config = ConfigDict(
@@ -48,7 +48,7 @@ class PersonasPersonaTraitCreateRequest(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of PersonasPersonaTraitCreateRequest from a JSON string"""
+        """Create an instance of KnowledgeEntityTypeUpsertRequest from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -80,7 +80,7 @@ class PersonasPersonaTraitCreateRequest(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of PersonasPersonaTraitCreateRequest from a dict"""
+        """Create an instance of KnowledgeEntityTypeUpsertRequest from a dict"""
         if obj is None:
             return None
 
@@ -88,7 +88,7 @@ class PersonasPersonaTraitCreateRequest(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "data": [PersonasPersonalityTraitIn.from_dict(_item) for _item in obj["data"]] if obj.get("data") is not None else None
+            "data": [KnowledgeEntityTypeUpsertRow.from_dict(_item) for _item in obj["data"]] if obj.get("data") is not None else None
         })
         return _obj
 

neurograph/v1/models/{knowledge_entity_upsert_row.py → knowledge_entity_type_upsert_row.py}

@@ -22,9 +22,9 @@ from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
-class KnowledgeEntityUpsertRow(BaseModel):
+class KnowledgeEntityTypeUpsertRow(BaseModel):
     """
-    KnowledgeEntityUpsertRow
+    KnowledgeEntityTypeUpsertRow
     """ # noqa: E501
     id: Optional[StrictInt] = None
     kind: Optional[StrictStr] = None
@@ -50,7 +50,7 @@ class KnowledgeEntityUpsertRow(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of KnowledgeEntityUpsertRow from a JSON string"""
+        """Create an instance of KnowledgeEntityTypeUpsertRow from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -75,7 +75,7 @@ class KnowledgeEntityUpsertRow(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of KnowledgeEntityUpsertRow from a dict"""
+        """Create an instance of KnowledgeEntityTypeUpsertRow from a dict"""
         if obj is None:
             return None
 

neurograph/v1/models/knowledge_entity_upsert_request.py

@@ -17,9 +17,9 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
-from neurograph.v1.models.knowledge_entity_upsert_row import KnowledgeEntityUpsertRow
+from neurograph.v1.models.knowledge_entity import KnowledgeEntity
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -27,8 +27,9 @@ class KnowledgeEntityUpsertRequest(BaseModel):
     """
     KnowledgeEntityUpsertRequest
     """ # noqa: E501
-    data: Optional[List[KnowledgeEntityUpsertRow]] = None
-    __properties: ClassVar[List[str]] = ["data"]
+    batch_id: Optional[StrictStr] = None
+    data: Optional[List[KnowledgeEntity]] = None
+    __properties: ClassVar[List[str]] = ["batch_id", "data"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -88,7 +89,8 @@ class KnowledgeEntityUpsertRequest(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "data": [KnowledgeEntityUpsertRow.from_dict(_item) for _item in obj["data"]] if obj.get("data") is not None else None
+            "batch_id": obj.get("batch_id"),
+            "data": [KnowledgeEntity.from_dict(_item) for _item in obj["data"]] if obj.get("data") is not None else None
         })
         return _obj