classDiagram
    AIEngine <|-- OAIEngine
    OAIEngine <|-- LocalOAIEngine
    LocalOAIEngine <|-- NitroExtension 
    LocalOAIEngine <|-- TensorRTLLMExtension
    OAIEngine <|--RemoteOAIEngine
    RemoteOAIEngine <|-- OpenAIExtension
    RemoteOAIEngine <|-- GroqAIExtension
    RemoteOAIEngine <|-- TritonTensorRTExtension 

class AIEngine {
<<Abstract>>
 string provider
 models()
 prePopulateModels()
}

class OAIEngine {
<<Abstract>>
 string inferenceUrl
 Model loadedModel
 inference()
 stopInference()
 headers()
}

class RemoteOAIEngine {
<<Abstract>>
 string apiKey
 headers()
}

class LocalOAIEngine {
<<abstract>>
 loadModel()
 unloadModel()
}

class OpenAIExtension {
 string inferenceUrl
 string apiKey
}
class GroqAIExtension{
 string inferenceUrl
 string apiKey
}
class TritonTensorRTExtension {
 string inferenceUrl
 string apiKey
}
class NitroExtension {
 spawnNitroProcess()
 loadModel()
 unloadModel()
 killSubprocess()
}
class TensorRTLLMExtension {
 spawnNitroProcess()
 loadModel()
 unloadModel()
 killSubprocess()
}