diff --git a/apps/chat/serializers/chat.py b/apps/chat/serializers/chat.py index 198569d6de7..0851235ae2e 100644 --- a/apps/chat/serializers/chat.py +++ b/apps/chat/serializers/chat.py @@ -168,7 +168,7 @@ def generate_prompt(self, instance: dict, with_valid=True): q = prompt.replace("{userInput}", message) messages[-1]['content'] = q - model_exist = QuerySet(Model).filter(workspace_id=workspace_id, + model_exist = QuerySet(Model).filter( id=model_id, model_type = "LLM" ).exists()