gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added release

Browse source
This commit is contained in:
Vasilije 2023-10-05 10:37:34 +02:00
parent 317223d866
commit 99ebfb3113
1 changed files with 0 additions and 51 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
level_3/schema/semantic/semantic_schema.py
View file

@ -1,51 +0,0 @@
from marshmallow import Schema, fields
class DocumentMetadataSchemaV1(Schema):
user_id = fields.Str(required=True)
memory_id = fields.Str(required=True)
ltm_memory_id = fields.Str(required=True)
st_memory_id = fields.Str(required=True)
buffer_id = fields.Str(required=True)
version = fields.Str(load_default="")
agreement_id = fields.Str(load_default="")
privacy_policy = fields.Str(load_default="")
terms_of_service = fields.Str(load_default="")
format = fields.Str(load_default="")
schema_version = fields.Str(load_default="")
checksum = fields.Str(load_default="")
owner = fields.Str(load_default="")
license = fields.Str(load_default="")
validity_start = fields.Str(load_default="")
validity_end = fields.Str(load_default="")
class DocumentMetadataSchemaV2(Schema):
user_id = fields.Str(required=True)
memory_id = fields.Str(required=True)
ltm_memory_id = fields.Str(required=True)
st_memory_id = fields.Str(required=True)
buffer_id = fields.Str(required=True)
version = fields.Str(load_default="")
agreement_id = fields.Str(load_default="")
privacy_policy = fields.Str(load_default="")
terms_of_service = fields.Str(load_default="")
format = fields.Str(load_default="")
schema_version = fields.Str(load_default="")
checksum = fields.Str(load_default="")
owner = fields.Str(load_default="")
license = fields.Str(load_default="")
validity_start = fields.Str(load_default="")
validity_end = fields.Str(load_default="")
random = fields.Str(load_default="")
class DocumentSchema(Schema):
metadata = fields.Nested(DocumentMetadataSchemaV1, required=True)
page_content = fields.Str(required=True)
SCHEMA_VERSIONS = {
"1.0": DocumentMetadataSchemaV1,
"2.0": DocumentMetadataSchemaV2
}
def get_schema_version(version):
return SCHEMA_VERSIONS.get(version, DocumentMetadataSchemaV1)