Update api/utils.py
Browse files- api/utils.py +2 -7
api/utils.py
CHANGED
@@ -17,7 +17,6 @@ from api.logger import setup_logger
|
|
17 |
|
18 |
logger = setup_logger(__name__)
|
19 |
|
20 |
-
|
21 |
def create_chat_completion_data(
|
22 |
content: str, model: str, timestamp: int, finish_reason: Optional[str] = None
|
23 |
) -> Dict[str, Any]:
|
@@ -36,13 +35,11 @@ def create_chat_completion_data(
|
|
36 |
"usage": None,
|
37 |
}
|
38 |
|
39 |
-
|
40 |
def verify_app_secret(credentials: HTTPAuthorizationCredentials = Depends(security)):
|
41 |
if credentials.credentials != APP_SECRET:
|
42 |
raise HTTPException(status_code=403, detail="Invalid APP_SECRET")
|
43 |
return credentials.credentials
|
44 |
|
45 |
-
|
46 |
def message_to_dict(message):
|
47 |
if isinstance(message.content, str):
|
48 |
return {"role": message.role, "content": message.content}
|
@@ -59,7 +56,6 @@ def message_to_dict(message):
|
|
59 |
else:
|
60 |
return {"role": message.role, "content": message.content}
|
61 |
|
62 |
-
|
63 |
async def process_streaming_response(request: ChatRequest):
|
64 |
agent_mode = AGENT_MODE.get(request.model, {})
|
65 |
trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
|
@@ -67,7 +63,7 @@ async def process_streaming_response(request: ChatRequest):
|
|
67 |
# Log reduced information
|
68 |
logger.info(
|
69 |
f"Streaming request for model: '{request.model}', "
|
70 |
-
f"
|
71 |
)
|
72 |
|
73 |
json_data = {
|
@@ -126,7 +122,6 @@ async def process_streaming_response(request: ChatRequest):
|
|
126 |
logger.error(f"Error occurred during request: {e}")
|
127 |
raise HTTPException(status_code=500, detail=str(e))
|
128 |
|
129 |
-
|
130 |
async def process_non_streaming_response(request: ChatRequest):
|
131 |
agent_mode = AGENT_MODE.get(request.model, {})
|
132 |
trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
|
@@ -134,7 +129,7 @@ async def process_non_streaming_response(request: ChatRequest):
|
|
134 |
# Log reduced information
|
135 |
logger.info(
|
136 |
f"Non-streaming request for model: '{request.model}', "
|
137 |
-
f"
|
138 |
)
|
139 |
|
140 |
json_data = {
|
|
|
17 |
|
18 |
logger = setup_logger(__name__)
|
19 |
|
|
|
20 |
def create_chat_completion_data(
|
21 |
content: str, model: str, timestamp: int, finish_reason: Optional[str] = None
|
22 |
) -> Dict[str, Any]:
|
|
|
35 |
"usage": None,
|
36 |
}
|
37 |
|
|
|
38 |
def verify_app_secret(credentials: HTTPAuthorizationCredentials = Depends(security)):
|
39 |
if credentials.credentials != APP_SECRET:
|
40 |
raise HTTPException(status_code=403, detail="Invalid APP_SECRET")
|
41 |
return credentials.credentials
|
42 |
|
|
|
43 |
def message_to_dict(message):
|
44 |
if isinstance(message.content, str):
|
45 |
return {"role": message.role, "content": message.content}
|
|
|
56 |
else:
|
57 |
return {"role": message.role, "content": message.content}
|
58 |
|
|
|
59 |
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, {})
|
|
|
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 = {
|
|
|
122 |
logger.error(f"Error occurred during request: {e}")
|
123 |
raise HTTPException(status_code=500, detail=str(e))
|
124 |
|
|
|
125 |
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, {})
|
|
|
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 = {
|