...
Package config
Overview ▹
Index ▹
func CreateUpdateConfigFileFromCLI ¶
func CreateUpdateConfigFileFromCLI(fileName string) (err error)
CreateUpdateConfigFileFromCLI reads and updates the configuration file based on command-line arguments.
Parameters:
- fileName: The name of the configuration file.
Returns:
- err: An error if there was an issue creating or updating the configuration file.
func GetGlobalConfigAsJSON ¶
func GetGlobalConfigAsJSON() string
GetGlobalConfigAsJSON returns the global configuration as a JSON string.
Returns:
- string: The global configuration as a JSON string.
func InitConfig ¶
func InitConfig(requiredProperties []string, optionalDefaultValues map[string]interface{})
InitConfig initializes the configuration for the Allie service.
Parameters:
- requiredProperties: The list of required properties.
- optionalDefaultValues: The map of optional properties and their default values.
func InitGlobalConfigFromAzureKeyVault ¶
func InitGlobalConfigFromAzureKeyVault() (err error)
InitGlobalConfigFromAzureKeyVault extracts the configuration from Azure Key Vault. It iterates over all secrets in the key vault and if the secret name matches a field in the Config struct, it sets the field to the value of the secret.
Returns:
- err: An error if there was an issue extracting the configuration.
func InitGlobalConfigFromFile ¶
func InitGlobalConfigFromFile(fileName string, requiredProperties []string, optionalDefaultValues map[string]interface{}) (err error)
InitGlobalConfigFromFile reads the configuration file and initializes the Config object.
Parameters:
- fileName: The name of the configuration file.
- requiredProperties: The list of required properties.
- optionalDefaultValues: The map of optional properties and their default values.
Returns:
- err: An error if there was an issue initializing the configuration.
func ValidateConfig ¶
func ValidateConfig(config Config, requiredProperties []string) (err error)
ValidateConfig checks for mandatory entries in the configuration and validates chosen models.
Parameters:
- config: The configuration object to validate.
- requiredProperties: The list of required properties.
Returns:
- err: An error if there was an issue validating the configuration.
func createFlags ¶
func createFlags(val reflect.Value, prefix string)
CreateFlags initializes command-line flags for configuration.
Parameters:
- val: The value to create flags for.
- prefix: The prefix to use for the flags.
func defineOptionalProperties ¶
func defineOptionalProperties(config *Config, optionalDefaultValues map[string]interface{}) (err error)
defineOptionalProperties sets optional properties for the configuration.
Parameters:
- config: The configuration object to validate.
- optionalDefaultValues: The map of optional properties and their default values.
Returns:
- err: An error if there was an issue setting the optional properties.
func isZeroValue ¶
func isZeroValue(v reflect.Value) bool
isZeroValue checks if a reflect.Value is zero for its type.
Parameters:
- v: The reflect.Value to check.
Returns:
- bool: True if the value is zero, false otherwise.
func timeToString ¶
func timeToString(t time.Time) string
timeToString converts a time.Time value to a formatted string.
Parameters:
- t: The time.Time value to convert.
Returns:
- string: The formatted string.
func writeInterfaceToFile ¶
func writeInterfaceToFile(data interface{}) error
writeInterfaceToFile writes interface data to an error log file.
Parameters:
- data: The data to write to the file.
Returns:
- error: An error if there was an issue writing to the file.
func writeStringToFile ¶
func writeStringToFile(data string) error
writeStringToFile writes string data to an error log file.
Parameters:
- data: The data to write to the file.
Returns:
- error: An error if there was an issue writing to the file.
type Config ¶
Config contains all the configuration settings for the Allie service.
type Config struct {
// Logging
///////////
LOG_LEVEL string `yaml:"LOG_LEVEL" json:"LOGLEVEL"`
// Local Logs
LOCAL_LOGS bool `yaml:"LOCAL_LOGS" json:"LOCALLOGS"`
LOCAL_LOGS_LOCATION string `yaml:"LOCAL_LOGS_LOCATION" json:"LOCALLOGSLOCATION"`
// Datadog Logs
DATADOG_LOGS bool `yaml:"DATADOG_LOGS" json:"DATADOGLOGS"`
STAGE string `yaml:"STAGE" json:"STAGE"`
VERSION string `yaml:"VERSION" json:"VERSION"`
SERVICE_NAME string `yaml:"SERVICE_NAME" json:"SERVICENAME"`
ERROR_FILE_LOCATION string `yaml:"ERROR_FILE_LOCATION" json:"ERRORFILELOCATION"`
LOGGING_URL string `yaml:"LOGGING_URL" json:"LOGGINGURL"`
LOGGING_API_KEY string `yaml:"LOGGING_API_KEY" json:"LOGGINGAPIKEY"`
DATADOG_SOURCE string `yaml:"DATADOG_SOURCE" json:"DATADOGSOURCE"`
// Datadog Metrics
DATADOG_METRICS bool `yaml:"DATADOG_METRICS" json:"DATADOGMETRICS"`
METRICS_URL string `yaml:"METRICS_URL" json:"METRICSURL"`
// SSL Settings
/////////////////
USE_SSL bool `yaml:"USE_SSL" json:"USESSL"`
SSL_CERT_PUBLIC_KEY_FILE string `yaml:"SSL_CERT_PUBLIC_KEY_FILE" json:"SSLCERTPUBLICKEYFILE"`
SSL_CERT_PRIVATE_KEY_FILE string `yaml:"SSL_CERT_PRIVATE_KEY_FILE" json:"SSLCERTPRIVATEKEYFILE"`
// Azure Key Vault Settings
////////////////////////////
EXTRACT_CONFIG_FROM_AZURE_KEY_VAULT bool `yaml:"EXTRACT_CONFIG_FROM_AZURE_KEY_VAULT" json:"EXTRACTCONFIGFROMAZUREKEYVAULT"`
AZURE_KEY_VAULT_NAME string `yaml:"AZURE_KEY_VAULT_NAME" json:"AZUREKEYVAULTNAME"`
AZURE_MANAGED_IDENTITY_ID string `yaml:"AZURE_MANAGED_IDENTITY_ID" json:"AZUREMANAGEDIDENTITYID"`
// Allie Agent
///////////////
AGENT_PORT string `yaml:"AGENT_PORT" json:"AGENTPORT"`
WORKFLOW_API_KEY string `yaml:"WORKFLOW_API_KEY" json:"WORKFLOWAPIKEY"`
WORKFLOW_STORE_PATH string `yaml:"WORKFLOW_STORE_PATH" json:"WORKFLOWSTOREPATH"`
BINARY_STORE_PATH string `yaml:"BINARY_STORE_PATH" json:"BINARYSTOREPATH"`
NUMBER_OF_WORKFLOW_WORKERS int `yaml:"NUMBER_OF_WORKFLOW_WORKERS" json:"NUMBEROFWORKFLOWWORKERS"`
// External Function Endpoints
EXTERNALFUNCTIONS_ENDPOINT string `yaml:"EXTERNALFUNCTIONS_ENDPOINT" json:"EXTERNALFUNCTIONSENDPOINT"`
FLOWKIT_PYTHON_ENDPOINT string `yaml:"FLOWKIT_PYTHON_ENDPOINT" json:"FLOWKITPYTHONENDPOINT"`
FLOWKIT_PYTHON_API_KEY string `yaml:"FLOWKIT_PYTHON_API_KEY" json:"FLOWKITPYTHONAPIKEY"`
// Azure AD Authentication
AZURE_AD_AUTHENTICATION bool `yaml:"AZURE_AD_AUTHENTICATION" json:"AZUREADAUTHENTICATION"`
AZURE_AD_AUTHENTICATION_URL string `yaml:"AZURE_AD_AUTHENTICATION_URL" json:"AZUREADAUTHENTICATIONURL"`
// Private Workflows
LOAD_PRIVATE_WORKFLOWS bool `yaml:"LOAD_PRIVATE_WORKFLOWS" json:"LOADPRIVATEWORKFLOWS"`
GITHUB_USER string `yaml:"GITHUB_USER" json:"GITHUBUSER"`
GITHUB_TOKEN string `yaml:"GITHUB_TOKEN" json:"GITHUBTOKEN"`
PRIVATE_WORKFLOWS_FOLDER string `yaml:"PRIVATE_WORKFLOWS_FOLDER" json:"PRIVATEWORKFLOWSFOLDER"`
// Exec Settings
EXEC_ENDPOINT string `yaml:"EXEC_ENDPOINT" json:"EXECENDPOINT"`
EXEC_AGENT_API_KEY string `yaml:"EXEC_AGENT_API_KEY" json:"EXECAGENTAPIKEY"`
MONGO_DB_FOR_MULTI_AGENT bool `yaml:"MONGO_DB_FOR_MULTI_AGENT" json:"MONGODBFORMULTIAGENT"`
MONGO_DB_ENDPOINT string `yaml:"MONGO_DB_ENDPOINT" json:"MONGODBENDPOINT"`
// Allie Flowkit
/////////////////
EXTERNALFUNCTIONS_GRPC_PORT string `yaml:"EXTERNALFUNCTIONS_GRPC_PORT" json:"EXTERNALFUNCTIONSGRPCPORT"`
FLOWKIT_API_KEY string `yaml:"FLOWKIT_API_KEY" json:"FLOWKITAPIKEY"`
// Allie Modules
LLM_HANDLER_ENDPOINT string `yaml:"LLM_HANDLER_ENDPOINT" json:"LLMHANDLERENDPOINT"`
KNOWLEDGE_DB_ENDPOINT string `yaml:"KNOWLEDGE_DB_ENDPOINT" json:"KNOWLEDGEDBENDPOINT"`
// Azure Cognitive Services
ACS_ENDPOINT string `yaml:"ACS_ENDPOINT" json:"ACSENDPOINT"`
ACS_API_KEY string `yaml:"ACS_API_KEY" json:"ACSAPIKEY"`
ACS_API_VERSION string `yaml:"ACS_API_VERSION" json:"ACSAPIVERSION"`
// Allie LLM
/////////////
WEBSERVER_PORT string `yaml:"WEBSERVER_PORT" json:"WEBSERVERPORT"`
MODELS_CONFIG_LOCATION string `yaml:"MODELS_CONFIG_LOCATION" json:"MODELSCONFIGLOCATION"`
// Allie DB
///////////
WEBSERVER_PORT_DB string `yaml:"WEBSERVER_PORT_DB" json:"WEBSERVERPORTDB"`
EMBEDDINGS_DIMENSIONS int `yaml:"EMBEDDINGS_DIMENSIONS" json:"EMBEDDINGSDIMENSIONS"`
MILVUS_INDEX_TYPE string `yaml:"MILVUS_INDEX_TYPE" json:"MILVUSINDEXTYPE"`
MILVUS_METRIC_TYPE string `yaml:"MILVUS_METRIC_TYPE" json:"MILVUSMETRICTYPE"` // cosine, l2 or ip
MILVUS_HOST string `yaml:"MILVUS_HOST" json:"MILVUSHOST"`
MILVUS_PORT string `yaml:"MILVUS_PORT" json:"MILVUSPORT"`
NEO4J_DB bool `yaml:"NEO4J_DB" json:"NEO4JDB"`
NEO4J_URI string `yaml:"NEO4J_URI" json:"NEO4JURI"`
NEO4J_USERNAME string `yaml:"NEO4J_USERNAME" json:"NEO4JUSERNAME"`
NEO4J_PASSWORD string `yaml:"NEO4J_PASSWORD" json:"NEO4JPASSWORD"`
TEMP_COLLECTION_NAME string `yaml:"TEMP_COLLECTION_NAME" json:"TEMPCOLLECTIONNAME"`
ELASTICSEARCH_HOST string `yaml:"ELASTICSEARCH_HOST" json:"ELASTICSEARCHHOST"`
ELASTICSEARCH_PORT string `yaml:"ELASTICSEARCH_PORT" json:"ELASTICSEARCHPORT"`
ELASTICSEARCH_USERNAME string `yaml:"ELASTICSEARCH_USERNAME" json:"ELASTICSEARCHUSERNAME"`
ELASTICSEARCH_PASSWORD string `yaml:"ELASTICSEARCH_PASSWORD" json:"ELASTICSEARCHPASSWORD"`
ELASTICSEARCH_INDEX_TYPE string `yaml:"ELASTICSEARCH_INDEX_TYPE" json:"ELASTICSEARCHINDEXTYPE"` // cosineSimilarity or dotProduct
ELASTICSEARCH_TRUSTED_CERTIFICATES []string `yaml:"ELASTICSEARCH_TRUSTED_CERTIFICATES" json:"ELASTICSEARCHTRUSTEDCERTIFICATES"` // list of paths to trusted certificates
ELASTICSEARCH_INSECURE_CONNECTION bool `yaml:"ELASTICSEARCH_INSECURE_CONNECTION" json:"ELASTICSEARCHINSECURECONNECTION"`
DATABASE_TYPE string `yaml:"DATABASE_TYPE" json:"DATABASETYPE"` // milvus or elasticsearch
// Allie Exec
//////////////
WEBSERVER_PORT_EXEC string `yaml:"WEBSERVER_PORT_EXEC" json:"WEBSERVERPORTEXEC"`
EXEC_API_KEY string `yaml:"EXEC_API_KEY" json:"EXECAPIKEY"`
// Python executable name
PYTHON_EXECUTABLE string `yaml:"PYTHON_EXECUTABLE" json:"PYTHONEXECUTABLE"`
// File transfer
WATCH_FOLDER_PATH string `yaml:"WATCH_FOLDER_PATH" json:"WATCHFOLDERPATH"`
MILLISECONDS_SINCE_LAST_CHANGE int `yaml:"MILLISECONDS_SINCE_LAST_CHANGE" json:"MILLISECONDSSINCELASTCHANGE"`
// Agent connection
AGENT_ENDPOINT string `yaml:"AGENT_ENDPOINT" json:"AGENTENDPOINT"`
}
Initialize conifg dict
var GlobalConfig *Config
func readYaml ¶
func readYaml(fileName string, configStruct Config) (extractedConfigStruct Config, err error)
readYaml reads the yaml specified in `fileName` parameter and saves it to `config_struct`
Parameters:
- fileName: The name of the configuration file.
- configStruct: Struct with the parameters of the YAML to read.
Returns:
- extractedConfigStruct: The extracted configuration struct.
- err: An error if there was an issue reading the YAML file.
type flagStringSlice ¶
flagStringSlice is a custom flag type for string slices.
type flagStringSlice []string
func (*flagStringSlice) Set ¶
func (fss *flagStringSlice) Set(value string) error
Set sets the value of the flagStringSlice.
Parameters:
- value: The value to set.
Returns:
- error: An error if there was an issue setting the value.
func (*flagStringSlice) String ¶
func (fss *flagStringSlice) String() string
String returns a string representation of the flagStringSlice.
Returns:
- string: The string representation of the flagStringSlice.