apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-04-26 01:06:25 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

update config (#93)

Browse source
This commit is contained in:
Adil Hafeez 2024-09-30 17:49:05 -07:00 • committed by GitHub
parent 4182879717
commit cc35eb0cd7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
13 changed files with 575 additions and 329 deletions
Download patch file Download diff file Expand all files Collapse all files

23
config_generator/config_generator.py
View file

@ -17,25 +17,28 @@ config_yaml = yaml.safe_load(katanemo_config)
inferred_clusters = {}
for prompt_target in config_yaml["prompt_targets"]:
cluster = prompt_target.get("endpoint", {}).get("cluster", "")
if cluster not in inferred_clusters:
inferred_clusters[cluster] = {
"name": cluster,
"address": cluster,
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)
clusters = config_yaml.get("clusters", {})
endpoints = config_yaml.get("endpoints", {})
# override the inferred clusters with the ones defined in the config
for name, cluster in clusters.items():
for name, endpoint_details in endpoints.items():
if name in inferred_clusters:
print("updating cluster", cluster)
inferred_clusters[name].update(cluster)
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] = cluster
inferred_clusters[name] = endpoint_details
print("updated clusters", inferred_clusters)