Merge pull request #13 from FacerAin/feat/retrieval

Update endpoint for similarity_search

app/agent/prompts.py

@@ -8,6 +8,9 @@
 If you don't know the answer, just say that you don't know, don't try to make up an answer.
 Users tend to want up-to-date information, so please refer to the current date to answer.
 Each context is separated by [SEP]. In most cases, the contents of each context are independent, so choose the most appropriate one and answer.
+If you think there are multiple appropriate answers, it’s okay to paste a reference link and a brief explanation at the bottom of the answer.
+In situations where the provided context is insufficient, it’s possible to list multiple candidate links to inform the user.
+When the context cannot be found, instead of saying that the context was not found, advise the user on suitable alternative actions.
 Attach the page link of the corresponding context at the bottom of the answer.
 You must answer in Korean.
 

app/api/api_v1/endpoints/chat.py

@@ -1,6 +1,7 @@
 from fastapi import APIRouter
 
 from app.agent import ChatAgent
+from app.agent.retriever import PineconeRetriever
 from app.schemas.chat import ResponseAnswer, ReuqestQuery
 
 router = APIRouter()
@@ -12,3 +13,9 @@
 def make_chat(req: ReuqestQuery):
     answer = chat_agent.run(req.query)
     return {"answer": answer}
+
+
+@router.get("/similarity-search")
+def get_similarity_search(query: str):
+    retreiver = PineconeRetriever(index_name="khugpt")
+    return retreiver.get_relevant_doc_string(query)