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

fix tets

Browse Source
This commit is contained in:
Brandon Hancock
2025-01-02 15:24:57 -05:00
parent 9429253ac2
commit 1d3b4fde04
3 changed files with 24 additions and 859 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/agent_test.py
View File

@@ -1457,7 +1457,7 @@ def test_agent_with_ollama_llama3():
assert agent.llm.model == "ollama/llama3.2:3b"
assert agent.llm.base_url == "http://localhost:11434"
task = "Respond in 20 words. Who which model are you?"
task = "Respond in 20 words. Which model are you?"
response = agent.llm.call([{"role": "user", "content": task}])
assert response
@@ -1473,7 +1473,9 @@ def test_llm_call_with_ollama_llama3():
temperature=0.7,
max_tokens=30,
)
messages = [{"role": "user", "content": "Respond in 20 words. Who are you?"}]
messages = [
{"role": "user", "content": "Respond in 20 words. Which model are you??"}
]
response = llm.call(messages)