어디서나 Genie 접속: Microsoft Teams 및 커스텀 앱에 대화형 분석 도입하기

from databricks.sdk import WorkspaceClient

# Exchange authorization code for tokens
token_response = client.oauth.get_token(
    client_id=CLIENT_ID,
    client_secret=CLIENT_SECRET,
    redirect_uri=REDIRECT_URI,
    code=auth_code,
    code_verifier=code_verifier,
    grant_type="authorization_code"
)

def get_user_client(access_token: str):
    return WorkspaceClient(
        host=DATABRICKS_HOST,
        token=access_token
    )

@app.post("/conversations")
def start_conversation(request: ConversationRequest, access_token: str = Depends(get_token)):
    client = get_user_client(access_token)
    response = client.genie.create_conversation(
        space_id=request.space_id,
        content=request.initial_prompt
    )
    return response