tanbushi commited on
Commit
30c165e
·
1 Parent(s): 07df554

erase prints

Browse files
Files changed (1) hide show
  1. routers/cohere.py +0 -24
routers/cohere.py CHANGED
@@ -15,72 +15,48 @@ print('db_module_filename',db_module_filename)
15
  @router.post("/embed")
16
  async def embed(request:Request, current_user: UserModel = Depends(get_current_user)):
17
  data = await request.json()
18
- print(data)
19
  model = data.get('model', '')
20
- print(model)
21
  if (model=='')or(model is None):
22
  return {'error': 'model is empty'}
23
  api_key_info = await get_api_key(model)
24
  api_key = api_key_info.get('api_key', '')
25
  base_url = api_key_info.get('base_url', '')
26
- print(api_key_info)
27
 
28
  headers = {
29
  'Content-Type': 'application/json',
30
  'Authorization': f'Bearer {api_key}',
31
  'User-Agent': 'PostmanRuntime/7.43.0'
32
  }
33
-
34
- print(f'base_url: {base_url}')
35
 
36
  try:
37
  response = requests.post(url=f"{base_url}/embed", headers=headers, json=data)
38
- print(response.json())
39
  return response.json()
40
  except Exception as e:
41
- print(e)
42
  return {'error': e}
43
 
44
 
45
  @router.post("/rerank")
46
  async def rerank(request:Request, current_user: UserModel = Depends(get_current_user)):
47
  data = await request.json()
48
- print(data)
49
  model = data.get('model', '')
50
- print(model)
51
  if (model=='')or(model is None):
52
  return {'error': 'model is empty'}
53
  api_key_info = await get_api_key(model)
54
  api_key = api_key_info.get('api_key', '')
55
  base_url = api_key_info.get('base_url', '')
56
- print(api_key_info)
57
 
58
  headers = {
59
  'Content-Type': 'application/json',
60
  'Authorization': f'Bearer {api_key}',
61
  'User-Agent': 'PostmanRuntime/7.43.0'
62
  }
63
-
64
- print(f'base_url: {base_url}')
65
 
66
  try:
67
  response = requests.post(url=f"{base_url}/rerank", headers=headers, json=data)
68
- print(response.json())
69
  return response.json()
70
  except Exception as e:
71
- print(e)
72
  return {'error': e}
73
 
74
- # # 从数据库获取默认模型
75
- # async def get_default_model():
76
- # query = f"SELECT * FROM api_names order by default_order limit 1"
77
- # response = TbsDb(db_module_filename, "Cloudflare").get_item(query)
78
- # try:
79
- # result = response['result'][0]['results'][0]['api_name']
80
- # except:
81
- # result = ''
82
- # return result
83
-
84
  async def get_api_key(model):
85
  query = f"""
86
  SELECT an.api_name, ak.api_key, an.base_url, ag.group_name
 
15
  @router.post("/embed")
16
  async def embed(request:Request, current_user: UserModel = Depends(get_current_user)):
17
  data = await request.json()
 
18
  model = data.get('model', '')
 
19
  if (model=='')or(model is None):
20
  return {'error': 'model is empty'}
21
  api_key_info = await get_api_key(model)
22
  api_key = api_key_info.get('api_key', '')
23
  base_url = api_key_info.get('base_url', '')
 
24
 
25
  headers = {
26
  'Content-Type': 'application/json',
27
  'Authorization': f'Bearer {api_key}',
28
  'User-Agent': 'PostmanRuntime/7.43.0'
29
  }
 
 
30
 
31
  try:
32
  response = requests.post(url=f"{base_url}/embed", headers=headers, json=data)
 
33
  return response.json()
34
  except Exception as e:
 
35
  return {'error': e}
36
 
37
 
38
  @router.post("/rerank")
39
  async def rerank(request:Request, current_user: UserModel = Depends(get_current_user)):
40
  data = await request.json()
 
41
  model = data.get('model', '')
 
42
  if (model=='')or(model is None):
43
  return {'error': 'model is empty'}
44
  api_key_info = await get_api_key(model)
45
  api_key = api_key_info.get('api_key', '')
46
  base_url = api_key_info.get('base_url', '')
 
47
 
48
  headers = {
49
  'Content-Type': 'application/json',
50
  'Authorization': f'Bearer {api_key}',
51
  'User-Agent': 'PostmanRuntime/7.43.0'
52
  }
 
 
53
 
54
  try:
55
  response = requests.post(url=f"{base_url}/rerank", headers=headers, json=data)
 
56
  return response.json()
57
  except Exception as e:
 
58
  return {'error': e}
59
 
 
 
 
 
 
 
 
 
 
 
60
  async def get_api_key(model):
61
  query = f"""
62
  SELECT an.api_name, ak.api_key, an.base_url, ag.group_name