title | titleSuffix | description | manager | ms.service | ms.subservice | ms.topic | ms.date | ms.reviewer | ms.author | author | ms.custom | ||
---|---|---|---|---|---|---|---|---|---|---|---|---|---|
Consume deployed serverless API endpoints from a different workspace |
Azure Machine Learning |
Learn how to consume a serverless API endpoint from a different workspace than the one where it was deployed. |
scottpolly |
azure-machine-learning |
inferencing |
how-to |
05/09/2024 |
None |
mopeakande |
msakande |
|
In this article, you learn how to configure an existing serverless API endpoint in a different workspace than the one where it was deployed.
Certain models in the model catalog can be deployed as serverless APIs. This kind of deployment provides a way to consume models as an API without hosting them on your subscription, while keeping the enterprise security and compliance that organizations need. This deployment option doesn't require quota from your subscription.
The need to consume a serverless API endpoint in a different workspace than the one that was used to create the deployment might arise in situations such as these:
- You want to centralize your deployments in a given workspace and consume them from different workspaces in your organization.
- You need to deploy a model in a workspace in a particular Azure region where serverless deployment for that model is available. However, you need to consume it from another region, where serverless deployment isn't available for the particular models.
-
An Azure subscription with a valid payment method. Free or trial Azure subscriptions won't work. If you don't have an Azure subscription, create a paid Azure account to begin.
-
An Azure Machine Learning workspace where you want to consume the existing deployment.
-
A model deployed to a serverless API endpoint. This article assumes that you previously deployed the Meta-Llama-3-8B-Instruct model. To learn how to deploy this model as a serverless API, see Deploy models as serverless APIs.
-
You need to install the following software to work with Azure Machine Learning:
You can use any compatible web browser to navigate Azure Machine Learning studio.
The Azure CLI and the ml extension for Azure Machine Learning.
az extension add -n ml
If you already have the extension installed, ensure the latest version is installed.
az extension update -n ml
Once the extension is installed, configure it:
az account set --subscription <subscription> az configure --defaults workspace=<workspace-name> group=<resource-group> location=<location>
Install the Azure Machine Learning SDK for Python.
pip install -U azure-ai-ml
Once installed, import necessary namespaces:
from azure.ai.ml import MLClient from azure.identity import InteractiveBrowserCredential from azure.ai.ml.entities import ServerlessEndpoint, ServerlessConnection
Follow these steps to create a connection:
-
Connect to the workspace where the endpoint is deployed:
Go to Azure Machine Learning studio and navigate to the workspace where the endpoint you want to connect to is deployed.
Configure the CLI to point to the workspace:
az account set --subscription <subscription> az configure --defaults workspace=<workspace-name> group=<resource-group> location=<location>
Create a client connected to your workspace:
client = MLClient( credential=InteractiveBrowserCredential(tenant_id="<tenant-id>"), subscription_id="<subscription-id>", resource_group_name="<resource-group>", workspace_name="<workspace-name>", )
-
Get the endpoint's URL and credentials for the endpoint you want to connect to. In this example, you get the details for an endpoint name meta-llama3-8b-qwerty.
-
Select Endpoints from the left sidebar.
-
Select the Serverless endpoints tab to display the serverless API endpoints.
-
Select the endpoint you want to connect to.
-
On the endpoint's Details tab, copy the values for Target URI and Key.
az ml serverless-endpoint get-credentials -n meta-llama3-8b-qwerty
endpoint_name = "meta-llama3-8b-qwerty" endpoint_keys = client.serverless_endpoints.get_keys(endpoint_name) print(endpoint_keys.primary_key) print(endpoint_keys.secondary_key)
-
-
Now, connect to the workspace where you want to create the connection and consume the endpoint.
-
Create the connection in the workspace:
-
Go to the workspace where the connection needs to be created to.
-
Go to the Manage section in the left navigation bar and select Connections.
-
Select Create.
-
Select Serverless Model.
-
For the Target URI, paste the value you copied previously.
-
For the Key, paste the value you copied previously.
-
Give the connection a name, in this case meta-llama3-8b-connection.
-
Select Add connection.
Create a connection definition:
connection.yml
name: meta-llama3-8b-connection type: serverless endpoint: https://meta-llama3-8b-qwerty-serverless.inference.ai.azure.com api_key: 1234567890qwertyuiop
az ml connection create -f connection.yml
client.connections.create_or_update(ServerlessConnection( name="meta-llama3-8b-connection", endpoint="https://meta-llama3-8b-qwerty-serverless.inference.ai.azure.com", api_key="1234567890qwertyuiop" ))
-
-
At this point, the connection is available for consumption.
-
To validate that the connection is working:
-
From the left navigation bar of Azure Machine Learning studio, go to Authoring > Prompt flow.
-
Select Create to create a new flow.
-
Select Create in the Chat flow box.
-
Give your Prompt flow a name and select Create.
-
Select the chat node from the graph to go to the chat section.
-
For Connection, open the dropdown list to select the connection you just created, in this case meta-llama3-8b-connection.
-
Select Start compute session from the top navigation bar, to start a prompt flow automatic runtime.
-
Select the Chat option. You can now send messages and get responses.
-