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

neurograph/v1/api/knowledge_api.py

@@ -33,11 +33,13 @@ from neurograph.v1.models.knowledge_entity_get_many_response import KnowledgeEnt
 from neurograph.v1.models.knowledge_entity_schema_upsert_request import KnowledgeEntitySchemaUpsertRequest
 from neurograph.v1.models.knowledge_entity_schemas_response import KnowledgeEntitySchemasResponse
 from neurograph.v1.models.knowledge_entity_schemas_upsert_response import KnowledgeEntitySchemasUpsertResponse
+from neurograph.v1.models.knowledge_entity_type_response import KnowledgeEntityTypeResponse
+from neurograph.v1.models.knowledge_entity_type_upsert_request import KnowledgeEntityTypeUpsertRequest
 from neurograph.v1.models.knowledge_entity_upsert_request import KnowledgeEntityUpsertRequest
+from neurograph.v1.models.knowledge_entity_upsert_response import KnowledgeEntityUpsertResponse
 from neurograph.v1.models.knowledge_ingest_raw_request import KnowledgeIngestRawRequest
 from neurograph.v1.models.knowledge_ingest_raw_response import KnowledgeIngestRawResponse
 from neurograph.v1.models.knowledge_kind_response import KnowledgeKindResponse
-from neurograph.v1.models.knowledge_type_response import KnowledgeTypeResponse
 
 from neurograph.v1.api_client import ApiClient, RequestSerialized
 from neurograph.v1.api_response import ApiResponse
@@ -2599,7 +2601,6 @@ class KnowledgeApi:
         _response_types_map: Dict[str, Optional[str]] = {
             '200': "KnowledgeEntityCreateResponse",
             '400': "KnowledgeEntityCreateResponse",
-            '503': "KnowledgeEntityCreateResponse",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -2668,7 +2669,6 @@ class KnowledgeApi:
         _response_types_map: Dict[str, Optional[str]] = {
             '200': "KnowledgeEntityCreateResponse",
             '400': "KnowledgeEntityCreateResponse",
-            '503': "KnowledgeEntityCreateResponse",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -2737,7 +2737,6 @@ class KnowledgeApi:
         _response_types_map: Dict[str, Optional[str]] = {
             '200': "KnowledgeEntityCreateResponse",
             '400': "KnowledgeEntityCreateResponse",
-            '503': "KnowledgeEntityCreateResponse",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -2823,6 +2822,283 @@ class KnowledgeApi:
 
 
 
+    @validate_call
+    def api_v1_knowledge_entity_upsert_post(
+        self,
+        request: Annotated[KnowledgeEntityUpsertRequest, Field(description="Body")],
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> KnowledgeEntityUpsertResponse:
+        """Bulk Uspert Entities
+
+        Upsert one or many entities, up to 10mb per request. If no batch_id is passed, one will be created. Behind the scenes, this inserts into a staging table that then upserts into the knowledge.entity table.
+
+        :param request: Body (required)
+        :type request: KnowledgeEntityUpsertRequest
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._api_v1_knowledge_entity_upsert_post_serialize(
+            request=request,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "KnowledgeEntityUpsertResponse",
+            '400': "KnowledgeEntityUpsertResponse",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+
+    @validate_call
+    def api_v1_knowledge_entity_upsert_post_with_http_info(
+        self,
+        request: Annotated[KnowledgeEntityUpsertRequest, Field(description="Body")],
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[KnowledgeEntityUpsertResponse]:
+        """Bulk Uspert Entities
+
+        Upsert one or many entities, up to 10mb per request. If no batch_id is passed, one will be created. Behind the scenes, this inserts into a staging table that then upserts into the knowledge.entity table.
+
+        :param request: Body (required)
+        :type request: KnowledgeEntityUpsertRequest
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._api_v1_knowledge_entity_upsert_post_serialize(
+            request=request,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "KnowledgeEntityUpsertResponse",
+            '400': "KnowledgeEntityUpsertResponse",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+
+    @validate_call
+    def api_v1_knowledge_entity_upsert_post_without_preload_content(
+        self,
+        request: Annotated[KnowledgeEntityUpsertRequest, Field(description="Body")],
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """Bulk Uspert Entities
+
+        Upsert one or many entities, up to 10mb per request. If no batch_id is passed, one will be created. Behind the scenes, this inserts into a staging table that then upserts into the knowledge.entity table.
+
+        :param request: Body (required)
+        :type request: KnowledgeEntityUpsertRequest
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._api_v1_knowledge_entity_upsert_post_serialize(
+            request=request,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "KnowledgeEntityUpsertResponse",
+            '400': "KnowledgeEntityUpsertResponse",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _api_v1_knowledge_entity_upsert_post_serialize(
+        self,
+        request,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {
+        }
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+        if request is not None:
+            _body_params = request
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'application/json'
+                ]
+            )
+
+        # set the HTTP header `Content-Type`
+        if _content_type:
+            _header_params['Content-Type'] = _content_type
+        else:
+            _default_content_type = (
+                self.api_client.select_header_content_type(
+                    [
+                        'application/json'
+                    ]
+                )
+            )
+            if _default_content_type is not None:
+                _header_params['Content-Type'] = _default_content_type
+
+        # authentication setting
+        _auth_settings: List[str] = [
+            'TokenAuth'
+        ]
+
+        return self.api_client.param_serialize(
+            method='POST',
+            resource_path='/api/v1/knowledge/entity/upsert/',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+
+
+
     @validate_call
     def api_v1_knowledge_ingest_raw_post(
         self,
@@ -3920,7 +4196,7 @@ class KnowledgeApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> KnowledgeTypeResponse:
+    ) -> KnowledgeEntityTypeResponse:
         """Get Knowledge Types
 
         Knowledge Types are a sub-classification under kind.
@@ -3955,9 +4231,9 @@ class KnowledgeApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "KnowledgeTypeResponse",
-            '400': "KnowledgeTypeResponse",
-            '503': "KnowledgeTypeResponse",
+            '200': "KnowledgeEntityTypeResponse",
+            '400': "KnowledgeEntityTypeResponse",
+            '503': "KnowledgeEntityTypeResponse",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -3985,7 +4261,7 @@ class KnowledgeApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[KnowledgeTypeResponse]:
+    ) -> ApiResponse[KnowledgeEntityTypeResponse]:
         """Get Knowledge Types
 
         Knowledge Types are a sub-classification under kind.
@@ -4020,9 +4296,9 @@ class KnowledgeApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "KnowledgeTypeResponse",
-            '400': "KnowledgeTypeResponse",
-            '503': "KnowledgeTypeResponse",
+            '200': "KnowledgeEntityTypeResponse",
+            '400': "KnowledgeEntityTypeResponse",
+            '503': "KnowledgeEntityTypeResponse",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -4085,9 +4361,9 @@ class KnowledgeApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "KnowledgeTypeResponse",
-            '400': "KnowledgeTypeResponse",
-            '503': "KnowledgeTypeResponse",
+            '200': "KnowledgeEntityTypeResponse",
+            '400': "KnowledgeEntityTypeResponse",
+            '503': "KnowledgeEntityTypeResponse",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -4161,7 +4437,7 @@ class KnowledgeApi:
     @validate_call
     def api_v1_knowledge_type_post(
         self,
-        request: Annotated[KnowledgeEntityUpsertRequest, Field(description="Body")],
+        request: Annotated[KnowledgeEntityTypeUpsertRequest, Field(description="Body")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -4174,13 +4450,13 @@ class KnowledgeApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> KnowledgeTypeResponse:
+    ) -> KnowledgeEntityTypeResponse:
         """Get Knowledge Types
 
         Knowledge Types are a sub-classification under kind.
 
         :param request: Body (required)
-        :type request: KnowledgeEntityUpsertRequest
+        :type request: KnowledgeEntityTypeUpsertRequest
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -4212,9 +4488,9 @@ class KnowledgeApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "KnowledgeTypeResponse",
-            '400': "KnowledgeTypeResponse",
-            '503': "KnowledgeTypeResponse",
+            '200': "KnowledgeEntityTypeResponse",
+            '400': "KnowledgeEntityTypeResponse",
+            '503': "KnowledgeEntityTypeResponse",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -4230,7 +4506,7 @@ class KnowledgeApi:
     @validate_call
     def api_v1_knowledge_type_post_with_http_info(
         self,
-        request: Annotated[KnowledgeEntityUpsertRequest, Field(description="Body")],
+        request: Annotated[KnowledgeEntityTypeUpsertRequest, Field(description="Body")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -4243,13 +4519,13 @@ class KnowledgeApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[KnowledgeTypeResponse]:
+    ) -> ApiResponse[KnowledgeEntityTypeResponse]:
         """Get Knowledge Types
 
         Knowledge Types are a sub-classification under kind.
 
         :param request: Body (required)
-        :type request: KnowledgeEntityUpsertRequest
+        :type request: KnowledgeEntityTypeUpsertRequest
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -4281,9 +4557,9 @@ class KnowledgeApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "KnowledgeTypeResponse",
-            '400': "KnowledgeTypeResponse",
-            '503': "KnowledgeTypeResponse",
+            '200': "KnowledgeEntityTypeResponse",
+            '400': "KnowledgeEntityTypeResponse",
+            '503': "KnowledgeEntityTypeResponse",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -4299,7 +4575,7 @@ class KnowledgeApi:
     @validate_call
     def api_v1_knowledge_type_post_without_preload_content(
         self,
-        request: Annotated[KnowledgeEntityUpsertRequest, Field(description="Body")],
+        request: Annotated[KnowledgeEntityTypeUpsertRequest, Field(description="Body")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -4318,7 +4594,7 @@ class KnowledgeApi:
         Knowledge Types are a sub-classification under kind.
 
         :param request: Body (required)
-        :type request: KnowledgeEntityUpsertRequest
+        :type request: KnowledgeEntityTypeUpsertRequest
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -4350,9 +4626,9 @@ class KnowledgeApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "KnowledgeTypeResponse",
-            '400': "KnowledgeTypeResponse",
-            '503': "KnowledgeTypeResponse",
+            '200': "KnowledgeEntityTypeResponse",
+            '400': "KnowledgeEntityTypeResponse",
+            '503': "KnowledgeEntityTypeResponse",
         }
         response_data = self.api_client.call_api(
             *_param,