Integrate the AI/ML API API with the Python API

import { axios } from "@pipedream/platform"
export default defineComponent({
  props: {
    ai_ml_api: {
      type: "app",
      app: "ai_ml_api",
    }
  },
  async run({steps, $}) {
    const data = {
          "model": "mistralai/Mistral-7B-v0.1",
          "messages":  [
            {
                "role": "system",
                "content": "You are an AI assistant who knows everything.",
            },
            {
                "role": "user",
                "content": "Tell me, why is the sky blue?"
            },
        ]
    }
    return await axios($, {
      method: "POST",
      url: `https://api.aimlapi.com/chat/completions`,
      headers: {
        Authorization: `Bearer ${this.ai_ml_api.$auth.api_key}`,
      },
      data,
    })
  },
})

def handler(pd: "pipedream"):
  # Reference data from previous steps
  print(pd.steps["trigger"]["context"]["id"])
  # Return data for use in future steps
  return {"foo": {"test":True}}