aidial-sdk 0.41.0.dev2__tar.gz → 0.41.0.dev4__tar.gz

{aidial_sdk-0.41.0.dev2 → aidial_sdk-0.41.0.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aidial-sdk
-Version: 0.41.0.dev2
+Version: 0.41.0.dev4
 Summary: Framework to create applications and model adapters for AI DIAL
 License-Expression: Apache-2.0
 License-File: LICENSE
@@ -32,8 +32,8 @@ Requires-Dist: prometheus-client (>=0.17.1,<1.0) ; extra == "telemetry"
 Requires-Dist: pydantic (>=1.10.17,<3)
 Requires-Dist: uvicorn (>=0.19,<1.0)
 Requires-Dist: wrapt (>=1.10,<3)
-Project-URL: Documentation, https://epam-rail.com/dial_api
-Project-URL: Homepage, https://epam-rail.com
+Project-URL: Documentation, https://dialx.ai/dial_api
+Project-URL: Homepage, https://dialx.ai
 Project-URL: Repository, https://github.com/epam/ai-dial-sdk
 Description-Content-Type: text/markdown
 
@@ -76,9 +76,9 @@ Description-Content-Type: text/markdown
 
 ## Overview
 
-Framework to create applications and model adapters for [AI DIAL](https://epam-rail.com).
+Framework to create applications and model adapters for [AI DIAL](https://dialx.ai).
 
-Applications and model adapters implemented using this framework will be compatible with [AI DIAL API](https://epam-rail.com/dial_api) that was designed based on [Azure OpenAI API](https://learn.microsoft.com/en-us/azure/ai-services/openai/reference).
+Applications and model adapters implemented using this framework will be compatible with [AI DIAL API](https://dialx.ai/dial_api) that was designed based on [Azure OpenAI API](https://learn.microsoft.com/en-us/azure/ai-services/openai/reference).
 
 ---
 

{aidial_sdk-0.41.0.dev2 → aidial_sdk-0.41.0.dev4}/README.md

@@ -37,9 +37,9 @@
 
 ## Overview
 
-Framework to create applications and model adapters for [AI DIAL](https://epam-rail.com).
+Framework to create applications and model adapters for [AI DIAL](https://dialx.ai).
 
-Applications and model adapters implemented using this framework will be compatible with [AI DIAL API](https://epam-rail.com/dial_api) that was designed based on [Azure OpenAI API](https://learn.microsoft.com/en-us/azure/ai-services/openai/reference).
+Applications and model adapters implemented using this framework will be compatible with [AI DIAL API](https://dialx.ai/dial_api) that was designed based on [Azure OpenAI API](https://learn.microsoft.com/en-us/azure/ai-services/openai/reference).
 
 ---
 

{aidial_sdk-0.41.0.dev2 → aidial_sdk-0.41.0.dev4}/aidial_sdk/chat_completion/choice.py

@@ -11,6 +11,7 @@ from aidial_sdk.chat_completion.chunks import (
     ContentChunk,
     EndChoiceChunk,
     FormSchemaChunk,
+    ReasoningContentChunk,
     StartChoiceChunk,
     StateChunk,
 )
@@ -96,6 +97,18 @@ class Choice(ChoiceBase):
         self.send_chunk(ContentChunk(content, self._index))
         self._last_finish_reason = FinishReason.STOP
 
+    def append_reasoning_content(self, content: str) -> None:
+        if not self._opened:
+            raise runtime_error(
+                "Trying to append reasoning content to an unopened choice"
+            )
+        if self._closed:
+            raise runtime_error(
+                "Trying to append reasoning content to a closed choice"
+            )
+
+        self.send_chunk(ReasoningContentChunk(content, self._index))
+
     @property
     def content_stream(self) -> ContentStream:
         return ContentStream(self)

{aidial_sdk-0.41.0.dev2 → aidial_sdk-0.41.0.dev4}/aidial_sdk/chat_completion/chunks.py

@@ -96,6 +96,27 @@ class ContentChunk(BaseChunk):
         }
 
 
+class ReasoningContentChunk(BaseChunk):
+    content: str
+    choice_index: int
+
+    def __init__(self, content: str, choice_index: int):
+        self.content = content
+        self.choice_index = choice_index
+
+    def to_dict(self):
+        return {
+            "choices": [
+                {
+                    "index": self.choice_index,
+                    "finish_reason": None,
+                    "delta": {"reasoning_content": self.content},
+                }
+            ],
+            "usage": None,
+        }
+
+
 class FunctionToolCallChunk(BaseChunk):
     choice_index: int
     call_index: int

{aidial_sdk-0.41.0.dev2 → aidial_sdk-0.41.0.dev4}/aidial_sdk/chat_completion/request.py

@@ -152,6 +152,7 @@ class Message(ExtraAllowModel):
     tool_call_id: StrictStr | None = None
     function_call: FunctionCall | None = None
     refusal: StrictStr | None = None
+    reasoning_content: StrictStr | None = None
 
     def text(self) -> str:
         """

{aidial_sdk-0.41.0.dev2 → aidial_sdk-0.41.0.dev4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aidial-sdk"
-version = "0.41.0.dev2"
+version = "0.41.0.dev4"
 description = "Framework to create applications and model adapters for AI DIAL"
 authors = [{ name = "EPAM RAIL", email = "SpecialEPM-DIALDevTeam@epam.com" }]
 license = "Apache-2.0"
@@ -14,8 +14,8 @@ dependencies = [
 ]
 
 [project.urls]
-Homepage = "https://epam-rail.com"
-Documentation = "https://epam-rail.com/dial_api"
+Homepage = "https://dialx.ai"
+Documentation = "https://dialx.ai/dial_api"
 Repository = "https://github.com/epam/ai-dial-sdk"
 
 [project.optional-dependencies]