andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-03 00:02:36 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

fix: pass tools to tests to simulate real scenario

Browse Source 
The tests were failing because they didn't pass tools to the call()
method, so self.tools was None and the function call processing
was skipped. In real usage, tools are always configured when
function calls are expected.

Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
Devin AI
2025-12-16 08:50:29 +00:00
parent 4938654ede
commit dac1010356
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
lib/crewai/tests/llms/google/test_google.py
View File

@@ -768,8 +768,11 @@ def test_gemini_function_call_without_text_returns_action_format():
mock_generate.return_value = mock_response mock_generate.return_value = mock_response
# Pass tools to simulate real scenario where tools are configured
# This sets self.tools in the completion object
tools = [{"name": "search_tool", "description": "Search for information"}]
messages = [{"role": "user", "content": "What's the weather in Tokyo?"}] messages = [{"role": "user", "content": "What's the weather in Tokyo?"}]
result = completion.call(messages=messages) result = completion.call(messages=messages, tools=tools)
assert result is not None assert result is not None
assert result != "" assert result != ""
@@ -817,8 +820,10 @@ def test_gemini_function_call_with_text_returns_text():
mock_generate.return_value = mock_response mock_generate.return_value = mock_response
# Pass tools to simulate real scenario where tools are configured
tools = [{"name": "search_tool", "description": "Search for information"}]
messages = [{"role": "user", "content": "What's the weather?"}] messages = [{"role": "user", "content": "What's the weather?"}]
result = completion.call(messages=messages) result = completion.call(messages=messages, tools=tools)
assert result is not None assert result is not None
assert result != "" assert result != ""
@@ -866,9 +871,12 @@ def test_gemini_function_call_with_available_functions_executes_tool():
mock_generate.return_value = mock_response mock_generate.return_value = mock_response
# Pass tools to simulate real scenario where tools are configured
tools = [{"name": "search_tool", "description": "Search for information"}]
messages = [{"role": "user", "content": "What's the weather in Tokyo?"}] messages = [{"role": "user", "content": "What's the weather in Tokyo?"}]
result = completion.call( result = completion.call(
messages=messages, messages=messages,
tools=tools,
available_functions=available_functions available_functions=available_functions
) )
@@ -910,8 +918,10 @@ def test_gemini_streaming_function_call_without_text_returns_action_format():
mock_stream.return_value = [mock_chunk] mock_stream.return_value = [mock_chunk]
# Pass tools to simulate real scenario where tools are configured
tools = [{"name": "calculator", "description": "Calculate expressions"}]
messages = [{"role": "user", "content": "Calculate 2 + 2"}] messages = [{"role": "user", "content": "Calculate 2 + 2"}]
result = completion.call(messages=messages) result = completion.call(messages=messages, tools=tools)
assert result is not None assert result is not None
assert result != "" assert result != ""