tests.chat.llm package#
Submodules#
tests.chat.llm.test_abs_llm module#
Test the AbsLlm class.
- class tests.chat.llm.test_abs_llm.DummyLlm[source]#
Bases:
AbsLlm
Dummy class for testing the abstract base class AbsLlm.
- predict(message: str)[source]#
Predict the next word based on the input message.
This method must be implemented by any subclass of AbsLlm.
- Parameters:
message (str) – The input message used to predict the next word.
num_of_response (int) – How many completions to generate for each prompt.
- Raises:
NotImplementedError – This method must be implemented by subclasses.
tests.chat.llm.test_constants module#
Test the constants of the LLM module.
tests.chat.llm.test_huggingface module#
tests.chat.llm.test_openai module#
Test the OpenAIModel class