neurograph-core 1.202508221357__py3-none-any.whl → 1.202509152210__py3-none-any.whl

neurograph/v1/models/knowledge_enrichment_artifact_create_request.py

@@ -0,0 +1,113 @@
+# coding: utf-8
+
+"""
+    Neurograph Core
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 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, StrictFloat, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional, Union
+from typing import Optional, Set
+from typing_extensions import Self
+
+class KnowledgeEnrichmentArtifactCreateRequest(BaseModel):
+    """
+    KnowledgeEnrichmentArtifactCreateRequest
+    """ # noqa: E501
+    cloud_run_job_name: Optional[StrictStr] = None
+    confidence: Optional[Union[StrictFloat, StrictInt]] = None
+    config_entity_id: Optional[StrictStr] = None
+    config_snapshot: Optional[Dict[str, Any]] = None
+    input_params: Optional[Dict[str, Any]] = None
+    job_name: Optional[StrictStr] = None
+    output_json: Optional[Dict[str, Any]] = None
+    output_schema_id: Optional[StrictStr] = None
+    output_text: Optional[StrictStr] = None
+    spec_version: Optional[StrictInt] = None
+    subject_entity_id: Optional[StrictStr] = None
+    supersedes_artifact_id: Optional[StrictInt] = None
+    task_name: Optional[StrictStr] = None
+    task_queue_name: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["cloud_run_job_name", "confidence", "config_entity_id", "config_snapshot", "input_params", "job_name", "output_json", "output_schema_id", "output_text", "spec_version", "subject_entity_id", "supersedes_artifact_id", "task_name", "task_queue_name"]
+
+    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"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of KnowledgeEnrichmentArtifactCreateRequest 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.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of KnowledgeEnrichmentArtifactCreateRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "cloud_run_job_name": obj.get("cloud_run_job_name"),
+            "confidence": obj.get("confidence"),
+            "config_entity_id": obj.get("config_entity_id"),
+            "config_snapshot": obj.get("config_snapshot"),
+            "input_params": obj.get("input_params"),
+            "job_name": obj.get("job_name"),
+            "output_json": obj.get("output_json"),
+            "output_schema_id": obj.get("output_schema_id"),
+            "output_text": obj.get("output_text"),
+            "spec_version": obj.get("spec_version"),
+            "subject_entity_id": obj.get("subject_entity_id"),
+            "supersedes_artifact_id": obj.get("supersedes_artifact_id"),
+            "task_name": obj.get("task_name"),
+            "task_queue_name": obj.get("task_queue_name")
+        })
+        return _obj
+
+

neurograph/v1/models/knowledge_enrichment_artifact_list_response.py

@@ -0,0 +1,103 @@
+# coding: utf-8
+
+"""
+    Neurograph Core
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 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, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from neurograph.v1.models.knowledge_enrichment_artifact import KnowledgeEnrichmentArtifact
+from neurograph.v1.models.knowledge_enrichment_query import KnowledgeEnrichmentQuery
+from typing import Optional, Set
+from typing_extensions import Self
+
+class KnowledgeEnrichmentArtifactListResponse(BaseModel):
+    """
+    KnowledgeEnrichmentArtifactListResponse
+    """ # noqa: E501
+    data: Optional[List[KnowledgeEnrichmentArtifact]] = None
+    error: Optional[StrictStr] = None
+    query: Optional[KnowledgeEnrichmentQuery] = None
+    __properties: ClassVar[List[str]] = ["data", "error", "query"]
+
+    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"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of KnowledgeEnrichmentArtifactListResponse 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.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of each item in data (list)
+        _items = []
+        if self.data:
+            for _item_data in self.data:
+                if _item_data:
+                    _items.append(_item_data.to_dict())
+            _dict['data'] = _items
+        # override the default output from pydantic by calling `to_dict()` of query
+        if self.query:
+            _dict['query'] = self.query.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of KnowledgeEnrichmentArtifactListResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "data": [KnowledgeEnrichmentArtifact.from_dict(_item) for _item in obj["data"]] if obj.get("data") is not None else None,
+            "error": obj.get("error"),
+            "query": KnowledgeEnrichmentQuery.from_dict(obj["query"]) if obj.get("query") is not None else None
+        })
+        return _obj
+
+

neurograph/v1/models/knowledge_enrichment_artifact_response.py

@@ -0,0 +1,93 @@
+# coding: utf-8
+
+"""
+    Neurograph Core
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 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, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from neurograph.v1.models.knowledge_enrichment_artifact import KnowledgeEnrichmentArtifact
+from typing import Optional, Set
+from typing_extensions import Self
+
+class KnowledgeEnrichmentArtifactResponse(BaseModel):
+    """
+    KnowledgeEnrichmentArtifactResponse
+    """ # noqa: E501
+    data: Optional[KnowledgeEnrichmentArtifact] = None
+    error: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["data", "error"]
+
+    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"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of KnowledgeEnrichmentArtifactResponse 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.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of data
+        if self.data:
+            _dict['data'] = self.data.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of KnowledgeEnrichmentArtifactResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "data": KnowledgeEnrichmentArtifact.from_dict(obj["data"]) if obj.get("data") is not None else None,
+            "error": obj.get("error")
+        })
+        return _obj
+
+

neurograph/v1/models/knowledge_enrichment_artifact_update_output_request.py

@@ -0,0 +1,93 @@
+# coding: utf-8
+
+"""
+    Neurograph Core
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 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, StrictFloat, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional, Union
+from typing import Optional, Set
+from typing_extensions import Self
+
+class KnowledgeEnrichmentArtifactUpdateOutputRequest(BaseModel):
+    """
+    KnowledgeEnrichmentArtifactUpdateOutputRequest
+    """ # noqa: E501
+    confidence: Optional[Union[StrictFloat, StrictInt]] = None
+    output_json: Optional[Dict[str, Any]] = None
+    output_text: Optional[StrictStr] = None
+    status: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["confidence", "output_json", "output_text", "status"]
+
+    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"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of KnowledgeEnrichmentArtifactUpdateOutputRequest 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.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of KnowledgeEnrichmentArtifactUpdateOutputRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "confidence": obj.get("confidence"),
+            "output_json": obj.get("output_json"),
+            "output_text": obj.get("output_text"),
+            "status": obj.get("status")
+        })
+        return _obj
+
+

neurograph/v1/models/knowledge_enrichment_artifact_update_status_request.py

@@ -0,0 +1,99 @@
+# coding: utf-8
+
+"""
+    Neurograph Core
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 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, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class KnowledgeEnrichmentArtifactUpdateStatusRequest(BaseModel):
+    """
+    KnowledgeEnrichmentArtifactUpdateStatusRequest
+    """ # noqa: E501
+    agent_engine_run_id: Optional[StrictStr] = None
+    attempt_count: Optional[StrictInt] = None
+    cloud_run_execution_id: Optional[StrictStr] = None
+    completed_at: Optional[StrictStr] = None
+    last_error: Optional[StrictStr] = None
+    started_at: Optional[StrictStr] = None
+    status: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["agent_engine_run_id", "attempt_count", "cloud_run_execution_id", "completed_at", "last_error", "started_at", "status"]
+
+    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"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of KnowledgeEnrichmentArtifactUpdateStatusRequest 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.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of KnowledgeEnrichmentArtifactUpdateStatusRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "agent_engine_run_id": obj.get("agent_engine_run_id"),
+            "attempt_count": obj.get("attempt_count"),
+            "cloud_run_execution_id": obj.get("cloud_run_execution_id"),
+            "completed_at": obj.get("completed_at"),
+            "last_error": obj.get("last_error"),
+            "started_at": obj.get("started_at"),
+            "status": obj.get("status")
+        })
+        return _obj
+
+

neurograph/v1/models/knowledge_enrichment_query.py

@@ -0,0 +1,99 @@
+# coding: utf-8
+
+"""
+    Neurograph Core
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 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, StrictBool, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class KnowledgeEnrichmentQuery(BaseModel):
+    """
+    KnowledgeEnrichmentQuery
+    """ # noqa: E501
+    client_id: Optional[StrictStr] = None
+    is_current: Optional[StrictBool] = None
+    job_name: Optional[StrictStr] = None
+    limit: Optional[StrictInt] = None
+    offset: Optional[StrictInt] = None
+    status: Optional[StrictStr] = None
+    subject_entity_id: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["client_id", "is_current", "job_name", "limit", "offset", "status", "subject_entity_id"]
+
+    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"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of KnowledgeEnrichmentQuery 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.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of KnowledgeEnrichmentQuery from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "client_id": obj.get("client_id"),
+            "is_current": obj.get("is_current"),
+            "job_name": obj.get("job_name"),
+            "limit": obj.get("limit"),
+            "offset": obj.get("offset"),
+            "status": obj.get("status"),
+            "subject_entity_id": obj.get("subject_entity_id")
+        })
+        return _obj
+
+