Function test_chat_completion

tests/models/test_chat_openai.py:38–44

(openai_completion_fixture)

def test_chat_completion(openai_completion_fixture):
    os.environ["OPENAI_API_KEY"] = "mock_api_key"
    model = ChatOpenAI(temperature=0)
    response = model.generate([UserMessage(content="test message")])
    assert isinstance(response, LLMResult)
    assert len(response.generations) == 1
    assert response.generations[0].message.content == "generated message"


def test_convert_tool_to_dict():