ffreemt commited on
Commit
33c1ed4
·
1 Parent(s): 2493442
Files changed (1) hide show
  1. app.py +12 -11
app.py CHANGED
@@ -1,11 +1,11 @@
1
  # pylint: disable=line-too-long, missing-module-docstring, missing-function-docstring,broad-exception-caught, too-many-statements
2
 
3
  import json
4
-
5
  import os
6
  import re
7
  import time
8
  from datetime import datetime, timedelta
 
9
 
10
  import requests
11
 
@@ -73,7 +73,7 @@ def fetch_tokens():
73
  },
74
  )
75
  cache["app_session"] = session.cookies.get("appSession")
76
- cache["app_session_time"] = datetime.now()
77
 
78
  # 检查并获取 accessToken
79
  if (
@@ -89,7 +89,7 @@ def fetch_tokens():
89
  logger.error("Failed to get access token")
90
  return None
91
  cache["access_token"] = response.json().get("accessToken")
92
- cache["access_token_time"] = datetime.now()
93
 
94
  y(cache)
95
 
@@ -98,13 +98,14 @@ def fetch_tokens():
98
 
99
  @app.route("/")
100
  def landing():
101
- return """
102
- query /hf/v1/chat/completions for a spin, e.g. curl -XPOST 127.0.0.1:7860/hf/v1/chat/completions -H "Authorization: Bearer Your_AUTHKEY"
103
-
104
- or
105
- curl -XPOST [127.0.0.1:7860|hf space url]/hf/v1/chat/completions -H "Authorization: Bearer Your_AUTHKEY" -H "Content-Type: application/json" --data "{\"model\": \"gpt-3.5-turbo\", \"messages\": [{\"role\": \"user\", \"content\": \"Say this is a test!\"}]}"
106
 
107
- """
 
 
 
108
 
109
 
110
  @app.route("/hf/v1/chat/completions", methods=["POST", "OPTIONS"])
@@ -138,8 +139,8 @@ def chat_completions():
138
  logger.error(f"Error parsing JSON body: {e}")
139
  return Response("Error parsing JSON body", status=400)
140
 
141
- messages = request_body.get("messages", [])
142
- model = request_body.get("model", "reka-core")
143
 
144
  conversation_history = [
145
  {
 
1
  # pylint: disable=line-too-long, missing-module-docstring, missing-function-docstring,broad-exception-caught, too-many-statements
2
 
3
  import json
 
4
  import os
5
  import re
6
  import time
7
  from datetime import datetime, timedelta
8
+ from textwrap import dedent
9
 
10
  import requests
11
 
 
73
  },
74
  )
75
  cache["app_session"] = session.cookies.get("appSession")
76
+ cache["app_session_time"] = datetime.now() # type: ignore
77
 
78
  # 检查并获取 accessToken
79
  if (
 
89
  logger.error("Failed to get access token")
90
  return None
91
  cache["access_token"] = response.json().get("accessToken")
92
+ cache["access_token_time"] = datetime.now() # type: ignore
93
 
94
  y(cache)
95
 
 
98
 
99
  @app.route("/")
100
  def landing():
101
+ return dedent(
102
+ """
103
+ query /hf/v1/chat/completions for a spin, e.g. curl -XPOST 127.0.0.1:7860/hf/v1/chat/completions -H "Authorization: Bearer Your_AUTHKEY"
 
 
104
 
105
+ or
106
+ curl -XPOST [127.0.0.1:7860 | hf-space-url(e.g. https://mikeee-reka.hf.space)]/hf/v1/chat/completions -H "Authorization: Bearer Your_AUTHKEY" -H "Content-Type: application/json" --data "{\"model\": \"gpt-3.5-turbo\", \"messages\": [{\"role\": \"user\", \"content\": \"Say this is a test!\"}]}"
107
+ """
108
+ )
109
 
110
 
111
  @app.route("/hf/v1/chat/completions", methods=["POST", "OPTIONS"])
 
139
  logger.error(f"Error parsing JSON body: {e}")
140
  return Response("Error parsing JSON body", status=400)
141
 
142
+ messages = request_body.get("messages", []) # type: ignore
143
+ model = request_body.get("model", "reka-core") # type: ignore
144
 
145
  conversation_history = [
146
  {