Extract Structured Equipment Diagnostics from LLMs with DSPy and Instructor

                      
                     
from typing import Dict, Any
import os
import requests
import json

# Configuration
class Config:
    API_URL: str = os.getenv('API_URL', 'https://api.example.com/diagnostics')
    API_KEY: str = os.getenv('API_KEY')

# Function to extract diagnostics
async def extract_diagnostics(equipment_id: str) -> Dict[str, Any]:
    headers = {'Authorization': f'Bearer {Config.API_KEY}', 'Content-Type': 'application/json'}
    payload = {'equipment_id': equipment_id}
    try:
        response = requests.post(Config.API_URL, headers=headers, json=payload)
        response.raise_for_status()
        return response.json()
    except requests.exceptions.RequestException as e:
        print(f'Error fetching diagnostics: {e}')
        return {'success': False, 'error': str(e)}

# Main execution
if __name__ == '__main__':
    equipment_id = '12345'
    diagnostics = extract_diagnostics(equipment_id)
    print(json.dumps(diagnostics, indent=2))