migrated to new db, improved references

app.py

@@ -1,6 +1,7 @@
 import discord
 import aiohttp
 
+import ast
 import os
 import threading
 
@@ -42,7 +43,7 @@ async def llm_output(question: str, context: str) -> str:
     # URL_LLM = 'http://localhost:11434'    # NOTE: FOR TESTING
 
     prompt = f"""
-    You are a senior developer. Answer the users question based on the context provided.
+    You are a senior FURY developer. FURY is a high level python graphics API similar to VTK.
 
     Question: {question}
 
@@ -107,12 +108,26 @@ async def on_message(message):
     question = message.content.replace("<@1243428204124045385>", "")
     
     embedding: list = await embedding_output(question)
+
     db_knn: dict = await db_output(embedding)
-    llm_answer: str = await llm_output(question, db_knn['matches'][0]['metadata']['text'])      # for the highest knn result (for the test only right now)  TODO: make this better
+    db_context = ""
+    references = ""
+    for i in range(len(db_knn['matches'])):
+        data = db_knn['matches'][i]['metadata']['data']
+        db_context += (data + "\n")
+        data = ast.literal_eval(data)
+        references += ("<https://github.com/fury-gl/fury/tree/master/" + data['path'] + ">")
+        if data.get("function_name"):
+            references += f"\tFunction Name: {data.get("function_name")}"
+        else:
+            references += f"\tClass Name: {data.get("class_name")}"
+        references += "\n"
+
+    llm_answer: str = await llm_output(question, db_context)      # for the highest knn result (for the test only right now)  TODO: make this better
 
     try:
         await message.reply(content=llm_answer[:1990], view=Like_Dislike())             # TODO: handle large responses (>2000)
-        await message.reply(content=db_knn['matches'][0]['metadata']['text'][:1990])    # TODO: handle large responses (>2000)
+        await message.reply(content=f"**References**\n{references}")                                  
     except Exception as e:                                                           # TODO: make exception handling better
         print(e)
         await message.reply("An error occurred. Retry again.")