small fix with entity

definitive-io · Jan 2, 2024 · 20a7bec · 20a7bec
1 parent 2bb2484
commit 20a7bec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/openassistants/openassistants/llm_function_calling/entity_resolution.py b/packages/openassistants/openassistants/llm_function_calling/entity_resolution.py
@@ -35,8 +35,8 @@ def entity_to_document(entity: IEntity) -> Document:
         metadata=dict(id=entity.get_identity()), page_content=entity.get_identity()
     )
 
-    if entity.get_description:
-        doc.page_content += f" ({entity.get_description})"
+    if entity.get_description():
+        doc.page_content += f" ({entity.get_description()})"
 
     return doc