Niansuh commited on
Commit
7e04107
·
verified ·
1 Parent(s): 9ca9231

Update api/utils.py

Browse files
Files changed (1) hide show
  1. 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"using agent mode: {agent_mode}, trending agent mode: {trending_agent_mode}"
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"using agent mode: {agent_mode}, trending agent mode: {trending_agent_mode}"
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 = {