Remove tools from LLMs (#2363)

2025-03-14 17:42:48 +05:30
parent 4be426f762
commit ee80a43810
21 changed files with 418 additions and 1071 deletions
--- a/mem0/llms/xai.py
+++ b/mem0/llms/xai.py
@@ -15,7 +15,11 @@ class XAILLM(LLMBase):
            self.config.model = "grok-2-latest"

        api_key = self.config.api_key or os.getenv("XAI_API_KEY")
-        base_url = self.config.xai_base_url or os.getenv("XAI_API_BASE") or "https://api.x.ai/v1"
+        base_url = (
+            self.config.xai_base_url
+            or os.getenv("XAI_API_BASE")
+            or "https://api.x.ai/v1"
+        )
        self.client = OpenAI(api_key=api_key, base_url=base_url)

    def generate_response(self, messages: List[Dict[str, str]], response_format=None):