Skip to content

Commit fcce735

Browse files
committed
fix: import missing
1 parent 037e390 commit fcce735

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

api/agentok_api/templates/import_mapping.j2

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
'DALLEAgent': 'from agentok_api.extensions.dalle_agent import DALLEAgent',
1717
} -%}
1818
{%- for node in nodes -%}
19-
{% set cls = node['data'].get('class') %}
19+
{% set cls = node['data'].get('class_type') %}
2020
{%- if cls and cls in import_dict and import_dict[cls] not in import_lines -%}
2121
{% do import_lines.append(import_dict[cls]) %}
2222
{%- endif -%}

0 commit comments

Comments
 (0)