klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: do not mock get (#598)

Browse Source
This commit is contained in:
cachho
2023-09-12 19:12:17 +02:00
committed by GitHub
parent 03146946fa
commit ac08638a63
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/embedchain/test_embedchain.py
View File

@@ -10,13 +10,11 @@ class TestChromaDbHostsLoglevel(unittest.TestCase):
os.environ["OPENAI_API_KEY"] = "test_key" os.environ["OPENAI_API_KEY"] = "test_key"
@patch("chromadb.api.models.Collection.Collection.add") @patch("chromadb.api.models.Collection.Collection.add")
@patch("chromadb.api.models.Collection.Collection.get")
@patch("embedchain.embedchain.EmbedChain.retrieve_from_database") @patch("embedchain.embedchain.EmbedChain.retrieve_from_database")
@patch("embedchain.llm.base.BaseLlm.get_answer_from_llm") @patch("embedchain.llm.base.BaseLlm.get_answer_from_llm")
@patch("embedchain.llm.base.BaseLlm.get_llm_model_answer") @patch("embedchain.llm.base.BaseLlm.get_llm_model_answer")
def test_whole_app( def test_whole_app(
self, self,
_mock_get,
_mock_add, _mock_add,
_mock_ec_retrieve_from_database, _mock_ec_retrieve_from_database,
_mock_get_answer_from_llm, _mock_get_answer_from_llm,