feat: pass decoded_token to llm and retrievers

application/agents/base.py

@@ -9,10 +9,21 @@ from application.llm.llm_creator import LLMCreator
 
 
 class BaseAgent:
-    def __init__(self, endpoint, llm_name, gpt_model, api_key, user_api_key=None):
+    def __init__(
+        self,
+        endpoint,
+        llm_name,
+        gpt_model,
+        api_key,
+        user_api_key=None,
+        decoded_token=None,
+    ):
         self.endpoint = endpoint
         self.llm = LLMCreator.create_llm(
-            llm_name, api_key=api_key, user_api_key=user_api_key
+            llm_name,
+            api_key=api_key,
+            user_api_key=user_api_key,
+            decoded_token=decoded_token,
         )
         self.llm_handler = get_llm_handler(llm_name)
         self.gpt_model = gpt_model

application/agents/classic_agent.py

@@ -17,8 +17,12 @@ class ClassicAgent(BaseAgent):
         user_api_key=None,
         prompt="",
         chat_history=None,
+        decoded_token=None,
     ):
-        super().__init__(endpoint, llm_name, gpt_model, api_key, user_api_key)
+        super().__init__(
+            endpoint, llm_name, gpt_model, api_key, user_api_key, decoded_token
+        )
+        self.user = decoded_token.get("sub")
         self.prompt = prompt
         self.chat_history = chat_history if chat_history is not None else []
 
@@ -73,7 +77,7 @@ class ClassicAgent(BaseAgent):
                         )
         messages_combine.append({"role": "user", "content": query})
 
-        tools_dict = self._get_user_tools()
+        tools_dict = self._get_user_tools(self.user)
         self._prepare_tools(tools_dict)
 
         resp = self._llm_gen(messages_combine, log_context)