Update api/utils.py
Browse files- 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": [
|
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": [
|
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,
|