mirror of
https://github.com/katanemo/plano.git
synced 2026-04-25 00:36:34 +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>
121 lines
3.7 KiB
Python
121 lines
3.7 KiB
Python
import subprocess
|
|
import os
|
|
import time
|
|
import select
|
|
import shlex
|
|
import yaml
|
|
import json
|
|
|
|
def run_docker_compose_ps(compose_file, env):
|
|
"""
|
|
Check if all Docker Compose services are in a healthy state.
|
|
|
|
Args:
|
|
path (str): The path where the docker-compose.yml file is located.
|
|
"""
|
|
try:
|
|
# Run `docker-compose ps` to get the health status of each service
|
|
ps_process = subprocess.Popen(
|
|
["docker", "compose", "-p", "arch", "ps", "--format", "table{{.Service}}\t{{.State}}\t{{.Ports}}"],
|
|
cwd=os.path.dirname(compose_file),
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
text=True,
|
|
env=env
|
|
)
|
|
# Capture the output of `docker-compose ps`
|
|
services_status, error_output = ps_process.communicate()
|
|
|
|
# Check if there is any error output
|
|
if error_output:
|
|
print(f"Error while checking service status:\n{error_output}", file=os.sys.stderr)
|
|
return {}
|
|
|
|
services = parse_docker_compose_ps_output(services_status)
|
|
return services
|
|
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Failed to check service status. Error:\n{e.stderr}")
|
|
return e
|
|
|
|
#Helper method to print service status
|
|
def print_service_status(services):
|
|
print(f"{'Service Name':<25} {'State':<20} {'Ports'}")
|
|
print("="*72)
|
|
for service_name, info in services.items():
|
|
status = info['STATE']
|
|
ports = info['PORTS']
|
|
print(f"{service_name:<25} {status:<20} {ports}")
|
|
|
|
#check for states based on the states passed in
|
|
def check_services_state(services, states):
|
|
for service_name, service_info in services.items():
|
|
status = service_info['STATE'].lower() # Convert status to lowercase for easier comparison
|
|
if any(state in status for state in states):
|
|
return True
|
|
|
|
return False
|
|
|
|
def get_llm_provider_access_keys(arch_config_file):
|
|
with open(arch_config_file, 'r') as file:
|
|
arch_config = file.read()
|
|
arch_config_yaml = yaml.safe_load(arch_config)
|
|
|
|
access_key_list = []
|
|
for llm_provider in arch_config_yaml.get("llm_providers", []):
|
|
acess_key = llm_provider.get("access_key")
|
|
if acess_key is not None:
|
|
access_key_list.append(acess_key)
|
|
|
|
return access_key_list
|
|
|
|
def load_env_file_to_dict(file_path):
|
|
env_dict = {}
|
|
|
|
# Open and read the .env file
|
|
with open(file_path, 'r') as file:
|
|
for line in file:
|
|
# Strip any leading/trailing whitespaces
|
|
line = line.strip()
|
|
|
|
# Skip empty lines and comments
|
|
if not line or line.startswith('#'):
|
|
continue
|
|
|
|
# Split the line into key and value at the first '=' sign
|
|
if '=' in line:
|
|
key, value = line.split('=', 1)
|
|
key = key.strip()
|
|
value = value.strip()
|
|
|
|
# Add key-value pair to the dictionary
|
|
env_dict[key] = value
|
|
|
|
return env_dict
|
|
|
|
def parse_docker_compose_ps_output(output):
|
|
# Split the output into lines
|
|
lines = output.strip().splitlines()
|
|
|
|
# Extract the headers (first row) and the rest of the data
|
|
headers = lines[0].split()
|
|
service_data = lines[1:]
|
|
|
|
# Initialize the result dictionary
|
|
services = {}
|
|
|
|
# Iterate over each line of data after the headers
|
|
for line in service_data:
|
|
# Split the line by tabs or multiple spaces
|
|
parts = line.split()
|
|
|
|
# Create a dictionary entry using the header names
|
|
service_info = {
|
|
headers[1]: parts[1], # State
|
|
headers[2]: parts[2] # Ports
|
|
}
|
|
|
|
# Add to the result dictionary using the service name as the key
|
|
services[parts[0]] = service_info
|
|
|
|
return services
|