prompture 1.0.44.dev1__tar.gz → 1.0.45.dev2__tar.gz

{prompture-1.0.44.dev1 → prompture-1.0.45.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompture
-Version: 1.0.44.dev1
+Version: 1.0.45.dev2
 Summary: Ask LLMs to return structured JSON and run cross-model tests. API-first.
 Author-email: Juan Denis <juan@vene.co>
 License-Expression: MIT
@@ -18,6 +18,7 @@ Requires-Dist: pydantic-settings>=2.0
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: python-dateutil>=2.9.0
 Requires-Dist: pyyaml>=6.0
+Requires-Dist: requests>=2.28
 Provides-Extra: openai
 Requires-Dist: openai>=1.55.0; extra == "openai"
 Provides-Extra: anthropic
@@ -31,10 +32,8 @@ Requires-Dist: python-toon>=0.1.0; extra == "toon"
 Requires-Dist: tukuy==0.0.30; extra == "toon"
 Provides-Extra: pandas
 Requires-Dist: pandas>=1.3.0; extra == "pandas"
-Provides-Extra: requests
-Requires-Dist: requests>=2.28; extra == "requests"
 Provides-Extra: all
-Requires-Dist: prompture[anthropic,google,groq,openai,pandas,requests,toon]; extra == "all"
+Requires-Dist: prompture[anthropic,google,groq,openai,pandas,toon]; extra == "all"
 Provides-Extra: test
 Requires-Dist: pytest>=7.0; extra == "test"
 Requires-Dist: pytest-asyncio>=0.23.0; extra == "test"

{prompture-1.0.44.dev1 → prompture-1.0.45.dev2}/prompture/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '1.0.44.dev1'
-__version_tuple__ = version_tuple = (1, 0, 44, 'dev1')
+__version__ = version = '1.0.45.dev2'
+__version_tuple__ = version_tuple = (1, 0, 45, 'dev2')
 
 __commit_id__ = commit_id = None

{prompture-1.0.44.dev1 → prompture-1.0.45.dev2}/prompture/agents/agent.py

@@ -279,6 +279,16 @@ class Agent(Generic[DepsType]):
         """Request graceful shutdown after the current iteration."""
         self._stop_requested = True
 
+    @property
+    def callbacks(self) -> AgentCallbacks:
+        """Current agent-level callbacks."""
+        return self._agent_callbacks
+
+    @callbacks.setter
+    def callbacks(self, value: AgentCallbacks) -> None:
+        """Set agent-level callbacks."""
+        self._agent_callbacks = value
+
     @property
     def conversation(self) -> Conversation | None:
         """The current persistent conversation, or ``None``."""
@@ -842,6 +852,10 @@ class Agent(Generic[DepsType]):
                 result = self._run_output_guardrails(ctx, result, conv, session, steps, all_tool_calls)
 
             # 12. Fire callbacks
+            if self._agent_callbacks.on_thinking:
+                for step in steps:
+                    if step.step_type == StepType.think and step.content:
+                        self._agent_callbacks.on_thinking(step.content)
             if self._agent_callbacks.on_step:
                 for step in steps:
                     self._agent_callbacks.on_step(step)
@@ -883,9 +897,7 @@ class Agent(Generic[DepsType]):
 
                 # Extract thinking content from <think> tags
                 thinking_text = self._extract_thinking(content)
-                if thinking_text and self._agent_callbacks.on_thinking:
-                    self._agent_callbacks.on_thinking(thinking_text)
-                    # Also record as a think step
+                if thinking_text:
                     steps.append(
                         AgentStep(
                             step_type=StepType.think,
@@ -1184,6 +1196,10 @@ class Agent(Generic[DepsType]):
                 result = self._run_output_guardrails(ctx, result, conv, session, steps, all_tool_calls)
 
             # 12. Fire callbacks
+            if self._agent_callbacks.on_thinking:
+                for step in steps:
+                    if step.step_type == StepType.think and step.content:
+                        self._agent_callbacks.on_thinking(step.content)
             if self._agent_callbacks.on_step:
                 for step in steps:
                     self._agent_callbacks.on_step(step)

{prompture-1.0.44.dev1 → prompture-1.0.45.dev2}/prompture/agents/async_agent.py

@@ -256,6 +256,16 @@ class AsyncAgent(Generic[DepsType]):
         """Request graceful shutdown after the current iteration."""
         self._stop_requested = True
 
+    @property
+    def callbacks(self) -> AgentCallbacks:
+        """Current agent-level callbacks."""
+        return self._agent_callbacks
+
+    @callbacks.setter
+    def callbacks(self, value: AgentCallbacks) -> None:
+        """Set agent-level callbacks."""
+        self._agent_callbacks = value
+
     @property
     def conversation(self) -> Any:
         """The current persistent conversation, or ``None``."""
@@ -902,6 +912,10 @@ class AsyncAgent(Generic[DepsType]):
                 result = await self._run_output_guardrails(ctx, result, conv, session, steps, all_tool_calls)
 
             # 11. Fire callbacks (async-aware)
+            if self._agent_callbacks.on_thinking:
+                for step in steps:
+                    if step.step_type == StepType.think and step.content:
+                        await self._invoke_callback(self._agent_callbacks.on_thinking, step.content)
             if self._agent_callbacks.on_step:
                 for step in steps:
                     await self._invoke_callback(self._agent_callbacks.on_step, step)
@@ -937,6 +951,20 @@ class AsyncAgent(Generic[DepsType]):
                 }
 
             if role == "assistant":
+                content = msg.get("content", "") or ""
+
+                # Extract thinking content from <think> tags
+                thinking_text = self._extract_thinking(content)
+                if thinking_text:
+                    steps.append(
+                        AgentStep(
+                            step_type=StepType.think,
+                            timestamp=now,
+                            content=thinking_text,
+                            usage=step_usage,
+                        )
+                    )
+
                 tc_list = msg.get("tool_calls", [])
                 if tc_list:
                     for tc in tc_list:
@@ -955,7 +983,7 @@ class AsyncAgent(Generic[DepsType]):
                             AgentStep(
                                 step_type=StepType.tool_call,
                                 timestamp=now,
-                                content=msg.get("content", ""),
+                                content=content,
                                 tool_name=name,
                                 tool_args=args,
                                 usage=step_usage,
@@ -967,7 +995,7 @@ class AsyncAgent(Generic[DepsType]):
                         AgentStep(
                             step_type=StepType.output,
                             timestamp=now,
-                            content=msg.get("content", ""),
+                            content=content,
                             usage=step_usage,
                         )
                     )
@@ -989,6 +1017,27 @@ class AsyncAgent(Generic[DepsType]):
                     )
                 )
 
+    def _extract_thinking(self, content: str) -> str | None:
+        """Extract thinking content from <think> tags.
+
+        Some models (like DeepSeek, Qwen) emit chain-of-thought reasoning
+        within <think>...</think> tags. This method extracts that content.
+
+        Args:
+            content: The assistant message content.
+
+        Returns:
+            The thinking text if found, None otherwise.
+        """
+        import re
+
+        # Match <think>...</think> tags (case-insensitive, allows multiline)
+        pattern = r"<think>(.*?)</think>"
+        matches = re.findall(pattern, content, re.DOTALL | re.IGNORECASE)
+        if matches:
+            return "\n".join(match.strip() for match in matches)
+        return None
+
     async def _parse_output(
         self,
         conv: Any,

{prompture-1.0.44.dev1 → prompture-1.0.45.dev2}/prompture.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompture
-Version: 1.0.44.dev1
+Version: 1.0.45.dev2
 Summary: Ask LLMs to return structured JSON and run cross-model tests. API-first.
 Author-email: Juan Denis <juan@vene.co>
 License-Expression: MIT
@@ -18,6 +18,7 @@ Requires-Dist: pydantic-settings>=2.0
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: python-dateutil>=2.9.0
 Requires-Dist: pyyaml>=6.0
+Requires-Dist: requests>=2.28
 Provides-Extra: openai
 Requires-Dist: openai>=1.55.0; extra == "openai"
 Provides-Extra: anthropic
@@ -31,10 +32,8 @@ Requires-Dist: python-toon>=0.1.0; extra == "toon"
 Requires-Dist: tukuy==0.0.30; extra == "toon"
 Provides-Extra: pandas
 Requires-Dist: pandas>=1.3.0; extra == "pandas"
-Provides-Extra: requests
-Requires-Dist: requests>=2.28; extra == "requests"
 Provides-Extra: all
-Requires-Dist: prompture[anthropic,google,groq,openai,pandas,requests,toon]; extra == "all"
+Requires-Dist: prompture[anthropic,google,groq,openai,pandas,toon]; extra == "all"
 Provides-Extra: test
 Requires-Dist: pytest>=7.0; extra == "test"
 Requires-Dist: pytest-asyncio>=0.23.0; extra == "test"

{prompture-1.0.44.dev1 → prompture-1.0.45.dev2}/prompture.egg-info/requires.txt

@@ -6,12 +6,13 @@ pydantic-settings>=2.0
 python-dotenv>=0.19.0
 python-dateutil>=2.9.0
 pyyaml>=6.0
+requests>=2.28
 
 [airllm]
 airllm>=2.8.0
 
 [all]
-prompture[anthropic,google,groq,openai,pandas,requests,toon]
+prompture[anthropic,google,groq,openai,pandas,toon]
 
 [anthropic]
 anthropic>=0.8.0
@@ -45,9 +46,6 @@ pandas>=1.3.0
 [redis]
 redis>=4.0
 
-[requests]
-requests>=2.28
-
 [scaffold]
 jinja2>=3.0
 

{prompture-1.0.44.dev1 → prompture-1.0.45.dev2}/pyproject.toml

@@ -23,6 +23,7 @@ dependencies = [
     "python-dotenv>=0.19.0",
     "python-dateutil>=2.9.0",
     "pyyaml>=6.0",
+    "requests>=2.28",
 ]
 
 [project.optional-dependencies]
@@ -32,9 +33,8 @@ google = ["google-genai>=1.0.0"]
 groq = ["groq>=0.4.0"]
 toon = ["python-toon>=0.1.0", "tukuy==0.0.30"]
 pandas = ["pandas>=1.3.0"]
-requests = ["requests>=2.28"]
 all = [
-    "prompture[openai,anthropic,google,groq,toon,pandas,requests]",
+    "prompture[openai,anthropic,google,groq,toon,pandas]",
 ]
 test = ["pytest>=7.0", "pytest-asyncio>=0.23.0", "prompture[all]"]
 dev = ["pytest>=7.0", "pytest-asyncio>=0.23.0", "ruff>=0.8.0", "prompture[all]"]