Fixes testing

Author: vmesel

sample_data/prompt.toml

@@ -24,9 +24,4 @@ Answer the following message to the user.
 [[endpoint]]
 path = "/lerolero"
 model_name = "lerolero"
-model_class_path = "dialog.llm.agents.lcel.runnable"
-
-[[endpoint]]
-path = "/lerolero2"
-model_name = "lerolero2"
-model_class_path = "dialog.llm.agents.default.DialogLLM"
+model_class_path = "dialog.llm.agents.lcel.runnable"

src/tests/test_views.py

@@ -80,8 +80,8 @@ def test_customized_openai_chat_completion_response_stream_false(client, llm_moc
 def test_multiple_models_load_on_setting_override(client_with_settings_override):
     response = client_with_settings_override.get("/openapi.json")
     assert response.status_code == 200
-    assert "/custom_model/chat/{chat_id}" in list(response.json().get("paths").keys())
-    assert "/custom_model/ask" in list(response.json().get("paths").keys())
+    assert "/lerolero/chat/{chat_id}" in list(response.json().get("paths").keys())
+    assert "/lerolero/ask" in list(response.json().get("paths").keys())
 
 def test_multiple_models_are_available_on_model_listing_for_webui(client_with_settings_override):
     response = client_with_settings_override.get("/openai/models")