From fc86b507b4e7fe7c47043b5b9676ca93ed5d06ed Mon Sep 17 00:00:00 2001 From: Nolan Tremelling <34580718+NolanTrem@users.noreply.github.com> Date: Tue, 22 Aug 2023 22:55:45 -0400 Subject: [PATCH] Automatabot run/8 14 2023 (#495) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * Diff results test_cli_scripts_run_doc_embedding.py * Manual fix—missed on instance of variable --- .../tests/unit/cli/test_cli_scripts_run_doc_embedding.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/automata/tests/unit/cli/test_cli_scripts_run_doc_embedding.py b/automata/tests/unit/cli/test_cli_scripts_run_doc_embedding.py index 039c0b157..80e29f658 100644 --- a/automata/tests/unit/cli/test_cli_scripts_run_doc_embedding.py +++ b/automata/tests/unit/cli/test_cli_scripts_run_doc_embedding.py @@ -56,14 +56,14 @@ def symbol_doc_embedding_handler_mock(): def test_initialize_providers( get_mock, set_overrides_mock, - OpenAIEmbeddingProvider_mock, + openai_embedding_provider_mock, ChromaSymbolEmbeddingVectorDatabase_mock, SymbolGraph_mock, symbol_graph_mock, ): SymbolGraph_mock.return_value = symbol_graph_mock ChromaSymbolEmbeddingVectorDatabase_mock.return_value = MagicMock() - OpenAIEmbeddingProvider_mock.return_value = MagicMock() + openai_embedding_provider_mock.return_value.return_value = MagicMock() symbol_code_embedding_handler = MagicMock() symbol_code_embedding_handler._get_sorted_supported_symbols.return_value = [ FakeSymbol("symbol1"), @@ -91,7 +91,7 @@ def test_initialize_providers( "symbol_graph": symbol_graph_mock, "code_embedding_db": ChromaSymbolEmbeddingVectorDatabase_mock.return_value, "doc_embedding_db": ChromaSymbolEmbeddingVectorDatabase_mock.return_value, - "embedding_provider": OpenAIEmbeddingProvider_mock.return_value, + "embedding_provider": openai_embedding_provider_mock.return_value, "disable_synchronization": True, } set_overrides_mock.assert_called_once_with(**overrides)