Skip to content

Commit 510a9fa

Browse files
committed
refactor: Generate prompt serializer
1 parent f1cd675 commit 510a9fa

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

apps/chat/serializers/chat.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,15 @@ class PromptGenerateSerializer(serializers.Serializer):
146146
model_id = serializers.CharField(required=False, allow_blank=True, allow_null=True, label=_("Model"))
147147
application_id = serializers.CharField(required=False, allow_blank=True, allow_null=True, label=_("Application"))
148148

149+
def is_valid(self, *, raise_exception=False):
150+
super().is_valid(raise_exception=True)
151+
workspace_id = self.data.get('workspace_id')
152+
query_set = QuerySet(Application).filter(id=self.data.get('application_id'))
153+
if workspace_id:
154+
query_set = query_set.filter(workspace_id=workspace_id)
155+
if not query_set.exists():
156+
raise AppApiException(500, _('Application id does not exist'))
157+
149158
def generate_prompt(self, instance: dict, with_valid=True):
150159
if with_valid:
151160
self.is_valid(raise_exception=True)

0 commit comments

Comments
 (0)