Skip to content

Commit b6aa6cd

Browse files
authored
fix: OpenAI interface call error (#4752)
1 parent 7fb8e25 commit b6aa6cd

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

apps/chat/serializers/chat.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from django.utils.translation import gettext_lazy as _
1717
from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
1818
from rest_framework import serializers
19+
1920
from application.chat_pipeline.pipeline_manage import PipelineManage
2021
from application.chat_pipeline.step.chat_step.i_chat_step import PostResponseHandler
2122
from application.chat_pipeline.step.chat_step.impl.base_chat_step import BaseChatStep
@@ -217,6 +218,8 @@ class OpenAIChatSerializer(serializers.Serializer):
217218
application_id = serializers.UUIDField(required=True, label=_("Application ID"))
218219
chat_user_id = serializers.CharField(required=True, label=_("Client id"))
219220
chat_user_type = serializers.CharField(required=True, label=_("Client Type"))
221+
ip_address = serializers.CharField(required=False, label=_("IP Address"))
222+
source = serializers.JSONField(required=False, label=_("Source"))
220223

221224
@staticmethod
222225
def get_message(instance):

apps/chat/views/chat.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,8 @@ def post(self, request: Request, application_id: str):
8282
ip_address = _get_ip_address(request)
8383
return OpenAIChatSerializer(data={'application_id': application_id, 'chat_user_id': request.auth.chat_user_id,
8484
'chat_user_type': request.auth.chat_user_type,
85-
'ip_address': ip_address, }).chat(request.data)
85+
'ip_address': ip_address,
86+
'source': {"type": ChatSourceChoices.API_CALL.value}}).chat(request.data)
8687

8788

8889
class AnonymousAuthentication(APIView):

0 commit comments

Comments
 (0)