llm function is custom function

mindsdb · Sep 18, 2024 · 6fa83da · 6fa83da
1 parent bea9557
commit 6fa83da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mindsdb_sql/planner/query_planner.py b/mindsdb_sql/planner/query_planner.py
@@ -201,7 +201,7 @@ def get_query_info(self, query):
 
         def find_objects(node, is_table, **kwargs):
             if isinstance(node, Function):
-                if node.namespace is not None:
+                if node.namespace is not None or node.op.lower() in ('llm',):
                     user_functions.append(node)
 
             if is_table: