Skip to content

Fix ABLE feedback UserMessage import#90

Open
DavidJBianco wants to merge 1 commit into
mainfrom
codex/fix-able-table-endpoint-importerror
Open

Fix ABLE feedback UserMessage import#90
DavidJBianco wants to merge 1 commit into
mainfrom
codex/fix-able-table-endpoint-importerror

Conversation

@DavidJBianco

Copy link
Copy Markdown
Collaborator

Motivation

  • The Streamlit ABLE feedback path imported UserMessage from autogen_agentchat.messages, which can raise an ImportError (and crash the /api/able-table flow) because the ABLE assistant and model client expect UserMessage from autogen_core.models.

Description

  • Replace the UserMessage import in peak_assistant/streamlit/util/helpers.py to from autogen_core.models import UserMessage while keeping TextMessage from autogen_agentchat.messages to preserve existing behavior.
  • Update tests/unit_tests/test_chat_history_conversion.py to import and assert against autogen_core.models.UserMessage so tests reflect the corrected message type.

Testing

  • Ran uv run --python 3.12 pytest tests/unit_tests/test_chat_history_conversion.py which passed (10 passed).
  • Ran uv run --python 3.12 ruff check tests/unit_tests/test_chat_history_conversion.py and linter issues were resolved before committing and checks passed.
  • Noted environment-specific issues when running pytest with the ambient Python 3.14 (missing autogen_agentchat dependency) and a failed pyarrow build when trying to run in the ambient 3.14 environment; these are dependency/build environment problems and not caused by the change.

Codex Task

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant