answer 2

.gitignore

@@ -1,3 +1,4 @@
+.DS_Store
 *.ipynb
 script/
 data/

main.py

@@ -39,4 +39,13 @@ def get_suggestion(inputs: str):
 
 @app.get("/api/answer")
 async def get_answer(inputs: str):
-    return StreamingResponse(api.get_answer(inputs))
+    return StreamingResponse(api.get_answer(inputs))
+
+class AnswerRequest(BaseModel):
+    inputs: str
+
+@app.post("/api/answer_v2")
+async def post_answer(request: AnswerRequest):
+    # 假设 api.get_answer() 需要 description 和 link 两个参数
+    response = api.get_answer_v2(request.inputs)
+    return StreamingResponse(response)

service/api.py

@@ -3,7 +3,7 @@ import time
 import json
 import requests
 from typing import Any
-from .llm import search_bot
+from .llm import search_bot,search_bot_v2
 from .vector_store import vector_store
 # from .logger import logger
 from util.time import TimeUtil
@@ -65,7 +65,32 @@ class API():
             text += i
             yield i
         database.write({"query":query,"api":'search',"result":text},dir=Dir.API)
-
+    
+    def get_answer_v2(self,query):
+        docs = vector_store.search(query)
+        source = []
+        context = []
+        for doc in docs:
+            if doc.metadata['score']<0.2:
+                continue
+            if 'title_llm' in doc.metadata:
+                source.append({
+                    "name":doc.metadata.get('title_llm',''),
+                    "url":doc.metadata.get('source',''),
+                    "description":doc.metadata.get('description_llm','')
+                })
+            description = doc.metadata.get('description','')
+            links = doc.metadata.get('source','')
+            context.append(f"<context>Description: {description}\nLinks: {links}\n<context/>")
+        # yield json.dumps(source)
+        # print(context)
+        context = "\n".join(context)
+        answer = search_bot_v2(query,context,stream=True)
+        text = ""
+        for i in answer:
+            text += i
+            yield i
+        database.write({"query":query,"api":'answer_v2',"result":text},dir=Dir.API)
 
     def get_filecoin_price(self):
         url="https://api.spacescope.io/v2/price/realtime"

service/llm.py

@@ -70,5 +70,24 @@ class SummerizeBot(Bot):
     def __init__(self):
         super().__init__(template=self.TEMPLATE)
 
+class SearchBotV2(Bot):
+    TEMPLATE = """
+        You are an expert tasked with summarizing search results and answering questions about FileCoin.
+        Generate a comprehensive and informative answer of 80 words or less for the
+        given question based solely on the provided search results in the context. You must
+        only use information from the provided search results. Include the most relevant link from the search results 
+        to support your answer.
+
+        Below are the structured search results retrieved from a knowledge bank. Each entry consists of a description and associated links.
+
+        {context}
+
+        Your task is to provide a precise and concise response to the question, and also include the most relevant link from the search results that substantiates your answer.
+    """
+
+    def __init__(self):
+        super().__init__(self.TEMPLATE)
+
 search_bot = SearchBot()
+search_bot_v2 = SearchBotV2()
 summerize_bot = SummerizeBot()