andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-24 15:48:23 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

updating specs

Browse Source
This commit is contained in:
João Moura
2023-12-20 17:20:55 -03:00
parent 8c785b2c72
commit e0270d6a68
3 changed files with 23 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/agent_test.py
View File

@@ -46,6 +46,7 @@ def test_custom_llm():
assert agent.llm.model_name == "gpt-4"
assert agent.llm.temperature == 0
@pytest.mark.vcr()
def test_agent_without_memory():
no_memory_agent = Agent(
role="test role",
@@ -71,7 +72,7 @@ def test_agent_without_memory():
result = no_memory_agent.execute_task("How much is 1 + 1?")
assert result == "2"
assert result == "1 + 1 equals 2."
assert no_memory_agent.agent_executor.memory is None
assert memory_agent.agent_executor.memory is not None
@@ -85,7 +86,7 @@ def test_agent_execution():
)
output = agent.execute_task("How much is 1 + 1?")
assert output == "1 + 1 equals 2."
assert output == "2"
@pytest.mark.vcr()
def test_agent_execution_with_tools():
@@ -93,9 +94,9 @@ def test_agent_execution_with_tools():
@tool
def multiplier(numbers) -> float:
"""Useful for when you need to multiply two numbers together.
The input to this tool should be a comma separated list of numbers of
length two, representing the two numbers you want to multiply together.
"""Useful for when you need to multiply two numbers together.
The input to this tool should be a comma separated list of numbers of
length two, representing the two numbers you want to multiply together.
For example, `1,2` would be the input if you wanted to multiply 1 by 2."""
a, b = numbers.split(',')
return int(a) * int(b)
@@ -109,7 +110,7 @@ def test_agent_execution_with_tools():
)
output = agent.execute_task("What is 3 times 4")
assert output == "3 times 4 equals to 12."
assert output == "12"
@pytest.mark.vcr()
def test_agent_execution_with_specific_tools():
@@ -117,9 +118,9 @@ def test_agent_execution_with_specific_tools():
@tool
def multiplier(numbers) -> float:
"""Useful for when you need to multiply two numbers together.
The input to this tool should be a comma separated list of numbers of
length two, representing the two numbers you want to multiply together.
"""Useful for when you need to multiply two numbers together.
The input to this tool should be a comma separated list of numbers of
length two, representing the two numbers you want to multiply together.
For example, `1,2` would be the input if you wanted to multiply 1 by 2."""
a, b = numbers.split(',')
return int(a) * int(b)