new db updates - new references + new error log

main.py

@@ -5,7 +5,6 @@ import ast
 import os
 import threading
 
-
 intents = discord.Intents.default()
 intents.message_content = True
 bot = discord.Bot(intents = intents)
@@ -50,15 +49,17 @@ async def llm_output(question: str, context: str) -> str:
     Context: {context}
     """
     obj = {
-            'model': 'llama3-70b-8192',
-            'prompt': prompt,
-            'stream': False
+            "model": "llama3-70b-8192",
+            "prompt": prompt,
+            "stream": False
         }
-    
+
     async with aiohttp.ClientSession() as session:
         async with session.post(URL_LLM + "/api/generate", json=obj) as response:
+            if response.status == 500:
+                return "Sorry, an internal Error happened. Please try again later.\nError 500."
             response_json = await response.json()
-    
+
     return response_json['choices'][0]['message']['content']
 
 async def embedding_output(message: str) -> list:
@@ -116,11 +117,15 @@ async def on_message(message):
         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'] + ">")
+        references += ("<https://github.com/fury-gl/fury/tree/master/" + data['path'] + ">").replace("//home/robin/Desktop/l/fury", "")
         if data.get("function_name"):
             references += f"\tFunction Name: {data.get('function_name')}"
-        else:
+        elif data.get("class_name"):
             references += f"\tClass Name: {data.get('class_name')}"
+        elif data['type'] == 'rst':
+            references += f"\tDocumentation: {data['path'].split("/")[-1]}"
+        elif data['type'] == 'documentation_examples':
+            references += f"\tDocumentation: {data['path'].split("/")[-1]}"
         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