code: def test_completion_together_ai_yi_chat():
      litellm.set_verbose = True
      model_name = "together_ai/zero-one-ai/Yi-34B-Chat"
      try:
          messages = [
              {"role": "user", "content": "What llm are you?"},
          ]
          response = completion(
              model=model_name, messages=messages, max_tokens=5
          )
          # Add any assertions here to check the response
          print(response)
          cost = completion_cost(completion_response=response)
          assert cost > 0.0
          print(
              "Cost for completion call together-computer/llama-2-70b: ",
              f"${float(cost):.10f}",
          )
      except litellm.Timeout as e:
          pass
      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 38 for calling a global function with example usage: test_completion_together_ai_yi_chat() without return types