andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 16:48:30 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Adding better support for open source tool calling models (#952)

Browse Source 
* Adding better support for open source tool calling models

* making sure the right tool is called

* fixing tests

* better support opensource models
This commit is contained in:
João Moura
2024-07-17 01:54:13 -07:00
committed by GitHub
parent d4d9d475b0
commit a1b37f073d
10 changed files with 5642 additions and 909 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/crew_test.py
View File

@@ -1355,6 +1355,7 @@ def test_crew_inputs_interpolate_both_agents_and_tasks_diff():
interpolate_task_inputs.assert_called()
@pytest.mark.vcr(filter_headers=["authorization"])
def test_crew_does_not_interpolate_without_inputs():
from unittest.mock import patch
@@ -1997,6 +1998,7 @@ def test_replay_task_with_context():
db_handler.reset()
@pytest.mark.vcr(filter_headers=["authorization"])
def test_replay_from_task_with_context():
agent = Agent(role="test_agent", backstory="Test Description", goal="Test Goal")
task1 = Task(
@@ -2114,6 +2116,7 @@ def test_replay_with_invalid_task_id():
crew.replay_from_task("bf5b09c9-69bd-4eb8-be12-f9e5bae31c2d")
@pytest.mark.vcr(filter_headers=["authorization"])
@patch.object(Crew, "_interpolate_inputs")
def test_replay_interpolates_inputs_properly(mock_interpolate_inputs):
agent = Agent(role="test_agent", backstory="Test Description", goal="Test Goal")