Feature/add missing abstractions (#449)

* fix * fix
SciPhi-AI · Jun 13, 2024 · efff384 · efff384
1 parent 5235957
commit efff384
Show file tree

Hide file tree

Showing 3 changed files with 441 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "r2r"
-version = "0.2.17"
+version = "0.2.18"
 description = "SciPhi R2R"
 authors = ["Owen Colegrove <owen@sciphi.ai>"]
 license = "MIT"

diff --git a/r2r/core/abstractions/document.py b/r2r/core/abstractions/document.py
@@ -146,9 +146,7 @@ def extract_entities(llm_payload: list[str]) -> dict[str, Entity]:
                 if colon_count == 1:
                     category, value = entry.split(":")
                     sub_category = None
-                elif colon_count == 2:
-                    category, sub_category, value = entry.split(":")
-                elif colon_count > 2:
+                elif colon_count >= 2:
                     parts = entry.split(":", 2)
                     category, sub_category, value = (
                         parts[0],