autogen.oai.openai_utils
get_key
def get_key(config)
Get a unique identifier of a configuration.
Arguments:
config
dict or list - A configuration.
Returns:
tuple
- A unique identifier which can be used as a key for a dict.
get_config_list
def get_config_list(api_keys: List, api_bases: Optional[List] = None, api_type: Optional[str] = None, api_version: Optional[str] = None) -> List[Dict]
Get a list of configs for openai api calls.
Arguments:
api_keys
list - The api keys for openai api calls.api_bases
list, optional - The api bases for openai api calls.api_type
str, optional - The api type for openai api calls.api_version
str, optional - The api version for openai api calls.
config_list_openai_aoai
def config_list_openai_aoai(key_file_path: Optional[str] = ".", openai_api_key_file: Optional[str] = "key_openai.txt", aoai_api_key_file: Optional[str] = "key_aoai.txt", aoai_api_base_file: Optional[str] = "base_aoai.txt", exclude: Optional[str] = None) -> List[Dict]
Get a list of configs for openai + azure openai api calls.
Arguments:
key_file_path
str, optional - The path to the key files.openai_api_key_file
str, optional - The file name of the openai api key.aoai_api_key_file
str, optional - The file name of the azure openai api key.aoai_api_base_file
str, optional - The file name of the azure openai api base.exclude
str, optional - The api type to exclude, "openai" or "aoai".
Returns:
list
- A list of configs for openai api calls.
config_list_from_models
def config_list_from_models(key_file_path: Optional[str] = ".", openai_api_key_file: Optional[str] = "key_openai.txt", aoai_api_key_file: Optional[str] = "key_aoai.txt", aoai_api_base_file: Optional[str] = "base_aoai.txt", exclude: Optional[str] = None, model_list: Optional[list] = None) -> List[Dict]
Get a list of configs for api calls with models in the model list.
Arguments:
key_file_path
str, optional - The path to the key files.openai_api_key_file
str, optional - The file name of the openai api key.aoai_api_key_file
str, optional - The file name of the azure openai api key.aoai_api_base_file
str, optional - The file name of the azure openai api base.exclude
str, optional - The api type to exclude, "openai" or "aoai".model_list
list, optional - The model list.
Returns:
list
- A list of configs for openai api calls.
config_list_gpt4_gpt35
def config_list_gpt4_gpt35(key_file_path: Optional[str] = ".", openai_api_key_file: Optional[str] = "key_openai.txt", aoai_api_key_file: Optional[str] = "key_aoai.txt", aoai_api_base_file: Optional[str] = "base_aoai.txt", exclude: Optional[str] = None) -> List[Dict]
Get a list of configs for gpt-4 followed by gpt-3.5 api calls.
Arguments:
key_file_path
str, optional - The path to the key files.openai_api_key_file
str, optional - The file name of the openai api key.aoai_api_key_file
str, optional - The file name of the azure openai api key.aoai_api_base_file
str, optional - The file name of the azure openai api base.exclude
str, optional - The api type to exclude, "openai" or "aoai".
Returns:
list
- A list of configs for openai api calls.
filter_config
def filter_config(config_list, filter_dict)
Filter the config list by provider and model.
Arguments:
config_list
list - The config list.filter_dict
dict, optional - The filter dict with keys corresponding to a field in each config, and values corresponding to lists of acceptable values for each key.
Returns:
list
- The filtered config list.
config_list_from_json
def config_list_from_json(env_or_file: str, file_location: Optional[str] = "", filter_dict: Optional[Dict[str, Union[List[Union[str, None]], Set[Union[str, None]]]]] = None) -> List[Dict]
Get a list of configs from a json parsed from an env variable or a file.
Arguments:
env_or_file
str - The env variable name or file name.file_location
str, optional - The file location.filter_dict
dict, optional - The filter dict with keys corresponding to a field in each config, and values corresponding to lists of acceptable values for each key. e.g.,
filter_dict = {
"api_type": ["open_ai", None], # None means a missing key is acceptable
"model": ["gpt-3.5-turbo", "gpt-4"],
}
Returns:
list
- A list of configs for openai api calls.