fix small bug

definitive-io · Dec 20, 2023 · 97026ab · 97026ab
1 parent 212f964
commit 97026ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/openassistants/openassistants/llm_function_calling/infilling.py b/packages/openassistants/openassistants/llm_function_calling/infilling.py
@@ -87,8 +87,8 @@ async def generate_argument_decisions(
 
 def entity_to_json_schema_obj(entity: IEntity):
     d = {"const": entity.get_identity()}
-    if entity.get_description is not None:
-        d["description"] = entity.get_description()
+    if (description := entity.get_description()) is not None:
+        d["description"] = description
     return d