Chame APIs REST diretamente do código do agente

Use o ponto de extremidade de proxy de conexões do Catálogo do Unity com o SDK do cliente próprio do serviço externo para chamar APIs REST diretamente do código do agente. Direcione a URL base do SDK para o ponto de extremidade do proxy, e use o token do Azure Databricks como sendo a chave de API. Azure Databricks autentica a solicitação e injeta automaticamente as credenciais do serviço externo a partir da conexão do Unity Catalog. Seu código não lida diretamente com os tokens do serviço externo.

Requirements

• Uma conexão HTTP do Unity Catalog para o serviço externo. O endpoint proxy usa essa conexão para autenticar e injetar as credenciais do serviço.
• USE CONNECTION no objeto de conexão.

OpenAI

Use DatabricksOpenAI para rotear chamadas para o OpenAI externo por meio do proxy de conexões do Unity Catalog. Primeiro, crie uma conexão HTTP do Catálogo do Unity usando sua chave de API OpenAI armazenada como um segredo do Databricks:

CREATE CONNECTION openai_connection TYPE HTTP
OPTIONS (
  host 'https://api.openai.com',
  base_path '/v1',
  bearer_token secret ('<secret-scope>', '<secret-key>')
);

Em seguida, instale o pacote databricks-openai e use a URL do proxy e o cliente de workspace no código do agente.

pip install databricks-openai

from databricks_openai import DatabricksOpenAI
from databricks.sdk import WorkspaceClient

w = WorkspaceClient()

client = DatabricksOpenAI(
    workspace_client=w,
    base_url=f"{w.config.host}/api/2.0/unity-catalog/connections/openai_connection/proxy/",
)

response = client.chat.completions.create(
    model="gpt-4o",
    messages=[{"role": "user", "content": "Hello!"}],
)
print(response.choices[0].message.content)

Slack

Configure o SDK do Slack para rotear por meio do proxy de conexões do Catálogo do Unity. Crie uma conexão HTTP do Catálogo do Unity com o host https://slack.com e o caminho base /api, e use a URL do proxy como a URL base do SDK:

from slack_sdk import WebClient
from databricks.sdk import WorkspaceClient

w = WorkspaceClient()

client = WebClient(
    token=w.config.authenticate()["Authorization"].split(" ")[1],
    base_url=f"{w.config.host}/api/2.0/unity-catalog/connections/slack_connection/proxy/",
)

result = client.chat_postMessage(channel="C123456", text="Hello from Databricks!")
print(result["message"]["text"])

HTTP Genérico

Para serviços sem um SDK dedicado, use a biblioteca requests com a URL do proxy diretamente.

import requests
from databricks.sdk import WorkspaceClient

w = WorkspaceClient()

response = requests.post(
    f"{w.config.host}/api/2.0/unity-catalog/connections/my_connection/proxy/api/v1/resource",
    headers={
        **w.config.authenticate(),
        "Content-Type": "application/json",
    },
    json={"key": "value"},
)

Para obter detalhes sobre o endpoint do proxy, os métodos de autenticação compatíveis e a configuração da conexão, consulte Encaminhar solicitações pelo proxy de conexão HTTP.