Upload folder using huggingface_hub

advanced_rag.py

@@ -196,19 +196,18 @@ class ElevatedRagChain:
             class MistralLLM(LLM):
                 temperature: float = 0.7
                 top_p: float = 0.95
-                _client: Any = PrivateAttr(default=None)  # Set default to None
-
+                _client: Any = PrivateAttr()  # Remove the default=None here
+                
                 def __init__(self, api_key: str, temperature: float = 0.7, top_p: float = 0.95, **kwargs: Any):
-                    # Do not pass api_key to super().__init__ since it's not a field
-                    super().__init__(**kwargs)
+                    # Initialize the private attributes before calling super().__init__
                     self._client = Mistral(api_key=api_key)
-                    self.temperature = temperature
-                    self.top_p = top_p
-
+                    # Now call super().__init__
+                    super().__init__(temperature=temperature, top_p=top_p, **kwargs)
+                    
                 @property
                 def _llm_type(self) -> str:
                     return "mistral_llm"
-
+                    
                 def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str:
                     response = self._client.chat.complete(
                         model="mistral-small-latest",
@@ -218,7 +217,7 @@ class ElevatedRagChain:
                         max_tokens=32000
                     )
                     return response.choices[0].message.content
-
+                    
                 @property
                 def _identifying_params(self) -> dict:
                     return {"model": "mistral-small-latest"}