mirror of
https://github.com/katanemo/plano.git
synced 2026-04-26 01:06:25 +02:00
0e5ea3d6db
* fixed environment variables issue with build. Now llm provider access keys are being written correctly * fixed and verified that keys are being properly set when archgw is booted up * removing leaf reference to a staged config file. not needed anymore * minor fixes to get the build in more stable state * minor fixes based on feedback --------- Co-authored-by: Salman Paracha <salmanparacha@MacBook-Pro-261.local>
101 lines
3.6 KiB
Python
101 lines
3.6 KiB
Python
import os
|
|
from jinja2 import Environment, FileSystemLoader
|
|
import yaml
|
|
from jsonschema import validate
|
|
|
|
ENVOY_CONFIG_TEMPLATE_FILE = os.getenv('ENVOY_CONFIG_TEMPLATE_FILE', 'envoy.template.yaml')
|
|
ARCH_CONFIG_FILE = os.getenv('ARCH_CONFIG_FILE', '/config/arch_config.yaml')
|
|
ENVOY_CONFIG_FILE_RENDERED = os.getenv('ENVOY_CONFIG_FILE_RENDERED', '/etc/envoy/envoy.yaml')
|
|
ARCH_CONFIG_SCHEMA_FILE = os.getenv('ARCH_CONFIG_SCHEMA_FILE', 'arch_config_schema.yaml')
|
|
|
|
def add_secret_key_to_llm_providers(config_yaml) :
|
|
llm_providers = []
|
|
for llm_provider in config_yaml.get("llm_providers", []):
|
|
access_key_env_var = llm_provider.get('access_key', False)
|
|
access_key_value = os.getenv(access_key_env_var, False)
|
|
if access_key_env_var and access_key_value:
|
|
llm_provider['access_key'] = access_key_value
|
|
llm_providers.append(llm_provider)
|
|
config_yaml["llm_providers"] = llm_providers
|
|
return config_yaml
|
|
|
|
def validate_and_render_schema():
|
|
env = Environment(loader=FileSystemLoader('./'))
|
|
template = env.get_template('envoy.template.yaml')
|
|
|
|
try:
|
|
validate_prompt_config(ARCH_CONFIG_FILE, ARCH_CONFIG_SCHEMA_FILE)
|
|
except Exception as e:
|
|
print(e)
|
|
exit(1) # validate_prompt_config failed. Exit
|
|
|
|
with open(ARCH_CONFIG_FILE, 'r') as file:
|
|
arch_config = file.read()
|
|
|
|
with open(ARCH_CONFIG_SCHEMA_FILE, 'r') as file:
|
|
arch_config_schema = file.read()
|
|
|
|
config_yaml = yaml.safe_load(arch_config)
|
|
config_schema_yaml = yaml.safe_load(arch_config_schema)
|
|
inferred_clusters = {}
|
|
|
|
for prompt_target in config_yaml["prompt_targets"]:
|
|
name = prompt_target.get("endpoint", {}).get("name", "")
|
|
if name not in inferred_clusters:
|
|
inferred_clusters[name] = {
|
|
"name": name,
|
|
"port": 80, # default port
|
|
}
|
|
|
|
print(inferred_clusters)
|
|
endpoints = config_yaml.get("endpoints", {})
|
|
|
|
# override the inferred clusters with the ones defined in the config
|
|
for name, endpoint_details in endpoints.items():
|
|
if name in inferred_clusters:
|
|
print("updating cluster", endpoint_details)
|
|
inferred_clusters[name].update(endpoint_details)
|
|
endpoint = inferred_clusters[name]['endpoint']
|
|
if len(endpoint.split(':')) > 1:
|
|
inferred_clusters[name]['endpoint'] = endpoint.split(':')[0]
|
|
inferred_clusters[name]['port'] = int(endpoint.split(':')[1])
|
|
else:
|
|
inferred_clusters[name] = endpoint_details
|
|
|
|
|
|
print("updated clusters", inferred_clusters)
|
|
|
|
config_yaml = add_secret_key_to_llm_providers(config_yaml)
|
|
arch_llm_providers = config_yaml["llm_providers"]
|
|
arch_config_string = yaml.dump(config_yaml)
|
|
|
|
data = {
|
|
'arch_config': arch_config_string,
|
|
'arch_clusters': inferred_clusters,
|
|
'arch_llm_providers': arch_llm_providers
|
|
}
|
|
|
|
rendered = template.render(data)
|
|
print(rendered)
|
|
print(ENVOY_CONFIG_FILE_RENDERED)
|
|
with open(ENVOY_CONFIG_FILE_RENDERED, 'w') as file:
|
|
file.write(rendered)
|
|
|
|
def validate_prompt_config(arch_config_file, arch_config_schema_file):
|
|
with open(arch_config_file, 'r') as file:
|
|
arch_config = file.read()
|
|
|
|
with open(arch_config_schema_file, 'r') as file:
|
|
arch_config_schema = file.read()
|
|
|
|
config_yaml = yaml.safe_load(arch_config)
|
|
config_schema_yaml = yaml.safe_load(arch_config_schema)
|
|
|
|
try:
|
|
validate(config_yaml, config_schema_yaml)
|
|
except Exception as e:
|
|
print(f"Error validating arch_config file: {arch_config_file}, error: {e.message}")
|
|
raise e
|
|
|
|
if __name__ == '__main__':
|
|
validate_and_render_schema()
|