prompture 1.9.1.dev2__tar.gz → 1.9.1.dev3__tar.gz

{prompture-1.9.1.dev2 → prompture-1.9.1.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompture
-Version: 1.9.1.dev2
+Version: 1.9.1.dev3
 Summary: Ask LLMs to return structured JSON and run cross-model tests. API-first.
 Author-email: Juan Denis <juan@vene.co>
 License-Expression: MIT

{prompture-1.9.1.dev2 → prompture-1.9.1.dev3}/prompture/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '1.9.1.dev2'
-__version_tuple__ = version_tuple = (1, 9, 1, 'dev2')
+__version__ = version = '1.9.1.dev3'
+__version_tuple__ = version_tuple = (1, 9, 1, 'dev3')
 
 __commit_id__ = commit_id = None

{prompture-1.9.1.dev2 → prompture-1.9.1.dev3}/prompture/agents/agent.py

@@ -387,6 +387,44 @@ class Agent(Generic[DepsType]):
         """Set agent-level callbacks."""
         self._agent_callbacks = value
 
+    @property
+    def driver(self):
+        """The driver instance this agent will run against, or ``None``."""
+        return self._driver
+
+    @driver.setter
+    def driver(self, value) -> None:
+        """Swap the driver (e.g. per-tenant credentials).
+
+        A persistent conversation built against the old driver is detected as
+        stale on the next run and rebuilt, so the swap takes effect there too.
+        """
+        self._driver = value
+
+    @property
+    def options(self) -> dict[str, Any]:
+        """Live per-call options dict (temperature, max_tokens, ...).
+
+        Mutations (``agent.options["temperature"] = 0.2``) apply from the
+        next run onward.
+        """
+        return self._options
+
+    @options.setter
+    def options(self, value: dict[str, Any] | None) -> None:
+        self._options = dict(value) if value else {}
+
+    @property
+    def system_prompt(self):
+        """The configured system prompt (str, Persona, or callable)."""
+        return self._system_prompt
+
+    @system_prompt.setter
+    def system_prompt(self, value) -> None:
+        """Replace the system prompt; re-resolved on every run, including
+        for persistent conversations."""
+        self._system_prompt = value
+
     @property
     def conversation(self) -> Conversation | None:
         """The current persistent conversation, or ``None``."""
@@ -890,6 +928,9 @@ class Agent(Generic[DepsType]):
                 # Respect the newly resolved system prompt for this run.
                 if system_prompt is not None:
                     conv.system_prompt = system_prompt
+                # Pick up option mutations made since the conversation was
+                # built (agent.options is a live public dict).
+                conv._options = dict(self._options)
                 # Update tools and callbacks for this run.
                 # NOTE: assigning ``callbacks`` mutates the driver instance,
                 # which may be shared (e.g. passed to several agents).  A

{prompture-1.9.1.dev2 → prompture-1.9.1.dev3}/prompture/agents/async_agent.py

@@ -321,6 +321,44 @@ class AsyncAgent(Generic[DepsType]):
         """Set agent-level callbacks."""
         self._agent_callbacks = value
 
+    @property
+    def driver(self):
+        """The driver instance this agent will run against, or ``None``."""
+        return self._driver
+
+    @driver.setter
+    def driver(self, value) -> None:
+        """Swap the driver (e.g. per-tenant credentials).
+
+        A persistent conversation built against the old driver is detected as
+        stale on the next run and rebuilt, so the swap takes effect there too.
+        """
+        self._driver = value
+
+    @property
+    def options(self) -> dict[str, Any]:
+        """Live per-call options dict (temperature, max_tokens, ...).
+
+        Mutations (``agent.options["temperature"] = 0.2``) apply from the
+        next run onward.
+        """
+        return self._options
+
+    @options.setter
+    def options(self, value: dict[str, Any] | None) -> None:
+        self._options = dict(value) if value else {}
+
+    @property
+    def system_prompt(self):
+        """The configured system prompt (str, Persona, or callable)."""
+        return self._system_prompt
+
+    @system_prompt.setter
+    def system_prompt(self, value) -> None:
+        """Replace the system prompt; re-resolved on every run, including
+        for persistent conversations."""
+        self._system_prompt = value
+
     @property
     def conversation(self) -> Any:
         """The current persistent conversation, or ``None``."""
@@ -921,6 +959,9 @@ class AsyncAgent(Generic[DepsType]):
                 # Respect the newly resolved system prompt for this run.
                 if system_prompt is not None:
                     conv.system_prompt = system_prompt
+                # Pick up option mutations made since the conversation was
+                # built (agent.options is a live public dict).
+                conv._options = dict(self._options)
                 # NOTE: assigning ``callbacks`` mutates the driver instance,
                 # which may be shared (e.g. passed to several agents).  A
                 # persistent-conversation agent should treat its driver as

{prompture-1.9.1.dev2 → prompture-1.9.1.dev3}/prompture/agents/async_conversation.py

@@ -187,6 +187,19 @@ class AsyncConversation:
         """The reasoning/thinking content from the last LLM response, if any."""
         return self._last_reasoning
 
+    def _assistant_message(self, content: str) -> dict[str, Any]:
+        """Assistant history entry for the response just consumed.
+
+        Attaches ``reasoning_content`` when the response carried it, so the
+        trace survives in history on every path (plain ask, tool loop,
+        streaming, JSON asks) -- not only the with-tool-calls branch -- and
+        providers that require reasoning to be echoed back keep working.
+        """
+        msg: dict[str, Any] = {"role": "assistant", "content": content}
+        if self._last_reasoning:
+            msg["reasoning_content"] = self._last_reasoning
+        return msg
+
     @property
     def messages(self) -> list[dict[str, Any]]:
         """Read-only view of the conversation history."""
@@ -546,7 +559,7 @@ class AsyncConversation:
         text: str = resp.get("text", "")
         self._last_reasoning = resp.get("reasoning_content")
         self._accumulate_usage(resp.get("meta", {}))
-        self._messages.append({"role": "assistant", "content": text})
+        self._messages.append(self._assistant_message(text))
         return text
 
     async def _final_answer_simulated(self, augmented_system: str, merged: dict[str, Any]) -> str:
@@ -642,7 +655,7 @@ class AsyncConversation:
 
         user_content = self._build_content_with_images(content, images)
         self._messages.append({"role": "user", "content": user_content})
-        self._messages.append({"role": "assistant", "content": text})
+        self._messages.append(self._assistant_message(text))
         self._accumulate_usage(meta)
 
         return text
@@ -680,7 +693,7 @@ class AsyncConversation:
 
             if not tool_calls:
                 self._last_reasoning = resp.get("reasoning_content")
-                self._messages.append({"role": "assistant", "content": text})
+                self._messages.append(self._assistant_message(text))
                 return text
 
             # Ensure every tool call has a usable id so parallel calls
@@ -782,7 +795,7 @@ class AsyncConversation:
 
             if not tool_calls:
                 self._last_reasoning = resp.get("reasoning_content")
-                self._messages.append({"role": "assistant", "content": text})
+                self._messages.append(self._assistant_message(text))
                 yield {"type": "text_delta", "text": text}
                 return
 
@@ -1259,7 +1272,7 @@ class AsyncConversation:
             )
             raise
 
-        self._messages.append({"role": "assistant", "content": full_text})
+        self._messages.append(self._assistant_message(full_text))
 
     async def ask_stream(
         self,
@@ -1382,7 +1395,7 @@ class AsyncConversation:
             else:
                 raise
 
-        self._messages.append({"role": "assistant", "content": cleaned})
+        self._messages.append(self._assistant_message(cleaned))
         self._accumulate_usage(meta)
 
         model_name = self._model_name

{prompture-1.9.1.dev2 → prompture-1.9.1.dev3}/prompture/agents/conversation.py

@@ -189,6 +189,19 @@ class Conversation:
         """The reasoning/thinking content from the last LLM response, if any."""
         return self._last_reasoning
 
+    def _assistant_message(self, content: str) -> dict[str, Any]:
+        """Assistant history entry for the response just consumed.
+
+        Attaches ``reasoning_content`` when the response carried it, so the
+        trace survives in history on every path (plain ask, tool loop,
+        streaming, JSON asks) -- not only the with-tool-calls branch -- and
+        providers that require reasoning to be echoed back keep working.
+        """
+        msg: dict[str, Any] = {"role": "assistant", "content": content}
+        if self._last_reasoning:
+            msg["reasoning_content"] = self._last_reasoning
+        return msg
+
     @property
     def messages(self) -> list[dict[str, Any]]:
         """Read-only view of the conversation history."""
@@ -550,7 +563,7 @@ class Conversation:
         text: str = resp.get("text", "")
         self._last_reasoning = resp.get("reasoning_content")
         self._accumulate_usage(resp.get("meta", {}))
-        self._messages.append({"role": "assistant", "content": text})
+        self._messages.append(self._assistant_message(text))
         return text
 
     def _final_answer_simulated(self, augmented_system: str, merged: dict[str, Any]) -> str:
@@ -656,7 +669,7 @@ class Conversation:
         # Record in history — store content with images for context
         user_content = self._build_content_with_images(content, images)
         self._messages.append({"role": "user", "content": user_content})
-        self._messages.append({"role": "assistant", "content": text})
+        self._messages.append(self._assistant_message(text))
         self._accumulate_usage(meta)
 
         return text
@@ -696,7 +709,7 @@ class Conversation:
             if not tool_calls:
                 # No tool calls -> final response
                 self._last_reasoning = resp.get("reasoning_content")
-                self._messages.append({"role": "assistant", "content": text})
+                self._messages.append(self._assistant_message(text))
                 return text
 
             # Ensure every tool call has a usable id so parallel calls
@@ -808,7 +821,7 @@ class Conversation:
             if not tool_calls:
                 # Final response — stream it if possible
                 self._last_reasoning = resp.get("reasoning_content")
-                self._messages.append({"role": "assistant", "content": text})
+                self._messages.append(self._assistant_message(text))
                 # Yield final text as deltas (single chunk since we already have the full text)
                 yield {"type": "text_delta", "text": text}
                 return
@@ -1296,7 +1309,7 @@ class Conversation:
             )
             raise
 
-        self._messages.append({"role": "assistant", "content": full_text})
+        self._messages.append(self._assistant_message(full_text))
 
     def ask_stream(
         self,
@@ -1517,7 +1530,7 @@ class Conversation:
                 raise
 
         # Store assistant response in history
-        self._messages.append({"role": "assistant", "content": cleaned})
+        self._messages.append(self._assistant_message(cleaned))
         self._accumulate_usage(meta)
 
         model_name = self._model_name

{prompture-1.9.1.dev2 → prompture-1.9.1.dev3}/prompture/groups/debate.py

@@ -98,11 +98,15 @@ class DebateGroup:
         *,
         state: dict[str, Any] | None = None,
         callbacks: GroupCallbacks | None = None,
+        deps: Any | None = None,
     ) -> None:
         self._agents = [(a, _agent_name(a, i)) for i, a in enumerate(agents)]
         self._config = config or DebateConfig()
         self._state: dict[str, Any] = dict(state) if state else {}
         self._callbacks = callbacks or GroupCallbacks()
+        # Forwarded to every debater and the judge as run(prompt, deps=...),
+        # so their tools receive a RunContext.deps like in plain agent runs.
+        self._deps = deps
         self._stop_requested = False
 
     def stop(self) -> None:
@@ -177,7 +181,7 @@ class DebateGroup:
 
                 step_t0 = time.perf_counter()
                 try:
-                    result = agent.run(effective)
+                    result = agent.run(effective, deps=self._deps) if self._deps is not None else agent.run(effective)
                     duration_ms = (time.perf_counter() - step_t0) * 1000
 
                     result_key = f"{name}_round{round_num}"
@@ -247,7 +251,7 @@ class DebateGroup:
 
             step_t0 = time.perf_counter()
             try:
-                judge_result = judge.run(judge_prompt)
+                judge_result = judge.run(judge_prompt, deps=self._deps) if self._deps is not None else judge.run(judge_prompt)
                 duration_ms = (time.perf_counter() - step_t0) * 1000
                 judge_verdict = judge_result.output_text
                 agent_results[judge_name] = judge_result
@@ -310,11 +314,15 @@ class AsyncDebateGroup:
         *,
         state: dict[str, Any] | None = None,
         callbacks: GroupCallbacks | None = None,
+        deps: Any | None = None,
     ) -> None:
         self._agents = [(a, _agent_name(a, i)) for i, a in enumerate(agents)]
         self._config = config or DebateConfig()
         self._state: dict[str, Any] = dict(state) if state else {}
         self._callbacks = callbacks or GroupCallbacks()
+        # Forwarded to every debater and the judge as run(prompt, deps=...),
+        # so their tools receive a RunContext.deps like in plain agent runs.
+        self._deps = deps
         self._stop_requested = False
 
     def stop(self) -> None:
@@ -384,7 +392,7 @@ class AsyncDebateGroup:
 
                 step_t0 = time.perf_counter()
                 try:
-                    result = await agent.run(effective)
+                    result = await (agent.run(effective, deps=self._deps) if self._deps is not None else agent.run(effective))
                     duration_ms = (time.perf_counter() - step_t0) * 1000
 
                     result_key = f"{name}_round{round_num}"
@@ -451,7 +459,7 @@ class AsyncDebateGroup:
 
             step_t0 = time.perf_counter()
             try:
-                judge_result = await judge.run(judge_prompt)
+                judge_result = await (judge.run(judge_prompt, deps=self._deps) if self._deps is not None else judge.run(judge_prompt))
                 duration_ms = (time.perf_counter() - step_t0) * 1000
                 judge_verdict = judge_result.output_text
                 agent_results[judge_name] = judge_result

{prompture-1.9.1.dev2 → prompture-1.9.1.dev3}/prompture.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompture
-Version: 1.9.1.dev2
+Version: 1.9.1.dev3
 Summary: Ask LLMs to return structured JSON and run cross-model tests. API-first.
 Author-email: Juan Denis <juan@vene.co>
 License-Expression: MIT