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

fixes from discussion

Browse Source
This commit is contained in:
Lorenze Jay
2024-11-27 10:38:20 -08:00
parent 3f87bf3ada
commit 5b03d6c8bc
7 changed files with 119 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/agent_test.py
View File

@@ -3,7 +3,6 @@
import os
from unittest import mock
from unittest.mock import patch
import pytest
from crewai import Agent, Crew, Task
@@ -11,7 +10,6 @@ from crewai.knowledge.source.base_knowledge_source import BaseKnowledgeSource
from crewai.agents.cache import CacheHandler
from crewai.agents.crew_agent_executor import CrewAgentExecutor
from crewai.agents.parser import AgentAction, CrewAgentParser, OutputParserException
from crewai.knowledge.knowledge import Knowledge
from crewai.llm import LLM
from crewai.tools import tool
from crewai.tools.tool_calling import InstructorToolCalling
@@ -1627,10 +1625,9 @@ def test_agent_with_knowledge_sources_context():
goal="Provide information based on knowledge sources",
backstory="You have access to specific knowledge sources.",
llm=LLM(model="gpt-4o-mini"),
knowledge=[string_source],
knowledge_sources=[string_source],
)
# Test that agent is properly initialized with knowledge sources
assert isinstance(agent.knowledge, Knowledge)
assert len(agent.knowledge.sources) == 1
assert isinstance(agent.knowledge.sources[0], BaseKnowledgeSource)
assert agent.knowledge.sources[0].metadata == {"preference": "personal"}
assert isinstance(agent.knowledge_sources, list)
assert len(agent.knowledge_sources) == 1
assert isinstance(agent.knowledge_sources[0], BaseKnowledgeSource)
assert agent.knowledge_sources[0].metadata == {"preference": "personal"}