diff --git a/api/app/models/document_model.py b/api/app/models/document_model.py index db9280c6..fb43d44d 100644 --- a/api/app/models/document_model.py +++ b/api/app/models/document_model.py @@ -26,6 +26,7 @@ class Document(Base): "html4excel": False, "graphrag": { "use_graphrag": False, + "scene_name": "", "entity_types": [ "organization", "person", @@ -33,7 +34,9 @@ class Document(Base): "event", "category" ], - "method": "general" + "method": "general", + "resolution": True, + "community": True } }, comment="default parser config") chunk_num = Column(Integer, default=0, comment="chunk num") diff --git a/api/app/models/knowledge_model.py b/api/app/models/knowledge_model.py index 6d3465f9..8f0909d3 100644 --- a/api/app/models/knowledge_model.py +++ b/api/app/models/knowledge_model.py @@ -65,6 +65,7 @@ class Knowledge(Base): "html4excel": False, "graphrag": { "use_graphrag": False, + "scene_name": "", "entity_types": [ "organization", "person", @@ -72,7 +73,9 @@ class Knowledge(Base): "event", "category" ], - "method": "general" + "method": "general", + "resolution": True, + "community": True } }, comment="default parser config") diff --git a/api/pyproject.toml b/api/pyproject.toml index 901858e6..2dcc706d 100644 --- a/api/pyproject.toml +++ b/api/pyproject.toml @@ -135,6 +135,8 @@ dependencies = [ "graspologic==3.4.5.dev2", "markdown-to-json==2.1.1", "valkey==6.0.2", + "python-calamine>=0.4.0", + "xlrd==2.0.2" ] [tool.pytest.ini_options] diff --git a/api/requirements.txt b/api/requirements.txt index 5530a9e3..99252e09 100644 --- a/api/requirements.txt +++ b/api/requirements.txt @@ -129,3 +129,5 @@ editdistance==0.8.1 graspologic==3.4.5.dev2 markdown-to-json==2.1.1 valkey==6.0.2 +python-calamine>=0.4.0 +xlrd==2.0.2