unique_toolkit 0.8.36__tar.gz → 0.8.37__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 (137) hide show
  1. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/CHANGELOG.md +6 -4
  2. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/PKG-INFO +7 -5
  3. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/pyproject.toml +1 -1
  4. unique_toolkit-0.8.37/unique_toolkit/_common/pydantic_helpers.py +29 -0
  5. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/language_model/infos.py +34 -11
  6. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/LICENSE +0 -0
  7. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/README.md +0 -0
  8. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/__init__.py +0 -0
  9. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/_base_service.py +0 -0
  10. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/_time_utils.py +0 -0
  11. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/chunk_relevancy_sorter/config.py +0 -0
  12. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/chunk_relevancy_sorter/exception.py +0 -0
  13. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/chunk_relevancy_sorter/schemas.py +0 -0
  14. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/chunk_relevancy_sorter/service.py +0 -0
  15. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/chunk_relevancy_sorter/tests/test_service.py +0 -0
  16. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/default_language_model.py +0 -0
  17. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/endpoint_builder.py +0 -0
  18. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/exception.py +0 -0
  19. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/feature_flags/schema.py +0 -0
  20. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/token/image_token_counting.py +0 -0
  21. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/token/token_counting.py +0 -0
  22. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/utils/structured_output/schema.py +0 -0
  23. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/validate_required_values.py +0 -0
  24. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/_common/validators.py +0 -0
  25. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/app/__init__.py +0 -0
  26. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/app/dev_util.py +0 -0
  27. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/app/init_logging.py +0 -0
  28. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/app/init_sdk.py +0 -0
  29. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/app/performance/async_tasks.py +0 -0
  30. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/app/performance/async_wrapper.py +0 -0
  31. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/app/schemas.py +0 -0
  32. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/app/unique_settings.py +0 -0
  33. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/app/verification.py +0 -0
  34. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/chat/__init__.py +0 -0
  35. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/chat/constants.py +0 -0
  36. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/chat/functions.py +0 -0
  37. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/chat/schemas.py +0 -0
  38. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/chat/service.py +0 -0
  39. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/chat/state.py +0 -0
  40. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/chat/utils.py +0 -0
  41. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/content/__init__.py +0 -0
  42. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/content/constants.py +0 -0
  43. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/content/functions.py +0 -0
  44. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/content/schemas.py +0 -0
  45. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/content/service.py +0 -0
  46. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/content/utils.py +0 -0
  47. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/debug_info_manager/debug_info_manager.py +0 -0
  48. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/embedding/__init__.py +0 -0
  49. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/embedding/constants.py +0 -0
  50. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/embedding/functions.py +0 -0
  51. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/embedding/schemas.py +0 -0
  52. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/embedding/service.py +0 -0
  53. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/embedding/utils.py +0 -0
  54. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/config.py +0 -0
  55. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/context_relevancy/prompts.py +0 -0
  56. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/context_relevancy/schema.py +0 -0
  57. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/context_relevancy/service.py +0 -0
  58. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/evaluation_manager.py +0 -0
  59. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/exception.py +0 -0
  60. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/hallucination/constants.py +0 -0
  61. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/hallucination/hallucination_evaluation.py +0 -0
  62. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/hallucination/prompts.py +0 -0
  63. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/hallucination/service.py +0 -0
  64. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/hallucination/utils.py +0 -0
  65. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/output_parser.py +0 -0
  66. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/schemas.py +0 -0
  67. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/tests/test_context_relevancy_service.py +0 -0
  68. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evals/tests/test_output_parser.py +0 -0
  69. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/__init__.py +0 -0
  70. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/config.py +0 -0
  71. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/constants.py +0 -0
  72. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/context_relevancy/constants.py +0 -0
  73. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/context_relevancy/prompts.py +0 -0
  74. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/context_relevancy/service.py +0 -0
  75. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/context_relevancy/utils.py +0 -0
  76. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/exception.py +0 -0
  77. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/hallucination/constants.py +0 -0
  78. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/hallucination/prompts.py +0 -0
  79. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/hallucination/service.py +0 -0
  80. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/hallucination/utils.py +0 -0
  81. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/output_parser.py +0 -0
  82. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/evaluators/schemas.py +0 -0
  83. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/framework_utilities/__init__.py +0 -0
  84. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/framework_utilities/langchain/client.py +0 -0
  85. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/framework_utilities/langchain/history.py +0 -0
  86. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/framework_utilities/openai/__init__.py +0 -0
  87. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/framework_utilities/openai/client.py +0 -0
  88. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/framework_utilities/openai/message_builder.py +0 -0
  89. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/framework_utilities/utils.py +0 -0
  90. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/history_manager/history_construction_with_contents.py +0 -0
  91. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/history_manager/history_manager.py +0 -0
  92. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/history_manager/loop_token_reducer.py +0 -0
  93. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/history_manager/utils.py +0 -0
  94. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/language_model/__init__.py +0 -0
  95. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/language_model/builder.py +0 -0
  96. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/language_model/constants.py +0 -0
  97. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/language_model/functions.py +0 -0
  98. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/language_model/prompt.py +0 -0
  99. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/language_model/reference.py +0 -0
  100. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/language_model/schemas.py +0 -0
  101. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/language_model/service.py +0 -0
  102. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/language_model/utils.py +0 -0
  103. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/postprocessor/postprocessor_manager.py +0 -0
  104. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/protocols/support.py +0 -0
  105. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/reference_manager/reference_manager.py +0 -0
  106. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/short_term_memory/__init__.py +0 -0
  107. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/short_term_memory/constants.py +0 -0
  108. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/short_term_memory/functions.py +0 -0
  109. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/short_term_memory/persistent_short_term_memory_manager.py +0 -0
  110. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/short_term_memory/schemas.py +0 -0
  111. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/short_term_memory/service.py +0 -0
  112. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/smart_rules/__init__.py +0 -0
  113. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/smart_rules/compile.py +0 -0
  114. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/thinking_manager/thinking_manager.py +0 -0
  115. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/a2a/__init__.py +0 -0
  116. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/a2a/config.py +0 -0
  117. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/a2a/manager.py +0 -0
  118. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/a2a/memory.py +0 -0
  119. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/a2a/schema.py +0 -0
  120. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/a2a/service.py +0 -0
  121. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/agent_chunks_hanlder.py +0 -0
  122. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/config.py +0 -0
  123. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/factory.py +0 -0
  124. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/mcp/__init__.py +0 -0
  125. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/mcp/manager.py +0 -0
  126. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/mcp/models.py +0 -0
  127. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/mcp/tool_wrapper.py +0 -0
  128. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/schemas.py +0 -0
  129. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/test/test_mcp_manager.py +0 -0
  130. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/test/test_tool_progress_reporter.py +0 -0
  131. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/tool.py +0 -0
  132. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/tool_manager.py +0 -0
  133. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/tool_progress_reporter.py +0 -0
  134. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/utils/execution/execution.py +0 -0
  135. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/utils/source_handling/schema.py +0 -0
  136. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/utils/source_handling/source_formatting.py +0 -0
  137. {unique_toolkit-0.8.36 → unique_toolkit-0.8.37}/unique_toolkit/tools/utils/source_handling/tests/test_source_formatting.py +0 -0
@@ -5,6 +5,10 @@ All notable changes to this project will be documented in this file.
5
5
  The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
6
6
  and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
7
7
 
8
+
9
+ ## [0.8.37] - 2025-09-01
10
+ - Adapt defaults and json-schema in `LanguageModelInfo`
11
+
8
12
  ## [0.8.36] - 2025-09-01
9
13
  - Added dependency `Pillow` and `Platformsdir`
10
14
 
@@ -15,12 +19,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
15
19
  - Automatic initializations of services and event generator
16
20
 
17
21
  ## [0.8.33] - 2025-08-31
18
-
19
- fixed tool for `web_search`
22
+ - fixed tool for `web_search`
20
23
 
21
24
  ## [0.8.32] - 2025-08-30
22
-
23
- moved over general packages for `web_search`
25
+ - moved over general packages for `web_search`
24
26
 
25
27
  ## [0.8.31] - 2025-08-29
26
28
  - Add various openai models to supported model list
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: unique_toolkit
3
- Version: 0.8.36
3
+ Version: 0.8.37
4
4
  Summary:
5
5
  License: Proprietary
6
6
  Author: Martin Fadler
@@ -116,6 +116,10 @@ All notable changes to this project will be documented in this file.
116
116
  The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
117
117
  and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
118
118
 
119
+
120
+ ## [0.8.37] - 2025-09-01
121
+ - Adapt defaults and json-schema in `LanguageModelInfo`
122
+
119
123
  ## [0.8.36] - 2025-09-01
120
124
  - Added dependency `Pillow` and `Platformsdir`
121
125
 
@@ -126,12 +130,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
126
130
  - Automatic initializations of services and event generator
127
131
 
128
132
  ## [0.8.33] - 2025-08-31
129
-
130
- fixed tool for `web_search`
133
+ - fixed tool for `web_search`
131
134
 
132
135
  ## [0.8.32] - 2025-08-30
133
-
134
- moved over general packages for `web_search`
136
+ - moved over general packages for `web_search`
135
137
 
136
138
  ## [0.8.31] - 2025-08-29
137
139
  - Add various openai models to supported model list
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "unique_toolkit"
3
- version = "0.8.36"
3
+ version = "0.8.37"
4
4
  description = ""
5
5
  authors = [
6
6
  "Martin Fadler <martin.fadler@unique.ch>",
@@ -0,0 +1,29 @@
1
+ import logging
2
+
3
+ import humps
4
+ from pydantic import ConfigDict
5
+ from pydantic.fields import ComputedFieldInfo, FieldInfo
6
+
7
+ logger = logging.getLogger(__name__)
8
+
9
+
10
+ def field_title_generator(
11
+ title: str,
12
+ info: FieldInfo | ComputedFieldInfo,
13
+ ) -> str:
14
+ return humps.decamelize(title).replace("_", " ").title()
15
+
16
+
17
+ def model_title_generator(model: type) -> str:
18
+ return humps.decamelize(model.__name__).replace("_", " ").title()
19
+
20
+
21
+ def get_configuration_dict(**kwargs) -> ConfigDict:
22
+ return ConfigDict(
23
+ # alias_generator=to_camel,
24
+ field_title_generator=field_title_generator,
25
+ model_title_generator=model_title_generator,
26
+ # populate_by_name=True,
27
+ # protected_namespaces=(),
28
+ **kwargs,
29
+ )
@@ -1,11 +1,12 @@
1
1
  from datetime import date
2
2
  from enum import StrEnum
3
- from typing import Any, ClassVar, Optional, Self
3
+ from typing import Annotated, Any, ClassVar, Optional, Self
4
4
 
5
- from pydantic import BaseModel
5
+ from pydantic import BaseModel, Field
6
6
  from pydantic.json_schema import SkipJsonSchema
7
7
  from typing_extensions import deprecated
8
8
 
9
+ from unique_toolkit._common.pydantic_helpers import get_configuration_dict
9
10
  from unique_toolkit.language_model.schemas import LanguageModelTokenLimits
10
11
 
11
12
 
@@ -130,9 +131,13 @@ class TemperatureBounds(BaseModel):
130
131
 
131
132
 
132
133
  class LanguageModelInfo(BaseModel):
133
- name: LanguageModelName | str
134
- version: str
135
- provider: LanguageModelProvider
134
+ model_config = get_configuration_dict()
135
+ name: (
136
+ Annotated[str, Field(title="Custom Model Name")]
137
+ | SkipJsonSchema[LanguageModelName]
138
+ ) = Field(title="Model Name", default=LanguageModelName.AZURE_GPT_4o_2024_1120)
139
+ provider: LanguageModelProvider = LanguageModelProvider.AZURE
140
+ version: str = Field(title="Model Version", default="")
136
141
 
137
142
  encoder_name: EncoderName = EncoderName.CL100K_BASE
138
143
 
@@ -140,16 +145,34 @@ class LanguageModelInfo(BaseModel):
140
145
  token_limits: LanguageModelTokenLimits = LanguageModelTokenLimits(
141
146
  token_limit_input=7_000, token_limit_output=1_000
142
147
  )
148
+
143
149
  capabilities: list[ModelCapabilities] = [ModelCapabilities.STREAMING]
144
150
 
145
- info_cutoff_at: date | SkipJsonSchema[None] = None
146
- published_at: date | SkipJsonSchema[None] = None
147
- retirement_at: date | SkipJsonSchema[None] = None
151
+ info_cutoff_at: (
152
+ Annotated[date, Field(title="Info Cutoff")]
153
+ | Annotated[None, Field(title="Info Cutoff Unknown")]
154
+ ) = None
155
+
156
+ published_at: (
157
+ Annotated[date, Field(title="Publishing Date")]
158
+ | Annotated[None, Field(title="Publishing Date Unknown")]
159
+ ) = None
160
+
161
+ retirement_at: (
162
+ Annotated[date, Field(title="Retirement Date")]
163
+ | Annotated[None, Field(title="Retirement Date Unknown")]
164
+ ) = date(2225, 12, 31)
165
+
166
+ deprecated_at: (
167
+ Annotated[date, Field(title="Deprecated Date")]
168
+ | Annotated[None, Field(title="Deprecated Date Unknown")]
169
+ ) = date(2225, 12, 31)
148
170
 
149
- deprecated_at: date | SkipJsonSchema[None] = None
150
- retirement_text: str | SkipJsonSchema[None] = None
171
+ retirement_text: str = "This model is no longer supported."
151
172
 
152
- temperature_bounds: TemperatureBounds | None = None
173
+ temperature_bounds: (
174
+ TemperatureBounds | Annotated[None, Field(title="Temperature Bounds Unknown")]
175
+ ) = None
153
176
 
154
177
  default_options: dict[str, Any] = {}
155
178
 
File without changes