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

pygpt_net/core/render/web/renderer.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.09.17 05:00:00                  #
+# Updated Date: 2025.09.26 17:00:00                  #
 # ================================================== #
 
 import json
@@ -16,15 +16,13 @@ import html as _html
 from dataclasses import dataclass, field
 
 from datetime import datetime
-from typing import Optional, List, Any, Dict, Tuple
+from typing import Optional, List, Any, Tuple
 from time import monotonic
 from io import StringIO
 
-from PySide6.QtCore import QTimer, QUrl, QCoreApplication, QEventLoop, QEvent
-from PySide6.QtWebEngineCore import QWebEnginePage
+from PySide6.QtCore import QTimer, QCoreApplication, QEventLoop, QEvent
 
 from pygpt_net.core.render.base import BaseRenderer
-from pygpt_net.core.text.utils import has_unclosed_code_tag
 from pygpt_net.item.ctx import CtxItem, CtxMeta
 from pygpt_net.ui.widget.textarea.input import ChatInput
 from pygpt_net.ui.widget.textarea.web import ChatWebOutput
@@ -155,6 +153,7 @@ class Renderer(BaseRenderer):
         app_path = self.window.core.config.get_app_path() if self.window else ""
         self._icon_expand = os.path.join(app_path, "data", "icons", "expand.svg")
         self._icon_sync = os.path.join(app_path, "data", "icons", "sync.svg")
+        self._agent_avatar = os.path.join(app_path, "data", "icons", "robot.svg")
         self._file_prefix = 'file:///' if self.window and self.window.core.platforms.is_windows() else 'file://'
 
         # Bridge readiness for node append/replace path
@@ -378,6 +377,15 @@ class Renderer(BaseRenderer):
         self.tool_output_end()
         self.prev_chunk_replace = False
 
+        # Ensure stream header identity is up-to-date (agent/preset override)
+        try:
+            header = self.get_name_header(ctx, stream=True)
+            if pid is not None:
+                self.pids[pid].header = header
+                self._stream_header[pid] = header or ""
+        except Exception:
+            pass
+
     def end(self, meta: CtxMeta, ctx: CtxItem, stream: bool = False):
         """
         Render end
@@ -1163,6 +1171,26 @@ class Renderer(BaseRenderer):
         meta = ctx.meta
         if meta is None:
             return ""
+
+        # Agent-provided display name override:
+        # If ctx.get_agent_name() returns a non-empty name, force "fake personalize":
+        # - use that name
+        # - optionally attach default avatar when enabled via config
+        # - treat as personalized header regardless of preset
+        agent_name = self._get_agent_name(ctx)
+        if agent_name:
+            avatar_html = ""
+            try:
+                use_default = self.window.core.config.get("agent.avatar.default", True)
+                if use_default and os.path.exists(self._agent_avatar):
+                    avatar_html = f"<img src=\"{self._file_prefix}{self._agent_avatar}\" class=\"avatar\"> "
+            except Exception:
+                pass
+            if stream:
+                return f"{avatar_html}{agent_name}"
+            else:
+                return f"<div class=\"name-header name-bot\">{avatar_html}{agent_name}</div>"
+
         preset_id = meta.preset
         if preset_id is None or preset_id == "":
             return ""
@@ -1904,23 +1932,60 @@ class Renderer(BaseRenderer):
 
     # ------------------------- Helpers: build JSON blocks -------------------------
 
+    def _get_agent_name(self, ctx: CtxItem) -> Optional[str]:
+        """
+        Resolve agent-provided name from ctx if available.
+
+        This is used to force "fake personalize" on the UI:
+        - when present and non-empty, we use this name,
+        - optionally attach default avatar when enabled via config,
+        - we set personalize flag to True in node payloads.
+        """
+        try:
+            if hasattr(ctx, "get_agent_name"):
+                name = ctx.get_agent_name()
+                if isinstance(name, str):
+                    name = name.strip()
+                return name or None
+        except Exception:
+            pass
+        return None
+
     def _output_identity(self, ctx: CtxItem) -> Tuple[str, Optional[str], bool]:
         """
-        Resolve output identity (name, avatar file:// path) based on preset.
+        Resolve output identity (name, avatar file:// path) based on preset or ctx-provided agent name.
 
         :param ctx: context item
         :return: (name, avatar, personalize)
         """
+        # 1) Agent-provided name override -> force personalize, optionally default avatar
+        agent_name = self._get_agent_name(ctx)
+        if agent_name:
+            avatar = None
+            try:
+                if self.window.core.config.get("agent.avatar.default", True) and os.path.exists(self._agent_avatar):
+                    avatar = f"{self._file_prefix}{self._agent_avatar}"
+            except Exception:
+                pass
+            return agent_name, avatar, True
+
+        # 2) Fallback to preset-based personalize
         meta = ctx.meta
         if meta is None:
-            return self.pids[self.get_or_create_pid(meta)].name_bot if meta else "", None, False
+            return "", None, False
+
+        pid = self.get_or_create_pid(meta)
+        default_name = self.pids[pid].name_bot if pid in self.pids else ""
+
         preset_id = meta.preset
         if not preset_id:
-            return self.pids[self.get_or_create_pid(meta)].name_bot, None, False
+            return default_name, None, False
+
         preset = self.window.core.presets.get(preset_id)
         if preset is None or not preset.ai_personalize:
-            return self.pids[self.get_or_create_pid(meta)].name_bot, None, False
-        name = preset.ai_name or self.pids[self.get_or_create_pid(meta)].name_bot
+            return default_name, None, False
+
+        name = preset.ai_name or default_name
         avatar = None
         if preset.ai_avatar:
             presets_dir = self.window.core.config.get_user_dir("presets")
@@ -1928,7 +1993,7 @@ class Renderer(BaseRenderer):
             avatar_path = os.path.join(avatars_dir, preset.ai_avatar)
             if os.path.exists(avatar_path):
                 avatar = f"{self._file_prefix}{avatar_path}"
-        return name, avatar, bool(preset.ai_personalize)
+        return name, avatar, True
 
     def _build_render_block(
             self,

pygpt_net/data/config/config.json

@@ -1,7 +1,7 @@
 {
   "__meta__": {
-    "version": "2.6.61",
-    "app.version": "2.6.61",
+    "version": "2.6.62",
+    "app.version": "2.6.62",
     "updated_at": "2025-09-26T00:00:00"
   },
   "access.audio.event.speech": false,

pygpt_net/data/config/models.json

@@ -1,7 +1,7 @@
 {
     "__meta__": {
-        "version": "2.6.61",
-        "app.version": "2.6.61",
+        "version": "2.6.62",
+        "app.version": "2.6.62",
         "updated_at": "2025-09-26T00:00:00"
     },
     "items": {

pygpt_net/data/css/style.dark.css

@@ -144,4 +144,22 @@ QCalendarWidget QAbstractItemView::item:hover {{
 QCalendarWidget QMenu::item:selected:focus,
 QCalendarWidget QMenu::item:selected {{
     background: #3a4045;
+}}
+
+/* Node editor */
+NodeEditor {{
+  qproperty-gridBackColor: #242629;
+  qproperty-gridPenColor: #3b3f46;
+
+  qproperty-nodeBackgroundColor: #2d2f34;
+  qproperty-nodeBorderColor: #4b4f57;
+  qproperty-nodeSelectionColor: #ff9900;
+  qproperty-nodeTitleColor: #3a3d44;
+
+  qproperty-portInputColor: #66b2ff;
+  qproperty-portOutputColor: #70e070;
+  qproperty-portConnectedColor: #ffd166;
+
+  qproperty-edgeColor: #c0c0c0;
+  qproperty-edgeSelectedColor: #ff8a5c;
 }}

pygpt_net/data/css/style.light.css

@@ -6,6 +6,7 @@
 QWidget {{
   color: #000;
   margin: 1px;
+  background-color: #eeeeee;
 }}
 
 QListView, 
@@ -248,7 +249,7 @@ QCalendarWidget QToolButton:hover {{
   height: 20px;
 }} 
 QMenu::indicator {{
-    border: 1px solid gray;   /* tu ustawiasz border */
+    border: 1px solid gray;
 }}
 QCalendarWidget QAbstractItemView::item:selected:focus,
 QCalendarWidget QAbstractItemView::item:selected {{
@@ -259,4 +260,22 @@ QCalendarWidget QAbstractItemView::item:hover {{
 }}
 .file-explorer QTreeView::branch {{
   background: #fff;
+}}
+
+/* Node editor */
+NodeEditor {{
+  qproperty-gridBackColor: #ffffff;
+  qproperty-gridPenColor: #eaeaea;
+
+  qproperty-nodeBackgroundColor: #2d2f34;
+  qproperty-nodeBorderColor: #4b4f57;
+  qproperty-nodeSelectionColor: #ff9900;
+  qproperty-nodeTitleColor: #3a3d44;
+
+  qproperty-portInputColor: #66b2ff;
+  qproperty-portOutputColor: #70e070;
+  qproperty-portConnectedColor: #ffd166;
+
+  qproperty-edgeColor: #c0c0c0;
+  qproperty-edgeSelectedColor: #ff8a5c;
 }}

pygpt_net/data/locale/locale.de.ini

@@ -658,6 +658,7 @@ event.control.voice_cmd.toggle = Sprachsteuerung: Umschalten
 event.control.voice_msg.start = Spracheingabe: Start
 event.control.voice_msg.stop = Spracheingabe: Stopp
 event.control.voice_msg.toggle = Spracheingabe: Umschalten
+exit.msg = Gefällt dir PyGPT? Unterstütze die Entwicklung des Projekts:
 expert.wait.failed: Aufruf des Experten fehlgeschlagen
 expert.wait.status: Warten auf Experten...
 files.delete.confirm = Datei/Verzeichnis löschen?
@@ -734,6 +735,7 @@ input.search.placeholder = Suchen...
 input.send_clear = Nach dem Senden löschen
 input.stream = Streamen
 input.tab = Eingabe
+input.tab.tooltip = {chars} Zeichen (~{tokens} Token)
 interpreter.all = Verlauf ausführen (alle)
 interpreter.auto_clear = Bei Senden löschen
 interpreter.btn.clear = Ausgabe löschen

pygpt_net/data/locale/locale.en.ini

@@ -663,6 +663,7 @@ event.control.voice_cmd.toggle = Voice control: Toggle
 event.control.voice_msg.start = Voice input: Start
 event.control.voice_msg.stop = Voice input: Stop
 event.control.voice_msg.toggle = Voice input: Toggle
+exit.msg = Do you like PyGPT? Support the development of the project:
 expert.wait.failed: Failed calling expert
 expert.wait.status: Waiting for expert...
 files.delete.confirm = Delete file/directory?
@@ -739,6 +740,7 @@ input.search.placeholder = Search...
 input.send_clear = Clear on send
 input.stream = Stream
 input.tab = Input
+input.tab.tooltip = {chars} chars (~{tokens} tokens)
 interpreter.all = Execute history (all)
 interpreter.auto_clear = Clear output on send
 interpreter.btn.clear = Clear output

pygpt_net/data/locale/locale.es.ini

@@ -659,6 +659,7 @@ event.control.voice_cmd.toggle = Control de voz: Conmutar
 event.control.voice_msg.start = Entrada de voz: Iniciar
 event.control.voice_msg.stop = Entrada de voz: Detener
 event.control.voice_msg.toggle = Entrada de voz: Conmutar
+exit.msg = ¿Te gusta PyGPT? Apoya el desarrollo del proyecto:
 expert.wait.failed: Error al llamar al experto
 expert.wait.status: Esperando al experto...
 files.delete.confirm = ¿Eliminar archivo/directorio?
@@ -735,6 +736,7 @@ input.search.placeholder = Buscar...
 input.send_clear = Limpiar al enviar
 input.stream = Transmisión
 input.tab = Entrada
+input.tab.tooltip = {chars} caracteres (~{tokens} fichas)
 interpreter.all = Ejecutar historial (todo)
 interpreter.auto_clear = Limpiar al enviar
 interpreter.btn.clear = Limpiar salida

pygpt_net/data/locale/locale.fr.ini

@@ -658,6 +658,7 @@ event.control.voice_cmd.toggle = Commande vocale : Basculer
 event.control.voice_msg.start = Entrée vocale : Commencer
 event.control.voice_msg.stop = Entrée vocale : Arrêter
 event.control.voice_msg.toggle = Entrée vocale : Basculer
+exit.msg = PyGPT vous plaît-il ? Soutenez le développement du projet :
 expert.wait.failed: Échec de l'appel à l'expert
 expert.wait.status: En attente de l'expert...
 files.delete.confirm = Supprimer le fichier/répertoire ?
@@ -734,6 +735,7 @@ input.search.placeholder = Rechercher...
 input.send_clear = Effacer après envoi
 input.stream = Flux
 input.tab = Entrée
+input.tab.tooltip = {chars} caractères (~{tokens} jetons)
 interpreter.all = Exécuter l'historique (tous)
 interpreter.auto_clear = Effacer à l'envoi
 interpreter.btn.clear = Effacer la sortie

pygpt_net/data/locale/locale.it.ini

@@ -658,6 +658,7 @@ event.control.voice_cmd.toggle = Comando vocale: Commuta
 event.control.voice_msg.start = Input vocale: Avvia
 event.control.voice_msg.stop = Input vocale: Arresta
 event.control.voice_msg.toggle = Input vocale: Commuta
+exit.msg = Ti piace PyGPT? Sostieni lo sviluppo del progetto:
 expert.wait.failed: Chiamata all'esperto non riuscita
 expert.wait.status: In attesa dell'esperto...
 files.delete.confirm = Eliminare il file/la cartella?
@@ -734,6 +735,7 @@ input.search.placeholder = Cerca...
 input.send_clear = Pulisci dopo l'invio
 input.stream = Flusso
 input.tab = Input
+input.tab.tooltip = {chars} caratteri (~{tokens} token)
 interpreter.all = Esegui cronologia (tutto)
 interpreter.auto_clear = Cancella all'invio
 interpreter.btn.clear = Cancella output

pygpt_net/data/locale/locale.pl.ini

@@ -244,7 +244,7 @@ clipboard.copied_to = Skopiowano do schowka:
 cmd.enabled = + Narzędzia
 cmd.tip = Wskazówka: Aby umożliwić wykonanie narzędzi z wtyczek, musisz włączyć opcję "+ Narzędzia".
 coming_soon = Dostępne wkrótce...
-common.down = Przesuń w dół
+common.down = Przesuń w dół
 common.up = Przesuń w górę
 confirm.assistant.delete = Na pewno usunąć asystenta?
 confirm.assistant.files.clear = Wyczyścić pliki (tylko lokalnie)?
@@ -659,6 +659,7 @@ event.control.voice_cmd.toggle = Kontrola głosowa: Przełącz
 event.control.voice_msg.start = Wejście głosowe: Rozpocznij
 event.control.voice_msg.stop = Wejście głosowe: Zatrzymaj
 event.control.voice_msg.toggle = Wejście głosowe: Przełącz
+exit.msg = Podoba Ci się PyGPT? Wesprzyj rozwój projektu:
 expert.wait.failed: Nie udało się wywołać eksperta
 expert.wait.status: Oczekiwanie na eksperta...
 files.delete.confirm = Usunąć plik/katalog?
@@ -735,6 +736,7 @@ input.search.placeholder = Szukaj...
 input.send_clear = Wyczyść po wysłaniu
 input.stream = Stream
 input.tab = Input
+input.tab.tooltip = {chars} znaków (~{tokens} tokenów)
 interpreter.all = Wykonaj historię (wszystko)
 interpreter.auto_clear = Wyczyść wyjście przy wysyłaniu
 interpreter.btn.clear = Wyczyść wyjście

pygpt_net/data/locale/locale.uk.ini

@@ -658,6 +658,7 @@ event.control.voice_cmd.toggle = Керування голосом: Перемк
 event.control.voice_msg.start = Голосовий вхід: Розпочати
 event.control.voice_msg.stop = Голосовий вхід: Зупинити
 event.control.voice_msg.toggle = Голосовий вхід: Перемкнути
+exit.msg = Вам подобається PyGPT? Підтримайте розвиток проєкту:
 expert.wait.failed: Виклик експерта не вдався
 expert.wait.status: Очікування експерта...
 files.delete.confirm = Видалити файл/директорію?
@@ -734,6 +735,7 @@ input.search.placeholder = Пошук...
 input.send_clear = Очистити після відправлення
 input.stream = Потік
 input.tab = Введення
+input.tab.tooltip = {chars} символів (~{tokens} токенів)
 interpreter.all = Виконати історію (все)
 interpreter.auto_clear = Очистити при відправці
 interpreter.btn.clear = Очистити вивід

pygpt_net/data/locale/locale.zh.ini

@@ -658,6 +658,7 @@ event.control.voice_cmd.toggle = 语音控制：切换
 event.control.voice_msg.start = 语音输入：开始
 event.control.voice_msg.stop = 语音输入：停止
 event.control.voice_msg.toggle = 语音输入：切换
+exit.msg = 你喜欢PyGPT吗？支持项目的发展：
 expert.wait.failed: 调用专家失败
 expert.wait.status: 等待专家...
 files.delete.confirm = 刪除文件/目錄？
@@ -734,6 +735,7 @@ input.search.placeholder = 搜索...
 input.send_clear = 發送後清除
 input.stream = 流
 input.tab = 輸入
+input.tab.tooltip = {chars} 字符 (~{tokens} 令牌)
 interpreter.all = 执行历史记录（全部）
 interpreter.auto_clear = 发送时清除
 interpreter.btn.clear = 清除输出

pygpt_net/item/ctx.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.09.05 18:00:00                  #
+# Updated Date: 2025.09.26 17:00:00                  #
 # ================================================== #
 
 import copy
@@ -209,6 +209,28 @@ class CtxItem:
             if src:
                 setattr(self, name, dp(src))
 
+    def set_agent_name(self, name: Optional[str]):
+        """
+        Set AI/agent name
+
+        :param name: name
+        """
+        self.ai_name = name
+        if name:
+            if self.extra is None:
+                self.extra = {}
+            self.extra['agent_name'] = name
+
+    def get_agent_name(self) -> Optional[str]:
+        """
+        Get AI/agent name
+
+        :return: name
+        """
+        if self.extra and isinstance(self.extra, dict):
+            return self.extra.get('agent_name', None)
+        return None
+
     def is_empty(self) -> bool:
         """
         Check if context item is empty

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

@@ -5,7 +5,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 14:00:00                  #
 # ================================================== #
 
 # >>> Based on LlamaIndex CodeActAgent implementation, with custom plugin tool support <<<
@@ -124,6 +124,9 @@ class CodeActAgent(BaseWorkflowAgent):
     _plugin_tool_fn: Union[Callable, Awaitable] = PrivateAttr(default=None)
     _on_stop: Optional[Callable] = PrivateAttr(default=None)
 
+    # Always emit this human-friendly agent name in workflow events for UI consumption.
+    _display_agent_name: str = PrivateAttr(default="CodeAct")
+
     def __init__(
         self,
         code_execute_fn: Union[Callable, Awaitable],
@@ -296,13 +299,13 @@ class CodeActAgent(BaseWorkflowAgent):
                 StepEvent(name=name, index=index, total=total, meta=meta or {})
             )
         except Exception:
-            # Fallback for older versions of AgentStream
+            # Fallback for environments lacking StepEvent wiring.
             try:
                 ctx.write_event_to_stream(
                     AgentStream(
                         delta="",
                         response="",
-                        current_agent_name="PlannerWorkflow",
+                        current_agent_name=self._display_agent_name,  # always "CodeAct"
                         tool_calls=[],
                         raw={"StepEvent": {"name": name, "index": index, "total": total, "meta": meta or {}}}
                     )
@@ -367,7 +370,7 @@ class CodeActAgent(BaseWorkflowAgent):
             current_llm_input.insert(0, ChatMessage(role="system", content=system_prompt))
 
         ctx.write_event_to_stream(
-            AgentInput(input=current_llm_input, current_agent_name=self.name)
+            AgentInput(input=current_llm_input, current_agent_name=self._display_agent_name)  # always "CodeAct"
         )
 
         if any(tool.metadata.name == "handoff" for tool in tools):
@@ -400,7 +403,7 @@ class CodeActAgent(BaseWorkflowAgent):
                     response=full_response_text,
                     tool_calls=[],
                     raw=raw,
-                    current_agent_name=self.name,
+                    current_agent_name=self._display_agent_name,  # always "CodeAct"
                 )
             )
 
@@ -443,7 +446,7 @@ class CodeActAgent(BaseWorkflowAgent):
             response=message,
             tool_calls=tool_calls,
             raw=raw,
-            current_agent_name=self.name,
+            current_agent_name=self._display_agent_name,  # always "CodeAct"
         )
 
     async def handle_tool_call_results(

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