jiarongqiu commited on
Commit
e9fffa8
·
1 Parent(s): 0995744
Files changed (1) hide show
  1. service/api.py +2 -1
service/api.py CHANGED
@@ -1,4 +1,5 @@
1
  import time
 
2
  from typing import Any
3
  from .llm import search_bot
4
  from .vector_store import vector_store
@@ -40,7 +41,7 @@ class API():
40
  "description":doc.metadata.get('description_llm','')
41
  })
42
  context.append(doc.metadata.get('description',''))
43
- yield source
44
  context = "\n".join(context)
45
  answer = search_bot(query,context,stream=True)
46
  text = ""
 
1
  import time
2
+ import json
3
  from typing import Any
4
  from .llm import search_bot
5
  from .vector_store import vector_store
 
41
  "description":doc.metadata.get('description_llm','')
42
  })
43
  context.append(doc.metadata.get('description',''))
44
+ yield json.dumps(source)
45
  context = "\n".join(context)
46
  answer = search_bot(query,context,stream=True)
47
  text = ""