Update uv.lock after environment fix

Co-Authored-By: João <joao@crewai.com>

pyproject.toml

@@ -48,7 +48,7 @@ Documentation = "https://docs.crewai.com"
 Repository = "https://github.com/crewAIInc/crewAI"
 
 [project.optional-dependencies]
-tools = ["crewai-tools~=0.59.0"]
+tools = ["crewai-tools~=0.58.0"]
 embeddings = [
     "tiktoken~=0.8.0"
 ]

src/crewai/__init__.py

@@ -54,7 +54,7 @@ def _track_install_async():
 
 _track_install_async()
 
-__version__ = "0.152.0"
+__version__ = "0.150.0"
 __all__ = [
     "Agent",
     "Crew",

src/crewai/cli/cli.py

@@ -3,7 +3,6 @@ from typing import Optional
 
 import click
 from crewai.cli.config import Settings
-from crewai.cli.settings.main import SettingsCommand
 from crewai.cli.add_crew_to_flow import add_crew_to_flow
 from crewai.cli.create_crew import create_crew
 from crewai.cli.create_flow import create_flow
@@ -228,7 +227,7 @@ def update():
 @crewai.command()
 def login():
     """Sign Up/Login to CrewAI Enterprise."""
-    Settings().clear_user_settings()
+    Settings().clear()
     AuthenticationCommand().login()
 
 
@@ -370,8 +369,8 @@ def org():
     pass
 
 
-@org.command("list")
-def org_list():
+@org.command()
+def list():
     """List available organizations."""
     org_command = OrganizationCommand()
     org_command.list()
@@ -392,34 +391,5 @@ def current():
     org_command.current()
 
 
-@crewai.group()
-def config():
-    """CLI Configuration commands."""
-    pass
-
-
-@config.command("list")
-def config_list():
-    """List all CLI configuration parameters."""
-    config_command = SettingsCommand()
-    config_command.list()
-
-
-@config.command("set")
-@click.argument("key")
-@click.argument("value")
-def config_set(key: str, value: str):
-    """Set a CLI configuration parameter."""
-    config_command = SettingsCommand()
-    config_command.set(key, value)
-
-
-@config.command("reset")
-def config_reset():
-    """Reset all CLI configuration parameters to default values."""
-    config_command = SettingsCommand()
-    config_command.reset_all_settings()
-
-
 if __name__ == "__main__":
     crewai()

src/crewai/cli/config.py

@@ -4,47 +4,10 @@ from typing import Optional
 
 from pydantic import BaseModel, Field
 
-from crewai.cli.constants import DEFAULT_CREWAI_ENTERPRISE_URL
-
 DEFAULT_CONFIG_PATH = Path.home() / ".config" / "crewai" / "settings.json"
 
-# Settings that are related to the user's account
-USER_SETTINGS_KEYS = [
-    "tool_repository_username",
-    "tool_repository_password",
-    "org_name",
-    "org_uuid",
-]
-
-# Settings that are related to the CLI
-CLI_SETTINGS_KEYS = [
-    "enterprise_base_url",
-]
-
-# Default values for CLI settings
-DEFAULT_CLI_SETTINGS = {
-    "enterprise_base_url": DEFAULT_CREWAI_ENTERPRISE_URL,
-}
-
-# Readonly settings - cannot be set by the user
-READONLY_SETTINGS_KEYS = [
-    "org_name",
-    "org_uuid",
-]
-
-# Hidden settings - not displayed by the 'list' command and cannot be set by the user
-HIDDEN_SETTINGS_KEYS = [
-    "config_path",
-    "tool_repository_username",
-    "tool_repository_password",
-]
-
 
 class Settings(BaseModel):
-    enterprise_base_url: Optional[str] = Field(
-        default=DEFAULT_CREWAI_ENTERPRISE_URL,
-        description="Base URL of the CrewAI Enterprise instance",
-    )
     tool_repository_username: Optional[str] = Field(
         None, description="Username for interacting with the Tool Repository"
     )
@@ -57,7 +20,7 @@ class Settings(BaseModel):
     org_uuid: Optional[str] = Field(
         None, description="UUID of the currently active organization"
     )
-    config_path: Path = Field(default=DEFAULT_CONFIG_PATH, frozen=True, exclude=True)
+    config_path: Path = Field(default=DEFAULT_CONFIG_PATH, exclude=True)
 
     def __init__(self, config_path: Path = DEFAULT_CONFIG_PATH, **data):
         """Load Settings from config path"""
@@ -74,16 +37,9 @@ class Settings(BaseModel):
         merged_data = {**file_data, **data}
         super().__init__(config_path=config_path, **merged_data)
 
-    def clear_user_settings(self) -> None:
-        """Clear all user settings"""
-        self._reset_user_settings()
-        self.dump()
-
-    def reset(self) -> None:
-        """Reset all settings to default values"""
-        self._reset_user_settings()
-        self._reset_cli_settings()
-        self.dump()
+    def clear(self) -> None:
+        """Clear all settings"""
+        self.config_path.unlink(missing_ok=True)
 
     def dump(self) -> None:
         """Save current settings to settings.json"""
@@ -96,13 +52,3 @@ class Settings(BaseModel):
         updated_data = {**existing_data, **self.model_dump(exclude_unset=True)}
         with self.config_path.open("w") as f:
             json.dump(updated_data, f, indent=4)
-
-    def _reset_user_settings(self) -> None:
-        """Reset all user settings to default values"""
-        for key in USER_SETTINGS_KEYS:
-            setattr(self, key, None)
-
-    def _reset_cli_settings(self) -> None:
-        """Reset all CLI settings to default values"""
-        for key in CLI_SETTINGS_KEYS:
-            setattr(self, key, DEFAULT_CLI_SETTINGS[key])

src/crewai/cli/constants.py

@@ -1,5 +1,3 @@
-DEFAULT_CREWAI_ENTERPRISE_URL = "https://app.crewai.com"
-
 ENV_VARS = {
     "openai": [
         {
@@ -322,4 +320,5 @@ DEFAULT_LLM_MODEL = "gpt-4o-mini"
 
 JSON_URL = "https://raw.githubusercontent.com/BerriAI/litellm/main/model_prices_and_context_window.json"
 
+
 LITELLM_PARAMS = ["api_key", "api_base", "api_version"]

src/crewai/cli/plus_api.py

@@ -1,3 +1,4 @@
+from os import getenv
 from typing import List, Optional
 from urllib.parse import urljoin
 
@@ -5,7 +6,6 @@ import requests
 
 from crewai.cli.config import Settings
 from crewai.cli.version import get_crewai_version
-from crewai.cli.constants import DEFAULT_CREWAI_ENTERPRISE_URL
 
 
 class PlusAPI:
@@ -29,10 +29,7 @@ class PlusAPI:
         settings = Settings()
         if settings.org_uuid:
             self.headers["X-Crewai-Organization-Id"] = settings.org_uuid
-
-        self.base_url = (
-            str(settings.enterprise_base_url) or DEFAULT_CREWAI_ENTERPRISE_URL
-        )
+        self.base_url = getenv("CREWAI_BASE_URL", "https://app.crewai.com")
 
     def _make_request(self, method: str, endpoint: str, **kwargs) -> requests.Response:
         url = urljoin(self.base_url, endpoint)
@@ -111,6 +108,7 @@ class PlusAPI:
 
     def create_crew(self, payload) -> requests.Response:
         return self._make_request("POST", self.CREWS_RESOURCE, json=payload)
-
+    
     def get_organizations(self) -> requests.Response:
         return self._make_request("GET", self.ORGANIZATIONS_RESOURCE)
+    

src/crewai/cli/settings/main.py

@@ -1,67 +0,0 @@
-from rich.console import Console
-from rich.table import Table
-from crewai.cli.command import BaseCommand
-from crewai.cli.config import Settings, READONLY_SETTINGS_KEYS, HIDDEN_SETTINGS_KEYS
-from typing import Any
-
-console = Console()
-
-
-class SettingsCommand(BaseCommand):
-    """A class to handle CLI configuration commands."""
-
-    def __init__(self, settings_kwargs: dict[str, Any] = {}):
-        super().__init__()
-        self.settings = Settings(**settings_kwargs)
-
-    def list(self) -> None:
-        """List all CLI configuration parameters."""
-        table = Table(title="CrewAI CLI Configuration")
-        table.add_column("Setting", style="cyan", no_wrap=True)
-        table.add_column("Value", style="green")
-        table.add_column("Description", style="yellow")
-
-        # Add all settings to the table
-        for field_name, field_info in Settings.model_fields.items():
-            if field_name in HIDDEN_SETTINGS_KEYS:
-                # Do not display hidden settings
-                continue
-
-            current_value = getattr(self.settings, field_name)
-            description = field_info.description or "No description available"
-            display_value = (
-                str(current_value) if current_value is not None else "Not set"
-            )
-
-            table.add_row(field_name, display_value, description)
-
-        console.print(table)
-
-    def set(self, key: str, value: str) -> None:
-        """Set a CLI configuration parameter."""
-
-        readonly_settings = READONLY_SETTINGS_KEYS + HIDDEN_SETTINGS_KEYS
-
-        if not hasattr(self.settings, key) or key in readonly_settings:
-            console.print(
-                f"Error: Unknown or readonly configuration key '{key}'",
-                style="bold red",
-            )
-            console.print("Available keys:", style="yellow")
-            for field_name in Settings.model_fields.keys():
-                if field_name not in readonly_settings:
-                    console.print(f"  - {field_name}", style="yellow")
-            raise SystemExit(1)
-
-        setattr(self.settings, key, value)
-        self.settings.dump()
-
-        console.print(f"Successfully set '{key}' to '{value}'", style="bold green")
-
-    def reset_all_settings(self) -> None:
-        """Reset all CLI configuration parameters to default values."""
-        self.settings.reset()
-        console.print(
-            "Successfully reset all configuration parameters to default values. It is recommended to run [bold yellow]'crewai login'[/bold yellow] to re-authenticate.",
-            style="bold green",
-        )

src/crewai/cli/templates/crew/pyproject.toml

@@ -5,7 +5,7 @@ description = "{{name}} using crewAI"
 authors = [{ name = "Your Name", email = "you@example.com" }]
 requires-python = ">=3.10,<3.14"
 dependencies = [
-    "crewai[tools]>=0.152.0,<1.0.0"
+    "crewai[tools]>=0.150.0,<1.0.0"
 ]
 
 [project.scripts]

src/crewai/cli/templates/flow/pyproject.toml

@@ -5,7 +5,7 @@ description = "{{name}} using crewAI"
 authors = [{ name = "Your Name", email = "you@example.com" }]
 requires-python = ">=3.10,<3.14"
 dependencies = [
-    "crewai[tools]>=0.152.0,<1.0.0",
+    "crewai[tools]>=0.150.0,<1.0.0",
 ]
 
 [project.scripts]

src/crewai/cli/templates/tool/pyproject.toml

@@ -5,7 +5,7 @@ description = "Power up your crews with {{folder_name}}"
 readme = "README.md"
 requires-python = ">=3.10,<3.14"
 dependencies = [
-    "crewai[tools]>=0.152.0"
+    "crewai[tools]>=0.150.0"
 ]
 
 [tool.crewai]

src/crewai/llm.py

@@ -308,6 +308,7 @@ class LLM(BaseLLM):
         api_version: Optional[str] = None,
         api_key: Optional[str] = None,
         callbacks: List[Any] = [],
+        reasoning: Optional[bool] = None,
         reasoning_effort: Optional[Literal["none", "low", "medium", "high"]] = None,
         stream: bool = False,
         **kwargs,
@@ -332,6 +333,7 @@ class LLM(BaseLLM):
         self.api_key = api_key
         self.callbacks = callbacks
         self.context_window_size = 0
+        self.reasoning = reasoning
         self.reasoning_effort = reasoning_effort
         self.additional_params = kwargs
         self.is_anthropic = self._is_anthropic_model(model)
@@ -406,10 +408,15 @@ class LLM(BaseLLM):
             "api_key": self.api_key,
             "stream": self.stream,
             "tools": tools,
-            "reasoning_effort": self.reasoning_effort,
             **self.additional_params,
         }
 
+        if self.reasoning is False:
+            # When reasoning is explicitly disabled, don't include reasoning_effort
+            pass
+        elif self.reasoning is True or self.reasoning_effort is not None:
+            params["reasoning_effort"] = self.reasoning_effort
+
         # Remove None values from params
         return {k: v for k, v in params.items() if v is not None}
 

src/crewai/memory/storage/mem0_storage.py

@@ -1,8 +1,7 @@
 import os
 from typing import Any, Dict, List
-from collections import defaultdict
+
 from mem0 import Memory, MemoryClient
-from crewai.utilities.chromadb import sanitize_collection_name
 
 from crewai.memory.storage.interface import Storage
 
@@ -71,32 +70,26 @@ class Mem0Storage(Storage):
         """
         Returns:
             dict: A filter dictionary containing AND conditions for querying data.
-                - Includes user_id and agent_id if both are present.
-                - Includes user_id if only user_id is present.
-                - Includes agent_id if only agent_id is present.
+                - Includes user_id if memory_type is 'external'.
                 - Includes run_id if memory_type is 'short_term' and mem0_run_id is present.
         """
-        filter = defaultdict(list)
+        filter = {
+            "AND": []
+        }
 
+        # Add user_id condition if the memory type is external
+        if self.memory_type == "external":
+            filter["AND"].append({"user_id": self.config.get("user_id", "")})
+
+        # Add run_id condition if the memory type is short_term and a run ID is set
         if self.memory_type == "short_term" and self.mem0_run_id:
             filter["AND"].append({"run_id": self.mem0_run_id})
-        else:
-            user_id = self.config.get("user_id", "")
-            agent_id = self.config.get("agent_id", "")
-
-            if user_id and agent_id:
-                filter["OR"].append({"user_id": user_id})
-                filter["OR"].append({"agent_id": agent_id})
-            elif user_id:
-                filter["AND"].append({"user_id": user_id})
-            elif agent_id:
-                filter["AND"].append({"agent_id": agent_id})
 
         return filter
 
     def save(self, value: Any, metadata: Dict[str, Any]) -> None:
         user_id = self.config.get("user_id", "")
-        assistant_message = [{"role" : "assistant","content" : value}]
+        assistant_message = [{"role" : "assistant","content" : value}] 
 
         base_metadata = {
             "short_term": "short_term",
@@ -111,32 +104,31 @@ class Mem0Storage(Storage):
             "infer": self.infer
         }
 
-        # MemoryClient-specific overrides
-        if isinstance(self.memory, MemoryClient):
-            params["includes"] = self.includes
-            params["excludes"] = self.excludes
-            params["output_format"] = "v1.1"
-            params["version"] = "v2"
-
-        if self.memory_type == "short_term" and self.mem0_run_id:
-            params["run_id"] = self.mem0_run_id
-
-        if user_id:
+        if self.memory_type == "external":
             params["user_id"] = user_id
 
-        if agent_id := self.config.get("agent_id", self._get_agent_name()):
-            params["agent_id"] = agent_id
+        
+        if params:
+            # MemoryClient-specific overrides
+            if isinstance(self.memory, MemoryClient):
+                params["includes"] = self.includes
+                params["excludes"] = self.excludes
+                params["output_format"] = "v1.1"
+                params["version"]="v2"
 
-        self.memory.add(assistant_message, **params)
+                if self.memory_type == "short_term":
+                    params["run_id"] = self.mem0_run_id
+
+            self.memory.add(assistant_message, **params)
 
     def search(self,query: str,limit: int = 3,score_threshold: float = 0.35) -> List[Any]:
         params = {
-            "query": query,
-            "limit": limit,
+            "query": query, 
+            "limit": limit, 
             "version": "v2",
             "output_format": "v1.1"
             }
-
+        
         if user_id := self.config.get("user_id", ""):
             params["user_id"] = user_id
 
@@ -146,7 +138,7 @@ class Mem0Storage(Storage):
             "entities": {"type": "entity"},
             "external": {"type": "external"},
         }
-
+        
         if self.memory_type in memory_type_map:
             params["metadata"] = memory_type_map[self.memory_type]
             if self.memory_type == "short_term":
@@ -159,29 +151,11 @@ class Mem0Storage(Storage):
         params['threshold'] = score_threshold
 
         if isinstance(self.memory, Memory):
-            params.pop("metadata", None)
-            params.pop("version", None)
-            params.pop("output_format", None)
-            params.pop("run_id", None)
+            del params["metadata"], params["version"], params["run_id"], params['output_format']
 
         results = self.memory.search(**params)
         return [r for r in results["results"]]
-
+    
     def reset(self):
         if self.memory:
             self.memory.reset()
-
-    def _sanitize_role(self, role: str) -> str:
-        """
-        Sanitizes agent roles to ensure valid directory names.
-        """
-        return role.replace("\n", "").replace(" ", "_").replace("/", "_")
-
-    def _get_agent_name(self) -> str:
-        if not self.crew:
-            return ""
-
-        agents = self.crew.agents
-        agents = [self._sanitize_role(agent.role) for agent in agents]
-        agents = "_".join(agents)
-        return sanitize_collection_name(name=agents, max_collection_length=MAX_AGENT_ID_LENGTH_MEM0)

tests/agent_reasoning_test.py

@@ -2,6 +2,7 @@
 
 import json
 import pytest
+from unittest.mock import patch
 
 from crewai import Agent, Task
 from crewai.llm import LLM
@@ -259,3 +260,31 @@ def test_agent_with_function_calling_fallback():
     assert result == "4"
     assert "Reasoning Plan:" in task.description
     assert "Invalid JSON that will trigger fallback" in task.description
+
+
+def test_agent_with_llm_reasoning_disabled():
+    """Test agent with LLM reasoning disabled."""
+    llm = LLM("gpt-3.5-turbo", reasoning=False)
+    
+    agent = Agent(
+        role="Test Agent",
+        goal="To test the LLM reasoning parameter",
+        backstory="I am a test agent created to verify the LLM reasoning parameter works correctly.",
+        llm=llm,
+        reasoning=False,
+        verbose=True
+    )
+    
+    task = Task(
+        description="Simple math task: What's 3+3?",
+        expected_output="The answer should be a number.",
+        agent=agent
+    )
+    
+    with patch.object(agent.llm, 'call') as mock_call:
+        mock_call.return_value = "6"
+        
+        result = agent.execute_task(task)
+        
+        assert result == "6"
+        assert "Reasoning Plan:" not in task.description

tests/cli/config_test.py

@@ -4,12 +4,7 @@ import tempfile
 import unittest
 from pathlib import Path
 
-from crewai.cli.config import (
-    Settings,
-    USER_SETTINGS_KEYS,
-    CLI_SETTINGS_KEYS,
-    DEFAULT_CLI_SETTINGS,
-)
+from crewai.cli.config import Settings
 
 
 class TestSettings(unittest.TestCase):
@@ -57,30 +52,6 @@ class TestSettings(unittest.TestCase):
         self.assertEqual(settings.tool_repository_username, "new_user")
         self.assertEqual(settings.tool_repository_password, "file_pass")
 
-    def test_clear_user_settings(self):
-        user_settings = {key: f"value_for_{key}" for key in USER_SETTINGS_KEYS}
-
-        settings = Settings(config_path=self.config_path, **user_settings)
-        settings.clear_user_settings()
-
-        for key in user_settings.keys():
-            self.assertEqual(getattr(settings, key), None)
-
-    def test_reset_settings(self):
-        user_settings = {key: f"value_for_{key}" for key in USER_SETTINGS_KEYS}
-        cli_settings = {key: f"value_for_{key}" for key in CLI_SETTINGS_KEYS}
-
-        settings = Settings(
-            config_path=self.config_path, **user_settings, **cli_settings
-        )
-
-        settings.reset()
-
-        for key in user_settings.keys():
-            self.assertEqual(getattr(settings, key), None)
-        for key in cli_settings.keys():
-            self.assertEqual(getattr(settings, key), DEFAULT_CLI_SETTINGS[key])
-
     def test_dump_new_settings(self):
         settings = Settings(
             config_path=self.config_path, tool_repository_username="user1"

tests/cli/organization/test_main.py

@@ -6,7 +6,7 @@ from click.testing import CliRunner
 import requests
 
 from crewai.cli.organization.main import OrganizationCommand
-from crewai.cli.cli import org_list, switch, current
+from crewai.cli.cli import list, switch, current
 
 
 @pytest.fixture
@@ -16,44 +16,44 @@ def runner():
 
 @pytest.fixture
 def org_command():
-    with patch.object(OrganizationCommand, "__init__", return_value=None):
+    with patch.object(OrganizationCommand, '__init__', return_value=None):
         command = OrganizationCommand()
         yield command
 
 
 @pytest.fixture
 def mock_settings():
-    with patch("crewai.cli.organization.main.Settings") as mock_settings_class:
+    with patch('crewai.cli.organization.main.Settings') as mock_settings_class:
         mock_settings_instance = MagicMock()
         mock_settings_class.return_value = mock_settings_instance
         yield mock_settings_instance
 
 
-@patch("crewai.cli.cli.OrganizationCommand")
+@patch('crewai.cli.cli.OrganizationCommand')
 def test_org_list_command(mock_org_command_class, runner):
     mock_org_instance = MagicMock()
     mock_org_command_class.return_value = mock_org_instance
 
-    result = runner.invoke(org_list)
+    result = runner.invoke(list)
 
     assert result.exit_code == 0
     mock_org_command_class.assert_called_once()
     mock_org_instance.list.assert_called_once()
 
 
-@patch("crewai.cli.cli.OrganizationCommand")
+@patch('crewai.cli.cli.OrganizationCommand')
 def test_org_switch_command(mock_org_command_class, runner):
     mock_org_instance = MagicMock()
     mock_org_command_class.return_value = mock_org_instance
 
-    result = runner.invoke(switch, ["test-id"])
+    result = runner.invoke(switch, ['test-id'])
 
     assert result.exit_code == 0
     mock_org_command_class.assert_called_once()
-    mock_org_instance.switch.assert_called_once_with("test-id")
+    mock_org_instance.switch.assert_called_once_with('test-id')
 
 
-@patch("crewai.cli.cli.OrganizationCommand")
+@patch('crewai.cli.cli.OrganizationCommand')
 def test_org_current_command(mock_org_command_class, runner):
     mock_org_instance = MagicMock()
     mock_org_command_class.return_value = mock_org_instance
@@ -67,18 +67,18 @@ def test_org_current_command(mock_org_command_class, runner):
 
 class TestOrganizationCommand(unittest.TestCase):
     def setUp(self):
-        with patch.object(OrganizationCommand, "__init__", return_value=None):
+        with patch.object(OrganizationCommand, '__init__', return_value=None):
             self.org_command = OrganizationCommand()
             self.org_command.plus_api_client = MagicMock()
 
-    @patch("crewai.cli.organization.main.console")
-    @patch("crewai.cli.organization.main.Table")
+    @patch('crewai.cli.organization.main.console')
+    @patch('crewai.cli.organization.main.Table')
     def test_list_organizations_success(self, mock_table, mock_console):
         mock_response = MagicMock()
         mock_response.raise_for_status = MagicMock()
         mock_response.json.return_value = [
             {"name": "Org 1", "uuid": "org-123"},
-            {"name": "Org 2", "uuid": "org-456"},
+            {"name": "Org 2", "uuid": "org-456"}
         ]
         self.org_command.plus_api_client = MagicMock()
         self.org_command.plus_api_client.get_organizations.return_value = mock_response
@@ -89,14 +89,16 @@ class TestOrganizationCommand(unittest.TestCase):
 
         self.org_command.plus_api_client.get_organizations.assert_called_once()
         mock_table.assert_called_once_with(title="Your Organizations")
-        mock_table.return_value.add_column.assert_has_calls(
-            [call("Name", style="cyan"), call("ID", style="green")]
-        )
-        mock_table.return_value.add_row.assert_has_calls(
-            [call("Org 1", "org-123"), call("Org 2", "org-456")]
-        )
+        mock_table.return_value.add_column.assert_has_calls([
+            call("Name", style="cyan"),
+            call("ID", style="green")
+        ])
+        mock_table.return_value.add_row.assert_has_calls([
+            call("Org 1", "org-123"),
+            call("Org 2", "org-456")
+        ])
 
-    @patch("crewai.cli.organization.main.console")
+    @patch('crewai.cli.organization.main.console')
     def test_list_organizations_empty(self, mock_console):
         mock_response = MagicMock()
         mock_response.raise_for_status = MagicMock()
@@ -108,32 +110,33 @@ class TestOrganizationCommand(unittest.TestCase):
 
         self.org_command.plus_api_client.get_organizations.assert_called_once()
         mock_console.print.assert_called_once_with(
-            "You don't belong to any organizations yet.", style="yellow"
+            "You don't belong to any organizations yet.",
+            style="yellow"
         )
 
-    @patch("crewai.cli.organization.main.console")
+    @patch('crewai.cli.organization.main.console')
     def test_list_organizations_api_error(self, mock_console):
         self.org_command.plus_api_client = MagicMock()
-        self.org_command.plus_api_client.get_organizations.side_effect = (
-            requests.exceptions.RequestException("API Error")
-        )
+        self.org_command.plus_api_client.get_organizations.side_effect = requests.exceptions.RequestException("API Error")
 
         with pytest.raises(SystemExit):
             self.org_command.list()
 
+
         self.org_command.plus_api_client.get_organizations.assert_called_once()
         mock_console.print.assert_called_once_with(
-            "Failed to retrieve organization list: API Error", style="bold red"
+            "Failed to retrieve organization list: API Error",
+            style="bold red"
         )
 
-    @patch("crewai.cli.organization.main.console")
-    @patch("crewai.cli.organization.main.Settings")
+    @patch('crewai.cli.organization.main.console')
+    @patch('crewai.cli.organization.main.Settings')
     def test_switch_organization_success(self, mock_settings_class, mock_console):
         mock_response = MagicMock()
         mock_response.raise_for_status = MagicMock()
         mock_response.json.return_value = [
             {"name": "Org 1", "uuid": "org-123"},
-            {"name": "Test Org", "uuid": "test-id"},
+            {"name": "Test Org", "uuid": "test-id"}
         ]
         self.org_command.plus_api_client = MagicMock()
         self.org_command.plus_api_client.get_organizations.return_value = mock_response
@@ -148,16 +151,17 @@ class TestOrganizationCommand(unittest.TestCase):
         assert mock_settings_instance.org_name == "Test Org"
         assert mock_settings_instance.org_uuid == "test-id"
         mock_console.print.assert_called_once_with(
-            "Successfully switched to Test Org (test-id)", style="bold green"
+            "Successfully switched to Test Org (test-id)",
+            style="bold green"
         )
 
-    @patch("crewai.cli.organization.main.console")
+    @patch('crewai.cli.organization.main.console')
     def test_switch_organization_not_found(self, mock_console):
         mock_response = MagicMock()
         mock_response.raise_for_status = MagicMock()
         mock_response.json.return_value = [
             {"name": "Org 1", "uuid": "org-123"},
-            {"name": "Org 2", "uuid": "org-456"},
+            {"name": "Org 2", "uuid": "org-456"}
         ]
         self.org_command.plus_api_client = MagicMock()
         self.org_command.plus_api_client.get_organizations.return_value = mock_response
@@ -166,11 +170,12 @@ class TestOrganizationCommand(unittest.TestCase):
 
         self.org_command.plus_api_client.get_organizations.assert_called_once()
         mock_console.print.assert_called_once_with(
-            "Organization with id 'non-existent-id' not found.", style="bold red"
+            "Organization with id 'non-existent-id' not found.",
+            style="bold red"
         )
 
-    @patch("crewai.cli.organization.main.console")
-    @patch("crewai.cli.organization.main.Settings")
+    @patch('crewai.cli.organization.main.console')
+    @patch('crewai.cli.organization.main.Settings')
     def test_current_organization_with_org(self, mock_settings_class, mock_console):
         mock_settings_instance = MagicMock()
         mock_settings_instance.org_name = "Test Org"
@@ -181,11 +186,12 @@ class TestOrganizationCommand(unittest.TestCase):
 
         self.org_command.plus_api_client.get_organizations.assert_not_called()
         mock_console.print.assert_called_once_with(
-            "Currently logged in to organization Test Org (test-id)", style="bold green"
+            "Currently logged in to organization Test Org (test-id)",
+            style="bold green"
         )
 
-    @patch("crewai.cli.organization.main.console")
-    @patch("crewai.cli.organization.main.Settings")
+    @patch('crewai.cli.organization.main.console')
+    @patch('crewai.cli.organization.main.Settings')
     def test_current_organization_without_org(self, mock_settings_class, mock_console):
         mock_settings_instance = MagicMock()
         mock_settings_instance.org_uuid = None
@@ -195,14 +201,16 @@ class TestOrganizationCommand(unittest.TestCase):
 
         assert mock_console.print.call_count == 3
         mock_console.print.assert_any_call(
-            "You're not currently logged in to any organization.", style="yellow"
+            "You're not currently logged in to any organization.",
+            style="yellow"
         )
 
-    @patch("crewai.cli.organization.main.console")
+    @patch('crewai.cli.organization.main.console')
     def test_list_organizations_unauthorized(self, mock_console):
         mock_response = MagicMock()
         mock_http_error = requests.exceptions.HTTPError(
-            "401 Client Error: Unauthorized", response=MagicMock(status_code=401)
+            "401 Client Error: Unauthorized",
+            response=MagicMock(status_code=401)
         )
 
         mock_response.raise_for_status.side_effect = mock_http_error
@@ -213,14 +221,15 @@ class TestOrganizationCommand(unittest.TestCase):
         self.org_command.plus_api_client.get_organizations.assert_called_once()
         mock_console.print.assert_called_once_with(
             "You are not logged in to any organization. Use 'crewai login' to login.",
-            style="bold red",
+            style="bold red"
         )
 
-    @patch("crewai.cli.organization.main.console")
+    @patch('crewai.cli.organization.main.console')
     def test_switch_organization_unauthorized(self, mock_console):
         mock_response = MagicMock()
         mock_http_error = requests.exceptions.HTTPError(
-            "401 Client Error: Unauthorized", response=MagicMock(status_code=401)
+            "401 Client Error: Unauthorized",
+            response=MagicMock(status_code=401)
         )
 
         mock_response.raise_for_status.side_effect = mock_http_error
@@ -231,5 +240,5 @@ class TestOrganizationCommand(unittest.TestCase):
         self.org_command.plus_api_client.get_organizations.assert_called_once()
         mock_console.print.assert_called_once_with(
             "You are not logged in to any organization. Use 'crewai login' to login.",
-            style="bold red",
+            style="bold red"
         )

tests/cli/test_plus_api.py

@@ -1,8 +1,8 @@
+import os
 import unittest
 from unittest.mock import MagicMock, patch, ANY
 
 from crewai.cli.plus_api import PlusAPI
-from crewai.cli.constants import DEFAULT_CREWAI_ENTERPRISE_URL
 
 
 class TestPlusAPI(unittest.TestCase):
@@ -30,41 +30,29 @@ class TestPlusAPI(unittest.TestCase):
         )
         self.assertEqual(response, mock_response)
 
-    def assert_request_with_org_id(
-        self, mock_make_request, method: str, endpoint: str, **kwargs
-    ):
+    def assert_request_with_org_id(self, mock_make_request, method: str, endpoint: str, **kwargs):
         mock_make_request.assert_called_once_with(
-            method,
-            f"{DEFAULT_CREWAI_ENTERPRISE_URL}{endpoint}",
-            headers={
-                "Authorization": ANY,
-                "Content-Type": ANY,
-                "User-Agent": ANY,
-                "X-Crewai-Version": ANY,
-                "X-Crewai-Organization-Id": self.org_uuid,
-            },
-            **kwargs,
+            method, f"https://app.crewai.com{endpoint}", headers={'Authorization': ANY, 'Content-Type': ANY, 'User-Agent':  ANY, 'X-Crewai-Version': ANY, 'X-Crewai-Organization-Id': self.org_uuid}, **kwargs
         )
 
     @patch("crewai.cli.plus_api.Settings")
     @patch("requests.Session.request")
-    def test_login_to_tool_repository_with_org_uuid(
-        self, mock_make_request, mock_settings_class
-    ):
+    def test_login_to_tool_repository_with_org_uuid(self, mock_make_request, mock_settings_class):
         mock_settings = MagicMock()
         mock_settings.org_uuid = self.org_uuid
-        mock_settings.enterprise_base_url = DEFAULT_CREWAI_ENTERPRISE_URL
         mock_settings_class.return_value = mock_settings
         # re-initialize Client
         self.api = PlusAPI(self.api_key)
-
+        
         mock_response = MagicMock()
         mock_make_request.return_value = mock_response
 
         response = self.api.login_to_tool_repository()
 
         self.assert_request_with_org_id(
-            mock_make_request, "POST", "/crewai_plus/api/v1/tools/login"
+            mock_make_request,
+            'POST',
+            '/crewai_plus/api/v1/tools/login'
         )
         self.assertEqual(response, mock_response)
 
@@ -78,27 +66,28 @@ class TestPlusAPI(unittest.TestCase):
             "GET", "/crewai_plus/api/v1/agents/test_agent_handle"
         )
         self.assertEqual(response, mock_response)
-
+        
     @patch("crewai.cli.plus_api.Settings")
     @patch("requests.Session.request")
     def test_get_agent_with_org_uuid(self, mock_make_request, mock_settings_class):
         mock_settings = MagicMock()
         mock_settings.org_uuid = self.org_uuid
-        mock_settings.enterprise_base_url = DEFAULT_CREWAI_ENTERPRISE_URL
         mock_settings_class.return_value = mock_settings
         # re-initialize Client
         self.api = PlusAPI(self.api_key)
-
+        
         mock_response = MagicMock()
         mock_make_request.return_value = mock_response
 
         response = self.api.get_agent("test_agent_handle")
 
         self.assert_request_with_org_id(
-            mock_make_request, "GET", "/crewai_plus/api/v1/agents/test_agent_handle"
+            mock_make_request,
+            "GET",
+            "/crewai_plus/api/v1/agents/test_agent_handle"
         )
         self.assertEqual(response, mock_response)
-
+    
     @patch("crewai.cli.plus_api.PlusAPI._make_request")
     def test_get_tool(self, mock_make_request):
         mock_response = MagicMock()
@@ -109,13 +98,12 @@ class TestPlusAPI(unittest.TestCase):
             "GET", "/crewai_plus/api/v1/tools/test_tool_handle"
         )
         self.assertEqual(response, mock_response)
-
+        
     @patch("crewai.cli.plus_api.Settings")
     @patch("requests.Session.request")
     def test_get_tool_with_org_uuid(self, mock_make_request, mock_settings_class):
         mock_settings = MagicMock()
         mock_settings.org_uuid = self.org_uuid
-        mock_settings.enterprise_base_url = DEFAULT_CREWAI_ENTERPRISE_URL
         mock_settings_class.return_value = mock_settings
         # re-initialize Client
         self.api = PlusAPI(self.api_key)
@@ -127,7 +115,9 @@ class TestPlusAPI(unittest.TestCase):
         response = self.api.get_tool("test_tool_handle")
 
         self.assert_request_with_org_id(
-            mock_make_request, "GET", "/crewai_plus/api/v1/tools/test_tool_handle"
+            mock_make_request,
+            "GET",
+            "/crewai_plus/api/v1/tools/test_tool_handle"
         )
         self.assertEqual(response, mock_response)
 
@@ -157,13 +147,12 @@ class TestPlusAPI(unittest.TestCase):
             "POST", "/crewai_plus/api/v1/tools", json=params
         )
         self.assertEqual(response, mock_response)
-
+        
     @patch("crewai.cli.plus_api.Settings")
     @patch("requests.Session.request")
     def test_publish_tool_with_org_uuid(self, mock_make_request, mock_settings_class):
         mock_settings = MagicMock()
         mock_settings.org_uuid = self.org_uuid
-        mock_settings.enterprise_base_url = DEFAULT_CREWAI_ENTERPRISE_URL
         mock_settings_class.return_value = mock_settings
         # re-initialize Client
         self.api = PlusAPI(self.api_key)
@@ -171,7 +160,7 @@ class TestPlusAPI(unittest.TestCase):
         # Set up mock response
         mock_response = MagicMock()
         mock_make_request.return_value = mock_response
-
+        
         handle = "test_tool_handle"
         public = True
         version = "1.0.0"
@@ -191,9 +180,12 @@ class TestPlusAPI(unittest.TestCase):
             "description": description,
             "available_exports": None,
         }
-
+        
         self.assert_request_with_org_id(
-            mock_make_request, "POST", "/crewai_plus/api/v1/tools", json=expected_params
+            mock_make_request,
+            "POST",
+            "/crewai_plus/api/v1/tools",
+            json=expected_params
         )
         self.assertEqual(response, mock_response)
 
@@ -319,11 +311,8 @@ class TestPlusAPI(unittest.TestCase):
             "POST", "/crewai_plus/api/v1/crews", json=payload
         )
 
-    @patch("crewai.cli.plus_api.Settings")
-    def test_custom_base_url(self, mock_settings_class):
-        mock_settings = MagicMock()
-        mock_settings.enterprise_base_url = "https://custom-url.com/api"
-        mock_settings_class.return_value = mock_settings
+    @patch.dict(os.environ, {"CREWAI_BASE_URL": "https://custom-url.com/api"})
+    def test_custom_base_url(self):
         custom_api = PlusAPI("test_key")
         self.assertEqual(
             custom_api.base_url,

tests/cli/test_settings_command.py

@@ -1,91 +0,0 @@
-import tempfile
-import unittest
-from pathlib import Path
-from unittest.mock import patch, MagicMock, call
-
-from crewai.cli.settings.main import SettingsCommand
-from crewai.cli.config import (
-    Settings,
-    USER_SETTINGS_KEYS,
-    CLI_SETTINGS_KEYS,
-    DEFAULT_CLI_SETTINGS,
-    HIDDEN_SETTINGS_KEYS,
-    READONLY_SETTINGS_KEYS,
-)
-import shutil
-
-
-class TestSettingsCommand(unittest.TestCase):
-    def setUp(self):
-        self.test_dir = Path(tempfile.mkdtemp())
-        self.config_path = self.test_dir / "settings.json"
-        self.settings = Settings(config_path=self.config_path)
-        self.settings_command = SettingsCommand(
-            settings_kwargs={"config_path": self.config_path}
-        )
-
-    def tearDown(self):
-        shutil.rmtree(self.test_dir)
-
-    @patch("crewai.cli.settings.main.console")
-    @patch("crewai.cli.settings.main.Table")
-    def test_list_settings(self, mock_table_class, mock_console):
-        mock_table_instance = MagicMock()
-        mock_table_class.return_value = mock_table_instance
-
-        self.settings_command.list()
-
-        # Tests that the table is created skipping hidden settings
-        mock_table_instance.add_row.assert_has_calls(
-            [
-                call(
-                    field_name,
-                    getattr(self.settings, field_name) or "Not set",
-                    field_info.description,
-                )
-                for field_name, field_info in Settings.model_fields.items()
-                if field_name not in HIDDEN_SETTINGS_KEYS
-            ]
-        )
-
-        # Tests that the table is printed
-        mock_console.print.assert_called_once_with(mock_table_instance)
-
-    def test_set_valid_keys(self):
-        valid_keys = Settings.model_fields.keys() - (
-            READONLY_SETTINGS_KEYS + HIDDEN_SETTINGS_KEYS
-        )
-        for key in valid_keys:
-            test_value = f"some_value_for_{key}"
-            self.settings_command.set(key, test_value)
-            self.assertEqual(getattr(self.settings_command.settings, key), test_value)
-
-    def test_set_invalid_key(self):
-        with self.assertRaises(SystemExit):
-            self.settings_command.set("invalid_key", "value")
-
-    def test_set_readonly_keys(self):
-        for key in READONLY_SETTINGS_KEYS:
-            with self.assertRaises(SystemExit):
-                self.settings_command.set(key, "some_readonly_key_value")
-
-    def test_set_hidden_keys(self):
-        for key in HIDDEN_SETTINGS_KEYS:
-            with self.assertRaises(SystemExit):
-                self.settings_command.set(key, "some_hidden_key_value")
-
-    def test_reset_all_settings(self):
-        for key in USER_SETTINGS_KEYS + CLI_SETTINGS_KEYS:
-            setattr(self.settings_command.settings, key, f"custom_value_for_{key}")
-        self.settings_command.settings.dump()
-
-        self.settings_command.reset_all_settings()
-
-        print(USER_SETTINGS_KEYS)
-        for key in USER_SETTINGS_KEYS:
-            self.assertEqual(getattr(self.settings_command.settings, key), None)
-
-        for key in CLI_SETTINGS_KEYS:
-            self.assertEqual(
-                getattr(self.settings_command.settings, key), DEFAULT_CLI_SETTINGS[key]
-            )

tests/llm_test.py

@@ -711,3 +711,99 @@ def test_ollama_does_not_modify_when_last_is_user(ollama_llm):
     formatted = ollama_llm._format_messages_for_provider(original_messages)
 
     assert formatted == original_messages
+
+
+def test_llm_reasoning_parameter_false():
+    """Test that reasoning=False disables reasoning mode."""
+    llm = LLM(model="ollama/qwen", reasoning=False)
+    
+    with patch("litellm.completion") as mock_completion:
+        mock_message = MagicMock()
+        mock_message.content = "Test response"
+        mock_choice = MagicMock()
+        mock_choice.message = mock_message
+        mock_response = MagicMock()
+        mock_response.choices = [mock_choice]
+        mock_response.usage = {"prompt_tokens": 5, "completion_tokens": 5, "total_tokens": 10}
+        mock_completion.return_value = mock_response
+        
+        llm.call("Test message")
+        
+        _, kwargs = mock_completion.call_args
+        assert "reasoning_effort" not in kwargs
+
+
+def test_llm_reasoning_parameter_true():
+    """Test that reasoning=True enables reasoning mode."""
+    llm = LLM(model="ollama/qwen", reasoning=True, reasoning_effort="medium")
+    
+    with patch("litellm.completion") as mock_completion:
+        mock_message = MagicMock()
+        mock_message.content = "Test response"
+        mock_choice = MagicMock()
+        mock_choice.message = mock_message
+        mock_response = MagicMock()
+        mock_response.choices = [mock_choice]
+        mock_response.usage = {"prompt_tokens": 5, "completion_tokens": 5, "total_tokens": 10}
+        mock_completion.return_value = mock_response
+        
+        llm.call("Test message")
+        
+        _, kwargs = mock_completion.call_args
+        assert kwargs["reasoning_effort"] == "medium"
+
+
+def test_llm_reasoning_parameter_none_with_reasoning_effort():
+    """Test that reasoning=None with reasoning_effort still includes reasoning_effort."""
+    llm = LLM(model="ollama/qwen", reasoning=None, reasoning_effort="high")
+    
+    with patch("litellm.completion") as mock_completion:
+        mock_message = MagicMock()
+        mock_message.content = "Test response"
+        mock_choice = MagicMock()
+        mock_choice.message = mock_message
+        mock_response = MagicMock()
+        mock_response.choices = [mock_choice]
+        mock_response.usage = {"prompt_tokens": 5, "completion_tokens": 5, "total_tokens": 10}
+        mock_completion.return_value = mock_response
+        
+        llm.call("Test message")
+        
+        _, kwargs = mock_completion.call_args
+        assert kwargs["reasoning_effort"] == "high"
+
+
+def test_llm_reasoning_false_overrides_reasoning_effort():
+    """Test that reasoning=False overrides reasoning_effort."""
+    llm = LLM(model="ollama/qwen", reasoning=False, reasoning_effort="high")
+    
+    with patch("litellm.completion") as mock_completion:
+        mock_message = MagicMock()
+        mock_message.content = "Test response"
+        mock_choice = MagicMock()
+        mock_choice.message = mock_message
+        mock_response = MagicMock()
+        mock_response.choices = [mock_choice]
+        mock_response.usage = {"prompt_tokens": 5, "completion_tokens": 5, "total_tokens": 10}
+        mock_completion.return_value = mock_response
+        
+        llm.call("Test message")
+        
+        _, kwargs = mock_completion.call_args
+        assert "reasoning_effort" not in kwargs
+
+
+@pytest.mark.vcr(filter_headers=["authorization"])
+def test_ollama_qwen_with_reasoning_disabled():
+    """Test Ollama Qwen model with reasoning disabled."""
+    if not os.getenv("OLLAMA_BASE_URL"):
+        pytest.skip("OLLAMA_BASE_URL not set; skipping test.")
+    
+    llm = LLM(
+        model="ollama/qwen",
+        base_url=os.getenv("OLLAMA_BASE_URL", "http://localhost:11434"),
+        reasoning=False
+    )
+    result = llm.call("What is 2+2?")
+    assert isinstance(result, str)
+    assert len(result.strip()) > 0

tests/storage/test_mem0_storage.py

@@ -191,39 +191,17 @@ def test_save_method_with_memory_oss(mem0_storage_with_mocked_config):
     """Test save method for different memory types"""
     mem0_storage, _, _ = mem0_storage_with_mocked_config
     mem0_storage.memory.add = MagicMock()
-
+    
     # Test short_term memory type (already set in fixture)
     test_value = "This is a test memory"
     test_metadata = {"key": "value"}
-
+    
     mem0_storage.save(test_value, test_metadata)
-
+    
     mem0_storage.memory.add.assert_called_once_with(
-        [{"role": "assistant" , "content": test_value}],
+        [{'role': 'assistant' , 'content': test_value}],
         infer=True,
         metadata={"type": "short_term", "key": "value"},
-        run_id="my_run_id",
-        user_id="test_user",
-        agent_id='Test_Agent'
-    )
-
-def test_save_method_with_multiple_agents(mem0_storage_with_mocked_config):
-    mem0_storage, _, _ = mem0_storage_with_mocked_config
-    mem0_storage.crew.agents = [MagicMock(role="Test Agent"), MagicMock(role="Test Agent 2"), MagicMock(role="Test Agent 3")]
-    mem0_storage.memory.add = MagicMock()
-
-    test_value = "This is a test memory"
-    test_metadata = {"key": "value"}
-
-    mem0_storage.save(test_value, test_metadata)
-
-    mem0_storage.memory.add.assert_called_once_with(
-        [{"role": "assistant" , "content": test_value}],
-        infer=True,
-        metadata={"type": "short_term", "key": "value"},
-        run_id="my_run_id",
-        user_id="test_user",
-        agent_id='Test_Agent_Test_Agent_2_Test_Agent_3'
     )
 
 
@@ -231,13 +209,13 @@ def test_save_method_with_memory_client(mem0_storage_with_memory_client_using_co
     """Test save method for different memory types"""
     mem0_storage = mem0_storage_with_memory_client_using_config_from_crew
     mem0_storage.memory.add = MagicMock()
-
+    
     # Test short_term memory type (already set in fixture)
     test_value = "This is a test memory"
     test_metadata = {"key": "value"}
-
+    
     mem0_storage.save(test_value, test_metadata)
-
+    
     mem0_storage.memory.add.assert_called_once_with(
         [{'role': 'assistant' , 'content': test_value}],
         infer=True,
@@ -246,9 +224,7 @@ def test_save_method_with_memory_client(mem0_storage_with_memory_client_using_co
         run_id="my_run_id",
         includes="include1",
         excludes="exclude1",
-        output_format='v1.1',
-        user_id='test_user',
-         agent_id='Test_Agent'
+        output_format='v1.1'
     )
 
 
@@ -261,10 +237,10 @@ def test_search_method_with_memory_oss(mem0_storage_with_mocked_config):
     results = mem0_storage.search("test query", limit=5, score_threshold=0.5)
 
     mem0_storage.memory.search.assert_called_once_with(
-        query="test query",
-        limit=5,
+        query="test query", 
+        limit=5, 
         user_id="test_user",
-        filters={'AND': [{'run_id': 'my_run_id'}]},
+        filters={'AND': [{'run_id': 'my_run_id'}]}, 
         threshold=0.5
     )
 
@@ -281,8 +257,8 @@ def test_search_method_with_memory_client(mem0_storage_with_memory_client_using_
     results = mem0_storage.search("test query", limit=5, score_threshold=0.5)
 
     mem0_storage.memory.search.assert_called_once_with(
-        query="test query",
-        limit=5,
+        query="test query", 
+        limit=5, 
         metadata={"type": "short_term"},
         user_id="test_user",
         version='v2',
@@ -310,108 +286,4 @@ def test_mem0_storage_default_infer_value(mock_mem0_memory_client):
         )
 
         mem0_storage = Mem0Storage(type="short_term", crew=crew)
-        assert mem0_storage.infer is True
-
-def test_save_memory_using_agent_entity(mock_mem0_memory_client):
-    config = {
-        "agent_id": "agent-123",
-    }
-
-    mock_memory = MagicMock(spec=Memory)
-    with patch.object(Memory, "__new__", return_value=mock_memory):
-        mem0_storage = Mem0Storage(type="external", config=config)
-        mem0_storage.save("test memory", {"key": "value"})
-        mem0_storage.memory.add.assert_called_once_with(
-            [{'role': 'assistant' , 'content': 'test memory'}],
-            infer=True,
-            metadata={"type": "external", "key": "value"},
-            agent_id="agent-123",
-        )
-
-def test_search_method_with_agent_entity():
-    mem0_storage = Mem0Storage(type="external", config={"agent_id": "agent-123"})
-    mock_results = {"results": [{"score": 0.9, "content": "Result 1"}, {"score": 0.4, "content": "Result 2"}]}
-    mem0_storage.memory.search = MagicMock(return_value=mock_results)
-
-    results = mem0_storage.search("test query", limit=5, score_threshold=0.5)
-
-    mem0_storage.memory.search.assert_called_once_with(
-        query="test query",
-        limit=5,
-        filters={"AND": [{"agent_id": "agent-123"}]},
-        threshold=0.5,
-    )
-
-    assert len(results) == 2
-    assert results[0]["content"] == "Result 1"
-
-
-def test_search_method_with_agent_id_and_user_id():
-    mem0_storage = Mem0Storage(type="external", config={"agent_id": "agent-123", "user_id": "user-123"})
-    mock_results = {"results": [{"score": 0.9, "content": "Result 1"}, {"score": 0.4, "content": "Result 2"}]}
-    mem0_storage.memory.search = MagicMock(return_value=mock_results)
-
-    results = mem0_storage.search("test query", limit=5, score_threshold=0.5)
-
-    mem0_storage.memory.search.assert_called_once_with(
-        query="test query",
-        limit=5,
-        user_id='user-123',
-        filters={"OR": [{"user_id": "user-123"}, {"agent_id": "agent-123"}]},
-        threshold=0.5,
-    )
-
-    assert len(results) == 2
-    assert results[0]["content"] == "Result 1"
-
-
-def test_search_method_with_memory_oss_missing_params():
-    """Test search method handles missing parameters gracefully when using Memory (OSS)"""
-    config = {
-        "user_id": "test_user",
-        "local_mem0_config": {
-            "vector_store": {"provider": "mock_vector_store"},
-            "llm": {"provider": "mock_llm"},
-        }
-    }
-    
-    mock_memory = MagicMock(spec=Memory)
-    mock_results = {"results": [{"score": 0.9, "content": "Result 1"}]}
-    mock_memory.search = MagicMock(return_value=mock_results)
-    
-    with patch("mem0.memory.main.Memory.from_config", return_value=mock_memory):
-        mem0_storage = Mem0Storage(type="external", config=config)
-        
-        results = mem0_storage.search("test query", limit=5, score_threshold=0.5)
-        
-        mock_memory.search.assert_called_once_with(
-            query="test query",
-            limit=5,
-            user_id="test_user",
-            filters={"AND": [{"user_id": "test_user"}]},
-            threshold=0.5
-        )
-        
-        assert len(results) == 1
-        assert results[0]["content"] == "Result 1"
-
-
-def test_search_method_with_memory_oss_no_optional_params():
-    """Test search method works when no optional parameters are present"""
-    mock_memory = MagicMock(spec=Memory)
-    mock_results = {"results": []}
-    mock_memory.search = MagicMock(return_value=mock_results)
-    
-    with patch("mem0.memory.main.Memory", return_value=mock_memory):
-        mem0_storage = Mem0Storage(type="external", config={})
-        
-        results = mem0_storage.search("test query")
-        
-        mock_memory.search.assert_called_once_with(
-            query="test query",
-            limit=3,
-            filters={},
-            threshold=0.35
-        )
-        
-        assert len(results) == 0
+        assert mem0_storage.infer is True