code: def test_completion_chat_sagemaker():
      try:
          messages = [
              {"role": "user", "content": "Hey, how's it going?"}
          ]
          litellm.set_verbose = True
          response = completion(
              model="sagemaker/berri-benchmarking-Llama-2-70b-chat-hf-4",
              messages=messages,
              max_tokens=100,
              temperature=0.7,
              stream=True,
          )
          # Add any assertions here to check the response
          complete_response = ""
          for chunk in response:
              complete_response += chunk.choices[0].delta.content or ""
          print(f"complete_response: {complete_response}")
          assert len(complete_response) > 0
      except Exception as e:
          pytest.fail(f"Error occurred: {e}")
  
imports: import sys, os
  import traceback
  from dotenv import load_dotenv
  import os, io
  import pytest
  import litellm
  from litellm import embedding, completion, completion_cost, Timeout
  from litellm import RateLimitError
  import openai
label_desc: Use image node_id 41 for calling a global function with example usage: test_completion_chat_sagemaker() without return types