Chat Stateful

ARK Platform Example / Chat Stateful

• Copy

  
  import requests
  
  ark_api_key = "API_KEY"
  ark_url = "https://api.ark-labs.cloud/api/v1/chat/completions"
  
  session = requests.Session()
  
  headers = {
      "Authorization": f"Bearer {ark_api_key}",
      "Content-Type": "application/json",
  }
  
  print("Waiting for the first response...")
  
  response = session.post(
      ark_url,
      json={
          "model": "gpt-4o",
          "messages": [
              {"role": "system", "content": "You are a helpful assistant."},
              {"role": "user", "content": "Tell me a story about a brave knight traversing space in a small rocket who's lost because GPS only works on Earth. 200 words."}
          ],
      },
      headers=headers,
  )
  
  if response.status_code == 200:
      print("Cookies received:", session.cookies.get_dict())
      print()
      print("First response:")
      data = response.json()
      print(data["choices"][0]["message"]["content"])
  else:
      print("Failed to get response:", response.text)
      exit()
  
  print()
  print("Waiting for the second response...")
  
  response = session.post(
      ark_url,
      json={
          "model": "gpt-4o",
          "messages": [
              {"role": "system", "content": "You are a helpful assistant."},
              # look, ma, no prompt history!
              {"role": "user", "content": "Translate the story to German, please."}
          ],
      },
      headers=headers,
  )
  
  print()
  print("Second response:")
  data = response.json()
  print(data["choices"][0]["message"]["content"])