Skip to content

fix: io inference for agent workflows#64

Merged
cristipufu merged 1 commit intomainfrom
fix/input_output_inference
Sep 5, 2025
Merged

fix: io inference for agent workflows#64
cristipufu merged 1 commit intomainfrom
fix/input_output_inference

Conversation

@cristipufu
Copy link
Copy Markdown
Member

@cristipufu cristipufu commented Sep 5, 2025

Description

This PR fixes input/output schema inference for agent workflows by adding special handling for BaseWorkflowAgent instances. The changes ensure that workflow agents have properly defined schemas for both input and output, with the input schema defaulting to a simple user message structure and the output schema using the agent's output_cls when available.

@cristipufu cristipufu self-assigned this Sep 5, 2025
Copy link
Copy Markdown

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull Request Overview

This PR fixes input/output schema inference for agent workflows by adding special handling for BaseWorkflowAgent instances. The changes ensure that workflow agents have properly defined schemas for both input and output, with the input schema defaulting to a simple user message structure and the output schema using the agent's output_cls when available.

  • Added BaseWorkflowAgent-specific schema generation logic
  • Enhanced output serialization to handle AgentOutput responses
  • Updated package version and dependencies

Reviewed Changes

Copilot reviewed 3 out of 4 changed files in this pull request and generated 2 comments.

File Description
src/uipath_llamaindex/_cli/cli_init.py Added BaseWorkflowAgent import and special schema generation logic for workflow agents
src/uipath_llamaindex/_cli/_runtime/_runtime.py Added AgentOutput handling in the output serialization process
pyproject.toml Bumped version to 0.0.35 and updated uipath dependency requirement

Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.

Comment thread src/uipath_llamaindex/_cli/cli_init.py Outdated
Comment thread src/uipath_llamaindex/_cli/_runtime/_runtime.py
@cristipufu cristipufu force-pushed the fix/input_output_inference branch 3 times, most recently from 9ecfa28 to 7a76be7 Compare September 5, 2025 16:17
@cristipufu cristipufu force-pushed the fix/input_output_inference branch from 7a76be7 to 945fa74 Compare September 5, 2025 16:21
@cristipufu cristipufu merged commit da22624 into main Sep 5, 2025
21 checks passed
@cristipufu cristipufu deleted the fix/input_output_inference branch November 30, 2025 11:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants