Azure / Azure/azure-sdk-for-python

Consume workspace_connection in pipeline job

Open
#40,063 2 comments 0 reactions 1 assignee Claimed by @kingernupur View on GitHub
AI customer-reported feature-request Machine Learning needs-team-attention Service Attention
Dominant language
Python
Stars
5.6k
Forks
3.4k
Avg merge
1d 21h
Merged PRs (30d)
193

Description

**Is your feature request related to a problem? Please describe.**
The customer we are working with wants to use AML for build an experimenation platform, and wants to use native openai sdk for the LLM components. To run the experiments on AML and keep parity between the deployed API and the experiment runner, we want to use the workspace connection. However, I have only seen how do use that within either a prompflow run or with the data import jobs.

**Describe the solution you'd like**
- We have created the connection in the Studio and now want to be able to connect to either our AzureOpenAI endpoint or our AzureCosmosDB within a pipeline component.

**Describe alternatives you've considered**
- Using keyvault

**Additional context**

![Image](https://github.com/user-attachments/assets/9f864496-3b9b-4591-9d26-b80501785519)

It would be nice if a pipeline could have an input of a workspace connection or include this in some other fashion for the components to securly consume secrets.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.