Add support for DeepSeek-R1

Author: RussellLuo

coagent/agents/aswarm/core.py

@@ -234,6 +234,7 @@ async def run_and_stream(
         while len(history) - init_len < max_turns:
             message = {
                 "content": "",
+                "reasoning_content": "",
                 # No `sender` param is supported by model
                 # "sender": agent.name,
                 "role": "assistant",
@@ -263,11 +264,16 @@ async def run_and_stream(
                 delta = json.loads(chunk.choices[0].delta.json())
                 if delta["role"] == "assistant":
                     delta["sender"] = active_agent.name
-                if delta["content"]:
+
+                delta_content = delta.get("content") or ""
+                delta_reasoning_content = delta.get("reasoning_content") or ""
+                if delta_content or delta_reasoning_content:
                     yield ChatMessage(
                         role=delta["role"] or "assistant",
-                        content=delta["content"],
+                        content=delta_content,
+                        reasoning_content=delta_reasoning_content,
                     )
+
                 delta.pop("role", None)
                 delta.pop("sender", None)
                 merge_chunk(message, delta)

coagent/agents/chat_agent.py

@@ -281,7 +281,7 @@ async def _handle_history(
             context_variables=msg.extensions,
         )
         async for resp in response:
-            if isinstance(resp, ChatMessage) and resp.content:
+            if isinstance(resp, ChatMessage) and resp.has_content:
                 yield resp
 
     async def update_user_confirmed(self, history: ChatHistory) -> None:

coagent/agents/messages.py

@@ -7,8 +7,17 @@
 
 
 class ChatMessage(Message):
-    role: str
-    content: str
+    role: str = Field(
+        ..., description="The role of the message. (e.g. `user`, `assistant`)"
+    )
+    content: str = Field(
+        default="",
+        description="The content of the message. For reasoning models, this is the content of the final answer.",
+    )
+    reasoning_content: str = Field(
+        default="",
+        description="The content of the CoT. Only available for reasoning models.",
+    )
 
     type: str = Field(default="", description="The type of the message. e.g. confirm")
     sender: str = Field(default="", description="The sending agent of the message.")
@@ -18,8 +27,13 @@ class ChatMessage(Message):
 
     def __add__(self, other: ChatMessage) -> ChatMessage:
         self.content += other.content
+        self.reasoning_content += other.reasoning_content
         return self
 
+    @property
+    def has_content(self) -> bool:
+        return bool(self.content or self.reasoning_content)
+
     def model_dump(self, **kwargs) -> dict[str, Any]:
         return super().model_dump(include={"role", "content"}, **kwargs)
 

poetry.lock

@@ -29,7 +29,7 @@ httpx-sse = "0.4.0"
 blinker = "1.9.0"
 loguru = "0.7.3"
 jq = "1.8.0"
-litellm = "1.55.12"
+litellm = "1.60.4"
 mcp = "1.2.0"
 
 [tool.pyright]