Skip to content

Commit dc6deef

Browse files
committed
allows HTK_OPENAI_SYSTEM_PROMPT_MODEL = None
1 parent 0a5dd8a commit dc6deef

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

lib/openai/models/fk_fields.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,15 @@
1212
def fk_openai_system_prompt(
1313
related_name: str, required: bool = False
1414
) -> models.ForeignKey:
15-
field = models.ForeignKey(
16-
htk_setting('HTK_OPENAI_SYSTEM_PROMPT_MODEL'),
17-
related_name=related_name,
18-
**build_kwargs(required=required),
19-
)
15+
fk_model = htk_setting('HTK_OPENAI_SYSTEM_PROMPT_MODEL')
16+
17+
if fk_model is None:
18+
field = None
19+
else:
20+
field = models.ForeignKey(
21+
fk_model,
22+
related_name=related_name,
23+
**build_kwargs(required=required),
24+
)
25+
2026
return field

0 commit comments

Comments
 (0)