pygpt-net 2.6.60__py3-none-any.whl → 2.6.62__py3-none-any.whl

pygpt_net/provider/agents/llama_index/workflow/openai.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.14 01:00:00                  #
+# Updated Date: 2025.09.26 15:00:00                  #
 # ================================================== #
 
 from __future__ import annotations
@@ -34,6 +34,7 @@ from llama_index.core.agent.workflow import (
     ToolCallResult,
     AgentStream,
     AgentOutput,
+    AgentInput,  # ensure AgentInput propagation includes agent name
 )
 
 # v12/v13 compatibility imports
@@ -167,6 +168,9 @@ class OpenAIWorkflowAgent(Workflow):
         self._on_stop = on_stop
         self.verbose = verbose
 
+        # human-friendly display name propagated to UI via workflow events
+        self._display_agent_name: str = "FunctionAgent"
+
         # construct FunctionAgent once, will override tools/system_prompt/memory per run
         self._agent = FunctionAgent(
             name="OpenAIWorkflowAgent",
@@ -287,9 +291,13 @@ class OpenAIWorkflowAgent(Workflow):
         :param total: Total number of steps (optional)
         :param meta: Optional metadata dictionary for the step event
         """
+        # Always include agent_name so UI can set it before first token arrives.
+        m = dict(meta or {})
+        m.setdefault("agent_name", self._display_agent_name)
+
         try:
             if StepEvent is not None:
-                ctx.write_event_to_stream(StepEvent(name=name, index=index, total=total, meta=meta or {}))
+                ctx.write_event_to_stream(StepEvent(name=name, index=index, total=total, meta=m))
                 return
         except Exception:
             pass
@@ -300,9 +308,9 @@ class OpenAIWorkflowAgent(Workflow):
                 AgentStream(
                     delta="",
                     response="",
-                    current_agent_name="OpenAIWorkflowAgent",
+                    current_agent_name=self._display_agent_name,
                     tool_calls=[],
-                    raw={"StepEvent": {"name": name, "index": index, "total": total, "meta": meta or {}}},
+                    raw={"StepEvent": {"name": name, "index": index, "total": total, "meta": m}},
                 )
             )
         except Exception:
@@ -499,14 +507,14 @@ class OpenAIWorkflowAgent(Workflow):
             self,
             ctx: Context,
             text: str,
-            agent_name: str = "OpenAIWorkflowAgent"
+            agent_name: str = "FunctionAgent"
     ):
         """
         Emit text to the context stream, handling validation errors gracefully.
 
         :param ctx: Context for the workflow
         :param text: Text to emit to the stream
-        :param agent_name: Name of the agent to set in the event (default: "OpenAIWorkflowAgent")
+        :param agent_name: Name of the agent to set in the event (default: "FunctionAgent")
         """
         try:
             ctx.write_event_to_stream(AgentStream(delta=text))
@@ -561,12 +569,31 @@ class OpenAIWorkflowAgent(Workflow):
                         pass
                     return last_answer
 
+                if isinstance(e, AgentInput):
+                    # Ensure the input event also carries the display name for UI
+                    try:
+                        e.current_agent_name = self._display_agent_name
+                    except Exception:
+                        pass
+                    ctx.write_event_to_stream(e)
+                    continue
+
                 if isinstance(e, AgentStream):
                     if getattr(e, "delta", None):
                         has_stream = True
-                    if not getattr(e, "current_agent_name", None):
+                    # Always enforce agent name for consistency in UI
+                    try:
+                        e.current_agent_name = self._display_agent_name
+                    except Exception:
+                        # If immutable, rebuild a compatible event object
                         try:
-                            e.current_agent_name = "OpenAIWorkflowAgent"
+                            e = AgentStream(
+                                delta=getattr(e, "delta", ""),
+                                response=getattr(e, "response", ""),
+                                current_agent_name=self._display_agent_name,
+                                tool_calls=getattr(e, "tool_calls", []),
+                                raw=getattr(e, "raw", {}),
+                            )
                         except Exception:
                             pass
                     ctx.write_event_to_stream(e)
@@ -581,9 +608,9 @@ class OpenAIWorkflowAgent(Workflow):
                             AgentStream(
                                 delta=content,
                                 response=content,
-                                current_agent_name="OpenAIWorkflowAgent",
-                                tool_calls=e.tool_calls,
-                                raw=e.raw,
+                                current_agent_name=self._display_agent_name,
+                                tool_calls=getattr(e, "tool_calls", []),
+                                raw=getattr(e, "raw", {}),
                             )
                         )
                     continue

pygpt_net/provider/agents/llama_index/workflow/planner.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.24 02:00:00                  #
+# Updated Date: 2025.09.26 15:20:00                  #
 # ================================================== #
 
 from typing import List, Optional, Callable
@@ -134,6 +134,10 @@ class PlannerWorkflow(Workflow):
         self._memory_char_limit = memory_char_limit
         self._on_stop = on_stop
 
+        # Human-friendly display names propagated to UI via workflow events.
+        self._display_planner_name: str = "PlannerWorkflow"
+        self._display_executor_name: str = "FunctionAgent"
+
         self._executor = FunctionAgent(
             name="PlannerExecutor",
             description="Executes planner sub-tasks using available tools.",
@@ -184,9 +188,13 @@ class PlannerWorkflow(Workflow):
         :param total: The total number of steps (optional).
         :param meta: Additional metadata for the step (optional).
         """
+        # Always pass a friendly agent name; for 'subtask' we signal the executor agent to the UI.
+        m = dict(meta or {})
+        m.setdefault("agent_name", self._display_executor_name if name == "subtask" else self._display_planner_name)
+
         try:
             ctx.write_event_to_stream(
-                StepEvent(name=name, index=index, total=total, meta=meta or {})
+                StepEvent(name=name, index=index, total=total, meta=m)
             )
         except Exception:
             # fallback for older versions of AgentStream
@@ -195,9 +203,9 @@ class PlannerWorkflow(Workflow):
                     AgentStream(
                         delta="",
                         response="",
-                        current_agent_name="PlannerWorkflow",
+                        current_agent_name=m.get("agent_name", self._display_planner_name),
                         tool_calls=[],
-                        raw={"StepEvent": {"name": name, "index": index, "total": total, "meta": meta or {}}}
+                        raw={"StepEvent": {"name": name, "index": index, "total": total, "meta": m}}
                     )
                 )
             except Exception:
@@ -308,9 +316,8 @@ class PlannerWorkflow(Workflow):
         :param text: The text message to emit.
         :param agent_name: The name of the agent emitting the text (default: "PlannerWorkflow").
         """
+        # Always try to include agent name; fall back to minimal event for older validators.
         try:
-            ctx.write_event_to_stream(AgentStream(delta=text))
-        except ValidationError:
             ctx.write_event_to_stream(
                 AgentStream(
                     delta=text,
@@ -320,6 +327,8 @@ class PlannerWorkflow(Workflow):
                     raw={},
                 )
             )
+        except ValidationError:
+            ctx.write_event_to_stream(AgentStream(delta=text))
 
     def _to_text(self, resp) -> str:
         """
@@ -443,9 +452,18 @@ class PlannerWorkflow(Workflow):
                     if delta:
                         has_stream = True
                         stream_buf.append(str(delta))
-                    if not getattr(e, "current_agent_name", None):
+                    # Always enforce a stable display name for executor events.
+                    try:
+                        e.current_agent_name = self._display_executor_name
+                    except Exception:
                         try:
-                            e.current_agent_name = self._executor.name
+                            e = AgentStream(
+                                delta=getattr(e, "delta", ""),
+                                response=getattr(e, "response", ""),
+                                current_agent_name=self._display_executor_name,
+                                tool_calls=getattr(e, "tool_calls", []),
+                                raw=getattr(e, "raw", {}),
+                            )
                         except Exception:
                             pass
                     ctx.write_event_to_stream(e)
@@ -460,7 +478,7 @@ class PlannerWorkflow(Workflow):
                             AgentStream(
                                 delta=last_answer,
                                 response=last_answer,
-                                current_agent_name=f"{self._executor.name} (subtask)",
+                                current_agent_name=self._display_executor_name,
                                 tool_calls=e.tool_calls,
                                 raw=e.raw,
                             )
@@ -484,7 +502,7 @@ class PlannerWorkflow(Workflow):
         try:
             return await _stream()
         except Exception as ex:
-            await self._emit_text(ctx, f"\n`Sub-task failed: {ex}`")
+            await self._emit_text(ctx, f"\n`Sub-task failed: {ex}`", agent_name=self._display_executor_name)
             return last_answer or ("".join(stream_buf).strip() if stream_buf else "")
 
     @step
@@ -520,7 +538,7 @@ class PlannerWorkflow(Workflow):
                 f"Expected output: {st.expected_output}\n"
                 f"Dependencies: {st.dependencies}\n\n"
             )
-        await self._emit_text(ctx, "\n".join(lines))
+        await self._emit_text(ctx, "\n".join(lines), agent_name=self._display_planner_name)
         return PlanReady(plan=plan, query=ev.query)
 
     @step
@@ -537,7 +555,7 @@ class PlannerWorkflow(Workflow):
         last_answer = ""
         completed: list[tuple[str, str]] = []  # (name, output)
 
-        await self._emit_text(ctx, "\n\n`Executing plan...`")
+        await self._emit_text(ctx, "\n\n`Executing plan...`", agent_name=self._display_planner_name)
 
         for i, st in enumerate(plan_sub_tasks, 1):
             self._emit_step_event(
@@ -550,6 +568,8 @@ class PlannerWorkflow(Workflow):
                     "expected_output": st.expected_output,
                     "dependencies": st.dependencies,
                     "input": st.input,
+                    # Signal that the next stream will come from the executor.
+                    "agent_name": self._display_executor_name,
                 },
             )
 
@@ -561,10 +581,10 @@ class PlannerWorkflow(Workflow):
 
             # stop callback
             if self._stopped():
-                await self._emit_text(ctx, "\n`Plan execution stopped.`")
+                await self._emit_text(ctx, "\n`Plan execution stopped.`", agent_name=self._display_planner_name)
                 return FinalEvent(result=last_answer or "Plan execution stopped.")
 
-            await self._emit_text(ctx, header)
+            await self._emit_text(ctx, header, agent_name=self._display_planner_name)
 
             # build context for sub-task
             ctx_text = self._build_context_for_subtask(
@@ -588,7 +608,7 @@ class PlannerWorkflow(Workflow):
             sub_answer = await self._run_subtask(ctx, composed_prompt)
             sub_answer = (sub_answer or "").strip()
 
-            await self._emit_text(ctx, f"\n\n`Finished Sub Task {i}/{total}: {st.name}`")
+            await self._emit_text(ctx, f"\n\n`Finished Sub Task {i}/{total}: {st.name}`", agent_name=self._display_planner_name)
 
             # save completed sub-task
             completed.append((st.name, sub_answer))
@@ -597,5 +617,5 @@ class PlannerWorkflow(Workflow):
 
             # TODO: refine plan if needed
 
-        await self._emit_text(ctx, "\n\n`Plan execution finished.`")
+        await self._emit_text(ctx, "\n\n`Plan execution finished.`", agent_name=self._display_planner_name)
         return FinalEvent(result=last_answer or "Plan finished.")

pygpt_net/provider/agents/llama_index/workflow/supervisor.py

@@ -1,3 +1,5 @@
+# workflow/supervisor.py
+
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 # ================================================== #
@@ -6,11 +8,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.17 02:00:00                  #
+# Updated Date: 2025.09.26 22:25:00                  #
 # ================================================== #
 
 import re
-from typing import Optional, Literal, List
+from typing import Optional, Literal, List, Callable, Any
 from pydantic import BaseModel, ValidationError
 from llama_index.core.workflow import Workflow, Context, StartEvent, StopEvent, Event, step
 from llama_index.core.agent.workflow import FunctionAgent, AgentStream
@@ -173,8 +175,6 @@ class SupervisorWorkflow(Workflow):
         :param agent_name: The name of the agent emitting the text (default: "PlannerWorkflow").
         """
         try:
-            ctx.write_event_to_stream(AgentStream(delta=text))
-        except ValidationError:
             ctx.write_event_to_stream(
                 AgentStream(
                     delta=text,
@@ -184,11 +184,47 @@ class SupervisorWorkflow(Workflow):
                     raw={},
                 )
             )
+        except ValidationError:
+            ctx.write_event_to_stream(AgentStream(delta=text))
+
+    async def _emit_step(self, ctx: Context, agent_name: str, index: int, total: int, meta: Optional[dict] = None):
+        """
+        Emit a StepEvent that your runner uses to split UI into blocks.
+        Mirrors the behavior used by the schema-driven workflow.
+        """
+        from pygpt_net.provider.agents.llama_index.workflow.events import StepEvent
+        try:
+            ctx.write_event_to_stream(
+                StepEvent(
+                    name="next",
+                    index=index,
+                    total=total,
+                    meta={"agent_name": agent_name, **(meta or {})},
+                )
+            )
+        except Exception:
+            pass
+
+    async def _run_muted(self, ctx: Context, awaitable) -> Any:
+        """
+        Execute an agent call while muting all events sent to ctx.
+        Matches schema-style emission: we control all UI events ourselves.
+        """
+        orig_write = ctx.write_event_to_stream
+
+        def _noop(ev: Any) -> None:
+            return None
+
+        ctx.write_event_to_stream = _noop
+        try:
+            return await awaitable
+        finally:
+            ctx.write_event_to_stream = orig_write
 
     @step
     async def supervisor_step(self, ctx: Context, ev: InputEvent) -> ExecuteEvent | OutputEvent:
         """
-        Supervisor step to process the user's input and generate an instruction for the Worker.
+        Supervisor step: run Supervisor silently, then emit exactly one UI block like schema.
 
         :param ctx: Context for the workflow
         :param ev: InputEvent containing the user's message and context.
@@ -206,21 +242,33 @@ class SupervisorWorkflow(Workflow):
             "Return ONE JSON following the schema.\n</control>"
         )
         sup_input = "\n".join(parts)
-        sup_resp = await self._supervisor.run(user_msg=sup_input, memory=self._supervisor_memory)
+
+        # Run Supervisor with stream muted to avoid extra blocks/finishes.
+        sup_resp = await self._run_muted(ctx, self._supervisor.run(user_msg=sup_input, memory=self._supervisor_memory))
         directive = parse_supervisor_json(str(sup_resp))
 
+        # Final/ask_user/max_rounds -> emit single Supervisor block and stop (schema-like).
         if directive.action == "final":
+            await self._emit_step(ctx, agent_name=self._supervisor.name, index=ev.round_idx + 1, total=ev.max_rounds)
             await self._emit_text(ctx, f"\n\n{directive.final_answer or str(sup_resp)}", agent_name=self._supervisor.name)
             return OutputEvent(status="final", final_answer=directive.final_answer or str(sup_resp), rounds_used=ev.round_idx)
+
         if directive.action == "ask_user" and ev.stop_on_ask_user:
+            await self._emit_step(ctx, agent_name=self._supervisor.name, index=ev.round_idx + 1, total=ev.max_rounds)
             q = directive.question or "I need more information, please clarify."
             await self._emit_text(ctx, f"\n\n{q}", agent_name=self._supervisor.name)
             return OutputEvent(status="ask_user", final_answer=q, rounds_used=ev.round_idx)
+
         if ev.round_idx >= ev.max_rounds:
+            await self._emit_step(ctx, agent_name=self._supervisor.name, index=ev.round_idx + 1, total=ev.max_rounds)
             await self._emit_text(ctx, "\n\nMax rounds exceeded.", agent_name=self._supervisor.name)
             return OutputEvent(status="max_rounds", final_answer="Exceeded maximum number of iterations.", rounds_used=ev.round_idx)
 
+        # Emit exactly one Supervisor block with the instruction (no JSON leakage, no duplicates).
         instruction = (directive.instruction or "").strip() or "Perform a step that gets closest to fulfilling the DoD."
+        await self._emit_step(ctx, agent_name=self._supervisor.name, index=ev.round_idx + 1, total=ev.max_rounds)
+        await self._emit_text(ctx, f"\n\n{instruction}", agent_name=self._supervisor.name)
+
         return ExecuteEvent(
             instruction=instruction,
             round_idx=ev.round_idx,
@@ -232,17 +280,19 @@ class SupervisorWorkflow(Workflow):
     @step
     async def worker_step(self, ctx: Context, ev: ExecuteEvent) -> InputEvent:
         """
-        Worker step to execute the Supervisor's instruction.
+        Worker step: run Worker silently and emit exactly one UI block like schema.
 
         :param ctx: Context for the workflow
         :param ev: ExecuteEvent containing the instruction and context.
         :return: InputEvent for the next round or final output.
         """
+        # Run Worker with stream muted; we will emit a single block with the final text.
         worker_input = f"Instruction from Supervisor:\n{ev.instruction}\n"
-        await self._emit_text(ctx, f"\n\n**Supervisor:** {ev.instruction}", agent_name=self._worker.name)
+        worker_resp = await self._run_muted(ctx, self._worker.run(user_msg=worker_input, memory=self._worker_memory))
 
-        worker_resp = await self._worker.run(user_msg=worker_input, memory=self._worker_memory)
-        await self._emit_text(ctx, f"\n\n**Worker:** {worker_resp}", agent_name=self._worker.name)
+        # Emit exactly one Worker block (schema-style: one AgentStream per node).
+        await self._emit_step(ctx, agent_name=self._worker.name, index=ev.round_idx + 1, total=ev.max_rounds)
+        await self._emit_text(ctx, f"\n\n{str(worker_resp)}", agent_name=self._worker.name)
 
         return InputEvent(
             user_msg="",

pygpt_net/provider/agents/openai/agent.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.24 03:00:00                  #
+# Updated Date: 2025.09.26 17:00:00                  #
 # ================================================== #
 
 from typing import Dict, Any, Tuple, Optional
@@ -133,6 +133,8 @@ class Agent(BaseAgent):
             "preset": preset,
             "is_expert_call": False,
         }
+
+        ctx.set_agent_name(agent.name)
         # call computer agent if computer tool is enabled
         if is_computer_tool(**tool_kwargs):
             computer = LocalComputer(window)

pygpt_net/provider/agents/openai/agent_b2b.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.26 01:00:00                  #
+# Updated Date: 2025.09.26 17:00:00                  #
 # ================================================== #
 
 import copy
@@ -265,6 +265,7 @@ class Agent(BaseAgent):
                     break
 
                 kwargs = self.prepare_model(model, window, previous_response_id, kwargs)
+                ctx.set_agent_name(bot_1.name)
                 result = await Runner.run(
                     bot_1,
                     **kwargs
@@ -296,6 +297,7 @@ class Agent(BaseAgent):
                 # -------- bot 2 --------
                 kwargs["input"] = input_items
                 kwargs = self.prepare_model(model_2, window, previous_response_id, kwargs)
+                ctx.set_agent_name(bot_2.name)
                 result = await Runner.run(
                     bot_2,
                     **kwargs
@@ -329,6 +331,7 @@ class Agent(BaseAgent):
                 # -------- bot 1 --------
                 kwargs["input"] = input_items
                 kwargs = self.prepare_model(model, window, previous_response_id, kwargs)
+                ctx.set_agent_name(bot_1.name)
                 result = Runner.run_streamed(
                     bot_1,
                     **kwargs
@@ -337,13 +340,13 @@ class Agent(BaseAgent):
                 handler.reset()
 
                 # bot 1 title
-                title = f"\n\n**{bot_1_name}**\n\n"
-                ctx.stream = title
+                # title = f"\n\n**{bot_1_name}**\n\n"
+                # ctx.stream = title
                 bridge.on_step(ctx, begin)
                 begin = False
                 handler.begin = begin
-                if not use_partial_ctx:
-                    handler.to_buffer(title)
+                # if not use_partial_ctx:
+                    # handler.to_buffer(title)
                 async for event in result.stream_events():
                     if bridge.stopped():
                         result.cancel()
@@ -376,6 +379,7 @@ class Agent(BaseAgent):
                 # -------- bot 2 --------
                 kwargs["input"] = input_items
                 kwargs = self.prepare_model(model_2, window, previous_response_id, kwargs)
+                ctx.set_agent_name(bot_2.name)
                 result = Runner.run_streamed(
                     bot_2,
                     **kwargs
@@ -383,11 +387,11 @@ class Agent(BaseAgent):
                 handler.reset()
 
                 # bot 2 title
-                title = f"\n\n**{bot_2_name}**\n\n"
-                ctx.stream = title
+                # title = f"\n\n**{bot_2_name}**\n\n"
+                # ctx.stream = title
                 bridge.on_step(ctx, False)
-                if not use_partial_ctx:
-                    handler.to_buffer(title)
+                # if not use_partial_ctx:
+                    # handler.to_buffer(title)
                 async for event in result.stream_events():
                     if bridge.stopped():
                         result.cancel()

pygpt_net/provider/agents/openai/agent_planner.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.26 01:00:00                  #
+# Updated Date: 2025.09.26 17:00:00                  #
 # ================================================== #
 
 from dataclasses import dataclass
@@ -153,7 +153,7 @@ class Agent(BaseAgent):
         :return: Agent provider instance
         """
         kwargs = {
-            "name": "evaluator",
+            "name": "Evaluator",
             "instructions": instructions,
             "model": window.core.agents.provider.get_openai_model(model),
             "output_type": EvaluationFeedback,
@@ -313,6 +313,7 @@ class Agent(BaseAgent):
         if not stream:
             while True:
                 kwargs["input"] = input_items
+                ctx.set_agent_name(agent.name)
                 if bridge.stopped():
                     bridge.on_stop(ctx)
                     break
@@ -333,6 +334,7 @@ class Agent(BaseAgent):
                     break
 
                 evaluator_result = await Runner.run(evaluator, input_items)
+                ctx.set_agent_name(evaluator.name)
                 result: EvaluationFeedback = evaluator_result.final_output
 
                 print(f"Evaluator score: {result.score}")
@@ -366,6 +368,7 @@ class Agent(BaseAgent):
             handler = StreamHandler(window, bridge, final_output)
             while True:
                 kwargs["input"] = input_items
+                ctx.set_agent_name(agent.name)
                 result = Runner.run_streamed(
                     agent,
                     **kwargs
@@ -386,6 +389,7 @@ class Agent(BaseAgent):
                     break
 
                 input_items = result.to_input_list()
+                ctx.set_agent_name(evaluator.name)
                 evaluator_result = await Runner.run(evaluator, input_items)
                 result: EvaluationFeedback = evaluator_result.final_output
 

pygpt_net/provider/agents/openai/agent_with_experts.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.24 03:00:00                  #
+# Updated Date: 2025.09.26 17:00:00                  #
 # ================================================== #
 
 from typing import Dict, Any, Tuple, Optional
@@ -123,6 +123,7 @@ class Agent(BaseAgent):
             agent_kwargs["handoffs"] = experts
 
         agent = self.get_agent(window, agent_kwargs)
+        ctx.set_agent_name(agent.name)
 
         kwargs = {
             "input": messages,
@@ -137,6 +138,7 @@ class Agent(BaseAgent):
                 agent,
                 **kwargs
             )
+            ctx.set_agent_name(agent.name)
             final_output, last_response_id = window.core.api.openai.responses.unpack_agent_response(result, ctx)
             response_id = result.last_response_id
             if verbose:
@@ -152,6 +154,7 @@ class Agent(BaseAgent):
                     result.cancel()
                     bridge.on_stop(ctx)
                     break
+                ctx.set_agent_name(agent.name)
                 final_output, response_id = handler.handle(event, ctx)
 
         return ctx, final_output, response_id

pygpt_net/provider/agents/openai/agent_with_experts_feedback.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.26 01:00:00                  #
+# Updated Date: 2025.09.26 17:00:00                  #
 # ================================================== #
 
 from dataclasses import dataclass
@@ -119,7 +119,7 @@ class Agent(BaseAgent):
         :return: Agent provider instance
         """
         kwargs = {
-            "name": "evaluator",
+            "name": "Evaluator",
             "instructions": instructions,
             "model": window.core.agents.provider.get_openai_model(model),
             "output_type": EvaluationFeedback,
@@ -212,6 +212,7 @@ class Agent(BaseAgent):
                     bridge.on_stop(ctx)
                     break
 
+                ctx.set_agent_name(agent.name)
                 result = await Runner.run(
                     agent,
                     **kwargs
@@ -259,6 +260,7 @@ class Agent(BaseAgent):
             handler = StreamHandler(window, bridge)
             while True:
                 kwargs["input"] = input_items
+                ctx.set_agent_name(agent.name)
                 result = Runner.run_streamed(
                     agent,
                     **kwargs

pygpt_net/provider/agents/openai/agent_with_feedback.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.26 01:00:00                  #
+# Updated Date: 2025.09.26 17:00:00                  #
 # ================================================== #
 
 from dataclasses import dataclass
@@ -119,7 +119,7 @@ class Agent(BaseAgent):
         :return: Agent provider instance
         """
         kwargs = {
-            "name": "evaluator",
+            "name": "Evaluator",
             "instructions": instructions,
             "model": window.core.agents.provider.get_openai_model(model),
             "output_type": EvaluationFeedback,
@@ -212,6 +212,7 @@ class Agent(BaseAgent):
                     bridge.on_stop(ctx)
                     break
 
+                ctx.set_agent_name(agent.name)
                 result = await Runner.run(
                     agent,
                     **kwargs
@@ -259,6 +260,7 @@ class Agent(BaseAgent):
             handler = StreamHandler(window, bridge)
             while True:
                 kwargs["input"] = input_items
+                ctx.set_agent_name(agent.name)
                 result = Runner.run_streamed(
                     agent,
                     **kwargs