Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Python: Improve azure assistant agent settings and retrieval operations. #10063

Merged
merged 4 commits into from
Jan 7, 2025

Conversation

moonbox3
Copy link
Contributor

@moonbox3 moonbox3 commented Jan 6, 2025

Motivation and Context

The AzureAssistantAgent retrieval path was not handling the optional ad_token. The agent constructor was handling it, to a degree, but there were improvements to make and a helper function to introduce so that we remove some code duplication.

Description

This PR:

  • Cleans up the code used to create the azure settings for an azure assistant agent by introducing a helper function to streamline the logic
  • Allows the AzureAssistantAgent.retrieve() method to use an ad_token, if desired.
  • Adds unit tests for the new logic. Other unit tests are still passing after the refactor.

Contribution Checklist

@moonbox3 moonbox3 requested a review from a team as a code owner January 6, 2025 07:27
@markwallace-microsoft markwallace-microsoft added the python Pull requests for the Python Semantic Kernel label Jan 6, 2025
@moonbox3 moonbox3 self-assigned this Jan 6, 2025
@moonbox3 moonbox3 added the agents label Jan 6, 2025
@markwallace-microsoft
Copy link
Member

markwallace-microsoft commented Jan 6, 2025

Python Test Coverage

Python Test Coverage Report •
FileStmtsMissCoverMissing
semantic_kernel/agents/open_ai
   azure_assistant_agent.py106298%271, 288
TOTAL16748176989% 

Python Unit Test Overview

Tests Skipped Failures Errors Time
3000 4 💤 0 ❌ 0 🔥 1m 16s ⏱️

@moonbox3 moonbox3 enabled auto-merge January 7, 2025 02:36
@moonbox3 moonbox3 added this pull request to the merge queue Jan 7, 2025
github-merge-queue bot pushed a commit that referenced this pull request Jan 7, 2025
…ns. (#10063)

### Motivation and Context

The AzureAssistantAgent retrieval path was not handling the optional
ad_token. The agent constructor was handling it, to a degree, but there
were improvements to make and a helper function to introduce so that we
remove some code duplication.

<!-- Thank you for your contribution to the semantic-kernel repo!
Please help reviewers and future users, providing the following
information:
  1. Why is this change required?
  2. What problem does it solve?
  3. What scenario does it contribute to?
  4. If it fixes an open issue, please link to the issue here.
-->

### Description

This PR:
- Cleans up the code used to create the azure settings for an azure
assistant agent by introducing a helper function to streamline the logic
- Allows the AzureAssistantAgent.retrieve() method to use an ad_token,
if desired.
- Adds unit tests for the new logic. Other unit tests are still passing
after the refactor.

<!-- Describe your changes, the overall approach, the underlying design.
These notes will help understanding how your code works. Thanks! -->

### Contribution Checklist

<!-- Before submitting this PR, please make sure: -->

- [X] The code builds clean without any errors or warnings
- [X] The PR follows the [SK Contribution
Guidelines](https://github.com/microsoft/semantic-kernel/blob/main/CONTRIBUTING.md)
and the [pre-submission formatting
script](https://github.com/microsoft/semantic-kernel/blob/main/CONTRIBUTING.md#development-scripts)
raises no violations
- [X] All unit tests pass, and I have added new tests where possible
- [X] I didn't break anyone 😄
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Jan 7, 2025
@moonbox3 moonbox3 added this pull request to the merge queue Jan 7, 2025
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Jan 7, 2025
@eavanvalkenburg eavanvalkenburg added this pull request to the merge queue Jan 7, 2025
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Jan 7, 2025
@eavanvalkenburg eavanvalkenburg force-pushed the azure-agent-improvements branch from 31f3ef4 to 4ded537 Compare January 7, 2025 18:31
@eavanvalkenburg eavanvalkenburg added this pull request to the merge queue Jan 7, 2025
Merged via the queue into microsoft:main with commit 5830081 Jan 7, 2025
25 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
agents documentation python Pull requests for the Python Semantic Kernel
Projects
Status: Sprint: Done
Development

Successfully merging this pull request may close these issues.

4 participants