more fixes

src/crewai/knowledge/storage/knowledge_storage.py

@@ -177,6 +177,19 @@ class KnowledgeStorage(BaseKnowledgeStorage):
         except Exception as e:
             Logger(verbose=True).log("error", f"Failed to upsert documents: {e}", "red")
             raise
+
+    def _create_default_embedding_function(self):
+        from chromadb.utils.embedding_functions.openai_embedding_function import (
+            OpenAIEmbeddingFunction,
+        )
+
+        return OpenAIEmbeddingFunction(
+            api_key=os.getenv("OPENAI_API_KEY"), model_name="text-embedding-3-small"
+        )
+
+    def _set_embedder_config(
+        self, embedder_config: Optional[Dict[str, Any]] = None
+    ) -> None:
         """Set the embedding configuration for the knowledge storage.
 
         Args:

tests/agent_test.py

@@ -1592,18 +1592,18 @@ def test_agent_execute_task_with_ollama():
     assert "AI" in result or "artificial intelligence" in result.lower()
 
 
-@pytest.mark.vcr(filter_headers=["authorization"])
+# @pytest.mark.vcr(filter_headers=["authorization"])
 def test_agent_with_knowledge_sources():
     # Create a knowledge source with some content
     content = "Brandon's favorite color is red and he likes Mexican food."
     string_source = StringKnowledgeSource(content=content)
 
-    # with patch(
-    #     "crewai.knowledge.storage.knowledge_storage.KnowledgeStorage"
-    # ) as MockKnowledge:
-    #     mock_knowledge_instance = MockKnowledge.return_value
-    #     mock_knowledge_instance.sources = [string_source]
-    #     mock_knowledge_instance.query.return_value = [{"content": content}]
+    with patch(
+        "crewai.knowledge.storage.knowledge_storage.KnowledgeStorage"
+    ) as MockKnowledge:
+        mock_knowledge_instance = MockKnowledge.return_value
+        mock_knowledge_instance.sources = [string_source]
+        mock_knowledge_instance.query.return_value = [{"content": content}]
 
         agent = Agent(
             role="Information Agent",