Niansuh commited on
Commit
a174705
·
verified ·
1 Parent(s): d4edf4d

Update api/utils.py

Browse files
Files changed (1) hide show
  1. api/utils.py +4 -4
api/utils.py CHANGED
@@ -60,14 +60,14 @@ async def process_streaming_response(request: ChatRequest):
60
  agent_mode = AGENT_MODE.get(request.model, {})
61
  trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
62
 
63
- # Log reduced information
64
  logger.info(
65
  f"Streaming request for model: '{request.model}', "
66
  f"agent mode: {agent_mode}, trending agent mode: {trending_agent_mode}"
67
  )
68
 
69
  json_data = {
70
- "messages": [message_to_dict(msg) for msg in request.messages],
71
  "previewToken": None,
72
  "userId": None,
73
  "codeModelMode": True,
@@ -126,14 +126,14 @@ async def process_non_streaming_response(request: ChatRequest):
126
  agent_mode = AGENT_MODE.get(request.model, {})
127
  trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
128
 
129
- # Log reduced information
130
  logger.info(
131
  f"Non-streaming request for model: '{request.model}', "
132
  f"agent mode: {agent_mode}, trending agent mode: {trending_agent_mode}"
133
  )
134
 
135
  json_data = {
136
- "messages": [message_to_dict(msg) for msg in request.messages],
137
  "previewToken": None,
138
  "userId": None,
139
  "codeModelMode": True,
 
60
  agent_mode = AGENT_MODE.get(request.model, {})
61
  trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
62
 
63
+ # Log reduced information, excluding user content
64
  logger.info(
65
  f"Streaming request for model: '{request.model}', "
66
  f"agent mode: {agent_mode}, trending agent mode: {trending_agent_mode}"
67
  )
68
 
69
  json_data = {
70
+ "messages": [{"role": "user", "content": "User message hidden"} for _ in request.messages],
71
  "previewToken": None,
72
  "userId": None,
73
  "codeModelMode": True,
 
126
  agent_mode = AGENT_MODE.get(request.model, {})
127
  trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
128
 
129
+ # Log reduced information, excluding user content
130
  logger.info(
131
  f"Non-streaming request for model: '{request.model}', "
132
  f"agent mode: {agent_mode}, trending agent mode: {trending_agent_mode}"
133
  )
134
 
135
  json_data = {
136
+ "messages": [{"role": "user", "content": "User message hidden"} for _ in request.messages],
137
  "previewToken": None,
138
  "userId": None,
139
  "codeModelMode": True,