lightspeed-stack 0.1.1__tar.gz → 0.1.2__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (128) hide show
  1. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/PKG-INFO +10 -2
  2. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/README.md +1 -1
  3. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/pyproject.toml +17 -1
  4. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/query.py +4 -2
  5. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/endpoints/test_query.py +17 -7
  6. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/LICENSE +0 -0
  7. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/__init__.py +0 -0
  8. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/.ruff_cache/.gitignore +0 -0
  9. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/.ruff_cache/0.9.1/5703048272820174433 +0 -0
  10. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/.ruff_cache/0.9.1/9961612457335986079 +0 -0
  11. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/.ruff_cache/CACHEDIR.TAG +0 -0
  12. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/__init__.py +0 -0
  13. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/authorized.py +0 -0
  14. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/config.py +0 -0
  15. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/conversations.py +0 -0
  16. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/feedback.py +0 -0
  17. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/health.py +0 -0
  18. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/info.py +0 -0
  19. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/metrics.py +0 -0
  20. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/models.py +0 -0
  21. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/root.py +0 -0
  22. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/endpoints/streaming_query.py +0 -0
  23. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/main.py +0 -0
  24. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/app/routers.py +0 -0
  25. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/auth/__init__.py +0 -0
  26. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/auth/interface.py +0 -0
  27. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/auth/jwk_token.py +0 -0
  28. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/auth/k8s.py +0 -0
  29. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/auth/noop.py +0 -0
  30. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/auth/noop_with_token.py +0 -0
  31. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/auth/utils.py +0 -0
  32. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/metrics/__init__.py +0 -0
  33. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/metrics/utils.py +0 -0
  34. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/models/__init__.py +0 -0
  35. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/models/config.py +0 -0
  36. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/models/requests.py +0 -0
  37. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/models/responses.py +0 -0
  38. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/runners/__init__.py +0 -0
  39. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/runners/data_collector.py +0 -0
  40. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/runners/uvicorn.py +0 -0
  41. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/services/__init__.py +0 -0
  42. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/services/data_collector.py +0 -0
  43. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/utils/.ruff_cache/.gitignore +0 -0
  44. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/utils/.ruff_cache/0.9.1/18446581155718949728 +0 -0
  45. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/utils/.ruff_cache/0.9.1/4991844299736624256 +0 -0
  46. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/utils/.ruff_cache/CACHEDIR.TAG +0 -0
  47. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/utils/__init__.py +0 -0
  48. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/utils/checks.py +0 -0
  49. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/utils/common.py +0 -0
  50. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/utils/endpoints.py +0 -0
  51. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/utils/mcp_headers.py +0 -0
  52. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/utils/suid.py +0 -0
  53. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/src/utils/types.py +0 -0
  54. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/__init__.py +0 -0
  55. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/configuration/lightspeed-stack.yaml +0 -0
  56. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/configuration/minimal-stack.yaml +0 -0
  57. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/configuration/password +0 -0
  58. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/configuration/run.yaml +0 -0
  59. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/configuration/server.crt +0 -0
  60. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/configuration/server.key +0 -0
  61. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/.pdm-python +0 -0
  62. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/.ruff_cache/.gitignore +0 -0
  63. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/.ruff_cache/0.9.1/6949908709306621198 +0 -0
  64. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/.ruff_cache/CACHEDIR.TAG +0 -0
  65. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/__init__.py +0 -0
  66. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/features/environment.py +0 -0
  67. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/features/llm_interface.feature +0 -0
  68. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/features/rest_api.feature +0 -0
  69. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/features/smoketests.feature +0 -0
  70. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/features/steps/__init__.py +0 -0
  71. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/features/steps/common.py +0 -0
  72. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/features/steps/common_http.py +0 -0
  73. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/features/steps/llm_query_response.py +0 -0
  74. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/gen_scenario_list.py +0 -0
  75. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/test_api.py +0 -0
  76. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/test_list.txt +0 -0
  77. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/e2e/utils/utils.py +0 -0
  78. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/integration/__init__.py +0 -0
  79. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/integration/test_configuration.py +0 -0
  80. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/integration/test_version.py +0 -0
  81. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/test_results/.coverage.integration +0 -0
  82. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/test_results/.coverage.unit +0 -0
  83. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/test_results/coverage_integration.json +0 -0
  84. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/test_results/coverage_unit.json +0 -0
  85. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/test_results/junit_integration.xml +0 -0
  86. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/test_results/junit_unit.xml +0 -0
  87. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/__init__.py +0 -0
  88. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/__init__.py +0 -0
  89. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/endpoints/__init__.py +0 -0
  90. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/endpoints/test_authorized.py +0 -0
  91. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/endpoints/test_config.py +0 -0
  92. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/endpoints/test_conversations.py +0 -0
  93. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/endpoints/test_feedback.py +0 -0
  94. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/endpoints/test_health.py +0 -0
  95. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/endpoints/test_info.py +0 -0
  96. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/endpoints/test_metrics.py +0 -0
  97. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/endpoints/test_models.py +0 -0
  98. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/endpoints/test_root.py +0 -0
  99. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/endpoints/test_streaming_query.py +0 -0
  100. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/app/test_routers.py +0 -0
  101. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/auth/__init__.py +0 -0
  102. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/auth/test_auth.py +0 -0
  103. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/auth/test_jwk_token.py +0 -0
  104. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/auth/test_k8s.py +0 -0
  105. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/auth/test_noop.py +0 -0
  106. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/auth/test_noop_with_token.py +0 -0
  107. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/auth/test_utils.py +0 -0
  108. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/metrics/__init__.py +0 -0
  109. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/metrics/test_utis.py +0 -0
  110. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/models/__init__.py +0 -0
  111. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/models/test_config.py +0 -0
  112. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/models/test_requests.py +0 -0
  113. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/models/test_responses.py +0 -0
  114. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/runners/__init__.py +0 -0
  115. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/runners/test_data_collector_runner.py +0 -0
  116. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/runners/test_uvicorn_runner.py +0 -0
  117. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/services/test_data_collector.py +0 -0
  118. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/test_client.py +0 -0
  119. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/test_configuration.py +0 -0
  120. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/test_lightspeed_stack.py +0 -0
  121. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/test_log.py +0 -0
  122. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/utils/__init__.py +0 -0
  123. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/utils/test_checks.py +0 -0
  124. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/utils/test_common.py +0 -0
  125. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/utils/test_endpoints.py +0 -0
  126. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/utils/test_mcp_headers.py +0 -0
  127. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/utils/test_suid.py +0 -0
  128. {lightspeed_stack-0.1.1 → lightspeed_stack-0.1.2}/tests/unit/utils/test_types.py +0 -0
@@ -1,7 +1,9 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: lightspeed-stack
3
- Version: 0.1.1
3
+ Version: 0.1.2
4
4
  Summary: LLM tooling stack
5
+ Keywords: LLM,RAG
6
+ Maintainer-Email: =?utf-8?b?UGF2ZWwgVGnFoW5vdnNrw70=?= <tisnik@centrum.cz>
5
7
  License: Apache License
6
8
  Version 2.0, January 2004
7
9
  http://www.apache.org/licenses/
@@ -204,6 +206,12 @@ License: Apache License
204
206
  See the License for the specific language governing permissions and
205
207
  limitations under the License.
206
208
 
209
+ Classifier: Development Status :: 4 - Beta
210
+ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
211
+ Classifier: Topic :: Software Development
212
+ Classifier: Programming Language :: Python :: 3
213
+ Classifier: Programming Language :: Python :: 3 :: Only
214
+ Classifier: License :: OSI Approved :: Apache Software License
207
215
  Project-URL: Homepage, https://github.com/lightspeed-core/lightspeed-stack
208
216
  Project-URL: Issues, https://github.com/lightspeed-core/lightspeed-stack/issues
209
217
  Requires-Python: <3.14,>=3.12
@@ -228,7 +236,7 @@ Description-Content-Type: text/markdown
228
236
  [![License](https://img.shields.io/badge/license-Apache-blue)](https://github.com/lightspeed-core/lightspeed-stack/blob/main/LICENSE)
229
237
  [![made-with-python](https://img.shields.io/badge/Made%20with-Python-1f425f.svg)](https://www.python.org/)
230
238
  [![Required Python version](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Flightspeed-core%2Flightspeed-stack%2Frefs%2Fheads%2Fmain%2Fpyproject.toml)](https://www.python.org/)
231
- [![Tag](https://img.shields.io/github/v/tag/lightspeed-core/lightspeed-stack)](https://github.com/lightspeed-core/lightspeed-stack/releases/tag/0.1.1)
239
+ [![Tag](https://img.shields.io/github/v/tag/lightspeed-core/lightspeed-stack)](https://github.com/lightspeed-core/lightspeed-stack/releases/tag/0.1.2)
232
240
 
233
241
  Lightspeed Core Stack (LCS) is an AI-powered assistant that provides answers to product questions using backend LLM services, agents, and RAG databases.
234
242
 
@@ -6,7 +6,7 @@
6
6
  [![License](https://img.shields.io/badge/license-Apache-blue)](https://github.com/lightspeed-core/lightspeed-stack/blob/main/LICENSE)
7
7
  [![made-with-python](https://img.shields.io/badge/Made%20with-Python-1f425f.svg)](https://www.python.org/)
8
8
  [![Required Python version](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Flightspeed-core%2Flightspeed-stack%2Frefs%2Fheads%2Fmain%2Fpyproject.toml)](https://www.python.org/)
9
- [![Tag](https://img.shields.io/github/v/tag/lightspeed-core/lightspeed-stack)](https://github.com/lightspeed-core/lightspeed-stack/releases/tag/0.1.1)
9
+ [![Tag](https://img.shields.io/github/v/tag/lightspeed-core/lightspeed-stack)](https://github.com/lightspeed-core/lightspeed-stack/releases/tag/0.1.2)
10
10
 
11
11
  Lightspeed Core Stack (LCS) is an AI-powered assistant that provides answers to product questions using backend LLM services, agents, and RAG databases.
12
12
 
@@ -3,6 +3,21 @@ name = "lightspeed-stack"
3
3
  dynamic = []
4
4
  description = "LLM tooling stack"
5
5
  authors = []
6
+ maintainers = [
7
+ { email = "tisnik@centrum.cz", name = "Pavel Tišnovský" },
8
+ ]
9
+ classifiers = [
10
+ "Development Status :: 4 - Beta",
11
+ "Topic :: Scientific/Engineering :: Artificial Intelligence",
12
+ "Topic :: Software Development",
13
+ "Programming Language :: Python :: 3",
14
+ "Programming Language :: Python :: 3 :: Only",
15
+ "License :: OSI Approved :: Apache Software License",
16
+ ]
17
+ keywords = [
18
+ "LLM",
19
+ "RAG",
20
+ ]
6
21
  requires-python = ">=3.12,<3.14"
7
22
  readme = "README.md"
8
23
  dependencies = [
@@ -18,7 +33,7 @@ dependencies = [
18
33
  "aiohttp>=3.12.14",
19
34
  "authlib>=1.6.0",
20
35
  ]
21
- version = "0.1.1"
36
+ version = "0.1.2"
22
37
 
23
38
  [project.license]
24
39
  file = "LICENSE"
@@ -77,6 +92,7 @@ dev = [
77
92
  "behave>=1.2.6",
78
93
  "types-cachetools>=6.1.0.20250717",
79
94
  "build>=1.2.2.post1",
95
+ "twine>=6.1.0",
80
96
  ]
81
97
  build = [
82
98
  "build>=1.2.2.post1",
@@ -216,10 +216,12 @@ def select_model_and_provider_id(
216
216
  },
217
217
  ) from e
218
218
 
219
+ llama_stack_model_id = f"{provider_id}/{model_id}"
219
220
  # Validate that the model_id and provider_id are in the available models
220
221
  logger.debug("Searching for model: %s, provider: %s", model_id, provider_id)
221
222
  if not any(
222
- m.identifier == model_id and m.provider_id == provider_id for m in models
223
+ m.identifier == llama_stack_model_id and m.provider_id == provider_id
224
+ for m in models
223
225
  ):
224
226
  message = f"Model {model_id} from provider {provider_id} not found in available models"
225
227
  logger.error(message)
@@ -231,7 +233,7 @@ def select_model_and_provider_id(
231
233
  },
232
234
  )
233
235
 
234
- return model_id, provider_id
236
+ return llama_stack_model_id, provider_id
235
237
 
236
238
 
237
239
  def _is_inout_shield(shield: Shield) -> bool:
@@ -191,10 +191,16 @@ def test_select_model_and_provider_id_from_request(mocker):
191
191
  )
192
192
 
193
193
  model_list = [
194
- mocker.Mock(identifier="model1", model_type="llm", provider_id="provider1"),
195
- mocker.Mock(identifier="model2", model_type="llm", provider_id="provider2"),
196
194
  mocker.Mock(
197
- identifier="default_model", model_type="llm", provider_id="default_provider"
195
+ identifier="provider1/model1", model_type="llm", provider_id="provider1"
196
+ ),
197
+ mocker.Mock(
198
+ identifier="provider2/model2", model_type="llm", provider_id="provider2"
199
+ ),
200
+ mocker.Mock(
201
+ identifier="default_provider/default_model",
202
+ model_type="llm",
203
+ provider_id="default_provider",
198
204
  ),
199
205
  ]
200
206
 
@@ -206,7 +212,7 @@ def test_select_model_and_provider_id_from_request(mocker):
206
212
  # Assert the model and provider from request take precedence from the configuration one
207
213
  model_id, provider_id = select_model_and_provider_id(model_list, query_request)
208
214
 
209
- assert model_id == "model2"
215
+ assert model_id == "provider2/model2"
210
216
  assert provider_id == "provider2"
211
217
 
212
218
 
@@ -222,9 +228,13 @@ def test_select_model_and_provider_id_from_configuration(mocker):
222
228
  )
223
229
 
224
230
  model_list = [
225
- mocker.Mock(identifier="model1", model_type="llm", provider_id="provider1"),
226
231
  mocker.Mock(
227
- identifier="default_model", model_type="llm", provider_id="default_provider"
232
+ identifier="provider1/model1", model_type="llm", provider_id="provider1"
233
+ ),
234
+ mocker.Mock(
235
+ identifier="default_provider/default_model",
236
+ model_type="llm",
237
+ provider_id="default_provider",
228
238
  ),
229
239
  ]
230
240
 
@@ -236,7 +246,7 @@ def test_select_model_and_provider_id_from_configuration(mocker):
236
246
  model_id, provider_id = select_model_and_provider_id(model_list, query_request)
237
247
 
238
248
  # Assert that the default model and provider from the configuration are returned
239
- assert model_id == "default_model"
249
+ assert model_id == "default_provider/default_model"
240
250
  assert provider_id == "default_provider"
241
251
 
242
252