Merge pull request #229 from topoteretes/feature/cog-620-bug-code_gra…

…ph_pipeline-cant-process-false-class fix: Fixes processing false Class keyword issue
topoteretes · Nov 15, 2024 · 8c0c0b9 · 8c0c0b9
2 parents b858f0b + 1df12c1
commit 8c0c0b9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cognee/shared/SourceCodeGraph.py b/cognee/shared/SourceCodeGraph.py
@@ -28,7 +28,7 @@ class Class(DataPoint):
     description: str
     constructor_parameters: List[Variable]
     extended_from_class: Optional["Class"] = None
-    has_methods: list["Function"]
+    has_methods: List["Function"]
 
     _metadata = {
         "index_fields": ["name"]
@@ -89,7 +89,8 @@ class SourceCodeGraph(DataPoint):
         Operator,
         Expression,
     ]]
-
 Class.model_rebuild()
 ClassInstance.model_rebuild()
 Expression.model_rebuild()
+FunctionCall.model_rebuild()
+SourceCodeGraph.model_rebuild()