dais-sdk 0.6.3__py3-none-any.whl → 0.6.4__py3-none-any.whl

dais_sdk/stream.py

@@ -74,6 +74,9 @@ class AssistantMessageCollector:
 
     def get_message(self) -> AssistantMessage:
         self.message_buf.tool_calls = self.tool_call_collector.get_tool_calls()
+        if self.message_buf.content is not None and self.message_buf.reasoning_content is None:
+            self.message_buf.content, self.message_buf.reasoning_content =\
+                AssistantMessage.extract_thinking_content(self.message_buf.content)
         return self.message_buf
 
     def clear(self):

dais_sdk/types/message.py

@@ -1,5 +1,6 @@
 import json
 import dataclasses
+import re
 import uuid
 from abc import ABC, abstractmethod
 from typing import Any, Literal, cast
@@ -92,6 +93,22 @@ class AssistantMessage(ChatMessage):
     usage: LiteLlmUsage | None = None
     role: Literal["assistant"] = "assistant"
 
+    @staticmethod
+    def extract_thinking_content(text: str) -> tuple[str, str | None]:
+        """
+        Extract thinking content from the start of model responsed text.
+        Returns:
+            A tuple of (extracted_content, optional thinking_content)
+        """
+        pattern = r"<(think|thinking)>(.*?)</\1>"
+        match = re.match(pattern, text, re.DOTALL)
+        if match:
+            start, end = match.span()
+            thinking_content = match.group(2)
+            remaining_content = text[:start] + text[end:]
+            return remaining_content.strip(), thinking_content
+        return text, None
+
     @classmethod
     def from_litellm_message(cls, response: LiteLlmModelResponse) -> "AssistantMessage":
         choices = cast(list[LiteLlmModelResponseChoices], response.choices)
@@ -108,9 +125,16 @@ class AssistantMessage(ChatMessage):
                 type="function",
             ) for tool_call in cast(list[ChatCompletionMessageToolCall], message_tool_calls)]
 
+        content: str | None = message.get("content")
+        reasoning_content: str | None = message.get("reasoning_content")
+        if content is not None:
+            content, thinking_block_content = cls.extract_thinking_content(content)
+            if reasoning_content is None:
+                reasoning_content = thinking_block_content
+
         return cls.model_construct(
-            content=message.get("content"),
-            reasoning_content=message.get("reasoning_content"),
+            content=content,
+            reasoning_content=reasoning_content,
             tool_calls=tool_calls,
             audio=message.get("audio"),
             images=message.get("images"),

{dais_sdk-0.6.3.dist-info → dais_sdk-0.6.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dais-sdk
-Version: 0.6.3
+Version: 0.6.4
 Summary: A wrapper of LiteLLM
 Author-email: BHznJNs <bhznjns@outlook.com>
 Requires-Python: >=3.14
@@ -11,11 +11,11 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Python :: 3.14
 License-File: LICENSE
-Requires-Dist: litellm>=1.80.0
+Requires-Dist: litellm>=1.81.0
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: httpx==0.28.1
 Requires-Dist: mcp==1.26.0
-Requires-Dist: starlette==0.50.0
+Requires-Dist: starlette==0.52.1
 Requires-Dist: uvicorn==0.40.0
 Project-URL: Source, https://github.com/Dais-Project/Dais-SDK
 Project-URL: Tracker, https://github.com/Dais-Project/Dais-SDK/issues

{dais_sdk-0.6.3.dist-info → dais_sdk-0.6.4.dist-info}/RECORD

@@ -2,7 +2,7 @@ dais_sdk/__init__.py,sha256=5QAoL8GyyFuKfe4HZ-TiRVj9VIpHc45i2xMMYORoWXU,11805
 dais_sdk/debug.py,sha256=T7qIy1BeeUGlF40l9JCMMVn8pvvMJAEQeG4adQbOydA,69
 dais_sdk/logger.py,sha256=99vJAQRKcu4CuHgZYAJ2zDQtGea6Bn3vJJrS-mtza7c,677
 dais_sdk/param_parser.py,sha256=QIxt1izv69r725pzU1qhq5bilcrGUgzlpiItHEWOrdc,1874
-dais_sdk/stream.py,sha256=yu9Zvr3CUrPD9sGsjqwNXy_233Tw25Hd72kwjrraMAM,3170
+dais_sdk/stream.py,sha256=97y6CysPFMq37uXryd4tNlvZYCZ7Rv3CZpntaw1-SF4,3426
 dais_sdk/mcp_client/__init__.py,sha256=B86aC4nnGzwfjk7H0CZ38YlMDiEx3EIDEAgJKUnwqIU,405
 dais_sdk/mcp_client/base_mcp_client.py,sha256=jWAfinzY00aL-qdNgyzYXKM-LhPHkmdqL24Uw439v-0,1055
 dais_sdk/mcp_client/local_mcp_client.py,sha256=JI7jdFyyL4VkrUuzwUybHJ3Y1vgEH0iCG3WqAgyLO1g,1951
@@ -18,10 +18,10 @@ dais_sdk/tool/toolset/python_toolset.py,sha256=p4QeBY_yZpzpnDT5w1S4EECtPOwt8dF8T
 dais_sdk/tool/toolset/toolset.py,sha256=X1xqWiWov4fboWQowB_YgJ_Tc-fIDmxbP8GreTj_7ME,322
 dais_sdk/types/__init__.py,sha256=-i1MYWIlUfjQIX0xZJta6phQNL44vXPSIx1eGyIYZXc,710
 dais_sdk/types/exceptions.py,sha256=ZlQW6QfzPatbfHJi_1s_2XBAHK4bZaeSreI3SUAX_5c,1439
-dais_sdk/types/message.py,sha256=a65Q6K3Hq1aVhsedUv2XjyJq2rQQJi9SkEAPPHcLtFw,7770
+dais_sdk/types/message.py,sha256=l54b-vcCCfUtFyTAfVKQd3Csso5KYuSXk1fidfrs2SM,8718
 dais_sdk/types/request_params.py,sha256=8Jq-aTeK933YENE-9ay_8q88hEr-oeZZGC7l52tOKEM,1635
 dais_sdk/types/tool.py,sha256=GT6gxjAGdnVzk6W84nPzZb24bLznzqilkHN6uvocf5M,1377
-dais_sdk-0.6.3.dist-info/licenses/LICENSE,sha256=Qwd-hsctqFqJErH4OWNNttmd0jccbH0ZCsf-YtAFNdo,1064
-dais_sdk-0.6.3.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-dais_sdk-0.6.3.dist-info/METADATA,sha256=Tm4_bgcDUtx7rp1rW82aLeA9VpvBnM-fctwQHiqyPgY,2612
-dais_sdk-0.6.3.dist-info/RECORD,,
+dais_sdk-0.6.4.dist-info/licenses/LICENSE,sha256=Qwd-hsctqFqJErH4OWNNttmd0jccbH0ZCsf-YtAFNdo,1064
+dais_sdk-0.6.4.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+dais_sdk-0.6.4.dist-info/METADATA,sha256=sVSP1GTTvQQbY5OzpBe3PjXQxnKYUNO9RwU46a8zTD0,2612
+dais_sdk-0.6.4.dist-info/RECORD,,