dllmforge.openai_api

Classes

OpenAIAPI([api_key, api_base, api_version, ...])

Class to interact with Azure OpenAI API.

class dllmforge.openai_api.OpenAIAPI(api_key=None, api_base=None, api_version=None, deployment_name='gpt-4o', embedding_deployment='text-embedding-3-large')

Class to interact with Azure OpenAI API.

Initialize the OpenAI API client with Azure configuration.

__init__(api_key=None, api_base=None, api_version=None, deployment_name='gpt-4o', embedding_deployment='text-embedding-3-large')

Initialize the OpenAI API client with Azure configuration.

check_server_status()

Check if the Azure OpenAI service is accessible.

list_available_models()

List available models from Azure OpenAI.

send_test_message(prompt='Hello, how are you?')

Send a test message to the model and get a response.

get_embeddings(text)

Get embeddings for the given text using Azure OpenAI.

chat_completion(messages, temperature=0.7, max_tokens=800)

Get a chat completion from the model.