Spaces:
Runtime error
Runtime error
jiarongqiu commited on
Commit ·
97994ad
1
Parent(s): 41761a8
add log
Browse files- service/api.py +4 -6
- 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 |
-
|
| 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 |
-
|
| 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()
|