File tree Expand file tree Collapse file tree 1 file changed +0
-4
lines changed Expand file tree Collapse file tree 1 file changed +0
-4
lines changed Original file line number Diff line number Diff line change @@ -140,19 +140,16 @@ def test_unknown_model_return_error_on_chat_completion(client_with_settings_over
140140 assert response .status_code == 404
141141
142142
143- # Test for get_memory_instance
144143@patch ('dialog.llm.agents.lcel.get_session' )
145144@patch ('dialog.llm.agents.lcel.generate_memory_instance' )
146145def test_get_memory_instance (mock_generate_memory_instance , mock_get_session ):
147146 mock_get_session .return_value = iter ([MagicMock ()])
148147 lcel .get_memory_instance ('test_session' )
149148 mock_generate_memory_instance .assert_called_once ()
150149
151- # Test for retriever
152150def test_retriever ():
153151 assert isinstance (lcel .retriever , DialogRetriever )
154152
155- # Test for format_docs
156153def test_format_docs ():
157154 docs = [Document (page_content = 'doc1' ), Document (page_content = 'doc2' )]
158155 result = lcel .format_docs (docs )
@@ -185,7 +182,6 @@ def psql_memory():
185182
186183
187184async def test_gen ():
188- #import gen from dialog/src/dialog/routers/openai and test it
189185 from src .dialog .routers .openai import ask_question_to_llm
190186
191187 _g = ask_question_to_llm (message = "Hello" )
You can’t perform that action at this time.
0 commit comments