ChatDev/tools/sync_vuegraphs.py

import os
import glob
import requests
import yaml
from pathlib import Path

# Configuration
API_URL = "http://localhost:6400/api/vuegraphs/upload/content"
YAML_DIR = "yaml_instance"

def sync_yaml_to_vuegraphs():
    """Reads all YAML files and uploads them to the VueGraph database."""
    print(f"Syncing YAML files from {YAML_DIR} to {API_URL}...")
    
    yaml_files = glob.glob(os.path.join(YAML_DIR, "*.yaml"))
    
    for file_path in yaml_files:
        try:
            filename = Path(file_path).stem  # simulation_hospital_lmstudio
            
            with open(file_path, "r") as f:
                content = f.read()
                
            # Basic validation to ensure it's a valid YAML
            try:
                yaml.safe_load(content)
            except yaml.YAMLError as e:
                print(f"Skipping {filename}: Invalid YAML - {e}")
                continue

            # Upload to VueGraph API
            payload = {
                "filename": filename,
                "content": content
            }
            
            response = requests.post(API_URL, json=payload)
            
            if response.status_code == 200:
                print(f"Synced: {filename}")
            else:
                print(f"Failed: {filename} - {response.status_code} {response.text}")
                
        except Exception as e:
            print(f"Error processing {file_path}: {e}")

if __name__ == "__main__":
    sync_yaml_to_vuegraphs()