Skip to content

Commit

Permalink
Remove unecessary tools
Browse files Browse the repository at this point in the history
  • Loading branch information
Dev-Khant committed Oct 22, 2024
1 parent 078aa66 commit 5a04e70
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 91 deletions.
57 changes: 0 additions & 57 deletions mem0/llms/utils/tools.py

This file was deleted.

34 changes: 0 additions & 34 deletions tests/llms/test_ollama.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@

from mem0.configs.llms.base import BaseLlmConfig
from mem0.llms.ollama import OllamaLLM
from mem0.llms.utils.tools import ADD_MEMORY_TOOL


@pytest.fixture
Expand Down Expand Up @@ -33,36 +32,3 @@ def test_generate_response_without_tools(mock_ollama_client):
model="llama3.1:70b", messages=messages, options={"temperature": 0.7, "num_predict": 100, "top_p": 1.0}
)
assert response == "I'm doing well, thank you for asking!"


def test_generate_response_with_tools(mock_ollama_client):
config = BaseLlmConfig(model="llama3.1:70b", temperature=0.7, max_tokens=100, top_p=1.0)
llm = OllamaLLM(config)
messages = [
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "Add a new memory: Today is a sunny day."},
]
tools = [ADD_MEMORY_TOOL]

mock_response = {
"message": {
"content": "I've added the memory for you.",
"tool_calls": [{"function": {"name": "add_memory", "arguments": {"data": "Today is a sunny day."}}}],
}
}

mock_ollama_client.chat.return_value = mock_response

response = llm.generate_response(messages, tools=tools)

mock_ollama_client.chat.assert_called_once_with(
model="llama3.1:70b",
messages=messages,
options={"temperature": 0.7, "num_predict": 100, "top_p": 1.0},
tools=tools,
)

assert response["content"] == "I've added the memory for you."
assert len(response["tool_calls"]) == 1
assert response["tool_calls"][0]["name"] == "add_memory"
assert response["tool_calls"][0]["arguments"] == {"data": "Today is a sunny day."}

0 comments on commit 5a04e70

Please sign in to comment.