class_docstr: Unit tests for OpenAI Info Callback
class_docstr_tok: ["Unit","tests","for","OpenAI","Info","Callback"]
code: def test_handler_unknown_model(
      self, handler: OpenAICallbackHandler
  ) -> None:
      response = OpenAIObject(
          {
              "usage": OpenAIObject(
                  {
                      "prompt_tokens": 2,
                      "completion_tokens": 1,
                      "total_tokens": 3,
                  }
              ),
              "model": "foo-bar",
          }
      )
  
      handler(response)
      assert handler.total_tokens == 3
      assert handler.prompt_tokens == 2
      assert handler.completion_tokens == 1
      # cost must be 0.0 for unknown model
      assert handler.total_cost == 0.0
  
file_path: pandas-ai/tests/helpers/test_openai_info.py
imports: import pytest
  from pandasai import SmartDataframe
  from pandasai.helpers import OpenAICallbackHandler, get_openai_callback
  import pandas
  from pandasai.llm.openai import OpenAI
label_desc: Use image node_id 2 for calling the TestOpenAIInfo obj's underlying member method code with example usage: obj.test_handler_unknown_model(handler) without return types