lm_polygraph.utils.openai_chat module

class lm_polygraph.utils.openai_chat.OpenAIChat(openai_model: str = 'gpt-4o', base_url: str = None, cache_path: str = '/home/docs/.cache', timeout: int = 600, max_tokens: int = None, rewrite_cache: bool = False)[source]

Bases: object

Allows for the implementation of a singleton class to chat with OpenAI model for dataset marking.

ask(message: str) str[source]