jiarongqiu commited on
Commit
97994ad
·
1 Parent(s): 41761a8
Files changed (2) hide show
  1. service/api.py +4 -6
  2. service/database.py +2 -0
service/api.py CHANGED
@@ -7,6 +7,8 @@ from .llm import search_bot
7
  from .vector_store import vector_store
8
  # from .logger import logger
9
  from util.time import TimeUtil
 
 
10
 
11
 
12
  class API():
@@ -38,7 +40,7 @@ class API():
38
  if name == '' or url == '':
39
  continue
40
  res.append({"name":name,"url":url})
41
- # logger.write_log({"query":query,"api":'auto_complete',"result":res})
42
  return res
43
 
44
  def get_answer(self,query):
@@ -62,11 +64,7 @@ class API():
62
  for i in answer:
63
  text += i
64
  yield i
65
- # logger.write_log({
66
- # "query":query,
67
- # "api":'search',
68
- # "source":source,
69
- # 'answer':text})
70
 
71
 
72
  def get_filecoin_price(self):
 
7
  from .vector_store import vector_store
8
  # from .logger import logger
9
  from util.time import TimeUtil
10
+ from .database import database
11
+ from util.schema import Dir
12
 
13
 
14
  class API():
 
40
  if name == '' or url == '':
41
  continue
42
  res.append({"name":name,"url":url})
43
+ database.write({"query":query,"api":'auto_complete',"result":res},dir=Dir.API)
44
  return res
45
 
46
  def get_answer(self,query):
 
64
  for i in answer:
65
  text += i
66
  yield i
67
+ database.write({"query":query,"api":'search',"result":text},dir=Dir.API)
 
 
 
 
68
 
69
 
70
  def get_filecoin_price(self):
service/database.py CHANGED
@@ -38,3 +38,5 @@ class Database:
38
  f.write(data)
39
  except Exception as e:
40
  print(e)
 
 
 
38
  f.write(data)
39
  except Exception as e:
40
  print(e)
41
+
42
+ database = Database()