made project openai key non retrievable

ohcnetwork · Sep 4, 2023 · 641d500 · 641d500
1 parent b1bc092
commit 641d500
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/ayushma/serializers/project.py b/ayushma/serializers/project.py
@@ -22,11 +22,20 @@ class Meta:
 
 
 class ProjectUpdateSerializer(serializers.ModelSerializer):
+    key_set = serializers.SerializerMethodField()
+
     class Meta:
         model = Project
-        fields = ProjectSerializer.Meta.fields + ("prompt", "open_ai_key")
+        fields = ProjectSerializer.Meta.fields + ("prompt", "open_ai_key", "key_set")
+        extra_kwargs = {
+            "open_ai_key": {"write_only": True},
+        }
+        read_only_fields = ("key_set",)
 
     def update(self, instance, validated_data):
         if validated_data.get("is_default", True):
             Project.objects.all().update(is_default=False)
         return super().update(instance, validated_data)
+
+    def get_key_set(self, obj):
+        return obj.open_ai_key is not None